switch to use the falkrdb-py client

docs/docs/use_cases/graph/integrations/graph_falkordb_qa.ipynb

@@ -16,7 +16,7 @@
     "FalkorDB is a low latency property graph database management system. You can simply run its docker locally:\n",
     "\n",
     "```bash\n",
-    "docker run -p 6379:6379 -it --rm falkordb/falkordb:edge\n",
+    "docker run -p 6379:6379 -it --rm falkordb/falkordb\n",
     "```\n",
     "\n",
     "Once launched, you can simply start creating a database on the local machine and connect to it."

libs/community/langchain_community/graphs/falkordb_graph.py

@@ -58,18 +58,21 @@ def __init__(
     ) -> None:
         """Create a new FalkorDB graph wrapper instance."""
         try:
-            import redis
-            from redis.commands.graph import Graph
+            from falkordb import FalkorDB
         except ImportError:
             raise ImportError(
-                "Could not import redis python package. "
-                "Please install it with `pip install redis`."
+                "Could not import falkordb python package. "
+                "Please install it with `pip install falkordb`."
             )
 
-        self._driver = redis.Redis(
-            host=host, port=port, username=username, password=password, ssl=ssl
-        )
-        self._graph = Graph(self._driver, database)
+        try:
+            self._driver = FalkorDB(
+                host=host, port=port, username=username, password=password, ssl=ssl
+            )
+        except Exception as e:
+            raise ConnectionError(f"Failed to connect to FalkorDB: {e}")
+
+        self._graph = self._driver.select_graph(database)
         self.schema: str = ""
         self.structured_schema: Dict[str, Any] = {}