From cde05795cad55f889e4bfb897bbce478a6568219 Mon Sep 17 00:00:00 2001 From: Cristian Pufu Date: Mon, 14 Apr 2025 08:14:07 +0300 Subject: [PATCH] feat: configurable recursion_limit and max_concurrency --- src/uipath_langchain/_cli/_runtime/_runtime.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/uipath_langchain/_cli/_runtime/_runtime.py b/src/uipath_langchain/_cli/_runtime/_runtime.py index ff567a07..62fa06f2 100644 --- a/src/uipath_langchain/_cli/_runtime/_runtime.py +++ b/src/uipath_langchain/_cli/_runtime/_runtime.py @@ -94,6 +94,14 @@ async def execute(self) -> Optional[UiPathRuntimeResult]: "callbacks": callbacks, } + recursion_limit = os.environ.get("LANGCHAIN_RECURSION_LIMIT", None) + max_concurrency = os.environ.get("LANGCHAIN_MAX_CONCURRENCY", None) + + if recursion_limit is not None: + graph_config["recursion_limit"] = int(recursion_limit) + if max_concurrency is not None: + graph_config["max_concurrency"] = int(max_concurrency) + # Stream the output at debug time if self.context.job_id is None: # Get final chunk while streaming