From 92d71063d943e489d86ea12f83532ae6e0eb8a96 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=B7=E9=9B=A8?= Date: Fri, 26 Sep 2025 16:05:07 +0800 Subject: [PATCH] =?UTF-8?q?feat:=E4=BB=8E=E5=90=91=E9=87=8F=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E5=BA=93=E7=A7=BB=E9=99=A4=E6=96=87=E6=A1=A3=E8=AF=B4?= =?UTF-8?q?=E6=98=8E=EF=BC=8C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- service/cus_vanna_srevice.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/service/cus_vanna_srevice.py b/service/cus_vanna_srevice.py index 94c4e92..986ede0 100644 --- a/service/cus_vanna_srevice.py +++ b/service/cus_vanna_srevice.py @@ -15,7 +15,7 @@ import json from template.template import get_base_template from datetime import datetime import logging - +from util import train_ddl logger = logging.getLogger(__name__) class OpenAICompatibleLLM(VannaBase): @@ -187,13 +187,13 @@ class OpenAICompatibleLLM(VannaBase): logger.info("Start to generate_sql_2 in cus_vanna_srevice") question_sql_list = self.get_similar_question_sql(question, **kwargs) ddl_list = self.get_related_ddl(question, **kwargs) - doc_list = self.get_related_documentation(question, **kwargs) + #doc_list = self.get_related_documentation(question, **kwargs) template = get_base_template() sql_temp = template['template']['sql'] char_temp = template['template']['chart'] # --------基于提示词，生成sql以及图表类型 sys_temp = sql_temp['system'].format(engine=config("DB_ENGINE", default='mysql'), lang='中文', - schema=ddl_list, documentation=doc_list, + schema=ddl_list, documentation=[train_ddl.train_document], data_training=question_sql_list) logger.info(f"sys_temp:{sys_temp}") user_temp = sql_temp['user'].format(question=question,