From cd03b14743fb01e128fa543742c64580039eeb04 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 1 Feb 2024 14:12:06 -0300 Subject: [PATCH] Refactor error handling and remove duplicate code --- src/backend/langflow/processing/base.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/backend/langflow/processing/base.py b/src/backend/langflow/processing/base.py index 06dc0a063..b887f2192 100644 --- a/src/backend/langflow/processing/base.py +++ b/src/backend/langflow/processing/base.py @@ -70,14 +70,13 @@ async def get_result_and_steps(langchain_object, inputs: Union[dict, str], **kwa except Exception as exc: logger.error(f"Error fixing memory inputs: {exc}") + trace_id = kwargs.pop("session_id", None) try: - trace_id = kwargs.pop("session_id", None) callbacks = setup_callbacks(sync=False, trace_id=trace_id, **kwargs) output = await langchain_object.acall(inputs, callbacks=callbacks) except Exception as exc: # make the error message more informative logger.debug(f"Error: {str(exc)}") - trace_id = kwargs.pop("session_id", None) callbacks = setup_callbacks(sync=True, trace_id=trace_id, **kwargs) output = langchain_object(inputs, callbacks=callbacks)