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
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=
|
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 |
)
|