31 lines
515 B
Python
31 lines
515 B
Python
|
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]
|