raise if no handler found

langchain-ai · Jan 8, 2025 · 0103122 · 0103122
1 parent 5edd223
commit 0103122
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libs/langgraph/langgraph/pregel/retry.py b/libs/langgraph/langgraph/pregel/retry.py
@@ -174,6 +174,8 @@ async def arun_with_retry(
                 exception_handler = retry_policy.retry_on.get(type(exc), False)
                 if callable(exception_handler):
                     exception_handler(task.input, exc)
+                else:
+                    raise
             else:
                 raise TypeError(
                     "retry_on must be an Exception class, a list or tuple of Exception classes, or a callable"