Fix SSE route dependency and align architecture docs
This commit is contained in:
@@ -1,13 +1,19 @@
|
||||
"""Define schema models for rag."""
|
||||
|
||||
from pydantic import BaseModel
|
||||
from typing import Optional
|
||||
# Group related schema definitions so validation rules stay consistent.
|
||||
|
||||
|
||||
|
||||
class RagChatRequest(BaseModel):
|
||||
"""Define the Rag Chat Request API model."""
|
||||
query: str
|
||||
top_k: int = 5
|
||||
|
||||
|
||||
class RetrievedDoc(BaseModel):
|
||||
"""Define the Retrieved Doc API model."""
|
||||
id: str
|
||||
doc_name: str
|
||||
clause_id: Optional[str] = None
|
||||
@@ -17,15 +23,18 @@ class RetrievedDoc(BaseModel):
|
||||
|
||||
|
||||
class SourceInfo(BaseModel):
|
||||
"""Define the Source Info API model."""
|
||||
name: str
|
||||
clause: Optional[str] = None
|
||||
|
||||
|
||||
class QuickQuestion(BaseModel):
|
||||
"""Define the Quick Question API model."""
|
||||
id: str
|
||||
question: str
|
||||
category: str
|
||||
|
||||
|
||||
class QuickQuestionsResponse(BaseModel):
|
||||
"""Define the Quick Questions Response API model."""
|
||||
questions: list[QuickQuestion]
|
||||
Reference in New Issue
Block a user