From 124a26335e2b0eb27e8f0d0a5ee94efac6752f1c Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Tue, 14 Jan 2025 13:58:56 +0100
Subject: [PATCH] feat: changes model independent edge method

---
 cognee/infrastructure/databases/graph/neo4j_driver/adapter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py b/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py
index 7988497c6..d632fbd81 100644
--- a/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py
+++ b/cognee/infrastructure/databases/graph/neo4j_driver/adapter.py
@@ -430,7 +430,7 @@ class Neo4jAdapter(GraphDBInterface):
         query_nodes = "MATCH (n) RETURN collect(n) AS nodes"
         nodes = await self.query(query_nodes)
 
-        query_edges = "MATCH ()-[r]->() RETURN collect(r) AS relationships"
+        query_edges = "MATCH (n)-[r]->(m) RETURN collect([n, r, m]) AS elements"
         edges = await self.query(query_edges)
 
         return (nodes, edges)