From af2033b2bdf61bbf539467a830b0ccc663b7c1da Mon Sep 17 00:00:00 2001 From: Danil Parunin 5f1b81b8-4f5d-11e8-9c2d-fa7ae01bbebc Date: Sat, 6 Dec 2025 15:05:12 +0300 Subject: [PATCH] =?UTF-8?q?=D0=97=D0=B0=D0=B3=D1=80=D1=83=D0=B7=D0=B8?= =?UTF-8?q?=D1=82=D1=8C=20=D1=84=D0=B0=D0=B9=D0=BB=D1=8B=20=D0=B2=20=C2=AB?= =?UTF-8?q?/=C2=BB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- schemas.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 schemas.py 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