From 26502e7dde01ef0a7944905b4c6b619e830e83ab 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 1309f59d..63354fa8 100644 --- a/backend/apps/chat/task/llm.py +++ b/backend/apps/chat/task/llm.py @@ -532,6 +532,9 @@ class LLMService: 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 @@ class LLMService: 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()