gg

2025-12-06 15:12:19 +03:00
parent 65608dc391
commit 76368175ca
21 changed files with 594 additions and 0 deletions
--- a/app/models/init.py
+++ b/app/models/init.py
--- a/app/models/pycache/init.cpython-311.pyc
+++ b/app/models/pycache/init.cpython-311.pyc
--- a/app/models/pycache/schemas.cpython-311.pyc
+++ b/app/models/pycache/schemas.cpython-311.pyc
--- a/app/models/schemas.py
+++ b/app/models/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