From 9e63e44de07be931cf6be0bd5c870a3460d3ca35 Mon Sep 17 00:00:00 2001 From: fit2cloud-chenyw Date: Fri, 19 Sep 2025 09:19:15 +0800 Subject: [PATCH] fix: Assistant select ds error with training template --- backend/apps/chat/task/llm.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py index f10b3cc3..f26da443 100644 --- a/backend/apps/chat/task/llm.py +++ b/backend/apps/chat/task/llm.py @@ -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()