2025-09-23 16:29:56 +08:00
|
|
|
from service.cus_vanna_srevice import CustomVanna
|
2025-09-24 14:17:07 +08:00
|
|
|
from util import train_ddl
|
2025-10-23 16:54:16 +08:00
|
|
|
from util import q_and_a_dict,q_and_a_test1,q_a_next
|
|
|
|
|
|
2025-10-20 16:55:13 +08:00
|
|
|
import random
|
2025-09-23 16:29:56 +08:00
|
|
|
table_ddls = [
|
2025-10-14 15:47:11 +08:00
|
|
|
train_ddl.person_database_ddl,train_ddl.person_status_ddl,
|
|
|
|
|
train_ddl.person_attendance_ddl,train_ddl.person_ac_area,
|
2025-11-01 10:16:06 +08:00
|
|
|
train_ddl.person_ac_position,
|
2025-10-14 10:31:00 +08:00
|
|
|
train_ddl.org_orgs_ddl
|
|
|
|
|
|
2025-09-23 16:29:56 +08:00
|
|
|
]
|
|
|
|
|
list_documentions = [
|
2025-09-24 14:17:07 +08:00
|
|
|
train_ddl.train_document,
|
2025-09-23 16:29:56 +08:00
|
|
|
]
|
2025-09-24 14:17:07 +08:00
|
|
|
|
|
|
|
|
|
2025-09-23 16:29:56 +08:00
|
|
|
def add_ddl(vn: CustomVanna):
|
|
|
|
|
for ddl in table_ddls:
|
|
|
|
|
vn.add_ddl(ddl)
|
|
|
|
|
|
2025-09-24 14:17:07 +08:00
|
|
|
|
2025-09-23 16:29:56 +08:00
|
|
|
def add_documentation(vn: CustomVanna):
|
|
|
|
|
for doc in list_documentions:
|
|
|
|
|
vn.add_documentation(doc)
|
2025-10-20 16:55:13 +08:00
|
|
|
dd=[]
|
2025-10-23 17:34:35 +08:00
|
|
|
dd.extend(q_and_a_test1.question_and_answer)
|
|
|
|
|
dd.extend(q_a_next.org_and_attend_q_a)
|
2025-10-20 16:55:13 +08:00
|
|
|
random.shuffle(dd)
|
2025-10-23 17:34:35 +08:00
|
|
|
# for d in dd:
|
|
|
|
|
# vn.add_question_sql(question=d['question'], sql=d['answer'])
|
2025-10-24 11:06:53 +08:00
|
|
|
## tag 待定，容易过拟合
|
|
|
|
|
for d in dd:
|
2025-10-23 16:52:25 +08:00
|
|
|
vn.add_question_sql(
|
2025-10-24 11:06:53 +08:00
|
|
|
question=f"{d['question']}",
|
2025-10-23 16:52:25 +08:00
|
|
|
sql=d['answer']
|
|
|
|
|
)
|
|
|
|
|
|
2025-09-28 16:44:58 +08:00
|
|
|
|
