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

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

Browse Source
This commit is contained in:
yujj128
2025-10-23 17:34:35 +08:00
parent cde7561951
commit 3b49ce3025
2 changed files with 33 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
util/load_ddl_doc.py
View File

@@ -23,17 +23,15 @@ def add_documentation(vn: CustomVanna):
for doc in list_documentions:
vn.add_documentation(doc)
dd=[]
dd.extend(q_and_a_dict.question_and_answer)
##dd.extend(q_a_next.org_and_attend_q_a)
dd.extend(q_and_a_test1.question_and_answer)
dd.extend(q_a_next.org_and_attend_q_a)
random.shuffle(dd)
for d in dd:
vn.add_question_sql(question=d['question'], sql=d['answer'])
# for d in dd:
# 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'])