inegrated the ai agent

app/schemas/code_scan.py

@@ -0,0 +1,29 @@
+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