初始化

app/schemas/rag.py

@@ -0,0 +1,31 @@
+from pydantic import BaseModel
+from typing import Optional
+
+
+class RagChatRequest(BaseModel):
+    query: str
+    top_k: int = 5
+
+
+class RetrievedDoc(BaseModel):
+    id: str
+    doc_name: str
+    clause_id: Optional[str] = None
+    score: float
+    content: str
+    preview: str
+
+
+class SourceInfo(BaseModel):
+    name: str
+    clause: Optional[str] = None
+
+
+class QuickQuestion(BaseModel):
+    id: str
+    question: str
+    category: str
+
+
+class QuickQuestionsResponse(BaseModel):
+    questions: list[QuickQuestion]