fix queries

yoheinakajima · Dec 20, 2023 · 3d179a0 · 3d179a0
1 parent 56a5c49
commit 3d179a0
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/drivers/falkordb.py b/drivers/falkordb.py
@@ -24,20 +24,14 @@ def get_graph_data(self) -> tuple[list[dict[str, Any]], list[dict[str, Any]]]:
         nodes = self.driver.query(
             """
         MATCH (n)
-        WITH collect(
-            {data: {id: n.id, label: n.label, color: n.color}}) AS node
-        RETURN node
+        RETURN {id: n.id, label: n.label, color: n.color}
         """
         )
-        nodes = [el["node"] for el in nodes][0]
 
         edges = self.driver.query(
             """
         MATCH (s)-[r]->(t)
-        WITH collect(
-            {data: {source: s.id, target: t.id, label:r.type, color: r.color}}
-        ) AS rel
-        RETURN rel
+        return {source: s.id, target: t.id, label:r.type, color: r.color}
         """
         )
         edges = [el["rel"] for el in edges][0]