Initial commit
This commit is contained in:
ruxia
+727
@@ -0,0 +1,727 @@
|
||||
[
|
||||
{
|
||||
"指标名称": "测试1",
|
||||
"指标描述": "xxx",
|
||||
"code": "xxxxxxxx"
|
||||
},
|
||||
{
|
||||
"指标名称": "测试2",
|
||||
"指标描述": "xxx",
|
||||
"code": "xxxxxxxx"
|
||||
},
|
||||
{
|
||||
"指标名称": "电压等级",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "工程编码",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "工程名称",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "工程时间",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "线路曲折系数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "线路长度合计_折单",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "路径长度_单回路长度",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "路径长度_双回路长度",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "路径长度_三回路长度",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "路径长度_四回路长度",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "杆塔总基数",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/基础工程】的【定额】下的【YX2-1/YX2-2/YX2-3/YX2-4/YX2-5/YX2-6/YX2-7】",
|
||||
"code": "def neo4j_find_function():\n project = ProjectBuilder.build()\n result = project.get_quantities_node_by_parent_and_code('工程数据/安装工程/安装/架空输电线路本体工程/基础工程', '定额', 'YX2-1/YX2-2/YX2-3/YX2-4/YX2-5/YX2-6/YX2-7')\n print(result)"
|
||||
},
|
||||
{
|
||||
"指标名称": "角钢塔_塔基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "角钢塔_塔材量",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【角钢】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "角钢塔_其中：高强钢塔材量",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【角钢、高强】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "角钢塔_塔材装材费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "角钢塔_塔材装材费_元",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【角钢】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "角钢塔_其中：高强钢塔材费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "角钢塔_其中：高强钢塔材费用_元",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【角钢、高强】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管塔_塔基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管塔_塔材量",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【钢管塔】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管塔_钢管价格",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管塔_钢管价格_元",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【钢管塔】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管杆_塔基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管杆_塔材量",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【钢管杆】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管杆_钢管价格",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "钢管杆_钢管价格_元",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的【主材】下的【钢管杆】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "水泥杆基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "直线塔基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "耐张转角塔基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "海拔",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_分裂数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_单根导线面积",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_导线量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_其中：节能导线量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_导线装材费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_导线装材费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_其中：节能导线费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_其中：节能导线费用_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "导线及线材_导线类型",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "设计风速",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "覆冰厚度",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_平地",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_丘陵",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_河网",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_泥沼",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_山地",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_高山",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_沙漠",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地形分布_峻岭",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_普通土",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_坚土",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_松砂石",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_水坑",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_泥水坑",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_流沙坑",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_岩石爆破",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "地质条件_岩石人工",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "土石方总量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "土石方量_基坑",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "土石方量_接地",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "土石方量_基面",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_台阶式",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_板式",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_插入式",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_掏挖",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_岩石嵌固",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_锚杆",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_灌注桩",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_人工挖孔桩",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "各类基础数量占总塔基数比例_其他",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "台阶式基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "板式基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "插入式基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "掏挖基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "岩石嵌固基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "锚杆基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "灌注桩基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "人工挖孔桩基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其他基础基数",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基础混凝土总量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "灌注桩基础混凝土量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "现浇基础混凝土量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "挖孔基础混凝土量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基础护壁混凝土用量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "预制混凝土用量",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基础钢材量",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/基础工程/基础砌筑】的【主材】下的【圆钢】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基础钢材价格",
|
||||
"指标描述": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/基础工程/基础砌筑】的【主材】下的【圆钢】",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体费用合计",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体工程人工费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体工程机械费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基础工程费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "杆塔工程费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "接地工程费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "架线工程费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "附件工程费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "辅助工程费用",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "辅助设施工程",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其他费用合计",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "建场费合计",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "项目建设管理费合计",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：工程监理费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "项目建设技术服务费合计",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：项目前期工作费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：勘察费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：设计费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：工程建设检测费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "生产准备费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：安全文明施工费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基本预备费",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "静态投资",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "建设期利息",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "动态投资",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "增值税抵扣税额",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体费用合计_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体工程人工费_本体_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体工程人工费_调试_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体工程机械费_本体_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "本体工程机械费_调试_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基础工程费用_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "杆塔工程费用_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "接地工程费用_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "架线工程费用_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "附件工程费用_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "辅助工程费用_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "辅助工程费用_调试_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "辅助设施工程_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其他费用合计_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "建场费合计_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "项目建设管理费合计_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：工程监理费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "项目建设技术服务费合计_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：项目前期工作费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：勘察费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：设计费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：工程建设检测费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "生产准备费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：安全文明施工费_线路_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "其中：安全文明施工费_调试_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "基本预备费_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "静态投资_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "建设期利息_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "动态投资_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
},
|
||||
{
|
||||
"指标名称": "增值税抵扣税额_元",
|
||||
"指标描述": "",
|
||||
"code": ""
|
||||
}
|
||||
]
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,577 @@
|
||||
from langchain.chains import LLMChain
|
||||
from langchain_openai import OpenAI
|
||||
from langchain_experimental.utilities import PythonREPL
|
||||
from project_implementation import ProjectBuilder
|
||||
from prompt_templates import FUNCTION_CALL_PROMPT
|
||||
import inspect
|
||||
import project
|
||||
import io
|
||||
import sys
|
||||
from parameter_rewriting import rewrite_query_parameters, KnowledgeGraphProcessor
|
||||
import json
|
||||
from langchain.agents import Tool, AgentExecutor, create_react_agent
|
||||
from langchain.prompts import PromptTemplate
|
||||
|
||||
from llm import llm
|
||||
|
||||
|
||||
# 获取ProjectTookiIt类的方法定义
|
||||
def get_project_class_methods():
|
||||
"""
|
||||
从project模块中提取ProjectTookiIt类的方法定义
|
||||
|
||||
Returns:
|
||||
str: 格式化后的方法定义字符串
|
||||
"""
|
||||
project_class_code = inspect.getsource(project.ProjectTookiIt)
|
||||
|
||||
lines = project_class_code.split("\n")
|
||||
result_lines = []
|
||||
|
||||
in_class = False
|
||||
skip_init = False
|
||||
|
||||
for line in lines:
|
||||
if line.strip().startswith("class ProjectTookiIt"):
|
||||
in_class = True
|
||||
result_lines.append(line)
|
||||
elif in_class:
|
||||
if line.strip().startswith("def __init__"):
|
||||
skip_init = True
|
||||
elif skip_init and line.strip() and not line.startswith(" " * 8):
|
||||
skip_init = False
|
||||
|
||||
if not skip_init:
|
||||
result_lines.append(line)
|
||||
|
||||
return "\n".join(result_lines)
|
||||
|
||||
|
||||
# 创建动态提示模板
|
||||
project_class_methods = get_project_class_methods()
|
||||
|
||||
# 创建 Chain
|
||||
function_call_chain = LLMChain(llm=llm, prompt=FUNCTION_CALL_PROMPT, output_key="code")
|
||||
|
||||
# Python 执行器
|
||||
repl = PythonREPL()
|
||||
|
||||
# 创建知识图谱处理器实例
|
||||
kg_processor = KnowledgeGraphProcessor()
|
||||
|
||||
|
||||
# 定义搜索知识库的工具
|
||||
def search_knowledge_base(query):
|
||||
"""
|
||||
在知识库中搜索关键词
|
||||
|
||||
Args:
|
||||
query (str): 搜索关键词
|
||||
|
||||
Returns:
|
||||
str: 搜索结果的JSON字符串
|
||||
"""
|
||||
found_data = kg_processor._search_in_kg(query)
|
||||
if found_data:
|
||||
return json.dumps(found_data, ensure_ascii=False, indent=2)
|
||||
else:
|
||||
return f"未找到与'{query}'相关的信息"
|
||||
|
||||
|
||||
# 定义获取节点定义的工具
|
||||
def get_node_definition(node_type):
|
||||
"""
|
||||
获取节点类型的定义
|
||||
|
||||
Args:
|
||||
node_type (str): 节点类型名称
|
||||
|
||||
Returns:
|
||||
str: 节点类型定义
|
||||
"""
|
||||
definition = kg_processor._get_node_definition(node_type)
|
||||
return definition
|
||||
|
||||
|
||||
# 创建工具列表
|
||||
tools = [
|
||||
Tool(
|
||||
name="search_knowledge_base",
|
||||
func=search_knowledge_base,
|
||||
description="在知识库中搜索关键词，返回相关信息。输入应该是一个搜索关键词。",
|
||||
),
|
||||
Tool(
|
||||
name="get_node_definition",
|
||||
func=get_node_definition,
|
||||
description="获取节点类型的定义，返回类型定义代码。输入应该是一个节点类型名称。",
|
||||
),
|
||||
]
|
||||
|
||||
# 创建Agent
|
||||
agent = create_react_agent(llm, tools, FUNCTION_CALL_PROMPT)
|
||||
|
||||
# 创建Agent执行器
|
||||
agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, handle_parsing_errors=True)
|
||||
|
||||
|
||||
def nl_query_to_function_call(input_data):
|
||||
"""
|
||||
将自然语言查询转换为函数调用并执行，或直接执行提供的代码
|
||||
|
||||
Args:
|
||||
input_data (dict): 包含type和value的字典
|
||||
{
|
||||
"type": "query|code",
|
||||
"value": "查询内容或代码"
|
||||
}
|
||||
|
||||
Returns:
|
||||
dict: 包含状态码、消息和数据的字典
|
||||
"""
|
||||
input_type = input_data.get("type", "query")
|
||||
input_value = input_data.get("value", "")
|
||||
max_retries = 1 # 设置最大重试次数
|
||||
current_retry = 0
|
||||
|
||||
original_query = input_value # 保存原始查询用于RAG
|
||||
|
||||
print(f"\n====== 开始处理查询 ======")
|
||||
print(f"查询类型: {input_type}")
|
||||
print(f"查询内容: {input_value}")
|
||||
|
||||
while current_retry <= max_retries:
|
||||
print(f"\n----- 尝试 #{current_retry + 1} -----")
|
||||
|
||||
# 如果type是query，使用LLM生成代码
|
||||
if input_type == "query" and current_retry == 0:
|
||||
# 从查询中提取关键部分
|
||||
import re
|
||||
|
||||
# 提取【】中的内容
|
||||
path_parts = re.findall(r"【([^】]+)】", input_value)
|
||||
|
||||
# 创建临时代码
|
||||
temp_code = f'search("{input_value}")'
|
||||
|
||||
# 对于每个路径，只添加最后一个部分
|
||||
for part in path_parts:
|
||||
if "/" in part:
|
||||
# 提取路径中的最后一个部分
|
||||
last_part = part.split("/")[-1].strip()
|
||||
if last_part:
|
||||
temp_code += f'\nsearch("{last_part}")'
|
||||
else:
|
||||
# 如果没有/，直接使用整个部分
|
||||
temp_code += f'\nsearch("{part}")'
|
||||
|
||||
# 获取知识库内容和节点定义
|
||||
knowledge_base, node_definitions = kg_processor._get_relevant_knowledge(temp_code)
|
||||
|
||||
# 使用Agent执行查询
|
||||
agent_response = agent_executor.invoke(
|
||||
{
|
||||
"query": input_value,
|
||||
"project_class_methods": project_class_methods,
|
||||
"KnowledgeBase": knowledge_base,
|
||||
"NodeDefinition": node_definitions,
|
||||
}
|
||||
)
|
||||
|
||||
# 从Agent响应中提取代码
|
||||
code = agent_response["output"]
|
||||
print(f"\n生成的代码:\n{code}")
|
||||
else:
|
||||
print(f"\n使用重写后的代码:\n{input_value}")
|
||||
code = input_value
|
||||
|
||||
# 保存原始代码用于返回
|
||||
original_code = code
|
||||
|
||||
# 执行生成的函数并捕获输出
|
||||
try:
|
||||
# 创建一个新的命名空间来执行代码，包含必要的导入
|
||||
namespace = {
|
||||
"ProjectBuilder": ProjectBuilder, # 添加ProjectBuilder到命名空间
|
||||
"project_implementation": __import__("project_implementation"),
|
||||
"project": __import__("project"),
|
||||
}
|
||||
|
||||
# 执行生成的代码，定义neo4j_find_function函数
|
||||
exec(code, namespace)
|
||||
|
||||
# 重定向stdout来捕获print输出
|
||||
old_stdout = sys.stdout
|
||||
redirected_output = io.StringIO()
|
||||
sys.stdout = redirected_output
|
||||
|
||||
try:
|
||||
# 执行函数
|
||||
print("\n执行代码...")
|
||||
result = namespace["neo4j_find_function"]()
|
||||
|
||||
# 获取捕获的输出
|
||||
output = redirected_output.getvalue().strip()
|
||||
print(f"\n原始输出:\n{output}")
|
||||
|
||||
# 检查结果是否为空
|
||||
is_empty_result = (
|
||||
not output
|
||||
or output.lower() == "none"
|
||||
or output == "[]"
|
||||
or "未找到" in output
|
||||
or "[]" in output
|
||||
or "None" in output
|
||||
or result is None
|
||||
)
|
||||
|
||||
# 如果结果为空，走重写流程
|
||||
if is_empty_result:
|
||||
print("\n查询未找到结果，尝试定位具体缺失节点...")
|
||||
|
||||
# 解析原始查询路径中的最后一个节点名
|
||||
import re
|
||||
|
||||
match = re.search(r"【([^】]+)】\s*$", original_query)
|
||||
missing_node = match.group(1) if match else "未知节点"
|
||||
|
||||
error_info = {
|
||||
"error_type": "NodeNotFoundError",
|
||||
"error_message": f"{missing_node} 未找到，请检查该节点是否存在。",
|
||||
"missing_node": missing_node,
|
||||
"original_query": original_query,
|
||||
"executed_code": original_code,
|
||||
}
|
||||
|
||||
print("结构化错误信息：")
|
||||
print(json.dumps(error_info, ensure_ascii=False, indent=2))
|
||||
|
||||
if current_retry < max_retries:
|
||||
print("\n尝试使用RAG重写查询...")
|
||||
try:
|
||||
# 使用RAG重写查询和代码，并传递错误信息
|
||||
rewritten = rewrite_query_parameters(original_query, original_code, error_info)
|
||||
|
||||
print(f"\nRAG重写结果: {json.dumps(rewritten, ensure_ascii=False, indent=2)}")
|
||||
|
||||
# 更新查询和代码
|
||||
if "query" in rewritten and "code" in rewritten and rewritten["code"] != original_code:
|
||||
print("\nRAG重写成功，使用新代码重试...")
|
||||
input_value = rewritten["code"] # 直接使用重写后的代码
|
||||
input_type = "code" # 切换到代码模式
|
||||
current_retry += 1
|
||||
continue # 继续下一次循环
|
||||
else:
|
||||
print("\nRAG重写未产生新代码，返回原始错误")
|
||||
except Exception as e:
|
||||
print(f"\nRAG重写失败: {e}")
|
||||
# 记录错误但继续执行
|
||||
|
||||
# RAG重写失败或未产生新代码，返回原始错误
|
||||
query_status = (
|
||||
"第一次查询失败，RAG重写也失败"
|
||||
if current_retry == 0
|
||||
else f"第{current_retry+1}次查询失败，RAG重写也失败"
|
||||
)
|
||||
print(f"\n{query_status}")
|
||||
return {
|
||||
"code": 1,
|
||||
"message": f"{missing_node} 未找到，请检查该节点是否存在。",
|
||||
"data": {"value": "", "code": original_code},
|
||||
"error_info": error_info,
|
||||
"query_status": query_status,
|
||||
}
|
||||
|
||||
# 清理输出，只保留有用的结果部分
|
||||
clean_output = output
|
||||
|
||||
# 如果输出包含查询结果数量和对象引用
|
||||
if "查询结果数量:" in output and "<project." in output:
|
||||
# 提取查询结果部分
|
||||
import re
|
||||
|
||||
# 尝试提取节点属性
|
||||
node_match = re.search(r"找到节点: <Node.*?properties=({.*?})>", output, re.DOTALL)
|
||||
if node_match:
|
||||
props_str = node_match.group(1).replace("'", '"')
|
||||
try:
|
||||
import ast
|
||||
|
||||
props = ast.literal_eval(props_str)
|
||||
clean_output = json.dumps(props, ensure_ascii=False, indent=2)
|
||||
except:
|
||||
pass
|
||||
|
||||
# 如果有查询结果数量信息
|
||||
count_match = re.search(r"查询结果数量: (\d+)", output)
|
||||
if count_match:
|
||||
count = count_match.group(1)
|
||||
if count == "0":
|
||||
clean_output = "未找到匹配的数据。"
|
||||
is_empty_result = True
|
||||
elif not node_match: # 如果没有提取到节点属性但有结果
|
||||
clean_output = f"找到 {count} 条匹配结果"
|
||||
|
||||
# 检查结果对象
|
||||
if result is not None:
|
||||
if isinstance(result, list):
|
||||
if not result: # 空列表
|
||||
is_empty_result = True
|
||||
else:
|
||||
# 处理非空列表
|
||||
formatted_items = []
|
||||
for item in result:
|
||||
if hasattr(item, "__dict__"):
|
||||
# 提取对象的所有属性
|
||||
attrs = {k: v for k, v in item.__dict__.items() if not k.startswith("_")}
|
||||
formatted_items.append(attrs)
|
||||
else:
|
||||
formatted_items.append(str(item))
|
||||
|
||||
if not is_empty_result: # 只有在不是空结果时才返回成功
|
||||
query_status = (
|
||||
"第一次查询成功"
|
||||
if current_retry == 0
|
||||
else f"第{current_retry+1}次查询成功(RAG重写后)"
|
||||
)
|
||||
print(f"\n{query_status}")
|
||||
return {
|
||||
"code": 0,
|
||||
"message": "成功",
|
||||
"data": {
|
||||
"value": json.dumps(formatted_items, ensure_ascii=False, indent=2),
|
||||
"code": original_code,
|
||||
},
|
||||
"query_status": query_status,
|
||||
}
|
||||
elif hasattr(result, "__dict__"):
|
||||
# 单个对象
|
||||
attrs = {k: v for k, v in result.__dict__.items() if not k.startswith("_")}
|
||||
|
||||
if not is_empty_result: # 只有在不是空结果时才返回成功
|
||||
query_status = (
|
||||
"第一次查询成功" if current_retry == 0 else f"第{current_retry+1}次查询成功(RAG重写后)"
|
||||
)
|
||||
print(f"\n{query_status}")
|
||||
return {
|
||||
"code": 0,
|
||||
"message": "成功",
|
||||
"data": {
|
||||
"value": json.dumps(attrs, ensure_ascii=False, indent=2),
|
||||
"code": original_code,
|
||||
},
|
||||
"query_status": query_status,
|
||||
}
|
||||
|
||||
# 如果没有对象属性但有清理后的输出，且不是空结果
|
||||
if (
|
||||
clean_output
|
||||
and clean_output.lower() != "none"
|
||||
and clean_output != "[]"
|
||||
and "未找到" not in clean_output
|
||||
and not is_empty_result
|
||||
):
|
||||
query_status = (
|
||||
"第一次查询成功" if current_retry == 0 else f"第{current_retry+1}次查询成功(RAG重写后)"
|
||||
)
|
||||
print(f"\n{query_status}")
|
||||
return {
|
||||
"code": 0,
|
||||
"message": "成功",
|
||||
"data": {"value": clean_output, "code": original_code},
|
||||
"query_status": query_status,
|
||||
}
|
||||
|
||||
finally:
|
||||
# 恢复stdout
|
||||
sys.stdout = old_stdout
|
||||
|
||||
except Exception as e:
|
||||
import traceback
|
||||
|
||||
error_details = traceback.format_exc()
|
||||
print(f"\n执行代码时出错: {error_details}")
|
||||
|
||||
# 如果走到这里，说明结果为空或未找到匹配项，应该执行RAG重写流程
|
||||
print("\n查询未找到结果，尝试定位具体缺失节点...")
|
||||
|
||||
# 解析原始查询路径中的最后一个节点名
|
||||
import re
|
||||
|
||||
match = re.search(r"【([^】]+)】\s*$", original_query)
|
||||
missing_node = match.group(1) if match else "未知节点"
|
||||
|
||||
error_info = {
|
||||
"error_type": "NodeNotFoundError",
|
||||
"error_message": f"{missing_node} 未找到，请检查该节点是否存在。",
|
||||
"missing_node": missing_node,
|
||||
"original_query": original_query,
|
||||
"executed_code": original_code,
|
||||
}
|
||||
|
||||
print("结构化错误信息：")
|
||||
print(json.dumps(error_info, ensure_ascii=False, indent=2))
|
||||
|
||||
if current_retry < max_retries:
|
||||
print("\n尝试使用RAG重写查询...")
|
||||
try:
|
||||
# 使用RAG重写查询和代码，并传递错误信息
|
||||
rewritten = rewrite_query_parameters(original_query, original_code, error_info)
|
||||
|
||||
print(f"\nRAG重写结果: {json.dumps(rewritten, ensure_ascii=False, indent=2)}")
|
||||
|
||||
# 更新查询和代码
|
||||
if "query" in rewritten and "code" in rewritten and rewritten["code"] != original_code:
|
||||
print("\nRAG重写成功，使用新代码重试...")
|
||||
input_value = rewritten["code"] # 直接使用重写后的代码
|
||||
input_type = "code" # 切换到代码模式
|
||||
current_retry += 1
|
||||
continue # 继续下一次循环
|
||||
else:
|
||||
print("\nRAG重写未产生新代码，返回原始错误")
|
||||
except Exception as e:
|
||||
print(f"\nRAG重写失败: {e}")
|
||||
# 记录错误但继续执行
|
||||
|
||||
# RAG重写失败或未产生新代码，返回原始错误
|
||||
query_status = (
|
||||
"第一次查询失败，RAG重写也失败"
|
||||
if current_retry == 0
|
||||
else f"第{current_retry+1}次查询失败，RAG重写也失败"
|
||||
)
|
||||
print(f"\n{query_status}")
|
||||
return {
|
||||
"code": 1,
|
||||
"message": f"{missing_node} 未找到，请检查该节点是否存在。",
|
||||
"data": {"value": "", "code": original_code},
|
||||
"error_info": error_info,
|
||||
"query_status": query_status,
|
||||
}
|
||||
|
||||
# 如果所有重试都失败
|
||||
print("\n所有重试都失败，无法找到匹配的结果")
|
||||
query_status = "所有重试都失败"
|
||||
return {
|
||||
"code": 1,
|
||||
"message": "所有重试都失败，无法找到匹配的结果",
|
||||
"data": {"value": "", "code": original_code},
|
||||
"query_status": query_status,
|
||||
}
|
||||
|
||||
|
||||
def format_result(result):
|
||||
"""
|
||||
格式化查询结果
|
||||
|
||||
Args:
|
||||
result: 查询结果（可能为 list、dict 或其他类型）
|
||||
|
||||
Returns:
|
||||
str: 格式化后的结果
|
||||
"""
|
||||
# 处理 project 对象
|
||||
if hasattr(result, "__module__") and result.__module__ == "project":
|
||||
# 这是一个 project 模块中的对象
|
||||
attrs = {k: v for k, v in result.__dict__.items() if not k.startswith("_")}
|
||||
return json.dumps(attrs, ensure_ascii=False, indent=2)
|
||||
|
||||
# 处理 project 对象列表
|
||||
if isinstance(result, list) and all(
|
||||
hasattr(item, "__module__") and item.__module__ == "project" for item in result if hasattr(item, "__module__")
|
||||
):
|
||||
formatted_items = []
|
||||
for item in result:
|
||||
if hasattr(item, "__dict__"):
|
||||
attrs = {k: v for k, v in item.__dict__.items() if not k.startswith("_")}
|
||||
formatted_items.append(attrs)
|
||||
else:
|
||||
formatted_items.append(str(item))
|
||||
|
||||
return json.dumps(formatted_items, ensure_ascii=False, indent=2)
|
||||
|
||||
# 如果结果是字符串，可能包含调试信息，需要提取有用部分
|
||||
if isinstance(result, str):
|
||||
# 尝试提取最终结果部分
|
||||
if "[]" in result:
|
||||
return "未找到匹配的数据。"
|
||||
|
||||
# 如果包含节点信息，提取关键部分
|
||||
import re
|
||||
|
||||
node_match = re.search(r"找到.*?labels=.*?properties=(.*?)>", result)
|
||||
if node_match:
|
||||
try:
|
||||
# 提取属性部分并格式化
|
||||
props_str = node_match.group(1).replace("'", '"')
|
||||
import ast
|
||||
|
||||
props = ast.literal_eval(props_str)
|
||||
|
||||
formatted = "找到节点:\n"
|
||||
for k, v in props.items():
|
||||
formatted += f" {k}: {v}\n"
|
||||
return formatted
|
||||
except:
|
||||
pass
|
||||
|
||||
# 如果包含查询结果数量
|
||||
count_match = re.search(r"查询结果数量: (\d+)", result)
|
||||
if count_match:
|
||||
count = count_match.group(1)
|
||||
if count == "0":
|
||||
return "未找到匹配的数据。"
|
||||
|
||||
# 如果是列表
|
||||
if isinstance(result, list):
|
||||
if not result:
|
||||
return "未找到匹配的数据。"
|
||||
|
||||
lines = [f"找到 {len(result)} 条匹配结果:"]
|
||||
for i, item in enumerate(result, 1):
|
||||
lines.append(f"\n结果 {i}:")
|
||||
if hasattr(item, "items"): # 检查是否有items方法（字典或类似字典的对象）
|
||||
try:
|
||||
for k, v in item.items():
|
||||
lines.append(f" {k}: {v}")
|
||||
except:
|
||||
lines.append(f" {item}")
|
||||
else:
|
||||
lines.append(f" {item}")
|
||||
return "\n".join(lines)
|
||||
|
||||
# 如果是字典
|
||||
elif isinstance(result, dict):
|
||||
lines = ["查询结果:"]
|
||||
for k, v in result.items():
|
||||
lines.append(f" {k}: {v}")
|
||||
return "\n".join(lines)
|
||||
|
||||
# 其他类型
|
||||
else:
|
||||
return str(result)
|
||||
|
||||
|
||||
def format_dict_or_item(item):
|
||||
"""
|
||||
格式化字典或其他对象
|
||||
|
||||
Args:
|
||||
item: 字典或其他对象
|
||||
|
||||
Returns:
|
||||
str: 格式化后的字符串
|
||||
"""
|
||||
if isinstance(item, dict):
|
||||
formatted = ""
|
||||
for key, value in item.items():
|
||||
formatted += f" {key}: {value}\n"
|
||||
return formatted
|
||||
|
||||
return str(item)
|
||||
|
||||
|
||||
question = {
|
||||
"type": "query",
|
||||
"value": "查找一下【工程数据/安装工程/安装/架空输电线路本体工程/杆塔工程/杆塔组立/铁塔、钢管杆组立】的类型为【主材】的【阿巴阿巴】",
|
||||
}
|
||||
result = nl_query_to_function_call(question)
|
||||
print(result)
|
||||
@@ -0,0 +1,111 @@
|
||||
import requests
|
||||
from typing import Any, Dict, List, Optional
|
||||
from langchain_core.language_models import BaseLLM
|
||||
from langchain_core.callbacks import CallbackManagerForLLMRun
|
||||
from langchain_core.outputs import LLMResult, Generation
|
||||
from openai import OpenAI
|
||||
import httpx
|
||||
import logging
|
||||
|
||||
|
||||
# class OpenAiLLM:
|
||||
|
||||
# def __init__(self, url, api_key, model_name):
|
||||
# self._api_key = api_key
|
||||
# self._url = url
|
||||
# self._model = model_name
|
||||
|
||||
# def generate(self, prompt):
|
||||
|
||||
# client = OpenAI(api_key=self._api_key, base_url=self._url)
|
||||
|
||||
# try:
|
||||
# # 创建 Completion 请求
|
||||
# completion = client.chat.completions.create(
|
||||
# model=self._model,
|
||||
# messages=[
|
||||
# {"role": "system", "content": "You are a helpful assistant"},
|
||||
# {"role": "user", "content": prompt}
|
||||
# ],
|
||||
# timeout=httpx.Timeout(300.0),
|
||||
# temperature=0.7,
|
||||
# )
|
||||
# return completion.choices[0].message.content
|
||||
# except Exception as e:
|
||||
# logging.error(f"LLM调用出错: {e}")
|
||||
# return f"模型调用失败: {str(e)}"
|
||||
|
||||
|
||||
# llm = OpenAiLLM(
|
||||
# url="http://172.20.0.145:9995/v1",
|
||||
# api_key="xxx",
|
||||
# model_name="deepseek-r1-distill-qwen2.5-32b",
|
||||
# )
|
||||
|
||||
|
||||
class Embedding:
|
||||
def __init__(self, url, api_key, model_name):
|
||||
self._api_key = api_key
|
||||
self._url = url
|
||||
self._model = model_name
|
||||
|
||||
def embed(self, text):
|
||||
# 使用OpenAI客户端
|
||||
client = OpenAI(api_key=self._api_key, base_url=self._url)
|
||||
|
||||
try:
|
||||
# 调用embeddings API
|
||||
response = client.embeddings.create(model=self._model, input=text, timeout=httpx.Timeout(60.0))
|
||||
# 返回嵌入向量
|
||||
return response.data[0].embedding
|
||||
except Exception as e:
|
||||
logging.error(f"嵌入模型调用出错: {e}")
|
||||
raise RuntimeError(f"嵌入请求失败: {str(e)}")
|
||||
|
||||
|
||||
embedding = Embedding(url="http://10.1.16.39:9995/v1", api_key="xxx", model_name="bge-m3")
|
||||
|
||||
|
||||
class SiliconFlowLLM(BaseLLM):
|
||||
"""自定义硅基流动大模型调用类"""
|
||||
|
||||
api_url: str
|
||||
api_key: str
|
||||
model: str
|
||||
|
||||
def _generate(
|
||||
self,
|
||||
prompts: List[str],
|
||||
stop: Optional[List[str]] = None,
|
||||
run_manager: Optional[CallbackManagerForLLMRun] = None,
|
||||
**kwargs: Any,
|
||||
) -> LLMResult:
|
||||
from langchain_core.outputs import Generation, LLMResult
|
||||
|
||||
generations = []
|
||||
for prompt in prompts:
|
||||
try:
|
||||
headers = {"Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json"}
|
||||
payload = {
|
||||
"model": self.model,
|
||||
"messages": [{"role": "user", "content": prompt}],
|
||||
}
|
||||
response = requests.post(self.api_url, json=payload, headers=headers)
|
||||
response.raise_for_status()
|
||||
text = response.json()["choices"][0]["message"]["content"]
|
||||
generations.append([Generation(text=text)])
|
||||
except Exception as e:
|
||||
raise Exception(f"调用硅基流动API失败: {str(e)}")
|
||||
|
||||
return LLMResult(generations=generations)
|
||||
|
||||
@property
|
||||
def _llm_type(self) -> str:
|
||||
return "siliconflow"
|
||||
|
||||
|
||||
llm = SiliconFlowLLM(
|
||||
api_url="https://api.siliconflow.cn/v1/chat/completions",
|
||||
api_key="sk-bbeamiumkouptsrueilgufqqyuumelcsivxwjbdugqwsqhwj",
|
||||
model="Qwen/Qwen2.5-72B-Instruct",
|
||||
)
|
||||
@@ -0,0 +1,46 @@
|
||||
from langchain_neo4j import nl_query_to_function_call
|
||||
|
||||
|
||||
def main():
|
||||
|
||||
input_str = "杆塔总基数"
|
||||
|
||||
import json
|
||||
|
||||
# 打开并读取 JSON 文件
|
||||
with open("./data/data.json", "r", encoding="utf-8") as f:
|
||||
json_data = json.load(f)
|
||||
|
||||
temp_result = None
|
||||
for item in json_data:
|
||||
if item["指标名称"] == input_str:
|
||||
temp_result = item
|
||||
break
|
||||
|
||||
if temp_result:
|
||||
if temp_result["code"] == "":
|
||||
question = {
|
||||
"type": "query",
|
||||
"value": temp_result["指标描述"],
|
||||
}
|
||||
result = nl_query_to_function_call(question)
|
||||
|
||||
if result["message"] == "成功":
|
||||
code_result = result["data"]["code"]
|
||||
temp_result["code"] = code_result
|
||||
with open("./data/data.json", "w", encoding="utf-8") as f:
|
||||
json.dump(json_data, f, ensure_ascii=False, indent=4)
|
||||
print(f"已更新 code 字段")
|
||||
else:
|
||||
question = {
|
||||
"type": "code",
|
||||
"value": temp_result["code"],
|
||||
}
|
||||
result = nl_query_to_function_call(question)
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
# 示例使用
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,383 @@
|
||||
import json
|
||||
import ast
|
||||
import inspect
|
||||
|
||||
from typing import Dict, Any, List, Optional, Union
|
||||
from langchain.agents import Tool, initialize_agent, AgentType
|
||||
from langchain.chat_models import ChatOpenAI
|
||||
from langchain.prompts import PromptTemplate
|
||||
from langchain.schema import BaseOutputParser
|
||||
from langchain.schema import BaseRetriever
|
||||
from langchain.callbacks.manager import CallbackManagerForRetrieverRun
|
||||
from langchain.schema import Document
|
||||
from langchain.chains import LLMChain
|
||||
from langchain.llms.base import BaseLLM
|
||||
|
||||
from prompt_templates import FUNCTION_RETURNS_LOOP_PROMPT
|
||||
from llm import llm as base_llm
|
||||
import project # 明确导入project模块
|
||||
|
||||
|
||||
class CodeOnlyOutputParser(BaseOutputParser):
|
||||
def parse(self, text: str) -> dict:
|
||||
if "Final Answer:" in text:
|
||||
code_part = text.split("Final Answer:")[-1].strip()
|
||||
else:
|
||||
code_part = text.strip()
|
||||
|
||||
return {"code": code_part}
|
||||
|
||||
def get_format_instructions(self) -> str:
|
||||
return "只输出最终的Python代码，不要包含其他解释或内容。"
|
||||
|
||||
|
||||
class KnowledgeGraphProcessor:
|
||||
"""知识图谱处理器"""
|
||||
|
||||
def __init__(self, kg_file="kg_simple_hierarchy.json"):
|
||||
"""初始化处理器"""
|
||||
self.kg_data = self._load_kg_data(kg_file)
|
||||
self.prompt = FUNCTION_RETURNS_LOOP_PROMPT
|
||||
|
||||
def _load_kg_data(self, file_path: str) -> Dict[str, Any]:
|
||||
"""加载知识图谱数据"""
|
||||
try:
|
||||
with open(file_path, "r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
except FileNotFoundError:
|
||||
print(f"警告: 未找到 {file_path} 文件")
|
||||
return {}
|
||||
except Exception as e:
|
||||
print(f"加载知识图谱数据时出错: {e}")
|
||||
return {}
|
||||
|
||||
def _extract_parameters_from_code(self, code: str) -> List[str]:
|
||||
"""从代码中提取字符串参数"""
|
||||
try:
|
||||
tree = ast.parse(code)
|
||||
parameters = []
|
||||
|
||||
for node in ast.walk(tree):
|
||||
if isinstance(node, ast.Str):
|
||||
parameters.append(node.s)
|
||||
elif isinstance(node, ast.Constant) and isinstance(node.value, str):
|
||||
parameters.append(node.value)
|
||||
|
||||
return parameters
|
||||
except Exception as e:
|
||||
print(f"解析代码时出错: {e}")
|
||||
return []
|
||||
|
||||
def _find_node_by_path(self, path: str, data: Union[Dict[str, Any], List[Any]] = None) -> Optional[Dict[str, Any]]:
|
||||
"""
|
||||
根据路径查找节点
|
||||
|
||||
Args:
|
||||
path: 节点路径，格式如 "工程数据/安装过程/安装/架空输电线路本体工程/杆塔工程"
|
||||
data: 当前数据节点
|
||||
|
||||
Returns:
|
||||
找到的节点及其子节点
|
||||
"""
|
||||
if data is None:
|
||||
data = self.kg_data
|
||||
|
||||
path_parts = path.split("/")
|
||||
current_part = path_parts[0]
|
||||
|
||||
# 如果data是列表，则遍历列表中的每个元素
|
||||
if isinstance(data, list):
|
||||
for item in data:
|
||||
result = self._find_node_by_path(path, item)
|
||||
if result:
|
||||
return result
|
||||
return None
|
||||
|
||||
# 如果data是字典，则按原来的逻辑处理
|
||||
if isinstance(data, dict):
|
||||
# 检查当前节点是否匹配
|
||||
found_node = None
|
||||
for key, value in data.items():
|
||||
if isinstance(value, str) and current_part in value:
|
||||
found_node = data
|
||||
break
|
||||
|
||||
# 如果在当前层级找到了节点
|
||||
if found_node:
|
||||
# 如果还有更深的路径部分，则继续递归查找
|
||||
if len(path_parts) > 1:
|
||||
# 查找子节点
|
||||
if "children" in found_node:
|
||||
for child in found_node["children"]:
|
||||
result = self._find_node_by_path("/".join(path_parts[1:]), child)
|
||||
if result:
|
||||
return result
|
||||
return None
|
||||
else:
|
||||
# 已经找到最终节点
|
||||
return found_node
|
||||
|
||||
# 如果当前层级没找到，则查找子节点
|
||||
if "children" in data:
|
||||
for child in data["children"]:
|
||||
result = self._find_node_by_path(path, child)
|
||||
if result:
|
||||
return result
|
||||
|
||||
return None
|
||||
|
||||
def _search_in_kg(self, parameter: str, data: Union[Dict[str, Any], List[Any]] = None) -> Optional[Dict[str, Any]]:
|
||||
"""在知识图谱中递归搜索参数"""
|
||||
# 先尝试按路径查找
|
||||
if "/" in parameter:
|
||||
result = self._find_node_by_path(parameter)
|
||||
if result:
|
||||
return result
|
||||
|
||||
# 如果路径查找失败，则按关键词搜索
|
||||
if data is None:
|
||||
data = self.kg_data
|
||||
|
||||
# 如果是列表，遍历每个元素
|
||||
if isinstance(data, list):
|
||||
for item in data:
|
||||
if isinstance(item, (dict, list)):
|
||||
result = self._search_in_kg(parameter, item)
|
||||
if result:
|
||||
return result
|
||||
return None
|
||||
|
||||
# 如果是字典，按原来的逻辑处理
|
||||
if isinstance(data, dict):
|
||||
for key, value in data.items():
|
||||
if isinstance(value, str) and parameter in value:
|
||||
return data
|
||||
elif isinstance(value, list):
|
||||
for item in value:
|
||||
if isinstance(item, (dict, list)):
|
||||
result = self._search_in_kg(parameter, item)
|
||||
if result:
|
||||
return result
|
||||
elif isinstance(value, dict):
|
||||
result = self._search_in_kg(parameter, value)
|
||||
if result:
|
||||
return result
|
||||
|
||||
return None
|
||||
|
||||
def _get_node_definition(self, node_type: str) -> str:
|
||||
"""获取节点类型定义"""
|
||||
try:
|
||||
# 检查project模块中是否存在该类
|
||||
if hasattr(project, node_type):
|
||||
cls = getattr(project, node_type)
|
||||
source = inspect.getsource(cls)
|
||||
return source
|
||||
else:
|
||||
return f"未找到类型 {node_type} 的定义"
|
||||
except NameError:
|
||||
return f"获取类型 {node_type} 定义时出错: project模块未正确导入"
|
||||
except Exception as e:
|
||||
return f"获取类型 {node_type} 定义时出错: {e}"
|
||||
|
||||
def _extract_node_types(self, data: Dict[str, Any], node_types: set):
|
||||
"""递归提取所有节点类型"""
|
||||
if not isinstance(data, dict):
|
||||
return
|
||||
|
||||
for key, value in data.items():
|
||||
# 所有键都可能是节点类型
|
||||
node_types.add(key)
|
||||
|
||||
if isinstance(value, dict):
|
||||
self._extract_node_types(value, node_types)
|
||||
elif isinstance(value, list):
|
||||
for item in value:
|
||||
if isinstance(item, dict):
|
||||
self._extract_node_types(item, node_types)
|
||||
|
||||
def _get_relevant_knowledge(self, code: str) -> tuple[str, str]:
|
||||
"""根据代码获取相关知识库内容和节点定义"""
|
||||
parameters = self._extract_parameters_from_code(code)
|
||||
|
||||
knowledge_base = ""
|
||||
node_definitions = ""
|
||||
|
||||
for param in parameters:
|
||||
# 处理【】格式的路径，提取最后一个部分
|
||||
if "【" in param:
|
||||
path_parts = []
|
||||
parts = param.split("【")
|
||||
for part in parts:
|
||||
if "】" in part:
|
||||
clean_part = part.split("】")[0].strip()
|
||||
if clean_part:
|
||||
path_parts.append(clean_part)
|
||||
|
||||
# 对于每个路径，只取最后一个部分
|
||||
for path in path_parts:
|
||||
if "/" in path:
|
||||
# 提取路径中的最后一个部分
|
||||
last_part = path.split("/")[-1].strip()
|
||||
if last_part:
|
||||
found_data = self._search_in_kg(last_part)
|
||||
if found_data:
|
||||
knowledge_base += f"节点 '{last_part}' 相关信息:\n"
|
||||
knowledge_base += json.dumps(found_data, ensure_ascii=False, indent=2) + "\n\n"
|
||||
|
||||
# 提取节点类型
|
||||
node_types = set()
|
||||
self._extract_node_types(found_data, node_types)
|
||||
|
||||
for node_type in node_types:
|
||||
definition = self._get_node_definition(node_type)
|
||||
if definition and "未找到" not in definition and "出错" not in definition:
|
||||
node_definitions += f"类型 {node_type} 定义:\n{definition}\n\n"
|
||||
else:
|
||||
# 如果没有/，直接使用整个部分
|
||||
found_data = self._search_in_kg(path)
|
||||
if found_data:
|
||||
knowledge_base += f"节点 '{path}' 相关信息:\n"
|
||||
knowledge_base += json.dumps(found_data, ensure_ascii=False, indent=2) + "\n\n"
|
||||
|
||||
# 提取节点类型
|
||||
node_types = set()
|
||||
self._extract_node_types(found_data, node_types)
|
||||
|
||||
for node_type in node_types:
|
||||
definition = self._get_node_definition(node_type)
|
||||
if definition and "未找到" not in definition and "出错" not in definition:
|
||||
node_definitions += f"类型 {node_type} 定义:\n{definition}\n\n"
|
||||
else:
|
||||
# 处理普通参数
|
||||
found_data = self._search_in_kg(param)
|
||||
if found_data:
|
||||
knowledge_base += f"参数 '{param}' 相关信息:\n"
|
||||
knowledge_base += json.dumps(found_data, ensure_ascii=False, indent=2) + "\n\n"
|
||||
|
||||
# 提取节点类型
|
||||
node_types = set()
|
||||
self._extract_node_types(found_data, node_types)
|
||||
|
||||
for node_type in node_types:
|
||||
definition = self._get_node_definition(node_type)
|
||||
if definition and "未找到" not in definition and "出错" not in definition:
|
||||
node_definitions += f"类型 {node_type} 定义:\n{definition}\n\n"
|
||||
|
||||
return knowledge_base, node_definitions
|
||||
|
||||
def process_query(
|
||||
self, original_query: str, original_code: str, error_info: Dict[str, str] = None
|
||||
) -> Dict[str, str]:
|
||||
"""
|
||||
处理查询并尝试修复错误
|
||||
|
||||
Args:
|
||||
original_query: 原始查询字符串
|
||||
original_code: 原始代码字符串
|
||||
error_info: 错误信息字典 (可选)
|
||||
|
||||
Returns:
|
||||
Dict: 包含修改后的query和code的字典
|
||||
"""
|
||||
print("\n====== RAG重写查询开始 ======")
|
||||
print(f"原始查询: {original_query}")
|
||||
print(f"原始代码: {original_code}")
|
||||
print(f"错误信息: {error_info}")
|
||||
|
||||
# 初始化工具
|
||||
tools = [
|
||||
Tool(
|
||||
name="search_knowledge_base",
|
||||
func=lambda x: json.dumps(self._search_in_kg(x), ensure_ascii=False),
|
||||
description="用于在知识图谱中搜索节点信息",
|
||||
),
|
||||
Tool(
|
||||
name="get_node_definition",
|
||||
func=lambda x: self._get_node_definition(x),
|
||||
description="获取指定类型的节点定义",
|
||||
),
|
||||
]
|
||||
|
||||
# 使用基础 LLM
|
||||
llm = base_llm
|
||||
|
||||
# 初始化 Agent
|
||||
agent_executor = initialize_agent(
|
||||
tools,
|
||||
llm,
|
||||
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
||||
verbose=True,
|
||||
handle_parsing_errors=True,
|
||||
prompt=self.prompt,
|
||||
output_parser=CodeOnlyOutputParser(),
|
||||
)
|
||||
|
||||
# 构造输入变量
|
||||
knowledge_base, node_definitions = self._get_relevant_knowledge(original_code)
|
||||
|
||||
print(f"知识库内容: {knowledge_base}")
|
||||
print(f"节点定义: {node_definitions}")
|
||||
|
||||
input_vars = {
|
||||
"input": original_query,
|
||||
"original_query": original_query,
|
||||
"error_info": json.dumps(error_info, ensure_ascii=False) if error_info else "",
|
||||
"original_code": original_code,
|
||||
"KnowledgeBase": knowledge_base,
|
||||
"NodeDefinition": node_definitions,
|
||||
}
|
||||
|
||||
print(f"Agent输入变量: {input_vars}")
|
||||
|
||||
# 调用 Agent 执行
|
||||
response = agent_executor.invoke(input_vars)
|
||||
|
||||
print(f"Agent响应: {response}")
|
||||
|
||||
# 处理不同的返回格式
|
||||
if isinstance(response, dict):
|
||||
if "output" in response:
|
||||
if isinstance(response["output"], dict) and "code" in response["output"]:
|
||||
return {"query": original_query, "code": response["output"]["code"]}
|
||||
elif isinstance(response["output"], str):
|
||||
return {"query": original_query, "code": response["output"]}
|
||||
|
||||
# 如果无法解析，返回原始代码
|
||||
print(f"警告: 无法解析 Agent 响应，返回原始代码")
|
||||
return {"query": original_query, "code": original_code}
|
||||
|
||||
|
||||
# 全局处理器实例（延迟初始化）
|
||||
_processor = None
|
||||
|
||||
|
||||
def _get_processor():
|
||||
"""获取处理器实例（单例模式）"""
|
||||
global _processor
|
||||
if _processor is None:
|
||||
_processor = KnowledgeGraphProcessor()
|
||||
return _processor
|
||||
|
||||
|
||||
def rewrite_query_parameters(
|
||||
original_query: str, original_code: str, error_info: Dict[str, str] = None
|
||||
) -> Dict[str, str]:
|
||||
"""
|
||||
重写查询参数的外部接口函数
|
||||
|
||||
Args:
|
||||
original_query: 原始查询字符串
|
||||
original_code: 原始代码字符串
|
||||
error_info: 错误信息字典 (可选)
|
||||
|
||||
Returns:
|
||||
Dict: 包含修改后的query和code的字典
|
||||
|
||||
Example:
|
||||
result = rewrite_query_parameters("查询动态费用", 'search_node("动态费用")')
|
||||
print(result["query"]) # 修改后的查询
|
||||
print(result["code"]) # 修改后的代码
|
||||
"""
|
||||
processor = _get_processor()
|
||||
return processor.process_query(original_query, original_code, error_info)
|
||||
+613
@@ -0,0 +1,613 @@
|
||||
"""
|
||||
软件知识图谱类定义
|
||||
根据Ontology_Layer.txt文件中的知识图谱信息创建
|
||||
"""
|
||||
|
||||
|
||||
class ProjectTookiIt:
|
||||
"""
|
||||
项目类
|
||||
描述: 代表整个项目结构的顶层容器，包含项目划分集
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.project_division_set = ProjectDivisionSet() # 项目划分集对象
|
||||
|
||||
def get_division_item_by_path(self, path):
|
||||
"""
|
||||
通过路径获取项目划分对象
|
||||
|
||||
Args:
|
||||
path (str): 以'/'分隔的多级项目划分名称路径
|
||||
|
||||
Returns:
|
||||
ProjectDivisionItem|None: 对应的项目划分对象，如果路径不存在返回None，成功找到返回项目划分对象
|
||||
|
||||
Note:
|
||||
当路径为空字符串时，会返回None
|
||||
"""
|
||||
names = path.split("/")
|
||||
if not names:
|
||||
raise ValueError("路径不能为空")
|
||||
|
||||
try:
|
||||
current = self.project_division_set.get_division_item(names[0])
|
||||
except KeyError:
|
||||
return None
|
||||
|
||||
for name in names[1:]:
|
||||
found = False
|
||||
for child in current.children:
|
||||
if child.项目名称 == name:
|
||||
current = child
|
||||
found = True
|
||||
break
|
||||
if not found:
|
||||
return None
|
||||
return current
|
||||
|
||||
def get_quantities_by_paths(self, paths_str, quantity_type=None, keyword=None):
|
||||
"""
|
||||
获取多个项目路径下的工程量对象
|
||||
|
||||
Args:
|
||||
paths_str (str): 以'@@'分隔的多个项目路径字符串，每个路径以'/'分隔层级
|
||||
quantity_type (str): 工程量类型('定额'、'主材'、'设备'或None表示所有类型)
|
||||
keyword (str): 工程量代码或名称中的关键字(可选)
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的工程量对象的列表
|
||||
"""
|
||||
paths = paths_str.split("@@") if paths_str else []
|
||||
results = []
|
||||
|
||||
for path in paths:
|
||||
item = self.get_division_item_by_path(path.strip())
|
||||
if item is None:
|
||||
continue
|
||||
|
||||
def collect_quantities(node):
|
||||
for quantity in node.get_quantities():
|
||||
# 检查类型匹配
|
||||
type_matched = True
|
||||
if quantity_type:
|
||||
if quantity_type == "定额" and not isinstance(quantity, Ration):
|
||||
type_matched = False
|
||||
elif quantity_type == "主材" and not isinstance(quantity, Material):
|
||||
type_matched = False
|
||||
elif quantity_type == "设备" and not isinstance(quantity, Equipment):
|
||||
type_matched = False
|
||||
|
||||
# 检查关键字匹配
|
||||
keyword_matched = True
|
||||
if keyword:
|
||||
keyword_matched = (keyword in (quantity.编码 or "")) or (keyword in (quantity.name or ""))
|
||||
|
||||
if type_matched and keyword_matched:
|
||||
results.append(quantity)
|
||||
|
||||
for child in node.children:
|
||||
collect_quantities(child)
|
||||
|
||||
collect_quantities(item)
|
||||
|
||||
return results
|
||||
|
||||
|
||||
class ProjectDivisionSet:
|
||||
"""
|
||||
项目划分集
|
||||
描述: 代表projectDivision下的第一层键和第二层建，例如"安装工程"、"安装"
|
||||
JSON对应: projectDivision下的直接键名，如"安装工程": {...}、"安装": [...]
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.category_name = None # xsd:string
|
||||
self.division_items = {} # 存储项目划分项的字典，键为项目名称
|
||||
|
||||
def get_division_item(self, item_name):
|
||||
"""
|
||||
Get project division item by name
|
||||
|
||||
Args:
|
||||
item_name (str): The name of item to find
|
||||
|
||||
Returns:
|
||||
ProjectDivisionItem: The corresponding division item object
|
||||
|
||||
Raises:
|
||||
KeyError: When item name does not exist
|
||||
"""
|
||||
if item_name not in self.division_items:
|
||||
raise KeyError(f'Item name "{item_name}" does not exist')
|
||||
return self.division_items[item_name]
|
||||
|
||||
# 工程属性查询方法
|
||||
@abstractmethod
|
||||
def get_project_property_by_name(self, name):
|
||||
"""
|
||||
通过名称获取工程属性
|
||||
|
||||
Args:
|
||||
name (str): 工程属性名称
|
||||
|
||||
Returns:
|
||||
str|None: 对应的工程属性值，如果不存在返回None
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_project_property_by_parent_and_name(self, parent_path, partial_name):
|
||||
"""
|
||||
通过父节点路径和模糊名称获取工程属性
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
|
||||
Returns:
|
||||
dict: 包含所有匹配的工程属性的字典，键为属性名，值为属性值
|
||||
"""
|
||||
pass
|
||||
|
||||
|
||||
class ProjectDivisionItem:
|
||||
"""
|
||||
项目划分项
|
||||
描述: 代表项目结构中的层级条目，具有自身的属性，并且可以包含子项目划分项或详细工作项
|
||||
JSON对应: ProjectDivisionSet数组中的对象，或ProjectDivisionItem的children数组中type为"项目划分"的对象
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.GUID = None # xsd:string
|
||||
self.id = None # xsd:string
|
||||
self.name = None # xsd:string
|
||||
self.代码 = None # xsd:string (可选)
|
||||
self.费率 = None # xsd:string
|
||||
self.单位 = None # xsd:string (可选)
|
||||
self.取费表id = None # xsd:string
|
||||
self.颜色标记 = None # xsd:string
|
||||
self.取费表 = None # xsd:string
|
||||
self.合价含税 = None # xsd:string (可选)
|
||||
self.type = None # xsd:string
|
||||
self.专业类型 = None # xsd:string
|
||||
self.资源库列表 = None # xsd:list (可选)
|
||||
self.notCheck = None # xsd:string (可选)
|
||||
self.children = [] # 存储子项目划分项的数组
|
||||
self.quantities = [] # 存储工程量对象的数组
|
||||
self.cost_set = None # xsd:CostSet
|
||||
|
||||
def get_cost_set(self):
|
||||
"""
|
||||
获取费用集对象
|
||||
|
||||
Returns:
|
||||
CostSet: 关联的费用集对象
|
||||
"""
|
||||
return self.cost_set
|
||||
|
||||
def add_child(self, child_item):
|
||||
"""
|
||||
添加子项目划分项
|
||||
|
||||
Args:
|
||||
child_item (ProjectDivisionItem): 子项目对象
|
||||
|
||||
Raises:
|
||||
ValueError: 当child_item不是ProjectDivisionItem类型时
|
||||
"""
|
||||
if not isinstance(child_item, ProjectDivisionItem):
|
||||
raise ValueError("child_item必须是ProjectDivisionItem类型")
|
||||
self.children.append(child_item)
|
||||
|
||||
def get_child(self, index):
|
||||
"""
|
||||
根据索引获取子项目划分项
|
||||
|
||||
Args:
|
||||
index (int): 子项目索引
|
||||
|
||||
Returns:
|
||||
ProjectDivisionItem: 对应的子项目对象
|
||||
|
||||
Raises:
|
||||
IndexError: 当索引超出范围时
|
||||
"""
|
||||
if index < 0 or index >= len(self.children):
|
||||
raise IndexError(f'索引"{index}"超出范围')
|
||||
return self.children[index]
|
||||
|
||||
def traverse(self, callback, depth=0):
|
||||
"""
|
||||
遍历树形结构
|
||||
|
||||
Args:
|
||||
callback (function): 回调函数，接收当前节点和深度作为参数
|
||||
depth (int): 当前深度，默认为0
|
||||
"""
|
||||
callback(self, depth)
|
||||
for child in self.children:
|
||||
child.traverse(callback, depth + 1)
|
||||
|
||||
def get_quantities(self):
|
||||
"""
|
||||
获取当前节点的ProjectQuantity及其子类(Ration, Material, Equipment)的实例
|
||||
|
||||
Returns:
|
||||
list: 包含当前节点ProjectQuantity实例的列表
|
||||
"""
|
||||
return self.quantities
|
||||
|
||||
|
||||
class ProjectQuantity:
|
||||
"""
|
||||
工程量
|
||||
描述: 代表项目划分项(ProjectDivisionItem)下的具体工作单元或物料项，是定额、主材、设备的父类型
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.id = None # xsd:string
|
||||
self.类型 = None # xsd:string ("0"为定额,"1"为主材,"5"为设备)
|
||||
self.name = None # xsd:string
|
||||
self.编码 = None # xsd:string
|
||||
self.单位 = None # xsd:string
|
||||
self.数量 = None # xsd:string
|
||||
self.资源库名称 = None # xsd:string
|
||||
self.投标数量 = None # xsd:string (可选)
|
||||
self.投标单价 = None # xsd:string (可选)
|
||||
self.特征段 = None # xsd:string (可选)
|
||||
self.关联父级量 = None # xsd:string (可选)
|
||||
self.颜色标记 = None # xsd:string (可选)
|
||||
self.单价不含税 = None
|
||||
self.cost_set = None # xsd:CostSet
|
||||
|
||||
def get_cost_set(self):
|
||||
"""
|
||||
获取费用集对象
|
||||
|
||||
Returns:
|
||||
CostSet: 关联的费用集对象
|
||||
"""
|
||||
return self.cost_set # xsd:string (可选)
|
||||
|
||||
|
||||
class Ration(ProjectQuantity):
|
||||
"""
|
||||
定额
|
||||
描述: 代表一种标准的工程量条目，通常包含详细的工、料、机消耗标准
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.计算式 = None # xsd:string
|
||||
self.中标计算式 = None # xsd:string
|
||||
self.人工费 = None # xsd:string
|
||||
self.机械费 = None # xsd:string
|
||||
self.甲供材料费不含税 = None # xsd:string
|
||||
self.材料费 = None # xsd:string
|
||||
self.定额系数 = None # xsd:string
|
||||
self.人工系数 = None # xsd:string
|
||||
self.材料系数 = None # xsd:string
|
||||
self.机械系数 = None # xsd:string
|
||||
self.定额范围 = None # xsd:string
|
||||
self.定额章节名称 = None # xsd:string
|
||||
self.费用类型 = None # xsd:string
|
||||
self.甲供材料费含税 = None # xsd:string
|
||||
self.投标合价 = None # xsd:string
|
||||
self.其中甲供材料费 = None # xsd:string
|
||||
self.合价不含税 = None # xsd:string
|
||||
self.基价 = None # xsd:string
|
||||
self.所属定额库 = None # xsd:string
|
||||
|
||||
|
||||
class Material(ProjectQuantity):
|
||||
"""
|
||||
主材
|
||||
描述: 代表工程中使用的主要材料
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.规格型号 = None # xsd:string
|
||||
self.损耗率 = None # xsd:string
|
||||
self.供货方 = None # xsd:string
|
||||
self.集中配送 = None # xsd:string (可选)
|
||||
self.单重 = None # xsd:string (可选)
|
||||
self.市场价不含税 = None # xsd:string
|
||||
self.市场价含税 = None # xsd:string
|
||||
self.单价含税 = None # xsd:string
|
||||
self.结算市场价不含税 = None # xsd:string (可选)
|
||||
self.结算市场价含税 = None # xsd:string (可选)
|
||||
self.基准价不含税 = None # xsd:string (可选)
|
||||
self.基准价含税 = None # xsd:string (可选)
|
||||
self.费用类型 = None # xsd:string
|
||||
self.增值税率 = None # xsd:string (可选)
|
||||
self.合价含税 = None # xsd:string
|
||||
self.合价不含税 = None # xsd:string
|
||||
self.线重 = None # xsd:string (可选)
|
||||
self.制造长度 = None # xsd:string (可选)
|
||||
self.截面积 = None # xsd:string (可选)
|
||||
|
||||
|
||||
class Equipment(ProjectQuantity):
|
||||
"""
|
||||
设备
|
||||
描述: 代表工程中安装或使用的设备
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.规格型号 = None # xsd:string
|
||||
self.供货方 = None # xsd:string
|
||||
self.运杂费率 = None # xsd:string (可选)
|
||||
self.单价含税 = None # xsd:string
|
||||
self.设备类型 = None # xsd:string (可选)
|
||||
self.增值税率 = None # xsd:string (可选)
|
||||
self.合价含税 = None # xsd:string
|
||||
self.合价不含税 = None # xsd:string
|
||||
|
||||
|
||||
class MaterialOrEquipment:
|
||||
"""
|
||||
材机
|
||||
描述: 代表DetailedWorkItem中所列出的具体材料、人工或机械设备及其详细信息
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.id = None # xsd:string
|
||||
self.编码 = None # xsd:string
|
||||
self.名称 = None # xsd:string
|
||||
self.单位 = None # xsd:string
|
||||
self.类型 = None # xsd:string
|
||||
self.供货方 = None # xsd:string
|
||||
self.预算价不含税 = None # xsd:string
|
||||
self.市场价不含税 = None # xsd:string
|
||||
self.预算价含税 = None # xsd:string
|
||||
self.市场价含税 = None # xsd:string
|
||||
self.结算预算价不含税 = None # xsd:string
|
||||
self.结算市场价不含税 = None # xsd:string
|
||||
self.结算预算价含税 = None # xsd:string
|
||||
self.结算市场价含税 = None # xsd:string
|
||||
self.暂估价 = None # xsd:string
|
||||
self.拆分 = None # xsd:string
|
||||
self.全口径市场价不含税 = None # xsd:string
|
||||
self.全口径市场价含税 = None # xsd:string
|
||||
self.商品砼 = None # xsd:string
|
||||
self.数量 = None # xsd:string
|
||||
self.是否未计价 = None # xsd:string
|
||||
|
||||
|
||||
class CostSet:
|
||||
"""
|
||||
费用集
|
||||
描述: 代表费用预览层级结构中的一个节点
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.GUID = None # xsd:string
|
||||
self.cost_items = {} # 存储费用项的字典，键为项目名称
|
||||
|
||||
def get_cost_item(self, item_name):
|
||||
"""
|
||||
根据名称获取费用项
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的费用项名称
|
||||
|
||||
Returns:
|
||||
CostItem: 对应的费用项对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当费用项名称不存在时
|
||||
"""
|
||||
if item_name not in self.cost_items:
|
||||
raise KeyError(f'费用项名称"{item_name}"不存在')
|
||||
return self.cost_items[item_name]
|
||||
|
||||
|
||||
class CostItem:
|
||||
"""
|
||||
费用项
|
||||
描述: 代表预览层级结构中的叶子节点，显示具体的费用项及其金额
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.id = None # xsd:string
|
||||
self.cost = None # xsd:string
|
||||
self.sub_items = {} # 存储子费用项的字典，键为项目名称
|
||||
|
||||
def get_sub_item(self, item_name):
|
||||
"""
|
||||
根据名称获取子费用项
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的子费用项名称
|
||||
|
||||
Returns:
|
||||
CostItem: 对应的子费用项对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当子费用项名称不存在时
|
||||
"""
|
||||
if item_name not in self.sub_items:
|
||||
raise KeyError(f'子费用项名称"{item_name}"不存在')
|
||||
return self.sub_items[item_name]
|
||||
|
||||
|
||||
class FeeTableTemplateSet:
|
||||
"""
|
||||
取费表模板集
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.feetable_template = {} # 存储费用项的字典，键为项目名称
|
||||
|
||||
def get_feetable_template_item(self, item_name):
|
||||
"""
|
||||
根据名称获取取费表模板项
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的取费表模板项名称
|
||||
|
||||
Returns:
|
||||
FeeTableTemplateItem: 对应的取费表模板项对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当取费表模板项名称不存在时
|
||||
"""
|
||||
if item_name not in self.feetable_template:
|
||||
raise KeyError(f'取费表模板项名称"{item_name}"不存在')
|
||||
return self.feetable_template[item_name]
|
||||
|
||||
|
||||
class FeeTableTemplateItem:
|
||||
"""
|
||||
取费表模板项
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.OutlayID = None # xsd:string
|
||||
self.type = None # xsd:string
|
||||
self.profession = None # xsd:string
|
||||
self.sub_feecollections = {} # 存储子费用项的字典，键为项目名称
|
||||
|
||||
def get_sub_fee(self, item_name):
|
||||
"""
|
||||
根据名称获取子取费
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的子取费名称
|
||||
|
||||
Returns:
|
||||
feecollections: 对应的子费用项对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当子费用项名称不存在时
|
||||
"""
|
||||
if item_name not in self.sub_feecollections:
|
||||
raise KeyError(f'取费名称"{item_name}"不存在')
|
||||
return self.sub_feecollections[item_name]
|
||||
|
||||
|
||||
class FeeCollection:
|
||||
"""
|
||||
取费
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.serialNumber = None # xsd:string
|
||||
self.code = None # xsd:string
|
||||
self.base = None # xsd:string
|
||||
self.rate = None # xsd:string
|
||||
self.sub_feecollections = {} # 存储子费用项的字典，键为项目名称
|
||||
|
||||
def get_sub_feecollections(self, item_name):
|
||||
"""
|
||||
根据名称获取子取费
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的子费用名称
|
||||
|
||||
Returns:
|
||||
feecollections: 对应的费用对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当子费用名称不存在时
|
||||
"""
|
||||
if item_name not in self.sub_feecollections:
|
||||
raise KeyError(f'取费名称"{item_name}"不存在')
|
||||
return self.sub_feecollections[item_name]
|
||||
|
||||
|
||||
class FeeScheduleSet:
|
||||
"""
|
||||
费用表集
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.feeschedule_template = {} # 存储费用项的字典，键为项目名称
|
||||
|
||||
def get_feetable_template_item(self, item_name):
|
||||
"""
|
||||
根据名称获取费用表项
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的费用表项名称
|
||||
|
||||
Returns:
|
||||
FeeScheduleItem: 对应的费用表项对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当费用表项名称不存在时
|
||||
"""
|
||||
if item_name not in self.feeschedule_template:
|
||||
raise KeyError(f'取费表模板项名称"{item_name}"不存在')
|
||||
return self.feeschedule_template[item_name]
|
||||
|
||||
|
||||
class FeeScheduleItem:
|
||||
"""
|
||||
费用表项
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.sub_fees = {} # 存储子费用项的字典，键为项目名称
|
||||
|
||||
def get_sub_fee(self, item_name):
|
||||
"""
|
||||
根据名称获取子费用
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的子费用名称
|
||||
|
||||
Returns:
|
||||
fees: 对应的子费用项对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当子费用项名称不存在时
|
||||
"""
|
||||
if item_name not in self.sub_fees:
|
||||
raise KeyError(f'费用名称"{item_name}"不存在')
|
||||
return self.sub_fees[item_name]
|
||||
|
||||
|
||||
class Fee:
|
||||
"""
|
||||
取费
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.serialNumber = None # xsd:string
|
||||
self.code = None # xsd:string
|
||||
self.rate = None # xsd:string
|
||||
self.amount = None # xsd:string
|
||||
self.sub_feecollections = {} # 存储子费用项的字典，键为项目名称
|
||||
|
||||
def get_sub_feecollections(self, item_name):
|
||||
"""
|
||||
根据名称获取子取费
|
||||
|
||||
Args:
|
||||
item_name (str): 要查找的子费用名称
|
||||
|
||||
Returns:
|
||||
feecollections: 对应的费用对象
|
||||
|
||||
Raises:
|
||||
KeyError: 当子费用名称不存在时
|
||||
"""
|
||||
if item_name not in self.sub_feecollections:
|
||||
raise KeyError(f'费用名称"{item_name}"不存在')
|
||||
return self.sub_feecollections[item_name]
|
||||
+455
@@ -0,0 +1,455 @@
|
||||
"""
|
||||
软件知识图谱类定义
|
||||
根据Ontology_Layer.txt文件中的知识图谱信息创建
|
||||
"""
|
||||
|
||||
from abc import ABC, abstractmethod
|
||||
import json
|
||||
|
||||
|
||||
class ProjectTookiIt(ABC):
|
||||
"""
|
||||
项目类(抽象基类)
|
||||
描述: 代表整个项目结构的顶层容器
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.project_division_set = ProjectDivisionItem() # 项目划分集对象
|
||||
|
||||
# 项目划分查询方法
|
||||
@abstractmethod
|
||||
def get_division_item_by_path(self, path):
|
||||
"""
|
||||
通过路径获取项目划分对象
|
||||
|
||||
Args:
|
||||
path (str): 以'/'分隔的多级项目划分名称路径
|
||||
|
||||
Returns:
|
||||
ProjectDivisionItem|None: 对应的项目划分对象，如果路径不存在返回None，成功找到返回项目划分对象
|
||||
|
||||
Note:
|
||||
当路径为空字符串时，会返回None
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_division_node_by_parent_and_name(self, parent_path, partial_name):
|
||||
"""
|
||||
通过父节点路径和模糊节点名称获取项目划分对象
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配节点的列表，如果没有匹配返回空列表
|
||||
"""
|
||||
pass
|
||||
|
||||
# 工程量查询方法
|
||||
@abstractmethod
|
||||
def get_quantities_by_paths(self, paths_str):
|
||||
"""
|
||||
获取指定项目路径下的工程量对象
|
||||
|
||||
Args:
|
||||
paths_str (str): 以'/'分隔的多级节点路径
|
||||
|
||||
|
||||
Returns:
|
||||
ProjectQuantity|None: 对应的工程量对象，如果路径不存在返回None，成功找到返回工程量对象
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_quantities_node_by_parent_and_code(self, parent_path, quantity_type=None, code=None):
|
||||
"""
|
||||
通过父节点路径和编码获取工程量对象（定额）
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
quantity_type (str): 工程量类型('定额'、'主材'、'设备'或None表示所有类型)
|
||||
code(str): 工程量编码，以'/'分隔的多个编码
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配节点的列表，如果没有匹配返回空列表
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_quantities_node_by_parent_and_name(self, parent_path, partial_name, quantity_type=None):
|
||||
"""
|
||||
通过父节点路径、模糊节点名称和类型获取工程量对象（主材或者设备）
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
quantity_type (str): 工程量类型('定额'、'主材'、'设备'或None表示所有类型)
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配节点的列表，如果没有匹配返回空列表
|
||||
"""
|
||||
pass
|
||||
|
||||
# 材机查询方法
|
||||
@abstractmethod
|
||||
def get_material_equipment_by_path(self, paths_str):
|
||||
"""
|
||||
通过路径获取材机对象
|
||||
|
||||
Args:
|
||||
path (str): 以'/'分隔的多级项目划分名称路径
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的材机对象的列表
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_material_equipment_by_parent_and_name(self, parent_path, partial_name):
|
||||
"""
|
||||
通过父节点路径和模糊名称获取材机对象
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的材机对象的列表
|
||||
"""
|
||||
pass
|
||||
|
||||
# 取费表模板查询方法
|
||||
@abstractmethod
|
||||
def get_fee_template_by_path(self, paths_str):
|
||||
"""
|
||||
通过路径获取取费表模板
|
||||
|
||||
Args:
|
||||
path (str): 以'/'分隔的多级项目划分名称路径
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的取费表模板对象的列表
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_fee_template_by_parent_and_name(self, parent_path, partial_name):
|
||||
"""
|
||||
通过父节点路径和模糊名称获取取费表模板
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的取费表模板对象的列表
|
||||
"""
|
||||
pass
|
||||
|
||||
# 费用表查询方法
|
||||
@abstractmethod
|
||||
def get_fee_schedule_on_auxiliary_expense_table(self, table_name, fee_name, fee: str):
|
||||
"""
|
||||
在辅助费用表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_fee_schedule_on_other_expense_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在其它费用表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_fee_schedule_on_land_acquisition_fee_table_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在其中：土地征用费表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_fee_schedule_on_installation_price_difference_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在安装价差表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_fee_schedule_on_Engineering_Cost_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在工程费用表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
pass
|
||||
|
||||
|
||||
class ProjectDivisionItem:
|
||||
"""
|
||||
项目划分项
|
||||
描述: 代表项目结构中的层级条目，具有自身的属性，并且可以包含子项目划分项或详细工作项
|
||||
JSON对应: ProjectDivisionSet数组中的对象，或ProjectDivisionItem的children数组中type为"项目划分"的对象
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.GUID = None # xsd:string
|
||||
self.id = None # xsd:string
|
||||
self.name = None # xsd:string
|
||||
self.代码 = None # xsd:string (可选)
|
||||
self.费率 = None # xsd:string
|
||||
self.单位 = None # xsd:string (可选)
|
||||
self.取费表id = None # xsd:string
|
||||
self.颜色标记 = None # xsd:string
|
||||
self.取费表 = None # xsd:string
|
||||
self.合价含税 = None # xsd:string (可选)
|
||||
self.type = None # xsd:string
|
||||
self.专业类型 = None # xsd:string
|
||||
self.资源库列表 = None # xsd:list (可选)
|
||||
self.notCheck = None # xsd:string (可选)
|
||||
|
||||
|
||||
class ProjectQuantity:
|
||||
"""
|
||||
工程量
|
||||
描述: 代表项目划分项(ProjectDivisionItem)下的具体工作单元或物料项，是定额、主材、设备的父类型
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.id = None # xsd:string
|
||||
self.类型 = None # xsd:string ("0"为定额,"1"为主材,"5"为设备)
|
||||
self.name = None # xsd:string
|
||||
self.编码 = None # xsd:string
|
||||
self.单位 = None # xsd:string
|
||||
self.数量 = None # xsd:string
|
||||
self.资源库名称 = None # xsd:string
|
||||
self.投标数量 = None # xsd:string (可选)
|
||||
self.投标单价 = None # xsd:string (可选)
|
||||
self.特征段 = None # xsd:string (可选)
|
||||
self.关联父级量 = None # xsd:string (可选)
|
||||
self.颜色标记 = None # xsd:string (可选)
|
||||
self.单价不含税 = None
|
||||
self.cost_set = None # xsd:CostSet
|
||||
|
||||
|
||||
class Ration(ProjectQuantity):
|
||||
"""
|
||||
定额
|
||||
描述: 代表一种标准的工程量条目，通常包含详细的工、料、机消耗标准
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.计算式 = None # xsd:string
|
||||
self.中标计算式 = None # xsd:string
|
||||
self.人工费 = None # xsd:string
|
||||
self.机械费 = None # xsd:string
|
||||
self.甲供材料费不含税 = None # xsd:string
|
||||
self.材料费 = None # xsd:string
|
||||
self.定额系数 = None # xsd:string
|
||||
self.人工系数 = None # xsd:string
|
||||
self.材料系数 = None # xsd:string
|
||||
self.机械系数 = None # xsd:string
|
||||
self.定额范围 = None # xsd:string
|
||||
self.定额章节名称 = None # xsd:string
|
||||
self.费用类型 = None # xsd:string
|
||||
self.甲供材料费含税 = None # xsd:string
|
||||
self.投标合价 = None # xsd:string
|
||||
self.其中甲供材料费 = None # xsd:string
|
||||
self.合价不含税 = None # xsd:string
|
||||
self.基价 = None # xsd:string
|
||||
self.所属定额库 = None # xsd:string
|
||||
|
||||
def __str__(self):
|
||||
"""返回定额的字符串表示"""
|
||||
attrs = {k: v for k, v in self.__dict__.items() if not k.startswith("_")}
|
||||
return json.dumps(attrs, ensure_ascii=False, indent=2)
|
||||
|
||||
|
||||
class Material(ProjectQuantity):
|
||||
"""
|
||||
主材
|
||||
描述: 代表工程中使用的主要材料
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.规格型号 = None # xsd:string
|
||||
self.损耗率 = None # xsd:string
|
||||
self.供货方 = None # xsd:string
|
||||
self.集中配送 = None # xsd:string (可选)
|
||||
self.单重 = None # xsd:string (可选)
|
||||
self.市场价不含税 = None # xsd:string
|
||||
self.市场价含税 = None # xsd:string
|
||||
self.单价含税 = None # xsd:string
|
||||
self.结算市场价不含税 = None # xsd:string (可选)
|
||||
self.结算市场价含税 = None # xsd:string (可选)
|
||||
self.基准价不含税 = None # xsd:string (可选)
|
||||
self.基准价含税 = None # xsd:string (可选)
|
||||
self.费用类型 = None # xsd:string
|
||||
self.增值税率 = None # xsd:string (可选)
|
||||
self.合价含税 = None # xsd:string
|
||||
self.合价不含税 = None # xsd:string
|
||||
self.线重 = None # xsd:string (可选)
|
||||
self.制造长度 = None # xsd:string (可选)
|
||||
self.截面积 = None # xsd:string (可选)
|
||||
|
||||
def __str__(self):
|
||||
"""返回材料的字符串表示"""
|
||||
attrs = {k: v for k, v in self.__dict__.items() if not k.startswith("_")}
|
||||
return json.dumps(attrs, ensure_ascii=False, indent=2)
|
||||
|
||||
|
||||
class Equipment(ProjectQuantity):
|
||||
"""
|
||||
设备
|
||||
描述: 代表工程中安装或使用的设备
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.规格型号 = None # xsd:string
|
||||
self.供货方 = None # xsd:string
|
||||
self.运杂费率 = None # xsd:string (可选)
|
||||
self.单价含税 = None # xsd:string
|
||||
self.设备类型 = None # xsd:string (可选)
|
||||
self.增值税率 = None # xsd:string (可选)
|
||||
self.合价含税 = None # xsd:string
|
||||
self.合价不含税 = None # xsd:string
|
||||
|
||||
|
||||
class MaterialOrEquipment:
|
||||
"""
|
||||
材机
|
||||
描述: 代表DetailedWorkItem中所列出的具体材料、人工或机械设备及其详细信息
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.id = None # xsd:string
|
||||
self.编码 = None # xsd:string
|
||||
self.名称 = None # xsd:string
|
||||
self.单位 = None # xsd:string
|
||||
self.类型 = None # xsd:string
|
||||
self.供货方 = None # xsd:string
|
||||
self.预算价不含税 = None # xsd:string
|
||||
self.市场价不含税 = None # xsd:string
|
||||
self.预算价含税 = None # xsd:string
|
||||
self.市场价含税 = None # xsd:string
|
||||
self.结算预算价不含税 = None # xsd:string
|
||||
self.结算市场价不含税 = None # xsd:string
|
||||
self.结算预算价含税 = None # xsd:string
|
||||
self.结算市场价含税 = None # xsd:string
|
||||
self.暂估价 = None # xsd:string
|
||||
self.拆分 = None # xsd:string
|
||||
self.全口径市场价不含税 = None # xsd:string
|
||||
self.全口径市场价含税 = None # xsd:string
|
||||
self.商品砼 = None # xsd:string
|
||||
self.数量 = None # xsd:string
|
||||
self.是否未计价 = None # xsd:string
|
||||
|
||||
|
||||
class FeeTableTemplateItem:
|
||||
"""
|
||||
取费表模板项
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.OutlayID = None # xsd:string
|
||||
self.type = None # xsd:string
|
||||
self.profession = None # xsd:string
|
||||
|
||||
|
||||
class FeeCollection:
|
||||
"""
|
||||
取费
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.serialNumber = None # xsd:string
|
||||
self.code = None # xsd:string
|
||||
self.base = None # xsd:string
|
||||
self.rate = None # xsd:string
|
||||
|
||||
|
||||
class FeeScheduleItem:
|
||||
"""
|
||||
费用表项
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
|
||||
|
||||
class Fee:
|
||||
"""
|
||||
取费
|
||||
描述:
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.name = None # xsd:string
|
||||
self.serialNumber = None # xsd:string
|
||||
self.code = None # xsd:string
|
||||
self.rate = None # xsd:string
|
||||
self.amount = None # xsd:string
|
||||
self.输出 = None # xsd:string (可选)
|
||||
self.取费基数 = None # xsd:string (可选)
|
||||
self.编制依据 = None # xsd:string (可选)
|
||||
self.编码 = None # xsd:string (可选)
|
||||
self.备注 = None # xsd:string (可选)
|
||||
self.表一显示 = None # xsd:string (可选)
|
||||
self.报表输出 = None # xsd:string (可选)
|
||||
self.建安合计费 = None # xsd:string (可选)
|
||||
self.合计费 = None # xsd:string (可选)
|
||||
self.占总计 = None # xsd:string (可选)
|
||||
self.建筑费 = None # xsd:string (可选)
|
||||
self.安装费 = None # xsd:string (可选)
|
||||
self.设备费 = None # xsd:string (可选)
|
||||
self.其他费 = None # xsd:string (可选)
|
||||
self.施工费 = None # xsd:string (可选)
|
||||
self.单位投资 = None # xsd:string (可选)
|
||||
@@ -0,0 +1,957 @@
|
||||
from neo4j import GraphDatabase
|
||||
from project import *
|
||||
import atexit
|
||||
|
||||
|
||||
class ProjectTookiItNeo4j(ProjectTookiIt):
|
||||
"""
|
||||
基于Neo4j数据库的项目类实现
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
"""
|
||||
初始化Neo4j连接
|
||||
|
||||
Args:
|
||||
uri (str): Neo4j数据库URI
|
||||
user (str): 用户名
|
||||
password (str): 密码
|
||||
"""
|
||||
uri = "bolt://172.20.0.145:7687"
|
||||
user = "neo4j"
|
||||
password = "password"
|
||||
|
||||
super().__init__()
|
||||
self.driver = GraphDatabase.driver(uri, auth=(user, password))
|
||||
self.session = self.driver.session()
|
||||
|
||||
# 初始化其他必要的数据结构
|
||||
self.material_equipment_dict = {} # 材机字典，键为ID
|
||||
self.fee_templates = {} # 取费表模板字典，键为ID
|
||||
self.fee_schedules = {} # 费用表字典，键为ID
|
||||
self.project_properties = {} # 工程属性字典
|
||||
|
||||
def close(self):
|
||||
"""
|
||||
关闭数据库连接
|
||||
"""
|
||||
if self.session:
|
||||
self.session.close()
|
||||
if self.driver:
|
||||
self.driver.close()
|
||||
|
||||
# 通用节点查询方法
|
||||
def get_node_by_path(self, path, node_labels=None):
|
||||
"""
|
||||
通过路径获取节点对象
|
||||
|
||||
Args:
|
||||
path (str): 以'/'分隔的多级节点路径
|
||||
node_labels (list): 节点标签列表，用于过滤结果
|
||||
|
||||
Returns:
|
||||
dict|None: 节点数据，如果路径不存在返回None
|
||||
"""
|
||||
if not path:
|
||||
return None
|
||||
|
||||
# 分割路径为各个部分
|
||||
path_parts = path.split("/")
|
||||
|
||||
# 构建查询
|
||||
if len(path_parts) == 1:
|
||||
# 只有一级路径，直接查询
|
||||
if node_labels:
|
||||
labels_str = ":" + "|:".join(node_labels)
|
||||
query = f"""
|
||||
MATCH (n{labels_str})
|
||||
WHERE n.name = $name
|
||||
RETURN n LIMIT 1
|
||||
"""
|
||||
else:
|
||||
query = """
|
||||
MATCH (n)
|
||||
WHERE n.name = $name
|
||||
RETURN n LIMIT 1
|
||||
"""
|
||||
params = {"name": path_parts[0]}
|
||||
else:
|
||||
# 多级路径，构建路径查询
|
||||
last_part = path_parts[-1]
|
||||
if node_labels:
|
||||
labels_str = ":" + "|:".join(node_labels)
|
||||
query = f"""
|
||||
MATCH path = (root)-[*]->(target{labels_str})
|
||||
WHERE target.name = $last_part
|
||||
RETURN target as n LIMIT 1
|
||||
"""
|
||||
else:
|
||||
query = """
|
||||
MATCH path = (root)-[*]->(target)
|
||||
WHERE target.name = $last_part
|
||||
RETURN target as n LIMIT 1
|
||||
"""
|
||||
params = {"last_part": last_part}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
record = result.single()
|
||||
|
||||
if not record:
|
||||
return None
|
||||
|
||||
return record["n"]
|
||||
except Exception as e:
|
||||
print(f"获取节点对象时出错: {e}")
|
||||
return None
|
||||
|
||||
# 项目划分查询方法
|
||||
def get_division_item_by_path(self, path):
|
||||
"""
|
||||
通过路径获取项目划分对象
|
||||
"""
|
||||
node_data = self.get_node_by_path(path, ["ProjectDivisionItem"])
|
||||
if not node_data:
|
||||
return None
|
||||
|
||||
item = ProjectDivisionItem()
|
||||
for key, value in node_data.items():
|
||||
if hasattr(item, key):
|
||||
setattr(item, key, value)
|
||||
|
||||
return item
|
||||
|
||||
def get_division_node_by_parent_and_name(self, parent_path, partial_name):
|
||||
"""
|
||||
通过父节点路径和模糊节点名称获取项目划分对象，包括子节点
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配项目划分节点的列表，如果没有匹配返回空列表
|
||||
"""
|
||||
if not partial_name:
|
||||
return []
|
||||
|
||||
# 使用通用方法获取父节点
|
||||
parent_node_data = self.get_node_by_path(parent_path, ["ProjectDivisionItem"])
|
||||
parent_node_id = parent_node_data["id"] if parent_node_data and "id" in parent_node_data else None
|
||||
|
||||
# 构建查询，根据是否有父节点ID调整查询条件，使用递归关系查询
|
||||
if parent_node_id:
|
||||
query = """
|
||||
MATCH (p)-[:CONTAINS|HAS|RELATED_TO*]-(n:ProjectDivisionItem)
|
||||
WHERE p.id = $parent_id AND n.name CONTAINS $partial_name
|
||||
RETURN n LIMIT 50
|
||||
"""
|
||||
params = {"parent_id": parent_node_id, "partial_name": partial_name}
|
||||
else:
|
||||
query = """
|
||||
MATCH (n:ProjectDivisionItem)
|
||||
WHERE n.name CONTAINS $partial_name
|
||||
RETURN n LIMIT 50
|
||||
"""
|
||||
params = {"partial_name": partial_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
items = []
|
||||
for record in result:
|
||||
node_data = record["n"]
|
||||
item = ProjectDivisionItem()
|
||||
for key, value in node_data.items():
|
||||
if hasattr(item, key):
|
||||
setattr(item, key, value)
|
||||
items.append(item)
|
||||
|
||||
return items
|
||||
except Exception as e:
|
||||
print(f"通过父节点路径和模糊名称获取项目划分对象时出错: {e}")
|
||||
return []
|
||||
|
||||
# 工程量查询方法
|
||||
def get_quantities_by_paths(self, paths_str):
|
||||
"""
|
||||
获取指定项目路径下的工程量对象
|
||||
|
||||
Args:
|
||||
paths_str (str): 以'/'分隔的多级节点路径
|
||||
|
||||
Returns:
|
||||
ProjectQuantity|None: 对应的工程量对象，如果路径不存在返回None，成功找到返回工程量对象
|
||||
"""
|
||||
if not paths_str:
|
||||
return None
|
||||
|
||||
# 使用通用方法获取节点，考虑所有可能的工程量类型
|
||||
node_data = self.get_node_by_path(paths_str, ["ProjectQuantity", "Quota", "MainMaterial", "Equipment"])
|
||||
|
||||
if not node_data:
|
||||
return None
|
||||
|
||||
# 根据节点标签或类型属性创建对应类型的对象
|
||||
quantity = self._create_quantity_object(node_data)
|
||||
|
||||
# 填充属性
|
||||
for key, value in node_data.items():
|
||||
if hasattr(quantity, key):
|
||||
setattr(quantity, key, value)
|
||||
|
||||
return quantity
|
||||
|
||||
def get_quantities_node_by_parent_and_code(self, parent_path, quantity_type=None, code=None):
|
||||
"""
|
||||
通过父节点路径和编码获取工程量对象（定额），包括子节点
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
quantity_type (str): 工程量类型('定额'、'主材'、'设备'或None表示所有类型)
|
||||
code(str): 工程量编码，以'/'分隔的多个编码
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配节点的列表，如果没有匹配返回空列表
|
||||
"""
|
||||
if not code:
|
||||
return []
|
||||
|
||||
# 使用通用方法获取父节点
|
||||
parent_node_data = self.get_node_by_path(parent_path)
|
||||
|
||||
# 从路径中获取父节点名称
|
||||
path_parts = parent_path.split("/")
|
||||
parent_name = path_parts[-1]
|
||||
|
||||
# 处理编码，可能有多个编码用/分隔
|
||||
code_parts = code.split("/")
|
||||
code_conditions = []
|
||||
for code_part in code_parts:
|
||||
if code_part:
|
||||
|
||||
code_conditions.append(f"q.编码 = '{code_part}'")
|
||||
|
||||
if not code_conditions:
|
||||
return []
|
||||
|
||||
code_query = " OR ".join(code_conditions)
|
||||
|
||||
# 根据工程量类型确定标签
|
||||
node_labels = []
|
||||
if quantity_type == "定额":
|
||||
node_labels = ["ProjectQuantity", "Quota"]
|
||||
elif quantity_type == "主材":
|
||||
node_labels = ["ProjectQuantity", "MainMaterial"]
|
||||
elif quantity_type == "设备":
|
||||
node_labels = ["ProjectQuantity", "Equipment"]
|
||||
else:
|
||||
node_labels = ["ProjectQuantity"]
|
||||
|
||||
# 构建标签字符串
|
||||
labels_str = ":" + ":".join(node_labels) if node_labels else ""
|
||||
|
||||
# 使用name属性进行匹配
|
||||
query = f"""
|
||||
MATCH (p)-[*1..5]->(q{labels_str})
|
||||
WHERE p.name = $parent_name AND ({code_query})
|
||||
RETURN q
|
||||
LIMIT 10
|
||||
"""
|
||||
params = {"parent_name": parent_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, params)
|
||||
quantities = []
|
||||
|
||||
for record in result:
|
||||
node_data = record["q"]
|
||||
quantity = self._create_quantity_object(node_data, quantity_type)
|
||||
|
||||
# 将节点属性赋值到对象
|
||||
for key, value in node_data.items():
|
||||
setattr(quantity, key, value)
|
||||
|
||||
# 转换为字典
|
||||
if hasattr(quantity, "to_dict"):
|
||||
quantities.append(quantity.to_dict())
|
||||
else:
|
||||
# 如果没有 to_dict 方法，就用 vars() 动态获取属性
|
||||
quantities.append(vars(quantity))
|
||||
|
||||
return quantities
|
||||
except Exception as e:
|
||||
print(f"通过编码获取工程量对象时出错: {e}")
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
return []
|
||||
|
||||
def get_quantities_node_by_parent_and_name(self, parent_path, partial_name, quantity_type=None):
|
||||
"""
|
||||
通过父节点路径、模糊节点名称和类型获取工程量对象（主材或者设备），包括子节点
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
quantity_type (str): 工程量类型('定额'、'主材'、'设备'或None表示所有类型)
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配节点的列表，如果没有匹配返回空列表
|
||||
"""
|
||||
if not partial_name:
|
||||
return []
|
||||
|
||||
# 调试输出
|
||||
print(f"搜索父路径: {parent_path}")
|
||||
print(f"搜索名称: {partial_name}")
|
||||
print(f"搜索类型: {quantity_type}")
|
||||
|
||||
# 使用通用方法获取父节点
|
||||
parent_node_data = self.get_node_by_path(parent_path)
|
||||
|
||||
if parent_node_data:
|
||||
print(f"找到父节点: {parent_node_data}")
|
||||
else:
|
||||
print("未找到父节点")
|
||||
|
||||
parent_node_id = parent_node_data["id"] if parent_node_data and "id" in parent_node_data else None
|
||||
|
||||
# 根据工程量类型确定标签和类型条件
|
||||
node_labels = []
|
||||
type_condition = ""
|
||||
|
||||
if quantity_type == "定额":
|
||||
node_labels = ["ProjectQuantity", "Quota"]
|
||||
type_condition = "q.类型 = '0'"
|
||||
elif quantity_type == "主材":
|
||||
node_labels = ["ProjectQuantity", "MainMaterial"]
|
||||
type_condition = "q.类型 = '1'"
|
||||
elif quantity_type == "设备":
|
||||
node_labels = ["ProjectQuantity", "Equipment"]
|
||||
type_condition = "q.类型 = '5'"
|
||||
else:
|
||||
node_labels = ["ProjectQuantity"]
|
||||
|
||||
# 构建标签字符串
|
||||
labels_str = ":" + ":".join(node_labels) if node_labels else ""
|
||||
|
||||
# 扩展关系类型
|
||||
relationship_types = "CONTAINS|HAS|RELATED_TO|USES|BELONGS_TO"
|
||||
|
||||
# 构建查询 - 使用递归关系查询
|
||||
if parent_node_id:
|
||||
query = f"""
|
||||
MATCH (p)-[:{relationship_types}*1..10]->(q{labels_str})
|
||||
WHERE p.id = $parent_id AND q.name CONTAINS $partial_name
|
||||
{f'AND {type_condition}' if type_condition else ''}
|
||||
RETURN q LIMIT 50
|
||||
"""
|
||||
params = {"parent_id": parent_node_id, "partial_name": partial_name}
|
||||
print(f"执行查询: {query}")
|
||||
print(f"参数: {params}")
|
||||
else:
|
||||
query = f"""
|
||||
MATCH (q{labels_str})
|
||||
WHERE q.name CONTAINS $partial_name
|
||||
{f'AND {type_condition}' if type_condition else ''}
|
||||
RETURN q LIMIT 50
|
||||
"""
|
||||
params = {"partial_name": partial_name}
|
||||
print(f"执行全局查询: {query}")
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
quantities = []
|
||||
for record in result:
|
||||
node_data = record["q"]
|
||||
print(f"找到节点: {node_data}")
|
||||
|
||||
# 创建对应类型的对象
|
||||
quantity = self._create_quantity_object(node_data, quantity_type)
|
||||
|
||||
# 填充属性
|
||||
for key, value in node_data.items():
|
||||
if hasattr(quantity, key):
|
||||
setattr(quantity, key, value)
|
||||
|
||||
quantities.append(quantity)
|
||||
|
||||
print(f"查询结果数量: {len(quantities)}")
|
||||
return quantities
|
||||
except Exception as e:
|
||||
print(f"通过名称获取工程量对象时出错: {e}")
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
return []
|
||||
|
||||
# 辅助方法，用于根据节点数据创建对应类型的工程量对象
|
||||
def _create_quantity_object(self, node_data, quantity_type=None):
|
||||
"""
|
||||
根据节点数据创建对应类型的工程量对象
|
||||
|
||||
Args:
|
||||
node_data (dict): 节点数据
|
||||
quantity_type (str): 工程量类型('定额'、'主材'、'设备'或None)
|
||||
|
||||
Returns:
|
||||
ProjectQuantity: 创建的工程量对象
|
||||
"""
|
||||
# 如果指定了类型，直接创建对应类型的对象
|
||||
if quantity_type == "定额":
|
||||
return Ration()
|
||||
elif quantity_type == "主材":
|
||||
return Material()
|
||||
elif quantity_type == "设备":
|
||||
return Equipment()
|
||||
|
||||
# 如果没有指定类型，尝试通过节点属性或标签判断
|
||||
if "类型" in node_data:
|
||||
if node_data["类型"] == "0":
|
||||
return Ration()
|
||||
elif node_data["类型"] == "1":
|
||||
return Material()
|
||||
elif node_data["类型"] == "5":
|
||||
return Equipment()
|
||||
|
||||
# 通过标签判断
|
||||
labels = list(node_data.labels) if hasattr(node_data, "labels") else []
|
||||
if "Quota" in labels:
|
||||
return Ration()
|
||||
elif "MainMaterial" in labels:
|
||||
return Material()
|
||||
elif "Equipment" in labels:
|
||||
return Equipment()
|
||||
|
||||
# 默认返回基类对象
|
||||
return ProjectQuantity()
|
||||
|
||||
# 材机查询方法实现
|
||||
def get_material_equipment_by_path(self, paths_str):
|
||||
"""
|
||||
通过路径获取材机对象
|
||||
|
||||
Args:
|
||||
paths_str (str): 以'/'分隔的多级项目划分名称路径
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的材机对象的列表
|
||||
"""
|
||||
if not paths_str:
|
||||
return []
|
||||
|
||||
# 使用通用方法获取节点
|
||||
node_data = self.get_node_by_path(paths_str)
|
||||
node_id = node_data["id"] if node_data and "id" in node_data else None
|
||||
|
||||
if not node_id:
|
||||
return []
|
||||
|
||||
# 查询与该节点关联的所有材机对象
|
||||
query = """
|
||||
MATCH (p)-[r]-(m)
|
||||
WHERE p.id = $node_id AND (m:MaterialOrEquipment OR m:Material OR m:Equipment)
|
||||
RETURN m LIMIT 50
|
||||
"""
|
||||
params = {"node_id": node_id}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
materials = []
|
||||
for record in result:
|
||||
material = self._create_material_object(record["m"])
|
||||
materials.append(material)
|
||||
|
||||
# 更新缓存
|
||||
if hasattr(material, "id") and material.id:
|
||||
self.material_equipment_dict[material.id] = material
|
||||
|
||||
return materials
|
||||
except Exception as e:
|
||||
print(f"通过路径获取材机对象时出错: {e}")
|
||||
return []
|
||||
|
||||
def get_material_equipment_by_parent_and_name(self, parent_path, partial_name):
|
||||
"""
|
||||
通过父节点路径和模糊名称获取材机对象
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的材机对象的列表
|
||||
"""
|
||||
if not partial_name:
|
||||
return []
|
||||
|
||||
# 使用通用方法获取父节点
|
||||
parent_node_data = self.get_node_by_path(parent_path)
|
||||
parent_node_id = parent_node_data["id"] if parent_node_data and "id" in parent_node_data else None
|
||||
|
||||
# 构建查询，根据是否有父节点ID调整查询条件
|
||||
if parent_node_id:
|
||||
# 如果找到了父节点，查找与父节点有关系的材机节点
|
||||
query = """
|
||||
MATCH (p)-[:CONTAINS|HAS|USES|RELATED_TO]-(m)
|
||||
WHERE p.id = $parent_id AND m.name CONTAINS $partial_name
|
||||
AND (m:MaterialOrEquipment OR m:Material OR m:Equipment)
|
||||
RETURN m LIMIT 20
|
||||
"""
|
||||
params = {"parent_id": parent_node_id, "partial_name": partial_name}
|
||||
else:
|
||||
# 如果没有找到父节点或没有提供父节点路径，只按名称查询
|
||||
query = """
|
||||
MATCH (m)
|
||||
WHERE m.name CONTAINS $partial_name
|
||||
AND (m:MaterialOrEquipment OR m:Material OR m:Equipment)
|
||||
RETURN m LIMIT 20
|
||||
"""
|
||||
params = {"partial_name": partial_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
materials = []
|
||||
for record in result:
|
||||
material = self._create_material_object(record["m"])
|
||||
materials.append(material)
|
||||
|
||||
# 更新缓存
|
||||
if hasattr(material, "id") and material.id:
|
||||
self.material_equipment_dict[material.id] = material
|
||||
|
||||
return materials
|
||||
except Exception as e:
|
||||
print(f"通过父节点路径和模糊名称获取材机对象时出错: {e}")
|
||||
return []
|
||||
|
||||
# 辅助方法，用于创建材机对象并填充属性
|
||||
def _create_material_object(self, node_data):
|
||||
"""
|
||||
根据节点数据创建材机对象并填充属性
|
||||
|
||||
Args:
|
||||
node_data (dict): 节点数据
|
||||
|
||||
Returns:
|
||||
MaterialOrEquipment: 创建的材机对象
|
||||
"""
|
||||
material = MaterialOrEquipment()
|
||||
|
||||
# 填充属性
|
||||
for key, value in node_data.items():
|
||||
if hasattr(material, key):
|
||||
setattr(material, key, value)
|
||||
|
||||
return material
|
||||
|
||||
# 取费表模板查询方法实现
|
||||
def get_fee_template_by_path(self, paths_str):
|
||||
"""
|
||||
通过路径获取取费表模板
|
||||
|
||||
Args:
|
||||
paths_str (str): 以'/'分隔的多级项目划分名称路径
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的取费表模板对象的列表
|
||||
"""
|
||||
if not paths_str:
|
||||
return []
|
||||
|
||||
# 使用通用方法获取节点
|
||||
node_data = self.get_node_by_path(paths_str)
|
||||
node_id = node_data["id"] if node_data and "id" in node_data else None
|
||||
|
||||
if not node_id:
|
||||
return []
|
||||
|
||||
# 查询与该节点关联的所有取费表模板
|
||||
query = """
|
||||
MATCH (p)-[r]-(t)
|
||||
WHERE p.id = $node_id AND (t:FeeTableTemplate OR t.type = 'FeeTableTemplate')
|
||||
RETURN t LIMIT 20
|
||||
"""
|
||||
params = {"node_id": node_id}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
templates = []
|
||||
for record in result:
|
||||
template = self._create_fee_template_object(record["t"])
|
||||
templates.append(template)
|
||||
|
||||
return templates
|
||||
except Exception as e:
|
||||
print(f"通过路径获取取费表模板时出错: {e}")
|
||||
return []
|
||||
|
||||
def get_fee_template_by_parent_and_name(self, parent_path, partial_name):
|
||||
"""
|
||||
通过父节点路径和模糊名称获取取费表模板
|
||||
|
||||
Args:
|
||||
parent_path (str): 父节点的路径，以'/'分隔的多级节点路径
|
||||
partial_name (str): 目标节点的模糊或不完整名称
|
||||
|
||||
Returns:
|
||||
list: 包含所有匹配的取费表模板对象的列表
|
||||
"""
|
||||
if not partial_name:
|
||||
return []
|
||||
|
||||
# 使用通用方法获取父节点
|
||||
parent_node_data = self.get_node_by_path(parent_path)
|
||||
parent_node_id = parent_node_data["id"] if parent_node_data and "id" in parent_node_data else None
|
||||
|
||||
# 构建查询，根据是否有父节点ID调整查询条件
|
||||
if parent_node_id:
|
||||
query = """
|
||||
MATCH (p)-[:CONTAINS|HAS|USES|RELATED_TO]-(t)
|
||||
WHERE p.id = $parent_id AND t.name CONTAINS $partial_name
|
||||
AND (t:FeeTableTemplate OR t.type = 'FeeTableTemplate')
|
||||
RETURN t LIMIT 20
|
||||
"""
|
||||
params = {"parent_id": parent_node_id, "partial_name": partial_name}
|
||||
else:
|
||||
query = """
|
||||
MATCH (t)
|
||||
WHERE t.name CONTAINS $partial_name
|
||||
AND (t:FeeTableTemplate OR t.type = 'FeeTableTemplate')
|
||||
RETURN t LIMIT 20
|
||||
"""
|
||||
params = {"partial_name": partial_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
templates = []
|
||||
for record in result:
|
||||
template = self._create_fee_template_object(record["t"])
|
||||
templates.append(template)
|
||||
|
||||
return templates
|
||||
except Exception as e:
|
||||
print(f"通过父节点路径和模糊名称获取取费表模板时出错: {e}")
|
||||
return []
|
||||
|
||||
# 辅助方法，用于创建取费表模板对象并填充属性
|
||||
def _create_fee_template_object(self, node_data):
|
||||
"""
|
||||
根据节点数据创建取费表模板对象并填充属性
|
||||
|
||||
Args:
|
||||
node_data (dict): 节点数据
|
||||
|
||||
Returns:
|
||||
FeeTableTemplateItem: 创建的取费表模板对象
|
||||
"""
|
||||
template = FeeTableTemplateItem()
|
||||
|
||||
# 填充属性
|
||||
for key, value in node_data.items():
|
||||
if hasattr(template, key):
|
||||
setattr(template, key, value)
|
||||
|
||||
# 更新缓存
|
||||
if hasattr(template, "OutlayID") and template.OutlayID:
|
||||
self.fee_templates[template.OutlayID] = template
|
||||
|
||||
return template
|
||||
|
||||
# 费用表查询方法实现
|
||||
def get_fee_schedule_on_auxiliary_expense_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在辅助费用表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值属性名
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
if not table_name or not fee_name or not fee:
|
||||
return None
|
||||
|
||||
# 构建查询，查找辅助费用表中的特定费用 - 使用任意关系类型
|
||||
query = """
|
||||
MATCH (t:FeeScheduleItem)-[r]->(f:Fee)
|
||||
WHERE t.name = $table_name AND f.name = $fee_name
|
||||
RETURN f LIMIT 1
|
||||
"""
|
||||
params = {"table_name": table_name, "fee_name": fee_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
all_records = result.data()
|
||||
|
||||
if len(all_records) > 0:
|
||||
fee_node = all_records[0]["f"]
|
||||
value = fee_node.get(fee)
|
||||
if value is not None:
|
||||
return value
|
||||
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"在辅助费用表中查找费用时出错: {e}")
|
||||
return None
|
||||
|
||||
def get_fee_schedule_on_other_expense_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在其它费用表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值属性名
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
if not table_name or not fee_name or not fee:
|
||||
return None
|
||||
|
||||
# 构建查询，查找其它费用表中的特定费用 - 使用任意关系类型
|
||||
query = """
|
||||
MATCH (t:FeeScheduleItem)-[r]->(f:Fee)
|
||||
WHERE t.name = $table_name AND f.name = $fee_name
|
||||
RETURN f LIMIT 1
|
||||
"""
|
||||
params = {"table_name": table_name, "fee_name": fee_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
all_records = result.data()
|
||||
|
||||
if len(all_records) > 0:
|
||||
fee_node = all_records[0]["f"]
|
||||
value = fee_node.get(fee)
|
||||
if value is not None:
|
||||
return value
|
||||
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"在其它费用表中查找费用时出错: {e}")
|
||||
return None
|
||||
|
||||
def get_fee_schedule_on_land_acquisition_fee_table_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在土地征用费表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值属性名
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
if not table_name or not fee_name or not fee:
|
||||
return None
|
||||
|
||||
# 构建查询，查找土地征用费表中的特定费用 - 使用任意关系类型
|
||||
query = """
|
||||
MATCH (t:FeeScheduleItem)-[r]->(f:Fee)
|
||||
WHERE t.name = $table_name AND f.name = $fee_name
|
||||
RETURN f LIMIT 1
|
||||
"""
|
||||
params = {"table_name": table_name, "fee_name": fee_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
all_records = result.data()
|
||||
|
||||
if len(all_records) > 0:
|
||||
fee_node = all_records[0]["f"]
|
||||
value = fee_node.get(fee)
|
||||
if value is not None:
|
||||
return value
|
||||
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"在土地征用费表中查找费用时出错: {e}")
|
||||
return None
|
||||
|
||||
def get_fee_schedule_on_installation_price_difference_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在安装价差表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值属性名
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
if not table_name or not fee_name or not fee:
|
||||
return None
|
||||
|
||||
# 构建查询，查找安装价差表中的特定费用 - 使用任意关系类型
|
||||
query = """
|
||||
MATCH (t:FeeScheduleItem)-[r]->(f:Fee)
|
||||
WHERE t.name = $table_name AND f.name = $fee_name
|
||||
RETURN f LIMIT 1
|
||||
"""
|
||||
params = {"table_name": table_name, "fee_name": fee_name}
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
|
||||
all_records = result.data()
|
||||
|
||||
if len(all_records) > 0:
|
||||
fee_node = all_records[0]["f"]
|
||||
value = fee_node.get(fee)
|
||||
if value is not None:
|
||||
return value
|
||||
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"在安装价差表中查找费用时出错: {e}")
|
||||
return None
|
||||
|
||||
def get_fee_schedule_on_Engineering_Cost_table(self, table_name, fee_name, fee):
|
||||
"""
|
||||
在工程费用表中查找费用
|
||||
|
||||
Args:
|
||||
table_name (str): 费用表名称
|
||||
fee_name (str): 要查找的费用名称
|
||||
fee (str): 匹配的费用值属性名
|
||||
|
||||
Returns:
|
||||
str: 匹配到的费用名称节点对应的费用值
|
||||
"""
|
||||
if not table_name or not fee_name or not fee:
|
||||
return None
|
||||
|
||||
# 调试输出
|
||||
print(f"查询费用表: {table_name}")
|
||||
print(f"查询费用名称: {fee_name}")
|
||||
print(f"查询费用属性: {fee}")
|
||||
|
||||
# 构建查询，使用递归关系查询，并扩展关系类型
|
||||
query = """
|
||||
MATCH (t:FeeScheduleItem)-[r*1..5]->(f:Fee)
|
||||
WHERE t.name = $table_name AND f.name = $fee_name
|
||||
RETURN f LIMIT 10
|
||||
"""
|
||||
params = {"table_name": table_name, "fee_name": fee_name}
|
||||
|
||||
print(f"执行查询: {query}")
|
||||
print(f"参数: {params}")
|
||||
|
||||
try:
|
||||
result = self.session.run(query, **params)
|
||||
all_records = result.data()
|
||||
|
||||
print(f"查询结果数量: {len(all_records)}")
|
||||
|
||||
if len(all_records) > 0:
|
||||
fee_node = all_records[0]["f"]
|
||||
print(f"找到费用节点: {fee_node}")
|
||||
|
||||
# 获取节点的所有属性，用于调试
|
||||
for key, value in fee_node.items():
|
||||
print(f"属性: {key} = {value}")
|
||||
|
||||
value = fee_node.get(fee)
|
||||
if value is not None:
|
||||
print(f"找到费用值: {value}")
|
||||
return value
|
||||
else:
|
||||
print(f"节点中没有属性: {fee}")
|
||||
|
||||
# 如果没有找到，尝试另一种查询方式，使用CONTAINS进行模糊匹配
|
||||
print("尝试使用模糊匹配...")
|
||||
backup_query = """
|
||||
MATCH (t:FeeScheduleItem)-[r*1..5]->(f:Fee)
|
||||
WHERE t.name CONTAINS $table_name AND f.name CONTAINS $fee_name
|
||||
RETURN f LIMIT 10
|
||||
"""
|
||||
backup_result = self.session.run(backup_query, **params)
|
||||
backup_records = backup_result.data()
|
||||
|
||||
print(f"模糊匹配结果数量: {len(backup_records)}")
|
||||
|
||||
if len(backup_records) > 0:
|
||||
fee_node = backup_records[0]["f"]
|
||||
print(f"找到费用节点(模糊匹配): {fee_node}")
|
||||
|
||||
# 获取节点的所有属性，用于调试
|
||||
for key, value in fee_node.items():
|
||||
print(f"属性: {key} = {value}")
|
||||
|
||||
value = fee_node.get(fee)
|
||||
if value is not None:
|
||||
print(f"找到费用值: {value}")
|
||||
return value
|
||||
else:
|
||||
print(f"节点中没有属性: {fee}")
|
||||
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"在工程费用表中查找费用时出错: {e}")
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
return None
|
||||
|
||||
|
||||
class ProjectBuilder:
|
||||
"""
|
||||
项目构建器
|
||||
描述: 用于构建项目对象的构建器
|
||||
"""
|
||||
|
||||
_instance = None
|
||||
|
||||
@staticmethod
|
||||
def build():
|
||||
"""
|
||||
构建并返回项目实例
|
||||
|
||||
Returns:
|
||||
ProjectTookiItNeo4j: 创建的项目实例
|
||||
"""
|
||||
# 如果已经有实例，先关闭它
|
||||
if ProjectBuilder._instance is not None:
|
||||
ProjectBuilder._instance.close()
|
||||
|
||||
# 创建新实例
|
||||
|
||||
ProjectBuilder._instance = ProjectTookiItNeo4j()
|
||||
|
||||
return ProjectBuilder._instance
|
||||
|
||||
@staticmethod
|
||||
def close():
|
||||
"""
|
||||
关闭当前项目实例的连接
|
||||
"""
|
||||
if ProjectBuilder._instance is not None:
|
||||
ProjectBuilder._instance.close()
|
||||
ProjectBuilder._instance = None
|
||||
|
||||
|
||||
# 注册退出处理函数，确保程序退出时自动关闭连接
|
||||
atexit.register(ProjectBuilder.close)
|
||||
|
||||
|
||||
# project = ProjectBuilder.build()
|
||||
# result = project.get_quantities_node_by_parent_and_code(
|
||||
# "工程数据/安装工程/安装/架空输电线路本体工程/基础工程", "定额", "YX2-1/YX2-2/YX2-3/YX2-6/YX2-7"
|
||||
# )
|
||||
|
||||
# print(result)
|
||||
@@ -0,0 +1,99 @@
|
||||
from langchain.prompts import PromptTemplate
|
||||
|
||||
FUNCTION_CALL_TEMPLATE = """
|
||||
你是一个专业的Python工程师。我会给你一个用户问题，你需要将其转换为对应的Python代码
|
||||
|
||||
可用工具:
|
||||
{tools}
|
||||
|
||||
工具名称: {tool_names}
|
||||
|
||||
你的任务是：
|
||||
1. 首先从用户的{query}中去提取出关键节点信息出来
|
||||
2. 将提取出来的信息在知识库中内容{KnowledgeBase}和节点属性{NodeDefinition}中进行查找对比是否正确，如果不正确，那么需要你从新解析用户的问题
|
||||
3. 根据用户输入解析出来的信息来从{project_class_methods}中定义的查找方法类确定使用哪个方法进行查询
|
||||
4. 每次只能调用一个方法，不要同时调用多个方法
|
||||
5. 模板中Function为需要根据用户输入来判断的方法名，parameter为用户输入中解析出来的方法的参数
|
||||
6. 使用固定的python模板来生成Python代码
|
||||
7. 代码应该完整，可以直接执行，并且能够返回查询结果
|
||||
|
||||
{agent_scratchpad}
|
||||
|
||||
输出代码模板：
|
||||
def neo4j_find_function():
|
||||
|
||||
project = ProjectBuilder.build()
|
||||
|
||||
result = project.Function(parameter)
|
||||
|
||||
print(result)
|
||||
|
||||
注意：
|
||||
- 必须要严格按照给定代码模板来输出代码，不得对模板进行修改等操作。
|
||||
- 代码应该简洁明了，只需要调用适当的方法并返回结果
|
||||
- 如果需要对结果进行处理（如单位换算，简单四则运算等），可以添加必要的代码
|
||||
- 只输出Python代码，不要添加任何解释或注释
|
||||
- 不要输出任何其他无关的内容出来其他内容（包括多余的符号，字符等）
|
||||
|
||||
使用以下格式：
|
||||
|
||||
思考: [你的推理过程]
|
||||
行动: 工具名称
|
||||
行动输入: 工具参数
|
||||
观察: 工具返回的结果
|
||||
|
||||
...重复以上步骤...
|
||||
|
||||
思考: 我现在已经收集到所有必要的信息，可以生成最终的Python代码。
|
||||
Final Answer:
|
||||
def neo4j_find_function():
|
||||
project = ProjectBuilder.build()
|
||||
result = project.Function(parameter)
|
||||
print(result)
|
||||
"""
|
||||
|
||||
FUNCTION_CALL_PROMPT = PromptTemplate.from_template(FUNCTION_CALL_TEMPLATE)
|
||||
|
||||
|
||||
###########################################################################################################################################################################
|
||||
|
||||
FUNCTION_RETURNS_LOOP_TEMPLATE = """
|
||||
|
||||
你是一个专业的Python工程师。我会给你一个用户问题和错误信息，你需要将其转换为对应的Python代码
|
||||
|
||||
你的任务是：
|
||||
1. 首先从用户的{original_query}中去提取出关键节点信息出来
|
||||
2. 将提取出来的信息在知识库中内容{KnowledgeBase}和节点属性{NodeDefinition}中进行查找对比是否正确，如果不正确，那么需要你从新解析用户的问题
|
||||
3. 根据用户输入解析出来的信息来、需要修改的代码{original_code}和代码的错误信息{error_info}来对代码和参数进行修改
|
||||
4. 每次只能调用一个方法，不要同时调用多个方法
|
||||
5. 模板中Function为需要根据用户输入来判断的方法名，parameter为用户输入中解析出来的方法的参数
|
||||
6. 使用固定的python模板来生成Python代码
|
||||
7. 代码应该完整，可以直接执行，并且能够返回查询结果
|
||||
|
||||
注意：
|
||||
- 你的任务是对错误的代码和错误参数进行修正，而不是返回错误信息和解释
|
||||
- 如果要查找的节点在知识图谱中不存在，那么请你返回一个类似的节点，而不是返回错误信息
|
||||
- 你的输出只能修改后的错误Python代码，不要添加任何解释或注释
|
||||
- 代码应该简洁明了，只需要调用适当的方法并返回结果
|
||||
- 如果需要对结果进行处理（如单位换算，简单四则运算等），可以添加必要的代码
|
||||
- 如果没有办法根据已有的信息对代码进行修改，那么只需要返回原本的错误代码即可，而不是去返回给我一段自然语言文本
|
||||
|
||||
请根据以下输入进行推理并修正代码：
|
||||
|
||||
思考: [你的推理过程]
|
||||
行动: 工具名称
|
||||
行动输入: 工具参数
|
||||
观察: 工具返回的结果
|
||||
|
||||
...重复以上步骤...
|
||||
|
||||
思考: 我已经根据错误信息修正了参数或路径，可以生成新的代码。
|
||||
Final Answer:
|
||||
def neo4j_find_function():
|
||||
project = ProjectBuilder.build()
|
||||
result = project.Function(parameter)
|
||||
print(result)
|
||||
"""
|
||||
|
||||
|
||||
FUNCTION_RETURNS_LOOP_PROMPT: PromptTemplate = PromptTemplate.from_template(FUNCTION_RETURNS_LOOP_TEMPLATE)
|
||||
+18
@@ -0,0 +1,18 @@
|
||||
from project_implementation import ProjectBuilder
|
||||
|
||||
|
||||
def find_specific_node():
|
||||
|
||||
project = ProjectBuilder.build()
|
||||
|
||||
table_name = "工程费用"
|
||||
fee_name = "动态费用"
|
||||
fee = "其他费(QTF)"
|
||||
|
||||
result = project.get_fee_schedule_on_Engineering_Cost_table(table_name, fee_name, fee)
|
||||
|
||||
print(result)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
find_specific_node()
|
||||
Reference in New Issue
Block a user