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: Assistant select ds error with training template

Browse Source
This commit is contained in:
fit2cloud-chenyw
2025-09-19 09:20:33 +08:00
committed by fit2cloud-chenyw
parent 449d65d050
commit 9e63e44de0
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/task/llm.py
+11 -10
View File
@@ -521,11 +521,12 @@ class LLMService:
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)
oid = self.ds.oid if isinstance(self.ds, CoreDatasource) else 1
dsId = self.ds.id if isinstance(self.ds, CoreDatasource) else None
self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question, 1)
self.chat_question.data_training = get_training_template(self.session, self.chat_question.question, dsId, oid)
self.init_messages()
@@ -936,11 +937,11 @@ class LLMService:
def run_task(self, in_chat: bool = True):
try:
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)
oid = self.ds.oid if isinstance(self.ds, CoreDatasource) else 1
dsId = self.ds.id if isinstance(self.ds, CoreDatasource) else None
self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question, oid)
self.chat_question.data_training = get_training_template(self.session, self.chat_question.question, dsId, oid)
self.init_messages()