weiweiw
/
Intention
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/dev' into dev

This commit is contained in:
jiang 2025-03-09 14:54:52 +08:00
parent 95843aae79 285865e9a8
commit 6debcbc1bd
1 changed files with 28 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
api/main.py
View File

@ -23,16 +23,18 @@ labels = [
# 标签映射
label_map = {
0: 'O', 1: 'B-date', 11: 'I-date',
2: 'B-projectName', 12: 'I-projectName',
3: 'B-projectType', 13: 'I-projectType',
4: 'B-constructionUnit', 14: 'I-constructionUnit',
5: 'B-implementationOrganization', 15: 'I-implementationOrganization',
6: 'B-projectDepartment', 16: 'I-projectDepartment',
7: 'B-projectManager', 17: 'I-projectManager',
8: 'B-subcontractor', 18: 'I-subcontractor',
9: 'B-teamLeader', 19: 'I-teamLeader',
10: 'B-riskLevel', 20: 'I-riskLevel'
0: 'O', # 非实体
1: 'B-date', 12: 'I-date',
2: 'B-project_name', 13: 'I-project_name',
3: 'B-project_type', 14: 'I-project_type',
4: 'B-construction_unit', 15: 'I-construction_unit',
5: 'B-implementation_organization', 16: 'I-implementation_organization',
6: 'B-project_department', 17: 'I-project_department',
7: 'B-project_manager', 18: 'I-project_manager',
8: 'B-subcontractor', 19: 'I-subcontractor',
9: 'B-team_leader', 20: 'I-team_leader',
10: 'B-risk_level', 21: 'I-risk_level',
11: 'B-page', 22: 'I-page',
}
# 初始化工具类
@ -208,22 +210,22 @@ def agent():
print(f"多轮意图识别后的label:{predicted_label}, id:{predicted_id},槽位抽取后的实体：{entities},message:{messages}")
#必须槽位缺失检查
status, sk = check_lost(predicted_id, entities)
if status == CheckResult.NEEDS_MORE_ROUNDS:
return jsonify({"code": 10001, "msg": "成功",
"answer": { "miss": sk},
})
#工程名和项目名标准化
print(f"start to check_project_standard_slot")
result, information = check_project_standard_slot(predicted_id, entities)
print(f"end check_project_standard_slot,{result},{information}")
if result == CheckResult.NEEDS_MORE_ROUNDS:
return jsonify({
"code": 10001, "msg": "成功",
"answer": {"miss": information},
})
# #必须槽位缺失检查
# status, sk = check_lost(predicted_id, entities)
# if status == CheckResult.NEEDS_MORE_ROUNDS:
# return jsonify({"code": 10001, "msg": "成功",
# "answer": { "miss": sk},
# })
#
# #工程名和项目名标准化
# print(f"start to check_project_standard_slot")
# result, information = check_project_standard_slot(predicted_id, entities)
# print(f"end check_project_standard_slot,{result},{information}")
# if result == CheckResult.NEEDS_MORE_ROUNDS:
# return jsonify({
# "code": 10001, "msg": "成功",
# "answer": {"miss": information},
# })
return jsonify({
"code": 200,"msg": "成功",