feat: configurable recursion_limit and max_concurrency

Author: cristipufu

src/uipath_langchain/_cli/_runtime/_runtime.py

@@ -1,9 +1,10 @@
 import json
 import logging
+import os
 from typing import List, Optional
 
 from langchain_core.callbacks.base import BaseCallbackHandler
-from langchain_core.runnables.config import RunnableConfig
+from langchain_core.runnables.config import DEFAULT_RECURSION_LIMIT, RunnableConfig
 from langchain_core.tracers.langchain import wait_for_all_tracers
 from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
 from langgraph.errors import EmptyInputError, GraphRecursionError, InvalidUpdateError
@@ -79,13 +80,29 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
                     )
                     callbacks = [tracer]
 
+                max_concurrency_env = os.environ.get("LANGCHAIN_MAX_CONCURRENCY", None)
+                max_concurrency = (
+                    int(max_concurrency_env)
+                    if max_concurrency_env is not None
+                    else None
+                )
+
+                recursion_limit_env = os.environ.get("LANGCHAIN_RECURSION_LIMIT", None)
+                recursion_limit = (
+                    int(recursion_limit_env)
+                    if recursion_limit_env is not None
+                    else DEFAULT_RECURSION_LIMIT
+                )
+
                 graph_config: RunnableConfig = {
                     "configurable": {
                         "thread_id": self.context.job_id
                         if self.context.job_id
                         else "default"
                     },
                     "callbacks": callbacks,
+                    "max_concurrency": max_concurrency,
+                    "recursion_limit": recursion_limit,
                 }
 
                 # Stream the output at debug time