gg

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