From a2b4193c0ae3fd677e488cfdcf91c7cfa41b3845 Mon Sep 17 00:00:00 2001 From: junjun Date: Fri, 27 Jun 2025 12:47:37 +0800 Subject: [PATCH] feat: mcp tools --- backend/apps/chat/api/chat.py | 14 +++++++------- backend/apps/chat/models/chat_model.py | 8 ++++++-- 2 files changed, 13 insertions(+), 9 deletions(-) diff --git a/backend/apps/chat/api/chat.py b/backend/apps/chat/api/chat.py index 733b5f70..bfc62011 100644 --- a/backend/apps/chat/api/chat.py +++ b/backend/apps/chat/api/chat.py @@ -8,7 +8,7 @@ from sqlmodel import select from apps.chat.curd.chat import list_chats, get_chat_with_records, create_chat, rename_chat, \ delete_chat, list_records -from apps.chat.models.chat_model import CreateChat, ChatRecord, RenameChat, Chat, ChatQuestion +from apps.chat.models.chat_model import CreateChat, ChatRecord, RenameChat, Chat, ChatQuestion, ChatToken from apps.chat.task.llm import LLMService from apps.datasource.crud.datasource import get_table_schema from apps.datasource.models.datasource import CoreDatasource @@ -58,15 +58,15 @@ async def delete(session: SessionDep, chart_id: int): @router.post("/mcp_start", operation_id="mcp_start") -async def mcp_start(session: SessionDep, token: str, create_chat_obj: CreateChat): - user = await get_current_user(session, token) - return await start_chat(session, user, create_chat_obj) +async def mcp_start(session: SessionDep, chat: ChatToken): + user = await get_current_user(session, chat.token) + return create_chat(session, user, CreateChat(), False) @router.post("/mcp_question", operation_id="mcp_question") -async def mcp_question(session: SessionDep, token: str, request_question: ChatQuestion): - user = await get_current_user(session, token) - return await stream_sql(session, user, request_question) +async def mcp_question(session: SessionDep, chat: ChatToken): + user = await get_current_user(session, chat.token) + return await stream_sql(session, user, chat) @router.post("/start") diff --git a/backend/apps/chat/models/chat_model.py b/backend/apps/chat/models/chat_model.py index 3b0824d6..981af51b 100644 --- a/backend/apps/chat/models/chat_model.py +++ b/backend/apps/chat/models/chat_model.py @@ -138,5 +138,9 @@ class AiModelQuestion(BaseModel): class ChatQuestion(AiModelQuestion): - question: str - chat_id: int + question: str = '' + chat_id: int = 0 + + +class ChatToken(ChatQuestion): + token:str = '' \ No newline at end of file