Catch and log exceptions occurring while executing a langchain agent …

…node
nv-morpheus · Feb 1, 2024 · b381084 · b381084
1 parent 6dae8fe
commit b381084
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/morpheus/llm/nodes/langchain_agent_node.py b/morpheus/llm/nodes/langchain_agent_node.py
@@ -66,7 +66,10 @@ async def _run_single(self, **kwargs: dict[str, typing.Any]) -> dict[str, typing
             return results
 
         # We are not dealing with a list, so run single
-        return await self._agent_executor.arun(**kwargs)
+        try:
+            return await self._agent_executor.arun(**kwargs)
+        except Exception as e:
+            logger.exception("Error running agent: %s: %s", kwargs, e)
 
     async def execute(self, context: LLMContext) -> LLMContext: