yangdx commited on
Commit
411c452
·
1 Parent(s): ab56eca

Convert node and edge IDs to f-strings for consistency.

Browse files

- Use f-strings for node IDs
- Use f-strings for edge IDs
- Ensure consistent ID formatting

Files changed (1) hide show
  1. lightrag/kg/neo4j_impl.py +5 -5
lightrag/kg/neo4j_impl.py CHANGED
@@ -865,17 +865,17 @@ class Neo4JStorage(BaseGraphStorage):
865
  if b_node.labels: # Only process if target node has labels
866
  # Create KnowledgeGraphNode for target
867
  target_node = KnowledgeGraphNode(
868
- id=target_id,
869
  labels=list(b_node.labels),
870
  properties=dict(b_node),
871
  )
872
 
873
  # Create KnowledgeGraphEdge
874
  target_edge = KnowledgeGraphEdge(
875
- id=edge_id,
876
  type=rel.type,
877
- source=node.id,
878
- target=target_id,
879
  properties=dict(rel),
880
  )
881
 
@@ -905,7 +905,7 @@ class Neo4JStorage(BaseGraphStorage):
905
 
906
  # Create initial KnowledgeGraphNode
907
  start_node = KnowledgeGraphNode(
908
- id=str(node_record["node_id"]),
909
  labels=list(node_record["n"].labels),
910
  properties=dict(node_record["n"]),
911
  )
 
865
  if b_node.labels: # Only process if target node has labels
866
  # Create KnowledgeGraphNode for target
867
  target_node = KnowledgeGraphNode(
868
+ id=f"{target_id}",
869
  labels=list(b_node.labels),
870
  properties=dict(b_node),
871
  )
872
 
873
  # Create KnowledgeGraphEdge
874
  target_edge = KnowledgeGraphEdge(
875
+ id=f"{edge_id}",
876
  type=rel.type,
877
+ source=f"{node.id}",
878
+ target=f"{target_id}",
879
  properties=dict(rel),
880
  )
881
 
 
905
 
906
  # Create initial KnowledgeGraphNode
907
  start_node = KnowledgeGraphNode(
908
+ id=f"{node_record['node_id']}",
909
  labels=list(node_record["n"].labels),
910
  properties=dict(node_record["n"]),
911
  )