File size: 577 Bytes
1ca8371 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
"""Schemas for the chat app."""
from pydantic import BaseModel, validator
class ChatResponse(BaseModel):
"""Chat response schema."""
sender: str
message: str
type: str
@validator("sender")
def sender_must_be_bot_or_you(cls, v):
if v not in ["bot", "you"]:
raise ValueError("sender must be bot or you")
return v
@validator("type")
def validate_message_type(cls, v):
if v not in ["start", "stream", "end", "error", "info"]:
raise ValueError("type must be start, stream or end")
return v |