community[patch]: Better error propagation for neo4jgraph (#17190)

There are other errors that could happen when refreshing the schema, so we want to propagate specific errors for more clarity
langchain-ai · Feb 8, 2024 · 7e4b676 · 7e4b676
1 parent d903fa3
commit 7e4b676
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/libs/community/langchain_community/graphs/neo4j_graph.py b/libs/community/langchain_community/graphs/neo4j_graph.py
@@ -133,12 +133,14 @@ def __init__(
         # Set schema
         try:
             self.refresh_schema()
-        except neo4j.exceptions.ClientError:
-            raise ValueError(
-                "Could not use APOC procedures. "
-                "Please ensure the APOC plugin is installed in Neo4j and that "
-                "'apoc.meta.data()' is allowed in Neo4j configuration "
-            )
+        except neo4j.exceptions.ClientError as e:
+            if e.code == "Neo.ClientError.Procedure.ProcedureNotFound":
+                raise ValueError(
+                    "Could not use APOC procedures. "
+                    "Please ensure the APOC plugin is installed in Neo4j and that "
+                    "'apoc.meta.data()' is allowed in Neo4j configuration "
+                )
+            raise e
 
     @property
     def get_schema(self) -> str: