'commit'

dsRag/Neo4j/N4_KG_Graph.py

@@ -7,10 +7,10 @@ from Config import Config
 # 连接Neo4j
 graph = Graph(Config.NEO4J_URI, auth=Config.NEO4J_AUTH)
 
-# 查询所有节点和关系
+# 修改后的查询语句，使用INCLUDES关系类型
 query = """
-MATCH (n)-[r]->(m)
-RETURN n, r, m
+MATCH p=()-[r:INCLUDES]->() 
+RETURN p LIMIT 250
 """
 data = graph.run(query).data()
 
@@ -19,17 +19,22 @@ net = Network(height="750px", width="100%", notebook=True, cdn_resources='in_lin
 
 # 添加节点和边
 for item in data:
-    net.add_node(item['n'].identity, 
-                label=item['n']['name'],
-                title=item['n'].get('description', ''),
-                group=item['n'].labels[0])
-    net.add_node(item['m'].identity, 
-                label=item['m']['name'],
-                title=item['m'].get('description', ''),
-                group=item['m'].labels[0])
-    net.add_edge(item['n'].identity, 
-                item['m'].identity, 
-                title=type(item['r']).__name__)
+    # 提取起始节点
+    start_node = item['p'].start_node
+    # 提取结束节点
+    end_node = item['p'].end_node
+    
+    net.add_node(start_node.identity,
+                label=start_node['name'],
+                title=start_node.get('description', ''),
+                group=list(start_node.labels)[0])
+    net.add_node(end_node.identity,
+                label=end_node['name'],
+                title=end_node.get('description', ''),
+                group=list(end_node.labels)[0])
+    net.add_edge(start_node.identity,
+                end_node.identity,
+                title="INCLUDES")
 
 # 生成HTML文件
 with open("knowledge_graph.html", "w", encoding='utf-8') as f: