mirror of
https://github.com/CyberSecurityUP/NeuroSploit.git
synced 2026-03-12 13:27:45 +00:00
e0935793c5
116 modules | 100 vuln types | 18 API routes | 18 frontend pages Major features: - VulnEngine: 100 vuln types, 526+ payloads, 12 testers, anti-hallucination prompts - Autonomous Agent: 3-stream auto pentest, multi-session (5 concurrent), pause/resume/stop - CLI Agent: Claude Code / Gemini CLI / Codex CLI inside Kali containers - Validation Pipeline: negative controls, proof of execution, confidence scoring, judge - AI Reasoning: ReACT engine, token budget, endpoint classifier, CVE hunter, deep recon - Multi-Agent: 5 specialists + orchestrator + researcher AI + vuln type agents - RAG System: BM25/TF-IDF/ChromaDB vectorstore, few-shot, reasoning templates - Smart Router: 20 providers (8 CLI OAuth + 12 API), tier failover, token refresh - Kali Sandbox: container-per-scan, 56 tools, VPN support, on-demand install - Full IA Testing: methodology-driven comprehensive pentest sessions - Notifications: Discord, Telegram, WhatsApp/Twilio multi-channel alerts - Frontend: React/TypeScript with 18 pages, real-time WebSocket updates
67 lines
2.1 KiB
Python
Executable File
67 lines
2.1 KiB
Python
Executable File
"""
|
|
NeuroSploit v3 - Agent Task Schemas
|
|
"""
|
|
from datetime import datetime
|
|
from typing import Optional, List
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class AgentTaskCreate(BaseModel):
|
|
"""Schema for creating an agent task"""
|
|
scan_id: str = Field(..., description="Scan ID this task belongs to")
|
|
task_type: str = Field(..., description="Task type: recon, analysis, testing, reporting")
|
|
task_name: str = Field(..., description="Human-readable task name")
|
|
description: Optional[str] = Field(None, description="Task description")
|
|
tool_name: Optional[str] = Field(None, description="Tool being used")
|
|
tool_category: Optional[str] = Field(None, description="Tool category")
|
|
|
|
|
|
class AgentTaskUpdate(BaseModel):
|
|
"""Schema for updating an agent task"""
|
|
status: Optional[str] = Field(None, description="Task status")
|
|
items_processed: Optional[int] = Field(None, description="Items processed")
|
|
items_found: Optional[int] = Field(None, description="Items found")
|
|
result_summary: Optional[str] = Field(None, description="Result summary")
|
|
error_message: Optional[str] = Field(None, description="Error message if failed")
|
|
|
|
|
|
class AgentTaskResponse(BaseModel):
|
|
"""Schema for agent task response"""
|
|
id: str
|
|
scan_id: str
|
|
task_type: str
|
|
task_name: str
|
|
description: Optional[str]
|
|
tool_name: Optional[str]
|
|
tool_category: Optional[str]
|
|
status: str
|
|
started_at: Optional[datetime]
|
|
completed_at: Optional[datetime]
|
|
duration_ms: Optional[int]
|
|
items_processed: int
|
|
items_found: int
|
|
result_summary: Optional[str]
|
|
error_message: Optional[str]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class AgentTaskListResponse(BaseModel):
|
|
"""Schema for list of agent tasks"""
|
|
tasks: List[AgentTaskResponse]
|
|
total: int
|
|
scan_id: str
|
|
|
|
|
|
class AgentTaskSummary(BaseModel):
|
|
"""Schema for agent task summary statistics"""
|
|
total: int
|
|
pending: int
|
|
running: int
|
|
completed: int
|
|
failed: int
|
|
by_type: dict # recon, analysis, testing, reporting counts
|
|
by_tool: dict # tool name -> count
|