chat.py 359 B

1234567891011121314151617
  1. from pydantic import BaseModel
  2. class ChatRequest(BaseModel):
  3. message: str
  4. class AskRequest(BaseModel):
  5. question: str
  6. class ChatResponse(BaseModel):
  7. answer: str
  8. latency_ms: int
  9. # Optional: Define AskResponse for consistency (though not currently used)
  10. class AskResponse(BaseModel):
  11. answer: str
  12. sources: list[str]
  13. latency_ms: int