You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.4 KiB
36 lines
1.4 KiB
import json
|
|
import os
|
|
|
|
def parse_relationships(file_path):
|
|
# 检查文件是否存在
|
|
if not os.path.exists(file_path):
|
|
raise FileNotFoundError(f"文件不存在: {file_path}")
|
|
|
|
# 读取并解析JSON文件
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|
data = json.load(f)
|
|
|
|
# 提取关键信息
|
|
result = {
|
|
"embedding_dim": data.get("embedding_dim", "未知维度"),
|
|
"relationship_count": len(data.get("data", [])),
|
|
"sample_relationships": data.get("data", [])[:3] # 显示前3条示例
|
|
}
|
|
return result
|
|
|
|
if __name__ == "__main__":
|
|
file_path = r"d:\dsWork\dsProject\dsLightRag\Topic\JiHe\vdb_relationships.json"
|
|
try:
|
|
relationships = parse_relationships(file_path)
|
|
print(f"嵌入维度: {relationships['embedding_dim']}")
|
|
print(f"关系总数: {relationships['relationship_count']}")
|
|
print("\n示例关系:\n")
|
|
for i, rel in enumerate(relationships['sample_relationships'], 1):
|
|
print(f"关系 {i}:")
|
|
print(f" ID: {rel['__id__']}")
|
|
print(f" 源实体: {rel['src_id']}")
|
|
print(f" 目标实体: {rel['tgt_id']}")
|
|
print(f" 关系描述: {rel['content'][:50]}...") # 截断长文本
|
|
print(f" 来源块ID: {rel['source_id']}\n")
|
|
except Exception as e:
|
|
print(f"解析错误: {str(e)}") |