Files
NeuroSploit/legacy/backend_fastapi/schemas/prompt.py
T
CyberSecurityUP a5badefc29 v3.3.0 GUI dashboard + reports + model expansion + root fix
Engine:
- Fix: inject IS_SANDBOX=1 so Claude Code's --dangerously-skip-permissions
  works under root (real backend runs were exiting rc=1 immediately)
- models: expand to 40 models / 13 providers, tagged CLI vs API
  (NVIDIA NIM, DeepSeek, Mistral, Qwen/DashScope, Groq, Together, OpenRouter,
  Ollama, Gemini) — Qwen/DeepSeek/Llama usable via API
- backends: on_start callback surfaces the exact argv ("what runs behind it")
- orchestrator: require a Playwright screenshot per confirmed finding; collect
  results/activity.json; auto-generate reports after a run
- report.py: HTML always + PDF via Typst engine (.typ source emitted too)

Web dashboard (webgui/, stdlib only — no npm/build):
- Sidebar dashboard (PentAGI-style): Run / Agents / Insights / Reports / Settings
- Multi-target runs; live execution console + per-task activity; finding cards
  with screenshots; backend+provider+model pickers (CLI & API)
- Agents tab: browse 213 + add new .md agents from the UI
- Insights: interactive RL-weight + severity charts
- Reports: download/preview PDF + HTML
- Settings/API: execution mode, per-provider API keys, orchestrator, verbosity
- Endpoints: /api/agents (GET/POST), /api/rl, /api/config, /api/reports,
  /reports/* + /shots/* static serving

Cleanup: retire replaced web stack (frontend React, FastAPI backend, core
orchestration, old test) to legacy/. Active engine + GUI are fully standalone.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-06-14 23:26:11 -03:00

78 lines
2.0 KiB
Python
Executable File

"""
NeuroSploit v3 - Prompt Schemas
"""
from datetime import datetime
from typing import Optional, List
from pydantic import BaseModel, Field
class PromptCreate(BaseModel):
"""Schema for creating a prompt"""
name: str = Field(..., max_length=255, description="Prompt name")
description: Optional[str] = Field(None, description="Prompt description")
content: str = Field(..., min_length=10, description="Prompt content")
category: Optional[str] = Field(None, description="Prompt category")
class PromptUpdate(BaseModel):
"""Schema for updating a prompt"""
name: Optional[str] = None
description: Optional[str] = None
content: Optional[str] = None
category: Optional[str] = None
class PromptParse(BaseModel):
"""Schema for parsing a prompt"""
content: str = Field(..., min_length=10, description="Prompt content to parse")
class VulnerabilityTypeExtracted(BaseModel):
"""Extracted vulnerability type from prompt"""
type: str
category: str
confidence: float
context: Optional[str] = None
class TestingScope(BaseModel):
"""Testing scope extracted from prompt"""
include_recon: bool = True
depth: str = "standard" # quick, standard, thorough, exhaustive
max_requests_per_endpoint: Optional[int] = None
time_limit_minutes: Optional[int] = None
class PromptParseResult(BaseModel):
"""Result of prompt parsing"""
vulnerabilities_to_test: List[VulnerabilityTypeExtracted]
testing_scope: TestingScope
special_instructions: List[str] = []
target_filters: dict = {}
output_preferences: dict = {}
class PromptResponse(BaseModel):
"""Schema for prompt response"""
id: str
name: str
description: Optional[str]
content: str
is_preset: bool
category: Optional[str]
parsed_vulnerabilities: List
created_at: datetime
updated_at: datetime
class Config:
from_attributes = True
class PromptPreset(BaseModel):
"""Schema for preset prompt"""
id: str
name: str
description: str
category: str
vulnerability_count: int