832 lines
55 KiB
Python
832 lines
55 KiB
Python
import json
|
||
import os
|
||
from itertools import product
|
||
|
||
# 目录路径
|
||
directory = "data"
|
||
|
||
# 确保目录存在
|
||
if not os.path.exists(directory):
|
||
os.makedirs(directory)
|
||
# 基础数据定义
|
||
BASE_DATA = {
|
||
# 实施组织
|
||
"implementation_organizations": ["宏源电力建设公司（变电）", "送一分公司", "变电分公司", "消防分公司", "安徽宏源电力建设有限公司(线路)", "检修试验分公司"],
|
||
# 工程性质
|
||
"project_types": ["基建", "技改大修", "用户工程", "小型基建"],
|
||
# 工程名
|
||
"project_names": [
|
||
"中心变",
|
||
"1号工程",
|
||
"国网安徽马鞍山供电公司220kV恒兴变电站220kV配电装置改造(PROJ-2024-0271)",
|
||
"国网供电公司220kV恒兴变电站220kV配电装置改造（调试部分）",
|
||
"安徽蚌埠濠州220kV变电站220千伏大唐凤阳红心镇光伏间隔扩建工程(电气安装）"
|
||
"滁州堰陈110千伏变电站新建工程",
|
||
"安徽蚌埠濠州220kV变电站220千伏大唐凤阳红心镇光伏间隔扩建工程(电气安装）(PROJ-2024-0794)",
|
||
"金牛500kV变电站新建工程（建筑）(PROJ-2023-0506)",
|
||
"谷岭220kV变电站220kV蕲城电厂、埇南间隔扩建工程(PROJ-2023-0466)",
|
||
"渝北±800千伏换流站工程电气安装A包(PROJ-2024-0057)",
|
||
"检修公司调相机一二次设备检修维护和改造服务框架-2025年南苑调相机检修（一期）(PROJ-2023-0179)"
|
||
"明生科创基地项目",
|
||
"安徽明生有限公司科创基地项目",
|
||
"无人机智能巡检技术实验室项目",
|
||
"九号线路项目",
|
||
"埇南间隔更换项目",
|
||
"国网滁州供电公司电流互感器更换项目",
|
||
"谷岭220kV变电站220kV蕲城电厂-埇南间隔扩建项目(PROJ-2023-0466)",
|
||
"安徽明生电力投资集团有限公司科创基地项目（一期）(PROJ-2024-1035)"
|
||
"九号工程",
|
||
"合州变电站",
|
||
"合州换流站"
|
||
],
|
||
# 建管单位
|
||
"construction_units": ["国网安徽省电力有限公司建设分公司", "国网安徽省电力有限公司马鞍山供电公司",
|
||
"中铁二局集团电务工程有限公司"],
|
||
# 项目部名称
|
||
"project_departments": ["调试一队", "第9项目管理部", "金上第十一项目部", "第八项目管理部（合肥）", "肥东9号项目部",
|
||
"金上第一项目部管理部(池州黄山)", "第一项目部管理部（肥东）", "调试四队","第一项目部"],
|
||
# 项目经理
|
||
"project_managers": ["陈少平项目经理", "范文立项目经理", "何东洋项目经理"],
|
||
# 分包单位
|
||
"subcontractors": ["安徽劦力建筑装饰有限责任公司", "安徽苏亚建设集团有限公司"],
|
||
# 班组名称
|
||
"team_names": ["张朵班组", "刘梁玉班组", "魏玉龙班组"],
|
||
# 班组长
|
||
"team_leaders": ["李元帅班组长", "刘雨豪班组长"],
|
||
# 风险等级
|
||
"risk_levels": ["1级", "一级", "二级", "5级", "四级"],
|
||
# 8+2工况
|
||
"operatings": ["8+2工况", "8加2工况"],
|
||
# 页面切换
|
||
"pages": ["风险管控", "日计划", "周风险", "日计划统计报表", "日计划推送", "生产管控中心", "考勤统计详情",
|
||
"今日作业计划", "周风险统计报表", "周风险推送"]
|
||
|
||
}
|
||
|
||
# 自然语言模板配置
|
||
TEMPLATE_CONFIG = {
|
||
"日计划数量查询": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天","2025-04-09"],
|
||
"templates": [
|
||
("{date}{project_name}有多少作业计划？", ["date", "project_name"]),
|
||
("{project_name}{date}有多少项作业计划？", ["project_name", "date"]),
|
||
("{date}公司{project_name}有多少作业计划？", ["date", "project_name"]),
|
||
("工程性质是{project_type}{date}有多少作业计划？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}有多少项作业计划？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}有多少条作业计划？", ["project_type", "date"]),
|
||
("{date}风险等级为{risk_level}的作业计划有多少？", ["date", "risk_level"]),
|
||
("公司{date}工程性质为{project_type}的有多少项作业计划？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}的有多少条作业计划？", ["date", "project_type"]),
|
||
("公司工程性质为{project_type}{date}有多少作业计划？", ["project_type", "date"]),
|
||
("工程性质为{project_type}{date}有多少项作业计划？", ["project_type", "date"]),
|
||
("查询{project_name}在{date}的作业计划数量", ["project_name", "date"]),
|
||
("{date}{project_type}类作业计划有多少？", ["date", "project_type"]),
|
||
("{project_type}类{date}作业计划有多少？", ["project_type", "date"]),
|
||
("{construction_unit}在{date}有多少作业计划？", ["construction_unit", "date"]),
|
||
("{construction_unit}在{date}有多少项作业计划？", ["construction_unit", "date"]),
|
||
("{date}{construction_unit}有多少作业计划？", ["date", "construction_unit"]),
|
||
("{date}公司有多少项作业计划？", ["date"]),
|
||
("{date}有多少条作业计划？", ["date"]),
|
||
("公司{date}有多少作业计划？", ["date"]),
|
||
("{date}{operating}有多少项作业计划？", ["date", "operating"]),
|
||
("{date}公司{operating}有多少项作业计划？", ["date", "operating"]),
|
||
("{date}{implementation_organization}{project_department}有多少项作业计划？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{date}{project_department}{implementation_organization}有多少项作业计划？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("{implementation_organization}{project_department}{date}有多少条作业计划？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{date}{implementation_organization}有多少项作业计划？", ["date", "implementation_organization"]),
|
||
("{date}公司{project_department}有多少作业计划？", ["date", "project_department"]),
|
||
("{date}{project_department}{implementation_organization}有多少作业计划？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("{date}{project_department}有多少项作业计划？", ["date", "project_department"]),
|
||
("公司{project_department}{date}有多少项{risk_level}风险作业计划？",
|
||
["project_department", "date", "risk_level"]),
|
||
("{project_department}{date}有多少项{risk_level}风险作业计划？",
|
||
["project_department", "date", "risk_level"]),
|
||
("{project_department}{date}有多少{risk_level}风险作业计划？", ["project_department", "date", "risk_level"]),
|
||
# 请帮我查一下
|
||
("请帮我查一下{date}{project_manager}作业计划是多少？", ["date", "project_manager"]),
|
||
("请帮我查一下{date}{subcontractor}有多少条作业计划？", ["date", "subcontractor"]),
|
||
("请帮我查一下{date}{team_leader}有多少作业计划？", ["date", "team_leader"]),
|
||
("请帮我查一下{date}风险等级为{risk_level}的作业计划有多少？", ["date", "risk_level"]),
|
||
("请帮我查一下{date}公司{project_department}有多少{risk_level}风险作业计划？",
|
||
["date", "project_department", "risk_level"]),
|
||
("请帮我查一下{date}{project_department}有多少{risk_level}风险作业计划？", ["date", "project_department", "risk_level"]),
|
||
("请帮我查一下{date}{project_type}类风险等级为{risk_level}的作业计划有多少？", ["date", "project_type", "risk_level"]),
|
||
("请帮我查一下{date}{construction_unit}有多少{risk_level}风险作业计划？", ["date", "construction_unit", "risk_level"]),
|
||
|
||
("请帮我查一下{date}存在{risk_level}风险的有多少", ["date", "risk_level"]),
|
||
("请帮我查一下{implementation_organization}{date}{risk_level}风险的有多少", ["implementation_organization","date", "risk_level"]),
|
||
("请帮我查一下{implementation_organization}{date}存在{risk_level}风险的有多少", ["implementation_organization","date", "risk_level"]),
|
||
|
||
("{date}{project_type}类{construction_unit}负责的作业计划有多少？",
|
||
["date", "project_type", "construction_unit"]),
|
||
("{date}{project_type}类{implementation_organization}组织实施的作业计划有多少？",
|
||
["date", "project_type", "implementation_organization"]),
|
||
("{date}{project_department}管理的{project_type}类作业计划有多少？",
|
||
["date", "project_department", "project_type"]),
|
||
("{date}{subcontractor}承包的{project_type}类作业计划有多少？", ["date", "subcontractor", "project_type"]),
|
||
("{date}{project_manager}负责的{project_type}类作业计划有多少？",
|
||
["date", "project_manager", "project_type"]),
|
||
("{date}{team_leader}带领的{project_type}类作业计划有多少？", ["date", "team_leader", "project_type"]),
|
||
("{date}{project_name}由{project_manager}作业计划有多少？", ["date", "project_name", "project_manager"]),
|
||
("{date}{project_name}中，风险等级为{risk_level}的作业计划有多少？", ["date", "project_name", "risk_level"]),
|
||
("{date}{project_manager}作业计划有多少？", ["date", "project_manager"]),
|
||
("{project_manager}在{date}作业计划有多少？", ["project_manager", "date"]),
|
||
|
||
("{date}{implementation_organization}{project_manager}的作业计划数量",
|
||
["date", "implementation_organization", "project_manager"]),
|
||
("{implementation_organization}{project_manager}在{date}的作业计划数量",
|
||
["implementation_organization", "project_manager", "date"]),
|
||
|
||
# 班组
|
||
("{date}{team_name}有多少项作业计划？", ["date", "team_name"]),
|
||
("{team_name}{date}有多少作业计划？", ["team_name", "date"]),
|
||
("{team_name}{date}有多少条作业计划？", ["team_name", "date"]),
|
||
("{team_name}{date}作业计划数量", ["team_name", "date"]),
|
||
("{date}{team_name}作业计划数量", ["date", "team_name"]),
|
||
#作业
|
||
("{date}{implementation_organization}{project_department}有多少作业？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{implementation_organization}{project_department}{date}有多少条作业？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{date}{project_department}{implementation_organization}有多少作业？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("{project_department}{implementation_organization}{date}有多少条作业？",
|
||
["project_department", "implementation_organization", "date"]),
|
||
|
||
("{date}{implementation_organization}有多少项作业？", ["date", "implementation_organization"]),
|
||
("{date}公司{project_department}有多少作业？", ["date", "project_department"]),
|
||
("{date}{project_department}有多少项作业？", ["date", "project_department"]),
|
||
#有多少
|
||
("{date}{implementation_organization}{project_department}有多少？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{project_department}{implementation_organization}{date}有多少？",
|
||
["project_department", "implementation_organization", "date"]),
|
||
("{date}{implementation_organization}有多少？", ["date", "implementation_organization"]),
|
||
("{date}公司{project_department}有多少？", ["date", "project_department"]),
|
||
("{date}{project_department}有多少？", ["date", "project_department"]),
|
||
("{date}{project_name}有多少", ["date", "project_name"]),
|
||
("{project_name}{date}有多少", ["project_name", "date"]),
|
||
("{date}公司{project_name}有多少？", ["date", "project_name"]),
|
||
("{date}工程性质是{project_type}有多少", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}有多少", ["project_type", "date"]),
|
||
("{date}存在{operating}的有多少", ["date", "operating"]),
|
||
("{date}{operating}的有多少", ["date", "operating"]),
|
||
]
|
||
},
|
||
|
||
"周计划数量查询": {
|
||
"date": ["本周", "上周", "上一周", "下周", "下一周", "最近一周", "本周内", "这一周"],
|
||
"templates": [
|
||
("{date}{project_name}作业计划有多少？", ["date", "project_name"]),
|
||
("{project_name}{date}作业计划有多少？", ["project_name", "date"]),
|
||
("公司{project_name}{date}作业计划有多少？", ["project_name", "date"]),
|
||
("{construction_unit}{date}作业计划有多少？", ["construction_unit", "date"]),
|
||
# 🎯 仅 date 维度
|
||
("{date}作业计划有多少？", ["date"]),
|
||
|
||
# 🎯 date + 其他单个维度
|
||
("{date}{project_name}有多少项作业计划？", ["date", "project_name"]),
|
||
|
||
("{date}{construction_unit}作业计划有多少？", ["date", "construction_unit"]),
|
||
("{date}{implementation_organization}作业计划有多少？", ["date", "implementation_organization"]),
|
||
("{date}{implementation_organization}{project_department}作业计划有多少？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{date}{project_department}{implementation_organization}作业计划有多少？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("{date}{project_department}作业计划有多少？", ["date", "project_department"]),
|
||
("{date}{project_manager}作业计划有多少？", ["date", "project_manager"]),
|
||
("{date}{subcontractor}作业计划有多少？", ["date", "subcontractor"]),
|
||
("{date}{team_leader}作业计划有多少？", ["date", "team_leader"]),
|
||
|
||
("{date}{project_department}作业计划数量", ["date", "project_department"]),
|
||
("{date}{subcontractor}作业计划数量？", ["date", "subcontractor"]),
|
||
|
||
# 🎯 date + 风险维度
|
||
|
||
("{date}有多少项{risk_level}风险作业计划？", ["date", "risk_level"]),
|
||
("{date}有多少{risk_level}风险作业计划？", ["date", "risk_level"]),
|
||
|
||
("{date}存在{risk_level}风险的作业计划有多少？", ["date", "risk_level"]),
|
||
("{date}{risk_level}风险作业计划有多少", ["date", "risk_level"]),
|
||
|
||
# 🎯 date + construction_unit + risk_level
|
||
("{construction_unit}{date}有多少项{risk_level}风险作业计划", ["construction_unit", "date", "risk_level"]),
|
||
|
||
# 🎯 date + implementation_organization + risk_level
|
||
("{date}{implementation_organization}风险等级为{risk_level}的作业计划有多少？",
|
||
["date", "implementation_organization", "risk_level"]),
|
||
|
||
# 🎯 date + project_name + project_manager
|
||
("{date}{project_name}{project_manager}负责的作业计划有多少？", ["date", "project_name", "project_manager"]),
|
||
|
||
# 🎯 date + project_name + risk_level
|
||
("{date}{project_name}有多少项{risk_level}风险作业计划？", ["date", "project_name", "risk_level"]),
|
||
|
||
# 🎯 project_manager 维度
|
||
("公司{project_manager}{date}作业计划数量？", ["project_manager", "date"]),
|
||
("{project_manager}在{date}作业计划有多少？", ["project_manager", "date"]),
|
||
("公司{project_manager}在{date}作业计划有多少？", ["project_manager", "date"]),
|
||
("{project_manager}在{date}负责的风险等级为{risk_level}的作业计划有多少？",
|
||
["project_manager", "date", "risk_level"]),
|
||
|
||
("{date}{team_name}有多少项作业计划？", ["date", "team_name"]),
|
||
("{team_name}{date}有多少作业计划？", ["team_name", "date"]),
|
||
("{team_name}{date}作业计划数量", ["team_name", "date"]),
|
||
("{date}{team_name}的作业计划数量", ["date", "team_name"]),
|
||
#有多少
|
||
("{date}{implementation_organization}{project_department}有多少？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{implementation_organization}{project_department}{date}有多少？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{date}{project_department}{implementation_organization}有多少项作业计划？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("请帮我查一下{date}{implementation_organization}有多少？", ["date", "implementation_organization"]),
|
||
("请帮我查一下{date}公司{project_department}有多少？", ["date", "project_department"]),
|
||
("请帮我查一下{date}{project_department}有多少？", ["date", "project_department"]),
|
||
("请帮我查一下{date}{project_name}有多少", ["date", "project_name"]),
|
||
("请帮我查一下{project_name}{date}有多少", ["project_name", "date"]),
|
||
("请帮我查一下{date}公司{project_name}有多少？", ["date", "project_name"]),
|
||
("请帮我查一下{date}工程性质是{project_type}有多少", ["project_type", "date"]),
|
||
("请帮我查一下工程性质是{project_type}{date}有多少", ["project_type", "date"])
|
||
]
|
||
},
|
||
"日计划作业内容": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天"],
|
||
"templates": [
|
||
("{date}{project_name}作业内容是什么？", ["date", "project_name"]),
|
||
("{project_name}在{date}的作业有哪些", ["project_name", "date"]),
|
||
("{date}{project_type}类作业有哪些？", ["date", "project_type"]),
|
||
("{project_type}类{date}具体作业内容是什么？", ["project_type", "date"]),
|
||
("{date}工程性质为{project_type}的作业内容是什么？", ["date", "project_type"]),
|
||
("工程性质为{project_type}的{date}作业计划分别是什么？", ["project_type", "date"]),
|
||
("工程性质为{project_type}的{date}4项作业计划分别是什么？", ["project_type", "date"]),
|
||
("{construction_unit}在{date}作业内容是什么？", ["construction_unit", "date"]),
|
||
# 1. 查询特定日期和项目的作业安排
|
||
("{date}{project_name}作业是什么？", ["date", "project_name"]),
|
||
("{date}属于{operating}作业内容是什么？", ["date", "operating"]),
|
||
("{date}存在{operating}作业是什么？", ["date", "operating"]),
|
||
# 3. 查询特定日期和项目类型的工程计划
|
||
("{date}{project_type}类作业有哪些？", ["date", "project_type"]),
|
||
|
||
("{date}{construction_unit}{risk_level}风险的作业内容是什么？", ["date", "construction_unit", "risk_level"]),
|
||
|
||
("{date}{implementation_organization}{risk_level}风险的作业是什么？",
|
||
["date", "implementation_organization", "risk_level"]),
|
||
|
||
# 5. 查询特定日期和项目经理的任务安排
|
||
("{project_manager}在{date}作业内容是什么？", ["project_manager", "date"]),
|
||
("{project_manager}在{date}作业计划分别是什么？", ["project_manager", "date"]),
|
||
("{project_manager}在{date}4项作业计划分别是什么？", ["project_manager", "date"]),
|
||
|
||
# 6. 查询特定日期和风险等级的任务
|
||
("{date}风险等级为{risk_level}的作业计划有哪些？", ["date", "risk_level"]),
|
||
("{date}风险等级为{risk_level}四项作业计划分别有哪些？", ["date", "risk_level"]),
|
||
|
||
# 7. 查询特定日期和实施单位的任务内容
|
||
("{implementation_organization}在{date}作业内容是什么？", ["implementation_organization", "date"]),
|
||
|
||
# 8. 查询特定日期和团队领导的任务安排
|
||
("{team_leader}在{date}作业内容是什么？", ["team_leader", "date"]),
|
||
|
||
# 9. 查询特定日期和项目类型下的高风险任务
|
||
("{date}的{project_type}类风险等级为{risk_level}的作业内容是什么？", ["date", "project_type", "risk_level"]),
|
||
|
||
# 10. 查询特定日期和风险等级的任务安排
|
||
("{date}风险等级为{risk_level}2项作业计划分别是什么？", ["date", "risk_level"]),
|
||
|
||
("{date}{risk_level}的作业内容是什么？", ["date", "risk_level"]),
|
||
|
||
# 11. 查询特定日期和施工单位的任务进展
|
||
("{construction_unit}{date}的作业计划是什么？", ["construction_unit", "date"]),
|
||
("{construction_unit}{date}作业有哪些", ["construction_unit", "date"]),
|
||
|
||
# 12. 查询特定日期和项目经理完成的任务
|
||
("{project_manager}在{date}作业内容是什么？", ["project_manager", "date"]),
|
||
("{project_manager}在{date}三项作业计划分别是什么？", ["project_manager", "date"]),
|
||
|
||
# 13. 查询特定日期和项目经理的高风险任务
|
||
("{project_manager}在{date}的风险等级为{risk_level}的作业内容是什么？",
|
||
["project_manager", "date", "risk_level"]),
|
||
|
||
# 15. 查询特定日期和所有任务安排
|
||
("{date}作业内容是什么？", ["date"]),
|
||
("{date}作业计划有哪些？", ["date"]),
|
||
("{date}作业计划是什么？", ["date"]),
|
||
("{date}三项作业计划分别是什么？", ["date"]),
|
||
("{date}作业是什么？", ["date"]),
|
||
("{date}作业有哪些？", ["date"]),
|
||
|
||
("{date}{implementation_organization}{project_department}作业内容是什么？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{implementation_organization}{project_department}{date}作业计划有哪些？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{implementation_organization}{project_department}{date}作业计划是什么？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{implementation_organization}{project_department}{date}作业是什么？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{implementation_organization}{project_department}{date}两项作业计划分别是什么？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{project_department}{implementation_organization}{date}作业计划是什么？",
|
||
["project_department", "implementation_organization", "date"]),
|
||
("{date}{project_department}{implementation_organization}作业是什么？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("{date}{project_department}{implementation_organization}两项作业计划分别是什么？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
|
||
# 16. 查询特定日期和项目进度
|
||
("请帮我查一下{date}{project_name}作业内容是什么？", ["date", "project_name"]),
|
||
# 班组
|
||
("请帮我查一下{date}{team_name}作业是什么？", ["date", "team_name"]),
|
||
("请帮我查一下{team_name}{date}作业内容", ["team_name", "date"]),
|
||
|
||
("{team_leader}在{date}具体的作业内容", ["team_leader", "date"]),
|
||
|
||
# 9. 查询特定日期和项目类型下的高风险任务
|
||
("{date}的{project_type}类风险等级为{risk_level}具体的作业计划", ["date", "project_type", "risk_level"]),
|
||
|
||
# 10. 查询特定日期和风险等级的任务安排
|
||
("{date}风险等级为{risk_level}具体的2项作业计划", ["date", "risk_level"]),
|
||
|
||
("{date}{risk_level}作业内容", ["date", "risk_level"]),
|
||
|
||
]
|
||
},
|
||
"周计划作业内容": {
|
||
"date": ["本周", "上周", "上一周", "下周", "下一周", "最近一周", "本周内", "这一周"],
|
||
"templates": [
|
||
("工程性质为{project_type}在{date}作业是什么？", ["project_type", "date"]),
|
||
("{date}工程性质为{project_type}作业内容是什么？", ["date", "project_type"]),
|
||
|
||
("{date}{construction_unit}作业有哪些？", ["date", "construction_unit"]),
|
||
|
||
("{implementation_organization}{date}的作业有哪些？", ["implementation_organization", "date"]),
|
||
|
||
("{implementation_organization}{project_department}{date}2项作业计划分别是什么？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
|
||
# 4. 查询某项目在指定周的所有作业计划
|
||
("{project_name}在{date}有哪些作业？", ["project_name", "date"]),
|
||
|
||
# 5. 查询指定周的所有项目类型作业内容
|
||
("{date}{project_type}类作业内容是什么？", ["date", "project_type"]),
|
||
|
||
# 6. 查询某施工单位在指定周的作业任务
|
||
("{construction_unit}在{date}作业计划分别是什么？", ["construction_unit", "date"]),
|
||
|
||
# 7. 查询某项目经理在指定周负责的作业内容
|
||
("{project_manager}在{date}作业内容是什么？", ["project_manager", "date"]),
|
||
|
||
# 8. 查询某团队负责人在指定周的作业安排
|
||
("{team_leader}在{date}作业内容是什么？", ["team_leader", "date"]),
|
||
|
||
# 9. 查询某项目类型在指定周的高风险作业内容
|
||
("{date}的{project_type}类并且风险等级为{risk_level}的作业内容是什么？",
|
||
["date", "project_type", "risk_level"]),
|
||
|
||
# 10. 查询某风险等级在指定周的作业内容
|
||
("{date}风险等级为{risk_level}的作业内容是什么？", ["date", "risk_level"]),
|
||
("{date}{risk_level}风险的作业计划分别是什么？", ["date", "risk_level"]),
|
||
("{date}{risk_level}有哪些作业", ["date", "risk_level"]),
|
||
|
||
# 11. 查询某施工单位在指定周的作业进展
|
||
("{construction_unit}在{date}作业内容是什么？", ["construction_unit", "date"]),
|
||
("{construction_unit}在{date}有哪些作业？", ["construction_unit", "date"]),
|
||
|
||
# 13. 查询某团队在指定周的作业安排
|
||
("{team_leader}领导的团队在{date}的作业有哪些？", ["team_leader", "date"]),
|
||
|
||
# 15. 查询某项目部门在指定周的作业安排
|
||
("{project_department}在{date}作业内容是什么？", ["project_department", "date"]),
|
||
|
||
("请帮我查一下{date}{team_name}作业内容是什么", ["date", "team_name"]),
|
||
("请帮我查一下{team_name}{date}2项作业计划", ["team_name", "date"]),
|
||
("请帮我查一下{team_name}{date}2项作业计划分别是什么", ["team_name", "date"]),
|
||
("请帮我查一下{team_name}{date}有哪些作业", ["team_name", "date"]),
|
||
|
||
("{date}{construction_unit}具体作业计划", ["date", "construction_unit"]),
|
||
|
||
("{implementation_organization}{date}的作业", ["implementation_organization", "date"]),
|
||
|
||
("{implementation_organization}{project_department}{date}具体2项作业计划",
|
||
["implementation_organization", "project_department", "date"]),
|
||
]
|
||
},
|
||
"施工人数": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天","2025-04-09"],
|
||
"templates": [
|
||
("{date}{project_name}施工人员有多少？", ["date", "project_name"]),
|
||
("{date}{project_name}施工人数是多少？", ["date", "project_name"]),
|
||
("{date}{project_name}现场有多少施工人员？", ["date", "project_name"]),
|
||
("{date}{project_name}现场施工人数是多少？", ["date", "project_name"]),
|
||
|
||
("{construction_unit}{date}的施工人数是多少？", ["construction_unit", "date"]),
|
||
("{construction_unit}{date}现场施工人数是多少？", ["construction_unit", "date"]),
|
||
# 2. 统计某施工单位在指定日期的施工总人数
|
||
("统计{construction_unit}在{date}的施工人数是多少？", ["construction_unit", "date"]),
|
||
("{date}属于{operating}的施工人数是多少？", ["date", "operating"]),
|
||
|
||
# 4. 查询某项目类型在指定日期的施工人员需求
|
||
("{date}{project_type}类有多少施工人员？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}有多少施工人员？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}有多少现场施工人员？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}的施工人数是多少？", ["date", "project_type"]),
|
||
("工程性质为{project_type}{date}的施工人数是多少？", ["project_type", "date"]),
|
||
("工程性质为{project_type}{date}的现场施工人数是多少？", ["project_type", "date"]),
|
||
("工程性质为{project_type}{date}有多少施工人员？", ["project_type", "date"]),
|
||
# 5. 统计某施工单位在指定日期的各项目施工人数
|
||
("{construction_unit}在{date}的施工人数是多少？", ["construction_unit", "date"]),
|
||
# 8. 统计某项目经理管理的项目在指定日期的施工总人数
|
||
("{project_manager}负责的项目在{date}的施工人数是多少？", ["project_manager", "date"]),
|
||
("{date}{project_manager}负责的项目的现场施工人数是多少？", ["date", "project_manager"]),
|
||
|
||
# 9. 查询某分包商在指定日期的施工人员投入
|
||
("{subcontractor}{date}施工人员有多少？", ["subcontractor", "date"]),
|
||
("{subcontractor}{date}的施工人数是多少？", ["subcontractor", "date"]),
|
||
("{date}{subcontractor}的施工人员有多少？", ["date", "subcontractor"]),
|
||
("{date}{subcontractor}的施工人数是多少？", ["date", "subcontractor"]),
|
||
|
||
("{implementation_organization}{date}现场施工人员有多少？", ["implementation_organization", "date"]),
|
||
("{implementation_organization}{date}现场有多少施工人员？", ["implementation_organization", "date"]),
|
||
("{implementation_organization}{date}的现场施工人数是多少？", ["implementation_organization", "date"]),
|
||
("{date}{implementation_organization}现场有多少施工人员？", ["date", "implementation_organization"]),
|
||
("{date}{implementation_organization}现场施工人数是多少？", ["date", "implementation_organization"]),
|
||
|
||
|
||
("{team_leader}{date}的施工人员有多少？", ["team_leader", "date"]),
|
||
("{team_leader}{date}的施工人数是多少？", ["team_leader", "date"]),
|
||
("{date}{team_leader}的施工人员有多少？", ["date", "team_leader"]),
|
||
("{date}{team_leader}的施工人数是多少？", ["date", "team_leader"]),
|
||
|
||
# 11. 查询某实施单位在指定日期的施工人员总数
|
||
("{implementation_organization}{date}的施工人数是多少？", ["implementation_organization", "date"]),
|
||
("{implementation_organization}{date}的施工人员有多少？", ["implementation_organization", "date"]),
|
||
("{date}{team_leader}的施工人员有多少？", ["date", "team_leader"]),
|
||
("{date}{team_leader}的施工人数是多少？", ["date", "team_leader"]),
|
||
|
||
# 16. 统计某项目部门在指定日期的施工人员数量
|
||
("{project_department}{date}的施工人员有多少？", ["project_department", "date"]),
|
||
("{project_department}{date}的施工人数是多少？", ["project_department", "date"]),
|
||
|
||
# 20. 统计某风险等级项目在指定日期的工种配置情况
|
||
("{date}{risk_level}风险的施工人数是多少？", ["date", "risk_level"]),
|
||
|
||
# 21. 查询某分包商在指定周的施工人员安排
|
||
("{subcontractor}{date}的施工人数是多少？", ["subcontractor", "date"]),
|
||
|
||
# 22. 统计某施工单位在指定周的高风险作业人员数量
|
||
("{construction_unit}{date}风险等级为{risk_level}的施工人数是多少？",
|
||
["construction_unit", "date", "risk_level"]),
|
||
|
||
("{date}{team_name}施工人数是多少", ["date", "team_name"]),
|
||
("{date}{team_name}施工人数是什么", ["date", "team_name"]),
|
||
("{team_name}{date}施工人数有多少", ["team_name", "date"]),
|
||
("{team_name}{date}施工人数是什么", ["team_name", "date"]),
|
||
|
||
#
|
||
("{date}{implementation_organization}{project_department}作业人数是多少？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{implementation_organization}{project_department}{date}作业人员有多少？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{implementation_organization}{project_department}{date}作业人数是多少？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{date}{implementation_organization}{project_department}有多少作业人员？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{implementation_organization}{project_department}{date}作业人数是多少？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{project_department}{implementation_organization}{date}有多少作业人员？",
|
||
["project_department", "implementation_organization", "date"]),
|
||
("{date}{project_department}{implementation_organization}有多少作业人数？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("请帮我查一下{date}{project_department}{implementation_organization}作业人员有多少？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
("请帮我查一下{project_department}{implementation_organization}{date}作业人员是多少",
|
||
["project_department", "implementation_organization", "date"]),
|
||
]
|
||
},
|
||
"作业考勤人数": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天"],
|
||
"templates": [
|
||
("{date}{project_name}作业考勤人数是多少", ["date", "project_name"]),
|
||
("{project_name}{date}作业考勤人数是多少", ["project_name", "date"]),
|
||
("查询{subcontractor}{date}的出勤记录", ["subcontractor", "date"]),
|
||
("查询{subcontractor}{date}的作业考勤人数是多少", ["subcontractor", "date"]),
|
||
#出勤人
|
||
("{date}{project_name}出勤人数是多少", ["date", "project_name"]),
|
||
("{project_name}{date}出勤人数是多少", ["project_name", "date"]),
|
||
("查询{subcontractor}{date}的出勤记录", ["subcontractor", "date"]),
|
||
("查询{subcontractor}{date}的作业出勤人数是多少", ["subcontractor", "date"]),
|
||
|
||
("{date}{operating}的作业考勤人数是多少？", ["date", "operating"]),
|
||
("{team_leader}{date}的作业考勤人数是多少", ["team_leader", "date"]),
|
||
# 4. 统计某施工单位在指定日期的考勤人数
|
||
("统计{construction_unit}{date}的考勤人数", ["construction_unit", "date"]),
|
||
|
||
# 5. 查询某实施单位在指定日期的考勤情况
|
||
("{implementation_organization}{date}的考勤情况如何？", ["implementation_organization", "date"]),
|
||
|
||
# 6. 查询某风险等级项目在指定日期的考勤详情
|
||
("{date}{risk_level}风险项目考勤详情", ["date", "risk_level"]),
|
||
|
||
# 7. 统计某项目类型在指定日期的出勤人数
|
||
("{date}{project_type}类出勤人数是多少？", ["date", "project_type"]),
|
||
|
||
# 10. 统计某项目在指定周的出勤总人数
|
||
("{project_name}{date}的出勤人数是多少？", ["project_name", "date"]),
|
||
("{project_name}{date}的考勤率是多少？", ["project_name", "date"]),
|
||
("{project_name}{date}的考勤率", ["project_name", "date"]),
|
||
|
||
# 11. 查询某分包商在指定周的出勤情况
|
||
("{subcontractor}在{date}的出勤情况如何？", ["subcontractor", "date"]),
|
||
("{subcontractor}在{date}的出勤情况怎么样？", ["subcontractor", "date"]),
|
||
|
||
("请帮我查一下{date}{team_name}考勤人数是多少", ["date", "team_name"]),
|
||
("请帮我查一下{team_name}{date}考勤人数", ["team_name", "date"]),
|
||
]
|
||
},
|
||
"页面切换": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天"],
|
||
"templates": [
|
||
("打开{page}页面", ["page"]),
|
||
("打开{page}", ["page"]),
|
||
("打开{page}模块", ["page"]),
|
||
("进入{page}", ["page"]),
|
||
("进入{page}模块", ["page"]),
|
||
("进入{page}页面", ["page"]),
|
||
("跳转到{page}", ["page"]),
|
||
("跳转到{page}模块", ["page"]),
|
||
("跳转到{page}页面", ["page"]),
|
||
("访问{page}页面", ["page"]),
|
||
("访问{page}模块", ["page"]),
|
||
("访问{page}", ["page"]),
|
||
("请打开{page}模块", ["page"]),
|
||
("请打开{page}", ["page"]),
|
||
("请打开{page}页面", ["page"]),
|
||
("加载{page}模块", ["page"]),
|
||
("加载{page}", ["page"]),
|
||
("加载{page}页面", ["page"]),
|
||
]
|
||
},
|
||
"作业面查询": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天"],
|
||
"templates": [
|
||
("{date}{project_name}有多少作业面？", ["date", "project_name"]),
|
||
|
||
("{date}{construction_unit}作业面有多少？", ["date", "construction_unit"]),
|
||
("{date}{implementation_organization}作业面是多少？", ["date", "implementation_organization"]),
|
||
("{date}{implementation_organization}{project_department}有多少作业面？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{date}{project_department}有多少作业面？", ["date", "project_department"]),
|
||
("{date}{project_manager}作业面是多少？", ["date", "project_manager"]),
|
||
("{date}{subcontractor}有多少作业面？", ["date", "subcontractor"]),
|
||
("{date}{team_leader}作业面是多少？", ["date", "team_leader"]),
|
||
("{date}{project_name}有多少作业面？", ["date", "project_name"]),
|
||
("{project_name}{date}有多少项作业面？", ["project_name", "date"]),
|
||
("{date}公司{project_name}有多少作业面？", ["date", "project_name"]),
|
||
("工程性质是{project_type}{date}有多少作业面？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}有多少项作业面？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}有多少条作业面？", ["project_type", "date"]),
|
||
("{date}风险等级为{risk_level}的作业面有多少？", ["date", "risk_level"]),
|
||
("请帮我查一下公司{date}工程性质为{project_type}作业面有多少？", ["date", "project_type"]),
|
||
("请帮我查一下{date}工程性质为{project_type}作业面有多少？", ["date", "project_type"]),
|
||
("请帮我查一下公司工程性质为{project_type}{date}作业面有多少？", ["project_type", "date"]),
|
||
]
|
||
},
|
||
"班组人数查询": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天","2025-04-09"],
|
||
"templates": [
|
||
("{date}{project_name}班组人员有多少？", ["date", "project_name"]),
|
||
("{date}{project_name}班组人数是多少？", ["date", "project_name"]),
|
||
("{date}{project_name}现场有多少班组人员？", ["date", "project_name"]),
|
||
("{date}{project_name}现场班组人数是多少？", ["date", "project_name"]),
|
||
|
||
("{construction_unit}{date}的班组人数是多少？", ["construction_unit", "date"]),
|
||
("{construction_unit}{date}现场班组人数是多少？", ["construction_unit", "date"]),
|
||
# 2. 统计某施工单位在指定日期的班组总人数
|
||
("统计{construction_unit}在{date}的班组人数是多少？", ["construction_unit", "date"]),
|
||
("{date}属于{operating}的班组人数是多少？", ["date", "operating"]),
|
||
|
||
# 4. 查询某项目类型在指定日期的班组工人员需求
|
||
("{date}{project_type}类有多少班组人员？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}有多少班组人员？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}有多少现场班组人员？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}的班组人数是多少？", ["date", "project_type"]),
|
||
("工程性质为{project_type}{date}有多少班组人员？", ["project_type", "date"]),
|
||
("工程性质为{project_type}{date}的现场班组人数是多少？", ["project_type", "date"]),
|
||
("工程性质为{project_type}{date}有多少班组人员？", ["project_type", "date"]),
|
||
("工程性质为{project_type}{date}的现场班组人数是多少？", ["project_type", "date"]),
|
||
# 5. 统计班组工单位在指定日期的各项目人数
|
||
("工程性质为{project_type}{date}的现场班组人数是多少？", ["project_type", "date"]),
|
||
("{construction_unit}在{date}的班组人数是多少？", ["construction_unit", "date"]),
|
||
("工程性质为{project_type}{date}的现场班组人数是多少？", ["project_type", "date"]),
|
||
# 8. 统计某项目经理管理的项目在指定日期的总人数
|
||
("工程性质为{project_type}{date}的现场班组人数是多少？", ["project_type", "date"]),
|
||
("{project_manager}负责的项目在{date}的班组人数是多少？", ["project_manager", "date"]),
|
||
("{date}{project_manager}负责的项目的现场班组人数是多少？", ["date", "project_manager"]),
|
||
#
|
||
("请帮我查一下{date}{implementation_organization}{project_department}班组人数是多少？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("请帮我查一下{implementation_organization}{project_department}{date}班组人员有多少？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("请帮我查一下{implementation_organization}{project_department}{date}班组人数是多少？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{date}{team_name}有多少？", ["date", "team_name"]),
|
||
("请帮我查一下{date}{team_name}有多少？", ["date", "team_name"]),
|
||
|
||
]
|
||
},
|
||
"班组数查询": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天","2025-04-09"],
|
||
"templates": [
|
||
("{date}{project_name}现场有多少班组？", ["date", "project_name"]),
|
||
("{date}{project_name}现场有多少个班组？", ["date", "project_name"]),
|
||
("{date}{project_name}现场班组有多少个？", ["date", "project_name"]),
|
||
("{date}{project_name}班组数是多少？", ["date", "project_name"]),
|
||
("{date}{project_name}班组数是什么？", ["date", "project_name"]),
|
||
("{date}{project_name}班组是多少？", ["date", "project_name"]),
|
||
("{date}{project_name}班组有多少个？", ["date", "project_name"]),
|
||
("{date}{construction_unit}班组数有多少？", ["date", "construction_unit"]),
|
||
("{date}{implementation_organization}现场班组数是多少？", ["date", "implementation_organization"]),
|
||
("{date}{implementation_organization}{project_department}有多少班组？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{date}{project_department}现场有多少个班组？", ["date", "project_department"]),
|
||
("{date}{project_manager}现场班组数是多少？", ["date", "project_manager"]),
|
||
("{date}{subcontractor}现场有多少班组？", ["date", "subcontractor"]),
|
||
("{date}{team_leader}班组数是多少？", ["date", "team_leader"]),
|
||
("{date}{team_leader}班组有多少？", ["date", "team_leader"]),
|
||
("{date}{project_name}有多少班组？", ["date", "project_name"]),
|
||
("{project_name}{date}有多少个班组？", ["project_name", "date"]),
|
||
("{date}公司{project_name}有多少班组？", ["date", "project_name"]),
|
||
("工程性质是{project_type}{date}有多少班组？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}有多少班组？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}有多少班组？", ["project_type", "date"]),
|
||
("请帮我查一下{date}风险等级为{risk_level}的班组有多少？", ["date", "risk_level"]),
|
||
("请帮我查一下公司{date}工程性质为{project_type}班组有多少个？", ["date", "project_type"]),
|
||
("请帮我查一下{date}工程性质为{project_type}现场班组有多少？", ["date", "project_type"]),
|
||
("请帮我查一下公司工程性质为{project_type}{date}有多少班组？", ["project_type", "date"]),
|
||
]
|
||
},
|
||
"作业面内容": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天","2025-04-09"],
|
||
"templates": [
|
||
("{date}{project_name}作业面是什么？", ["date", "project_name"]),
|
||
("{project_name}在{date}的作业面有哪些", ["project_name", "date"]),
|
||
("{date}{project_type}类作业面有哪些？", ["date", "project_type"]),
|
||
("{project_type}类{date}具体作业面内容是什么？", ["project_type", "date"]),
|
||
("{date}工程性质为{project_type}的作业面内容是什么？", ["date", "project_type"]),
|
||
("工程性质为{project_type}的{date}作业面分别是什么？", ["project_type", "date"]),
|
||
("工程性质为{project_type}的{date}4项作业面分别是什么？", ["project_type", "date"]),
|
||
("{construction_unit}在{date}作业面内容是什么？", ["construction_unit", "date"]),
|
||
# 1. 查询特定日期和项目的作业安排
|
||
("{date}{project_name}作业面是什么？", ["date", "project_name"]),
|
||
("{date}属于{operating}作业面内容是什么？", ["date", "operating"]),
|
||
("{date}存在{operating}作业面是什么？", ["date", "operating"]),
|
||
# 3. 查询特定日期和项目类型的工程计划
|
||
("{date}{project_type}类具体作业面有哪些？", ["date", "project_type"]),
|
||
|
||
("{date}{construction_unit}{risk_level}风险的作业面内容是什么？", ["date", "construction_unit", "risk_level"]),
|
||
|
||
("{date}{implementation_organization}{risk_level}风险的作业面是什么？",
|
||
["date", "implementation_organization", "risk_level"]),
|
||
|
||
# 5. 查询特定日期和项目经理的任务安排
|
||
("{project_manager}在{date}作业面内容是什么？", ["project_manager", "date"]),
|
||
("{project_manager}在{date}作业面分别是什么？", ["project_manager", "date"]),
|
||
("{project_manager}在{date}4项作业面分别是什么？", ["project_manager", "date"]),
|
||
|
||
# 6. 查询特定日期和风险等级的任务
|
||
("{date}风险等级为{risk_level}的作业面有哪些？", ["date", "risk_level"]),
|
||
("{date}风险等级为{risk_level}四项作业面具体分别有哪些？", ["date", "risk_level"]),
|
||
|
||
|
||
# 10. 查询特定日期和风险等级的任务安排
|
||
("{date}风险等级为{risk_level}2项作业面分别是什么？", ["date", "risk_level"]),
|
||
|
||
("{date}{risk_level}的作业面具体内容是什么？", ["date", "risk_level"]),
|
||
|
||
# 11. 查询特定日期和施工单位的任务进展
|
||
("{construction_unit}{date}的作业面是什么？", ["construction_unit", "date"]),
|
||
("{construction_unit}{date}作业面具体内容有哪些", ["construction_unit", "date"]),
|
||
|
||
# 12. 查询特定日期和项目经理完成的任务
|
||
("{project_manager}在{date}三项作业面分别是什么？", ["project_manager", "date"]),
|
||
|
||
# 13. 查询特定日期和项目经理的高风险任务
|
||
("{project_manager}在{date}的风险等级为{risk_level}的作业面内容是什么？",
|
||
["project_manager", "date", "risk_level"]),
|
||
|
||
# 15. 查询特定日期和所有任务安排
|
||
("{date}作业面内容是什么？", ["date"]),
|
||
("{date}作业面有哪些？", ["date"]),
|
||
("{date}作业面是什么？", ["date"]),
|
||
|
||
("{date}{implementation_organization}{project_department}具体作业面内容是什么？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{implementation_organization}{project_department}{date}具体作业面有哪些？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{implementation_organization}{project_department}{date}作业面是什么？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{implementation_organization}{project_department}{date}两项作业面分别是什么？",
|
||
["implementation_organization", "project_department", "date"]),
|
||
("{project_department}{implementation_organization}{date}具体作业面是什么？",
|
||
["project_department", "implementation_organization", "date"]),
|
||
("{date}{project_department}{implementation_organization}两项作业面分别是什么？",
|
||
["date", "project_department", "implementation_organization"]),
|
||
]
|
||
},
|
||
"班组详情": {
|
||
"date": ["今日", "昨日", "2024年5月24日", "5月24日", "今天", "昨天","2025-04-09"],
|
||
"templates": [
|
||
("{date}{project_name}现场具体有哪些班组？", ["date", "project_name"]),
|
||
("{date}{project_name}现场班组详情是什么？", ["date", "project_name"]),
|
||
("{date}{project_name}现场班组情况", ["date", "project_name"]),
|
||
("{date}{project_name}具体班组情况是什么？", ["date", "project_name"]),
|
||
("{date}{project_name}具体班组详情是什么？", ["date", "project_name"]),
|
||
("{date}{project_name}班组详细情况是什么？", ["date", "project_name"]),
|
||
("{date}{project_name}班组详情", ["date", "project_name"]),
|
||
("{date}{construction_unit}具体有哪些班组", ["date", "construction_unit"]),
|
||
("{date}{implementation_organization}班组有哪些？", ["date", "implementation_organization"]),
|
||
("{date}{implementation_organization}{project_department}现场具体有哪些班组？",
|
||
["date", "implementation_organization", "project_department"]),
|
||
("{date}{project_department}现场班组详情是什么？", ["date", "project_department"]),
|
||
("{date}{project_manager}现场班组情况？", ["date", "project_manager"]),
|
||
("{date}{subcontractor}具体班组情况是什么？", ["date", "subcontractor"]),
|
||
("{date}{team_leader}具体班组详情是什么？", ["date", "team_leader"]),
|
||
("{date}{team_leader}班组详细情况是什么？", ["date", "team_leader"]),
|
||
("{date}{project_name}班组详情？", ["date", "project_name"]),
|
||
("{project_name}{date}具体有哪些班组？", ["project_name", "date"]),
|
||
("{date}公司{project_name}班组有哪些？", ["date", "project_name"]),
|
||
("工程性质是{project_type}{date}现场具体有哪些班组？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}现场班组详情是什么？", ["project_type", "date"]),
|
||
("工程性质是{project_type}{date}现场班组情况？", ["project_type", "date"]),
|
||
("{date}风险等级为{risk_level}具体班组情况是什么？", ["date", "risk_level"]),
|
||
("公司{date}工程性质为{project_type}具体班组详情是什么？", ["date", "project_type"]),
|
||
("{date}工程性质为{project_type}班组详情？", ["date", "project_type"]),
|
||
("公司工程性质为{project_type}{date}具体有哪些班组？", ["project_type", "date"]),
|
||
("公司工程性质为{project_type}{date}班组有哪些？", ["project_type", "date"]),
|
||
]
|
||
}
|
||
}
|
||
|
||
|
||
def generate_natural_samples(config, label):
|
||
"""生成自然语言样本"""
|
||
samples = []
|
||
variable_pool = {
|
||
"project_name": BASE_DATA["project_names"],
|
||
"project_type": BASE_DATA["project_types"],
|
||
"construction_unit": BASE_DATA["construction_units"],
|
||
"implementation_organization": BASE_DATA["implementation_organizations"],
|
||
"subcontractor": BASE_DATA["subcontractors"],
|
||
"team_leader": [f"{tl}" for tl in BASE_DATA["team_leaders"]],
|
||
"risk_level": BASE_DATA["risk_levels"],
|
||
"date": config["date"],
|
||
"project_department": BASE_DATA["project_departments"],
|
||
"project_manager": BASE_DATA["project_managers"],
|
||
"page": BASE_DATA["pages"],
|
||
"operating": BASE_DATA["operatings"],
|
||
"team_name": BASE_DATA["team_names"]
|
||
}
|
||
|
||
for template, variables in config["templates"]:
|
||
for values in product(*[variable_pool[var] for var in variables]):
|
||
text = template.format(**dict(zip(variables, values)))
|
||
|
||
# 生成标注信息
|
||
annotations = []
|
||
pos = 0
|
||
for var, val in zip(variables, values):
|
||
start = text.find(val, pos)
|
||
if start == -1:
|
||
continue
|
||
end = start + len(val)
|
||
annotations.append({
|
||
"text": val,
|
||
"start": start,
|
||
"end": end,
|
||
"label": var
|
||
})
|
||
pos = end # 更新查找位置避免重复
|
||
|
||
samples.append({
|
||
"text": text,
|
||
"annotations": annotations,
|
||
"prompt": label
|
||
})
|
||
|
||
# 保存文件
|
||
filename = f"data/{label}.json"
|
||
with open(filename, "w", encoding="utf-8") as f:
|
||
json.dump(samples, f, ensure_ascii=False, indent=2)
|
||
|
||
print(f"已生成 {len(samples)} 条自然语言 {label} 数据")
|
||
|
||
|
||
# 主执行流程
|
||
if __name__ == "__main__":
|
||
for label, config in TEMPLATE_CONFIG.items():
|
||
generate_natural_samples(config, label)
|