feat：修改ddql，增加qa问答

2025-09-28 16:44:58 +08:00
parent dc16bfaa8e
commit 6064710f4f
7 changed files with 396 additions and 641 deletions
--- a/util/load_ddl_doc.py
+++ b/util/load_ddl_doc.py
@@ -1,8 +1,8 @@
 from service.cus_vanna_srevice import CustomVanna
 from util import train_ddl
-
+from  util import q_and_a_dict
 table_ddls = [
-    train_ddl.ddl_sql,train_ddl.attendance_ddl
+    train_ddl.person_ddl_sql,train_ddl.rule_ddl,train_ddl.user_status_ddl
 ]
 list_documentions = [
    train_ddl.train_document,
@@ -17,3 +17,6 @@ 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'])
+