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

feat: add SQL examples in prompt

Browse Source
This commit is contained in:
ulleo
2025-09-18 16:49:22 +08:00
committed by ulleo
parent ebb9d4c06d
commit 26502e7dde
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/task/llm.py
+5 -2
View File
@@ -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()