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

- Use f-strings for node IDs
- Use f-strings for edge IDs
- Ensure consistent ID formatting
This commit is contained in:
yangdx
2025-03-08 10:20:10 +08:00
parent 84222b8b76
commit 78f8d7a1ce

View File

@@ -865,17 +865,17 @@ class Neo4JStorage(BaseGraphStorage):
if b_node.labels: # Only process if target node has labels
# Create KnowledgeGraphNode for target
target_node = KnowledgeGraphNode(
id=target_id,
id=f"{target_id}",
labels=list(b_node.labels),
properties=dict(b_node),
)
# Create KnowledgeGraphEdge
target_edge = KnowledgeGraphEdge(
id=edge_id,
id=f"{edge_id}",
type=rel.type,
source=node.id,
target=target_id,
source=f"{node.id}",
target=f"{target_id}",
properties=dict(rel),
)
@@ -905,7 +905,7 @@ class Neo4JStorage(BaseGraphStorage):
# Create initial KnowledgeGraphNode
start_node = KnowledgeGraphNode(
id=str(node_record["node_id"]),
id=f"{node_record['node_id']}",
labels=list(node_record["n"].labels),
properties=dict(node_record["n"]),
)