修复枚举对应错误,增加问答示例,优化提示词,优化问答示例结构增强检索效果

This commit is contained in:
yujj128
2025-10-23 16:52:25 +08:00
parent 75af77cf4a
commit a02f637994
5 changed files with 897 additions and 72 deletions

View File

@@ -1,6 +1,6 @@
from service.cus_vanna_srevice import CustomVanna
from util import train_ddl
from util import q_and_a_dict
from util import q_and_a_dict,q_and_a_test1
table_ddls = [
train_ddl.person_database_ddl,train_ddl.person_status_ddl,
train_ddl.person_attendance_ddl,train_ddl.person_ac_area,
@@ -20,6 +20,12 @@ def add_ddl(vn: CustomVanna):
def add_documentation(vn: CustomVanna):
for doc in list_documentions:
vn.add_documentation(doc)
for d in q_and_a_dict.question_and_answer:
vn.add_question_sql(question=d['question'], sql=d['answer'])
for d in q_and_a_test1.question_and_answer:
vn.add_question_sql(
question=f"{d['question']} [分类:{d['category']}] [标签:{','.join(d['tags'])}]",
sql=d['answer']
)
# vn.add_question_sql(question=d['question'], sql=d['answer'])