Spaces:
Running
Running
File size: 1,256 Bytes
b68e1c5 1005046 b68e1c5 1080531 b68e1c5 1005046 b68e1c5 1005046 b68e1c5 1005046 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from enum import Enum
from typing import List, Optional
from pydantic import BaseModel, field_validator, model_validator
class ChatMessage(BaseModel):
role: str
content: str
class MediaType(str, Enum):
MOVIE = "movie"
TV = "tv"
class DeviceInfo(BaseModel):
device_type: Optional[str] = None
platform: Optional[str] = None
user_agent: Optional[str] = None
class ChatRequest(BaseModel):
question: str
history: List[ChatMessage] = []
media_type: MediaType = MediaType.MOVIE
genres: List[str] = []
providers: List[str] = []
year_range: List[int] = [1920, 2025]
session_id: str
query_id: str
device_info: Optional[DeviceInfo] = None
@field_validator("question")
def validate_question(cls, v):
if not v.strip():
raise ValueError("Question cannot be empty")
return v
@model_validator(mode="after")
def validate_year_range(self) -> "ChatRequest":
if len(self.year_range) != 2:
raise ValueError("year_range must be a list of exactly two integers: [start, end]")
return self
class FinalRec(BaseModel):
media_id: int
why: str
class FinalRecsRequest(BaseModel):
query_id: str
final_recs: List[FinalRec] |