feat: add SQL examples in prompt

Author: ulleo

backend/apps/chat/task/llm.py

@@ -29,6 +29,7 @@
     get_chat_chart_data, list_generate_sql_logs, list_generate_chart_logs, start_log, end_log, \
     get_last_execute_sql_error
 from apps.chat.models.chat_model import ChatQuestion, ChatRecord, Chat, RenameChat, ChatLog, OperationEnum
+from apps.data_training.curd.data_training import get_training_template
 from apps.datasource.crud.datasource import get_table_schema
 from apps.datasource.crud.permission import get_row_permission_filters, is_normal_user
 from apps.datasource.models.datasource import CoreDatasource
@@ -935,6 +936,9 @@ 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.init_messages()
 
             # return id

backend/apps/data_training/curd/data_training.py

@@ -204,7 +204,7 @@ def save_embeddings(session: Session, ids: List[int]):
 """
 
 
-def select_terminology_by_question(session: SessionDep, question: str, oid: int, datasource: int):
+def select_training_by_question(session: SessionDep, question: str, oid: int, datasource: int):
     if question.strip() == "":
         return []
 
@@ -303,7 +303,7 @@ def get_training_template(session: SessionDep, question: str, datasource: int, o
         oid = 1
     if not datasource:
         return ''
-    _results = select_terminology_by_question(session, question, oid, datasource)
+    _results = select_training_by_question(session, question, oid, datasource)
     if _results and len(_results) > 0:
         data_training = to_xml_string(_results)
         template = get_base_data_training_template().format(data_training=data_training)