hamba-ho's picture
Préparation pour déploiement Hugging Face (Docker + UI)
fab35c8
raw
history blame contribute delete
446 Bytes
from pydantic import BaseModel, Field
from typing import Optional, List
class QuestionRequest(BaseModel):
question: str = Field(..., min_length=2, max_length=500)
session_id: str = Field(..., description="Identifiant unique de la session de cours")
class Source(BaseModel):
document: str
page: Optional[int]
class QuestionResponse(BaseModel):
question: str
answer: str
sources: List[Source]
cached: bool = False