mirror of
https://github.com/SkalaraAI/langchain-chatbot.git
synced 2025-04-03 20:10:17 -04:00
23 lines
578 B
Python
23 lines
578 B
Python
"""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
|