update
This commit is contained in:
31
backend/app/schemas/rag.py
Normal file
31
backend/app/schemas/rag.py
Normal file
@@ -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]
|
||||
Reference in New Issue
Block a user