From 2b7cde64993e54a35d6c6a74115118d25ae9c286 Mon Sep 17 00:00:00 2001 From: junjun Date: Mon, 28 Jul 2025 17:09:07 +0800 Subject: [PATCH] feat: mcp param --- backend/apps/chat/models/chat_model.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/backend/apps/chat/models/chat_model.py b/backend/apps/chat/models/chat_model.py index f095a65d..7271502a 100644 --- a/backend/apps/chat/models/chat_model.py +++ b/backend/apps/chat/models/chat_model.py @@ -24,7 +24,8 @@ class Chat(SQLModel, table=True): chat_type: str = Field(max_length=20, default="chat") # chat, datasource datasource: int = Field(sa_column=Column(BigInteger, nullable=True)) engine_type: str = Field(max_length=64) - origin: Optional[int] = Field(sa_column=Column(Integer, nullable=False, default=0)) # 0: default, 1: mcp, 2: assistant + origin: Optional[int] = Field( + sa_column=Column(Integer, nullable=False, default=0)) # 0: default, 1: mcp, 2: assistant class ChatRecord(SQLModel, table=True): @@ -152,11 +153,12 @@ class AiModelQuestion(BaseModel): class ChatQuestion(AiModelQuestion): - question: str = '' - chat_id: int = 0 + question: str + chat_id: int + class ChatMcp(ChatQuestion): - token: str = '' + token: str class ChatStart(BaseModel):