from enum import StrEnum
from pydantic import BaseModel, Field

class AgentOption(StrEnum):
    AUTO = "Авто"
    LAWYER = "Юрист"
    ECONOMIST = "Экономист"
    WEB_DEVELOPER = "Web_developer"
    ACCOUNTANT = "Бухгалтер"
    PSYCHOLOGIST = "Психолог"

class ChatRequest(BaseModel):
    message: str = Field(..., min_length=1, max_length=4000)
    agent: AgentOption = AgentOption.AUTO

class ChatResponse(BaseModel):
    reply: str
    agent_used: str