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

fix: MCP auto select datasource error

Browse Source
This commit is contained in:
ulleo
2025-09-18 19:01:08 +08:00
committed by ulleo
parent 144c67a3e3
commit 449d65d050
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/task/llm.py
+8 -7
View File
@@ -520,14 +520,14 @@ class LLMService:
answer=orjson.dumps({'content': full_text}).decode(),
datasource=_datasource,
engine_type=_engine_type)
if self.ds:
self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question,
self.ds.oid if isinstance(self.ds,
CoreDatasource) else 1)
self.chat_question.data_training = get_training_template(self.session, self.chat_question.question,
self.ds.id, self.ds.oid)
self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question,
self.ds.oid if isinstance(self.ds,
CoreDatasource) else 1)
self.chat_question.data_training = get_training_template(self.session, self.chat_question.question,
self.ds.id, self.ds.oid)
self.init_messages()
self.init_messages()
if _error:
raise _error
@@ -1110,6 +1110,7 @@ class LLMService:
SQLBotLogUtil.info(image_url)
yield f'![{chart["type"]}]({image_url})'
except Exception as e:
traceback.print_exc()
error_msg: str
if isinstance(e, SingleMessageError):
error_msg = str(e)