db_model/chat.py

from pydantic import BaseModel

# 定义消息模型
class MessageModel(BaseModel):
    role: str
    content: str

# 定义聊天模型
class ChatModel(BaseModel):
    model: str = None
    messages: list[MessageModel]