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:
#	main_service.py
This commit is contained in:
yujj128
2025-10-15 10:29:26 +08:00
parent cfc694b821 b79ec73c0f
commit 987db89d7d
2 changed files with 52 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main_service.py
View File

@@ -1,6 +1,8 @@
import copy import copy
import logging import logging
from functools import wraps from functools import wraps
import util.utils
from logging_config import LOGGING_CONFIG
from service.cus_vanna_srevice import CustomVanna, QdrantClient, TTLCacheWrapper from service.cus_vanna_srevice import CustomVanna, QdrantClient, TTLCacheWrapper
from decouple import config from decouple import config
import flask import flask

59
util/q_and_a_dict.py
View File

@@ -1,13 +1,18 @@
question_and_answer = [ question_and_answer = [
{"question": "所有员工男女各有多少人", {"question": "综合处有多少员工，男女员工分别有多少",
"answer": ''' "answer": '''
SELECT CASE WHEN "gender" = '1' THEN '男' WHEN "gender" = '2' THEN '女' END AS "gender", SELECT COUNT(*) AS "总人数",
COUNT(*) AS "person_count" SUM(CASE WHEN gender = '1' THEN 1 ELSE 0 END) AS "男员工数",
FROM "YJOA_APPSERVICE_DB"."t_pr3rl2oj_yj_person_database" SUM(CASE WHEN gender = '2' THEN 1 ELSE 0 END) AS "女员工数"
WHERE "dr" = 0 FROM YJOA_APPSERVICE_DB.t_pr3rl2oj_yj_person_database p
GROUP BY "gender" WHERE p.internal_dept IN (SELECT "id"
ORDER BY "gender" ASC LIMIT 1000 FROM "IUAP_APDOC_BASEDOC"."org_orgs" START
WITH "name" LIKE '%综合处%'
CONNECT BY PRIOR "id" = "parentid"
)
AND p.dr = 0
LIMIT 1000;
''' '''
}, },
{ {
@@ -187,7 +192,7 @@ question_and_answer = [
END LIMIT 1000 END LIMIT 1000
''' '''
}, { }, {
"question": "数信中心 部门下有多少员工", "question": "数信中心下有多少员工",
"answer": ''' "answer": '''
select count(*) select count(*)
from YJOA_APPSERVICE_DB.t_pr3rl2oj_yj_person_database from YJOA_APPSERVICE_DB.t_pr3rl2oj_yj_person_database
@@ -309,6 +314,42 @@ question_and_answer = [
END END
ORDER BY "age_group" ASC LIMIT 1000 ORDER BY "age_group" ASC LIMIT 1000
''' '''
}, },{
"question": "查询综合处下面的员工以及他们的年龄",
"answer": '''
SELECT p."id" AS "员工ID",
p."name" AS "姓名",
o."name" AS "所属部门",
CAST(SUBSTR(p."birthday", 1, 4) AS INT) AS "出生年份",
2025 - CAST(SUBSTR(p."birthday", 1, 4) AS INT) AS "年龄",
CASE WHEN p."gender" = '1' THEN '男' WHEN p."gender" = '2' THEN '女' END AS "性别"
FROM "YJOA_APPSERVICE_DB"."t_pr3rl2oj_yj_person_database" p
JOIN "IUAP_APDOC_BASEDOC"."org_orgs" o ON p."internal_dept" = o."id"
WHERE p."dr" = 0
AND o."id" IN (SELECT "id"
FROM "IUAP_APDOC_BASEDOC"."org_orgs" START
WITH "name" LIKE '%综合处%'
CONNECT BY PRIOR "id" = "parentid")
AND p."birthday" IS NOT NULL
AND p."birthday" != ''
ORDER BY o."name" ASC, p."birthday" DESC
LIMIT 1000;
'''
},{
"question": "查询综合处下面的员工的平均年龄",
"answer": '''
SELECT ROUND(AVG(2025 - CAST(SUBSTR(p."birthday", 1, 4) AS INT)), 2) AS "平均年龄"
FROM "YJOA_APPSERVICE_DB"."t_pr3rl2oj_yj_person_database" p
JOIN "IUAP_APDOC_BASEDOC"."org_orgs" o ON p."internal_dept" = o."id"
WHERE p."dr" = 0
AND o."id" IN (SELECT "id"
FROM "IUAP_APDOC_BASEDOC"."org_orgs" START
WITH "name" LIKE '%综合处%'
CONNECT BY PRIOR "id" = "parentid")
AND p."birthday" IS NOT NULL
AND p."birthday" != ''
LIMIT 1000;
'''
}
] ]