init. project

rag-web-ui/backend/app/schemas/tooling.py

@@ -0,0 +1,52 @@
+from datetime import datetime
+from typing import Any, Dict, List, Optional
+
+from pydantic import BaseModel
+
+
+class ToolDefinitionResponse(BaseModel):
+    name: str
+    version: str
+    description: str
+    input_schema: Dict[str, Any]
+    output_schema: Dict[str, Any]
+
+
+class SRSToolCreateJobResponse(BaseModel):
+    job_id: int
+    status: str
+
+
+class SRSToolJobStatusResponse(BaseModel):
+    job_id: int
+    tool_name: str
+    status: str
+    error_message: Optional[str] = None
+    extraction_id: Optional[int] = None
+    started_at: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
+
+
+class SRSToolRequirementItem(BaseModel):
+    id: str
+    title: str
+    description: str
+    priority: str
+    acceptanceCriteria: List[str]
+    sourceField: str
+    sectionNumber: Optional[str] = None
+    sectionTitle: Optional[str] = None
+    requirementType: Optional[str] = None
+    sortOrder: int
+
+
+class SRSToolResultResponse(BaseModel):
+    jobId: int
+    documentName: str
+    generatedAt: str
+    statistics: Dict[str, Any]
+    requirements: List[SRSToolRequirementItem]
+
+
+class SRSToolRequirementsSaveRequest(BaseModel):
+    requirements: List[SRSToolRequirementItem]