chat.py 284 B

12345678910111213141516
  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. class AskResponse(BaseModel):
  10. answer: str
  11. sources: list[str]
  12. latency_ms: int