diff --git a/libs/langchain_v1/tests/unit_tests/agents/test_injected_runtime_create_agent.py b/libs/langchain_v1/tests/unit_tests/agents/test_injected_runtime_create_agent.py index 05efc0b289d..ff8c88788c5 100644 --- a/libs/langchain_v1/tests/unit_tests/agents/test_injected_runtime_create_agent.py +++ b/libs/langchain_v1/tests/unit_tests/agents/test_injected_runtime_create_agent.py @@ -254,9 +254,10 @@ def test_tool_runtime_config_access() -> None: config_data["has_configurable"] = ( "configurable" in runtime.config if runtime.config else False ) - # Config may have run_id or other fields depending on execution context if runtime.config: config_data["config_keys"] = list(runtime.config.keys()) + config_data["recursion_limit"] = runtime.config.get("recursion_limit") + config_data["metadata"] = runtime.config.get("metadata") return f"Config accessed for {x}" agent = create_agent( @@ -270,13 +271,26 @@ def test_tool_runtime_config_access() -> None: system_prompt="You are a helpful assistant.", ) - result = agent.invoke({"messages": [HumanMessage("Test config")]}) + result = agent.invoke( + {"messages": [HumanMessage("Test config")]}, + ) - # Verify config was accessible assert config_data["config_exists"] is True assert "config_keys" in config_data + assert config_data["recursion_limit"] == 9999 + assert config_data["metadata"]["ls_integration"] == "langchain_create_agent" + + tool_message = result["messages"][2] + assert isinstance(tool_message, ToolMessage) + assert tool_message.content == "Config accessed for 5" + + result = agent.invoke( + {"messages": [HumanMessage("Test config again")]}, + config={"recursion_limit": 7}, + ) + + assert config_data["recursion_limit"] == 7 - # Verify tool executed tool_message = result["messages"][2] assert isinstance(tool_message, ToolMessage) assert tool_message.content == "Config accessed for 5"