lei_y601/sqlbot_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'dev' of gitlab-devt.yced.com.cn:lei_y601/sqlbot_agent into dev

Browse Source 
# Conflicts:
#	util/train_ddl.py
This commit is contained in:
yujj128
2025-09-26 17:39:34 +08:00
parent fe4705fea1 92d71063d9
commit dc16bfaa8e
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
service/cus_vanna_srevice.py
View File

@@ -15,7 +15,7 @@ import json
from template.template import get_base_template from template.template import get_base_template
from datetime import datetime from datetime import datetime
import logging import logging
from util import train_ddl
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class OpenAICompatibleLLM(VannaBase): class OpenAICompatibleLLM(VannaBase):
@@ -187,13 +187,13 @@ class OpenAICompatibleLLM(VannaBase):
logger.info("Start to generate_sql_2 in cus_vanna_srevice") logger.info("Start to generate_sql_2 in cus_vanna_srevice")
question_sql_list = self.get_similar_question_sql(question, **kwargs) question_sql_list = self.get_similar_question_sql(question, **kwargs)
ddl_list = self.get_related_ddl(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() template = get_base_template()
sql_temp = template['template']['sql'] sql_temp = template['template']['sql']
char_temp = template['template']['chart'] char_temp = template['template']['chart']
# --------基于提示词，生成sql以及图表类型 # --------基于提示词，生成sql以及图表类型
sys_temp = sql_temp['system'].format(engine=config("DB_ENGINE", default='mysql'), lang='中文', 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) data_training=question_sql_list)
logger.info(f"sys_temp:{sys_temp}") logger.info(f"sys_temp:{sys_temp}")
user_temp = sql_temp['user'].format(question=question, user_temp = sql_temp['user'].format(question=question,

2
util/load_ddl_doc.py
View File

@@ -2,7 +2,7 @@ from service.cus_vanna_srevice import CustomVanna
from util import train_ddl from util import train_ddl
table_ddls = [ table_ddls = [
train_ddl.ddl_sql, train_ddl.ddl_sql,train_ddl.attendance_ddl
] ]
list_documentions = [ list_documentions = [
train_ddl.train_document, train_ddl.train_document,