chentianrui/langchain_KG
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

上传文件至 kg_lab_6.13

Browse Source 
6.16 更新 修复None bug
This commit is contained in:
zoujiwen
2025-06-16 14:18:43 +08:00
parent 2cd09e6528
commit 18aa0281bc
3 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kg_lab_6.13/main.py
+1 -1
View File
@@ -38,7 +38,7 @@ while True:
print(f"🧠 提取关键词：{keywords}") print(f"🧠 提取关键词：{keywords}")
# 预映射为图数据库结构 # 预映射为图数据库结构
input_neo4j = pre_mapping2(keywords, data) input_neo4j = pre_mapping(keywords, data)
print(f"📊 图谱相关知识输出：\n{input_neo4j}\n") print(f"📊 图谱相关知识输出：\n{input_neo4j}\n")
kg_lab_6.13/main2.py
+1 -1
View File
@@ -48,7 +48,7 @@ while True:
print(f"🧠 提取关键词：{keywords}") print(f"🧠 提取关键词：{keywords}")
# 预映射为图数据库结构 # 预映射为图数据库结构
input_neo4j = pre_mapping2(keywords, data) input_neo4j = pre_mapping(keywords, data)
print(f"📊 图谱相关知识输出：\n{input_neo4j}\n") print(f"📊 图谱相关知识输出：\n{input_neo4j}\n")
kg_lab_6.13/utils.py
+8 -5
View File
@@ -74,7 +74,10 @@ def pre_mapping(keywords, data):
else: else:
if len(ceshi["指标描述"]["指标映射"]) == 1: if len(ceshi["指标描述"]["指标映射"]) == 1:
temp0, temp1 = find_target_item(ceshi["指标描述"]["指标映射"][0], data) temp0, temp1 = find_target_item(ceshi["指标描述"]["指标映射"][0], data)
if "【" in temp0:
return f"{temp1}，换算规则：【{temp0}】" return f"{temp1}，换算规则：【{temp0}】"
else:
return f"查找一下【{temp1}】，换算规则：【{temp0}】"
elif len(ceshi["指标描述"]["指标映射"]) > 1: elif len(ceshi["指标描述"]["指标映射"]) > 1:
@@ -95,11 +98,11 @@ def pre_mapping(keywords, data):
if temp0 is None and temp1 is None: if temp0 is None and temp1 is None:
pass pass
elif temp0 != None and temp1 is None: elif temp1 != None and temp0 is None:
if judge_exists(item, data): if judge_exists(item, data):
temp1 = item temp0 = item
# temp0 = find_target_items(ceshi["指标描述"]["映射规则"], item, data) # temp0 = find_target_items(ceshi["指标描述"]["映射规则"], item, data)
result.append(f"模糊查找一下【{temp0}】，换算规则：【{temp1}】") result.append(f"模糊查找一下【{temp0}】，换算规则：【{temp1}】,233")
else: else:
continue continue
@@ -172,7 +175,7 @@ def pre_mapping2(keywords, data):
desc = ceshi.get("指标描述") desc = ceshi.get("指标描述")
if isinstance(desc, str): if isinstance(desc, str):
return f"{desc}" return f"测试：{desc}"
elif isinstance(desc, dict): elif isinstance(desc, dict):
mapping_rule = desc.get("映射规则") mapping_rule = desc.get("映射规则")
@@ -207,7 +210,7 @@ def pre_mapping2(keywords, data):
if judge_exists(item_name, data): if judge_exists(item_name, data):
if temp1 is None: if temp1 is None:
temp1 = item_name temp0 = item_name
result.append(f"模糊查找一下【{temp0}】，换算规则：【{temp1}】") result.append(f"模糊查找一下【{temp0}】，换算规则：【{temp1}】")
return result or "未匹配到有效的映射项" return result or "未匹配到有效的映射项"