必须槽位检查

api/mian.py

@@ -11,6 +11,8 @@ from pydantic import ValidationError
 
 from intentRecognition import IntentRecognition
 from slotRecognition import SlotRecognition
+from fuzzywuzzy import process
+import utils
 
 # 常量
 MODEL_ERNIE_PATH = R"../ernie/output/checkpoint-4160"
@@ -42,6 +44,11 @@ intent_recognizer = IntentRecognition(MODEL_ERNIE_PATH, labels)
 # 初始化槽位识别工具类
 slot_recognizer = SlotRecognition(MODEL_UIE_PATH, label_map)
 # 设置Flask应用
+
+#标准工程名
+standard_project_name_list = utils.load_standard_name('./standard_data/standard_project.txt')
+#标准项目名
+standard_program_name_list = utils.load_standard_name('./standard_data/standard_program.txt')
 app = Flask(__name__)
 
 
@@ -230,8 +237,6 @@ def agent():
         return jsonify({"error": str(e)}), 500  # 捕捉其他错误并返回
 
 
-# def check_lost(int_res, slot):
-#     return 0, ""
 def check_lost(int_res, slot):
     #labels: ["天气查询","互联网查询","页面切换","日计划数量查询","周计划数量查询","日计划作业内容","周计划作业内容","施工人数","作业考勤人数","知识问答"]
     mapping = {
@@ -273,5 +278,13 @@ def check_lost(int_res, slot):
         return 1, f"{apologize_str}请问你想查询什么时间的{intention_mapping[int_res]}？"
 
 
+#标准化工程名
+def check_project_standard_slot(int_res, slot)->tuple:
+    return 0, ''
+
+def fuzzy_match(user_input, standard_name):
+    result = process.extract(user_input, standard_name)
+    return result[0], result[1]/100
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=18074, debug=True)

api/standard_data/standard_project.txt

@@ -0,0 +1,199 @@
+杨柳220kV变电站220kV南坪间隔扩建工程
+阜阳东坡110kV变电站新建工程
+杨柳圩110kV变电站110kV配电装置改造工程
+阜阳太和城南110kV变电站新建工程
+安庆石化110kV变电站新建 工程
+华南城220kV变电站新建工程
+马鞍山万济220kV变电站新建工程
+紫蓬500kV变电站间隔扩建工程
+安徽阜阳阜四500kV变电站新建工程
+安徽合肥杜岗220kV开关站新建工程
+庐北220kV变电站新建工程
+渝北±800千伏换流站工程电气安装A包
+宿州城东220kV变电站新建工程
+马鞍山金河口110kV变电站新建工程
+安庆四500kV变电站新建工程
+合肥循环园220kV变电站新建工程
+马鞍山郑蒲220kV变电站新建工程
+国网安徽宣城供电公司500kV河沥变加装固定融冰装置
+滁州堰陈110千伏变电站新建工程
+安徽河沥500kV变电站3号主变扩建工程
+安徽合肥义兴220kV变电站新建工程
+宣城中港110kV变电站新建工程
+安庆城南110kV变电站新建工程
+国网安徽安庆供电公司500kV双岭变电站加装固定融冰装置
+芜湖蛟矶220kV变电站新建工程
+国网安徽马鞍山供电公司220kV恒兴变电站220kV配电装置改造
+马鞍山横龙110kV变电站新建工程
+金牛500kV变电站新建工程
+安庆双港220kV变电站新建工程
+安庆银山220kV变电站新建工程
+芜湖高村220kV变电站新建工程
+滁州红桥220kV变电站新建工程
+安徽合肥方兴II110kV变电站新建工程
+银联黄山园区220kV变电站工程
+陕北-安徽±800千伏特高压直流输电工程陶大庄村接地极工程
+松滋-安丰220kV线路工程
+伯阳500千伏变电站220千伏间隔扩建工程
+道轩汇流站-伯阳 220kV 线路工程
+香涧-梨花220kV线路工程
+埇桥-萧砀Ⅰ回500千伏线路工程
+埇桥-萧砀Ⅱ回500千伏线路工程
+石岗-施桥110kV线路工程
+芦集-古沟π入潘集变电站220kV线路工程
+黄阳-仙河110kV线路工程
+淮南芦集220千伏变电站220千伏配电装置改造工程
+伯阳-谯城500kV线路工程
+魏武-桐花π入祝集变电站110kV线路工程
+耿皇-焦楼220kV线路工程
+蕴山-沙埂110kV线路工程
+蚌埠孝仪220kV变电站新建工程（建筑安装部分）
+安徽六安春秋塘-山七π入汤池变电站110kV线路工程
+亳州城北110kV变电站新建工程
+蚌埠潼河110kV变电站新建工程
+沪宁合高铁安徽段500kV清同5737线等2处电力线路迁改工程
+伯阳-涡阳π入真源变电站220kV线路工程
+濉淮5379濉洪5380线147#-148#跨阜淮铁路迁改工程
+汤庄-平圩500kV线路工程
+换流站-金牛500kV线路工程
+包河10kV玉龙18网架结构化工程施工
+淮南安丰220千伏变电站新建工程
+祝集220kV变电站新建工程
+田家庵电厂秦集改接孝仪变电站220kV线路工程
+谯城500千伏变电站新建工程（建筑）
+金牛500kV变电站新建工程（建筑）
+陕北-安徽直流工程合州±800千伏换流站土建A包
+锁库500kV变电站新建工程
+安徽宿州萧砀500千伏变电站新建工程（建筑部分）
+合肥中心变B包
+中心500kV变电站新建工程（A包）
+国网北京检修公司2024年±500kV延庆换流站直流主设备及辅助设备不停电检修维护
+香涧-鹭岛500kV线路工程（一般线路东段）
+香涧-鹭岛500kV线路工程（淮河大跨越段）
+安徽滁州清流-环漪220kV线路改造工程
+香涧-鹭岛500kV线路工程（一般线路西段）
+谢桥电厂-原鹿220kV线路工程
+阜四-椿树220千伏线路工程
+寨西-苦竹溪T接金桥变电站110kV线路工程
+韦寨-范兴集π入阜四变电站220kV线路工程
+金牛-福渡500kV线路工程（巢湖、无为段）
+金牛-福渡500kV线路工程（庐江段）
+安徽马鞍山当涂-万济220kV线路工程
+黄山巷联-水西220kV线路工程
+苍山-横龙110kV架空线路工程
+安徽马鞍山围屏-万济220kV线路工程
+甘肃-浙江±800千伏特高压直流输电线路工程（皖3标段）
+广德-敬亭π入桂花（广五）变220kV线路工程
+绿雪-莲塘220kV线路工程
+芜湖福渡-蛟矶220kV线路工程
+塔岗-西梁山T接杨柳圩变电站110kV架空线路工程
+芜湖江北-通江π入蛟矶变电站220kV线路工程
+含山-环峰T接横龙110kV架空线路工程
+安徽宣城莲塘-敬亭220kV线路改造工程
+月桥-火龙岗π入高村变电站220kV线路工程
+太和-李腰π入城南变电站110kV架空线路工程
+文昌宫-淮北西牵引站220kV线路工程
+杨柳四铺π入况楼变110kV电缆线路工程
+陕北-安徽±800kV特高压直流输电线路工程（皖2标）
+安徽合肥晶合220kV外部供电工程（合肥杜岗Ⅱ220kV输变电工程）
+肥西-中心500kV线路工程（电缆部分）
+池州电厂二期-涓桥220kV线路工程
+陶楼-下塘、陶楼-航锂（凯博）π入下塘变电站220kV电缆线路工程
+显通-白杨（T接刘桥）、显通-溪河（T接刘桥）π入凌云变电站110kV架空线路工程
+南屏-蓬莱路π入派河变电站110kV线路工程
+茗南-熙湖T接城南变电站110kV架空线路工程
+显通-碱河π入凌云变电站220kV电缆线路工程
+灵泗-奎河220kV线路工程
+合肥二电厂-长临河π入循环园变电站220kV线路工程
+安徽合肥铭传-华南城220kV线路工程
+孔店-龙门500kV线路增容改造工程
+游乐-南岗、游乐-湖光路T接方兴Ⅱ变电站110kV电缆线路工程
+桐城-独秀π入双港变电站220kV线路工程
+杨柳-南坪改接至双堆集牵引站220kV线路工程
+合肥二电厂-彭郢π入长临河变电站220kV线路工程
+合肥长临河-义兴220kV架空线路工程
+陂塘-草庙乡牵引站220kV架空线路工程
+安徽官山-涓桥I、II回π入池州二变500千伏线路工程
+南坪-双堆集牵引站220kV线路工程
+肥西-中心500kV线路工程（架空部分）
+先锋-泉河π入安庆四变电站220kV线路工程
+况楼-杨柳220kV线路工程
+安庆四-涓桥500kV线路工程（一般线路段）
+藕池-漆园π入杨柳变电站220kV线路工程
+金牛-紫蓬500kV线路工程
+合肥金牛-庐江π入庐江北变电站220KV线路工程
+独秀-谭桥牵引站π入和平变电站220kV线路工程
+龙门500kV变电站间隔扩建工程
+安徽合肥下塘220kV变电站新建工程
+芜湖潘坛220kV变电站新建工程
+安徽滁州护桥220kV变电站2号主变扩建工程
+岩歙220kV变电站新建工程
+安徽淮北凌云220千伏变电站新建工程
+六安汤池 110kV 变电站新建工程
+白莲-邵岗π入尧塘变电站35kV架空线路工程
+淮南至桐城高速公路淮南段500kV电力迁改工程
+国网安徽电力营销服务计量用房
+国网安徽亳州供电公司运检营销综合用房
+安庆四500kV变电站新建工程（建筑）
+安徽沙河变至原鹿变500kV双回线路开断接入阜四变工程
+黄栗树-儒林π入堰陈变电站110kV线路工程
+金上-湖北线路工程（川12标）
+安徽阜阳薛桥-花园220kV线路改造工程
+章塘-邰桥、石桥-邰桥π入万济变电站110kV线路工程
+围屏-石桥π入万济变电站110kV线路工程
+安庆-龙山π入安庆四变电站220kV线路工程
+夏湖-华都改接入真源变电站110kV线路工程
+蓼城-冯井π入尧塘变电站110kV线路工程
+汤池-河棚35kV架空线路工程
+埇桥-萧砀回500千伏线路工程
+安徽官山-涓桥、回π入池州二变500千伏线路工程
+众兴-草庙乡牵引站220kV架空线路工程
+湖站1000千伏1号主变A相局放配合项目
+肥西-繁昌Ⅰ、Ⅱ回500kV线路改造工程
+宁芜铁扩能改造(安徽段)涉及500kV线路跨越宁芜铁路迁改工程
+S334峨山路东延伸(沿江高速至芜宣高速)新建工程二期500kV电力迁改工程
+芜湖抖音220kV线路涉500kV峨峰5914、峨廻5904线改造工程
+包河10kV延安30与兴集18网架结构优化工程施工
+国网安徽马鞍山供电公司500kV峨廻5904线#92等拉线杆塔改造工程
+国网安徽合肥供电公司2024年500kV肥西变电站一键顺控系统完善提升
+淮南芦集 220千伏变电站220千伏配电装置改造工程
+国网安徽马鞍山供电公司220kV恒兴变电站220kV配电装置改造 
+和襄高速机电工程总承包项目500kV-1100kV特高压电力线路迁建工程
+阜四500千伏变电站新建工程(建筑部分)
+蚌埠孝仪220kV变电站新建工程
+陕北~安徽±800千伏特高压直流输电工程受端换流站四通一平工程
+无人机智能巡检技术实验室项目
+国网安徽淮北供电公司500kV濉溪变电站1号主变5022电流互感器改造
+况楼220kV变电站间隔扩建工程
+国网北京检修公司2024年±500kV延庆换流站阀冷系统设备驻站
+安庆武昌220kV变电站220kV江调间隔扩建工程
+香涧-梨花π入固镇南牵引站220kV线路工程
+安徽合肥杜岗Ⅱ220kV开关站新建工程
+国网安徽合肥供电公司2024年220kV板桥变电站一键顺控系统完善提升
+科学城-长岗Ⅰ回π入塘稍变电站110kV架空线路工程
+牛草山风电场-仙河T接黄阳变电站110kV线路 工程
+世袭庄园二期居配工程
+谷岭220kV变电站220kV蕲城电厂、埇南间隔扩建工程
+官塘-燕山π入孝仪变电站220kV线路工程
+甘肃-浙江线路工程施工包1（芜湖市无为市-芜湖市南陵县）
+新河-建阳、明都-建阳π入红桥变电站220kV线路工程
+霸王-厉阳T接金河口110kV线路工程
+双港-独秀π入和平变电站220kV线路工程
+程集-牛庄π入阜四变220千伏线路工程
+合州换流站-文都500千伏线路工程
+安庆四-涓桥500kV线路工程（长江大跨越段）
+和平-邓村π入双港变电站220kV线路工程
+安徽合肥包河区10kV王郢19开关网架结构优化工程施工
+合肥二电厂-彭郢π入长临河变电站 220kV 线路工程
+安徽合肥肥西-游乐π入华南城变220kV架空线路工程
+安徽阜阳颍上县110kV半岗-润河35kV线路工程
+游乐-南岗、游乐-湖光路T接方兴Ⅱ变电站110kV架空线路工程
+合肥轨道7号线10kV杆线迁改工程
+科学城-长岗（双回T接空港）改接塘稍变电站110kV架空线路工程
+安徽蚌埠濠州220kV变电站220千伏大唐凤阳红心镇光伏间隔扩建工程(电气安装）
+余桥-银山220kV线路工程
+六庆城际铁路穿±500kV宜华线改造工程
+魏岗-古井π入祝集110kV线路工程
+灵泗500kV变电站新建工程
+蚌埠大唐滁州电厂220kV送出工程
+国网北京检修公司2024年±500kV延庆换流站直流主设备年度检修维护

api/utils.py

@@ -0,0 +1,18 @@
+def load_standard_name(file_path:str):
+    try:
+        # f = open(file_path, 'r', encoding='utf-8')
+        with open(file_path, 'r', encoding='utf-8') as f:
+            data = f.read()
+            data = data.split('\n')
+            works = {}
+            for d in data:
+                wk = d[:d.find('(')]
+                works[wk] = d
+            keys = list(works.keys())
+            return keys
+    except FileNotFoundError:
+        print(f"错误：文件 {file_path} 不存在")
+        raise FileNotFoundError(f"错误：文件 {file_path} 不存在")
+    except Exception as e:
+        print(f"读取文件时发生错误：{e}")
+        raise Exception(f"错误：文件 {file_path} 不存在")