From b934677a81d81e74aab3b3705f581aebb5450ebb Mon Sep 17 00:00:00 2001 From: Mark Pors Date: Sun, 11 Jun 2023 18:49:26 +0200 Subject: [PATCH] Obey handler.raise_error in _ahandle_event_for_handler (#6001) Obey `handler.raise_error` in `_ahandle_event_for_handler` Exceptions for async callbacks were only logged as warnings, also when `raise_error = True` #### Who can review? @hwchase17 @agola11 --- langchain/callbacks/manager.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/langchain/callbacks/manager.py b/langchain/callbacks/manager.py index 2069195f..2c935003 100644 --- a/langchain/callbacks/manager.py +++ b/langchain/callbacks/manager.py @@ -238,6 +238,8 @@ async def _ahandle_event_for_handler( else: logger.warning(f"Error in {event_name} callback: {e}") except Exception as e: + if handler.raise_error: + raise e logger.warning(f"Error in {event_name} callback: {e}")