From 5108a4ff38b43af343a60451e8858f72be3426f3 Mon Sep 17 00:00:00 2001 From: ulleo Date: Thu, 18 Sep 2025 16:47:20 +0800 Subject: [PATCH] feat: add SQL examples in prompt --- backend/apps/chat/task/llm.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py index 1309f59da..63354fa88 100644 --- a/backend/apps/chat/task/llm.py +++ b/backend/apps/chat/task/llm.py @@ -532,6 +532,9 @@ def select_datasource(self): 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() if _error: @@ -944,8 +947,8 @@ def run_task(self, in_chat: bool = True): 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.data_training = get_training_template(self.session, self.chat_question.question, + self.ds.id, self.ds.oid) self.init_messages()