From 0c2eeb873b3a01d15d5767b2ac0870cfb6ea9fec Mon Sep 17 00:00:00 2001 From: Boris Arzentar Date: Fri, 14 Nov 2025 12:20:46 +0100 Subject: [PATCH] fix: revert neo4j get_graph_data change --- .../databases/graph/neo4j_driver/adapter.py | 17 +---------------- cognee/tests/test_delete_dataset_neo4j.py | 7 +++++-- 2 files changed, 6 insertions(+), 18 deletions(-) diff --git a/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py b/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py index 6ff8f7645..ae2dd0edd 100644 --- a/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py +++ b/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py @@ -955,22 +955,7 @@ class Neo4jAdapter(GraphDBInterface): f"Retrieved {len(nodes)} nodes and {len(edges)} edges in {retrieval_time:.2f} seconds" ) - return ( - [ - ( - node_id, - { - **node_data, - "id": node_id, - "metadata": json.loads(node_data["metadata"]) - if "metadata" in node_data - else {}, - }, - ) - for (node_id, node_data) in nodes - ], - edges, - ) + return (nodes, edges) except Exception as e: logger.error(f"Error during graph data retrieval: {str(e)}") diff --git a/cognee/tests/test_delete_dataset_neo4j.py b/cognee/tests/test_delete_dataset_neo4j.py index 55e17432c..4f064484d 100644 --- a/cognee/tests/test_delete_dataset_neo4j.py +++ b/cognee/tests/test_delete_dataset_neo4j.py @@ -1,4 +1,5 @@ import os +import json import pathlib import pytest from unittest.mock import AsyncMock, patch @@ -149,7 +150,8 @@ async def main(mock_create_structured_output: AsyncMock): for node in initial_nodes: node_data = node[1] - collection_name = node_data["type"] + "_" + node_data["metadata"]["index_fields"][0] + node_metadata = json.loads(node_data["metadata"]) + collection_name = node_data["type"] + "_" + node_metadata["index_fields"][0] if collection_name not in initial_nodes_by_vector_collection: initial_nodes_by_vector_collection[collection_name] = [] initial_nodes_by_vector_collection[collection_name].append(node) @@ -172,7 +174,8 @@ async def main(mock_create_structured_output: AsyncMock): after_delete_nodes_by_vector_collection = {} for node in initial_nodes: node_data = node[1] - collection_name = node_data["type"] + "_" + node_data["metadata"]["index_fields"][0] + node_metadata = json.loads(node_data["metadata"]) + collection_name = node_data["type"] + "_" + node_metadata["index_fields"][0] if collection_name not in after_delete_nodes_by_vector_collection: after_delete_nodes_by_vector_collection[collection_name] = [] after_delete_nodes_by_vector_collection[collection_name].append(node)