feat: add AgentConfig top lvl for create_react_agent

Author: cotovanu-cristian

src/uipath_langchain/agent/react/__init__.py

@@ -1,7 +1,7 @@
 """UiPath ReAct Agent implementation"""
 
 from .agent import create_agent
-from .state import AgentGraphNode, AgentGraphState
+from .types import AgentGraphNode, AgentGraphState
 from .utils import resolve_output_model
 
 __all__ = [

src/uipath_langchain/agent/react/agent.py

@@ -18,11 +18,11 @@
 from .router import (
     route_agent,
 )
-from .state import AgentGraphNode, AgentGraphState
 from .terminate_node import (
     create_terminate_node,
 )
 from .tools import create_flow_control_tools
+from .types import AgentGraphConfig, AgentGraphNode, AgentGraphState
 
 
 def create_agent(
@@ -32,13 +32,16 @@ def create_agent(
     *,
     state_schema: Type[AgentGraphState] = AgentGraphState,
     response_format: type[BaseModel] | None = None,
-    recursion_limit: int = 50,
+    config: AgentGraphConfig | None = None,
 ) -> StateGraph[AgentGraphState]:
     """Build agent graph with INIT -> AGENT <-> TOOLS loop, terminated by control flow tools.
 
     Control flow tools (end_execution, raise_error) are auto-injected alongside regular tools.
     """
-    os.environ["LANGCHAIN_RECURSION_LIMIT"] = str(recursion_limit)
+    if config is None:
+        config = AgentGraphConfig()
+
+    os.environ["LANGCHAIN_RECURSION_LIMIT"] = str(config.recursion_limit)
 
     agent_tools = list(tools)
     flow_control_tools: list[BaseTool] = create_flow_control_tools(response_format)

src/uipath_langchain/agent/react/init_node.py

@@ -4,7 +4,7 @@
 
 from langchain_core.messages import HumanMessage, SystemMessage
 
-from .state import AgentGraphState
+from .types import AgentGraphState
 
 
 def create_init_node(

src/uipath_langchain/agent/react/llm_node.py

@@ -7,7 +7,7 @@
 from langchain_core.tools import BaseTool
 
 from .constants import MAX_SUCCESSIVE_COMPLETIONS
-from .state import AgentGraphState
+from .types import AgentGraphState
 from .utils import count_successive_completions
 
 

src/uipath_langchain/agent/react/router.py

@@ -7,7 +7,7 @@
 
 from .constants import MAX_SUCCESSIVE_COMPLETIONS
 from .exceptions import AgentNodeRoutingException
-from .state import AgentGraphNode, AgentGraphState
+from .types import AgentGraphNode, AgentGraphState
 from .utils import count_successive_completions
 
 FLOW_CONTROL_TOOLS = [END_EXECUTION_TOOL.name, RAISE_ERROR_TOOL.name]

src/uipath_langchain/agent/react/terminate_node.py

@@ -11,7 +11,7 @@
     AgentNodeRoutingException,
     AgentTerminationException,
 )
-from .state import AgentGraphState
+from .types import AgentGraphState
 
 
 def create_terminate_node(

src/uipath_langchain/agent/react/state.py src/uipath_langchain/agent/react/types.pysrc/uipath_langchain/agent/react/state.py renamed to src/uipath_langchain/agent/react/types.py

@@ -3,6 +3,7 @@
 from enum import StrEnum
 
 from langgraph.graph import MessagesState
+from pydantic import BaseModel, Field
 
 
 class AgentGraphState(MessagesState):
@@ -16,3 +17,9 @@ class AgentGraphNode(StrEnum):
     AGENT = "agent"
     TOOLS = "tools"
     TERMINATE = "terminate"
+
+
+class AgentGraphConfig(BaseModel):
+    recursion_limit: int = Field(
+        default=50, ge=1, description="Maximum recursion limit for the agent graph"
+    )