feat:增加强制提示词,增加新的问答示例

This commit is contained in:
雷雨
2025-10-20 16:55:13 +08:00
parent 52d572d4e7
commit db76d6f7fb
5 changed files with 202 additions and 30 deletions

View File

@@ -1,6 +1,7 @@
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_a_next
import random
table_ddls = [
train_ddl.person_database_ddl,train_ddl.person_status_ddl,
train_ddl.person_attendance_ddl,train_ddl.person_ac_area,
@@ -20,6 +21,10 @@ 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:
dd=[]
dd.extend(q_and_a_dict.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'])