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-09-28 16:44:58 +08:00
|
|
|
from util import q_and_a_dict
|
2025-09-23 16:29:56 +08:00
|
|
|
table_ddls = [
|
2025-09-28 16:44:58 +08:00
|
|
|
train_ddl.person_ddl_sql,train_ddl.rule_ddl,train_ddl.user_status_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-09-28 16:44:58 +08:00
|
|
|
for d in q_and_a_dict.question_and_answer:
|
|
|
|
|
vn.add_question_sql(question=d['question'], sql=d['answer'])
|
|
|
|
|
|
