diff --git a/schemas.py b/schemas.py new file mode 100644 index 0000000..9754198 --- /dev/null +++ b/schemas.py @@ -0,0 +1,18 @@ +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 \ No newline at end of file