修复枚举对应错误,增加问答示例,优化提示词,优化问答示例结构增强检索效果
This commit is contained in:
@@ -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'])
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user