lei_y601/sqlbot_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'dev' of http://106.13.42.156:33077/lei_y601/sqlbot_agent into dev

Browse Source 
# Conflicts:
#	util/load_ddl_doc.py
This commit is contained in:
yujj128
2025-10-23 16:54:16 +08:00
parent a02f637994 c40e36a2ba
commit cde7561951
6 changed files with 229 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
util/load_ddl_doc.py
View File

@@ -1,6 +1,8 @@
from service.cus_vanna_srevice import CustomVanna
from util import train_ddl
from util import q_and_a_dict,q_and_a_test1
from util import q_and_a_dict,q_and_a_test1,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 +22,12 @@ def add_ddl(vn: CustomVanna):
def add_documentation(vn: CustomVanna):
for doc in list_documentions:
vn.add_documentation(doc)
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'])
for d in q_and_a_test1.question_and_answer:
vn.add_question_sql(