18 lines
506 B
Python
18 lines
506 B
Python
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 |