handle connection error

FalkorDB · Feb 13, 2024 · e47f6fe · e47f6fe
1 parent 7fcbe8d
commit e47f6fe
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/libs/community/langchain_community/graphs/falkordb_graph.py b/libs/community/langchain_community/graphs/falkordb_graph.py
@@ -65,9 +65,13 @@ def __init__(
                 "Please install it with `pip install falkordb`."
             )
 
-        self._driver = FalkorDB(
-            host=host, port=port, username=username, password=password, ssl=ssl
-        )
+        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] = {}