File size: 220 Bytes
10bcd3f
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from pydantic import BaseModel

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

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