from pydantic import BaseModel
from typing import Optional,Literal

class RequestChat(BaseModel):
    user_id: Optional[str]
    role: Optional[str]
    languages: Optional[str]
    user_input: Optional[str]


class RecommendRequest(BaseModel):
    number:int

class RequestHistoryChat(BaseModel):
    user_id: Optional[str]
    
    
class RequestDetailChat(BaseModel):
    chat_id: Optional[str]


class ChatRequest(BaseModel):
    user_id: int

class DeleteChatRequest(BaseModel):
    chat_id: str

class UpdateNameChat(BaseModel):
    chat_id: str
    name_chat: str
    
class Regenerate(BaseModel):
    chat_id: str
    question_new: str
    languages: Optional[str]
    
class ChatWithServer(BaseModel):
    user_input: str
    language: Literal["EN", "VN"]
    chat_history_id: str