Files
securelens-backend/app/schemas/code_scan.py
2026-04-25 19:27:06 +05:30

30 lines
697 B
Python

from pydantic import BaseModel, HttpUrl
from typing import List, Optional, Dict, Any
class CodeScanRequest(BaseModel):
repo_url: str
github_token: str
# branch or commit hash optional
branch: Optional[str] = "main"
class VulnerabilityIssue(BaseModel):
file_path: str
severity: str # High, Medium, Low, Critical
issue: str
explanation: str
suggested_fix: Optional[str] = None
line_number: Optional[int] = None
class CodeScanResponse(BaseModel):
scan_id: str
repo_url: str
summary: str
issues: List[VulnerabilityIssue]
class CodeChatRequest(BaseModel):
scan_id: str
message: str
class CodeChatResponse(BaseModel):
reply: str