klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: mcp tools

Browse Source
This commit is contained in:
junjun
2025-06-27 13:58:53 +08:00
parent 6b6a1c5803
commit 7ae095da96
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/api/chat.py
+3 -3
View File
@@ -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, ChatToken
from apps.chat.models.chat_model import CreateChat, ChatRecord, RenameChat, Chat, ChatQuestion, ChatMcp
from apps.chat.task.llm import LLMService
from apps.datasource.crud.datasource import get_table_schema
from apps.datasource.models.datasource import CoreDatasource
@@ -58,13 +58,13 @@ async def delete(session: SessionDep, chart_id: int):
@router.post("/mcp_start", operation_id="mcp_start")
async def mcp_start(session: SessionDep, chat: ChatToken):
async def mcp_start(session: SessionDep, chat: ChatMcp):
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, chat: ChatToken):
async def mcp_question(session: SessionDep, chat: ChatMcp):
user = await get_current_user(session, chat.token)
return await stream_sql(session, user, chat)
backend/apps/chat/models/chat_model.py
+1 -1
View File
@@ -142,5 +142,5 @@ class ChatQuestion(AiModelQuestion):
chat_id: int = 0
class ChatToken(ChatQuestion):
class ChatMcp(ChatQuestion):
token:str = ''