revert changes in legacy code

FalkorDB · Apr 2, 2024 · e254ab3 · e254ab3
1 parent 1efeff8
commit e254ab3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/llama-index-legacy/llama_index/legacy/graph_stores/falkordb.py b/llama-index-legacy/llama_index/legacy/graph_stores/falkordb.py
@@ -27,14 +27,14 @@ def __init__(
         **kwargs: Any,
     ) -> None:
         try:
-            import falkordb
+            import redis
         except ImportError:
-            raise ImportError("Please install falkordb client: pip install falkordb")
+            raise ImportError("Please install redis client: pip install redis")
 
         """Initialize params."""
         self._node_label = node_label
 
-        self._driver = falkordb.FalkorDB.from_url(url).select_graph(database)
+        self._driver = redis.Redis.from_url(url).graph(database)
         self._driver.query(f"CREATE INDEX FOR (n:`{self._node_label}`) ON (n.id)")
 
         self._database = database