langchain Architecture Docs
Home / Function/ test_composition_preserves_state_and_runtime() — langchain Function Reference

test_composition_preserves_state_and_runtime() — langchain Function Reference

Architecture documentation for the test_composition_preserves_state_and_runtime() function in test_composition.py from the langchain codebase.

Function python LangChainCore Runnables calls 2

Entity Profile

Dependency Diagram

graph TD
  83445f37_7449_c29b_4f01_f92b178a8e2f["test_composition_preserves_state_and_runtime()"]
  f1e766c6_5a64_9e6c_f2a8_d41c64f6f142["TestChainModelCallHandlers"]
  83445f37_7449_c29b_4f01_f92b178a8e2f -->|defined in| f1e766c6_5a64_9e6c_f2a8_d41c64f6f142
  3988450e_19ed_2280_0a85_e9520f7f556b["create_test_request()"]
  83445f37_7449_c29b_4f01_f92b178a8e2f -->|calls| 3988450e_19ed_2280_0a85_e9520f7f556b
  1cdbb163_cfa8_588a_643a_640bbf32247f["create_mock_base_handler()"]
  83445f37_7449_c29b_4f01_f92b178a8e2f -->|calls| 1cdbb163_cfa8_588a_643a_640bbf32247f
  style 83445f37_7449_c29b_4f01_f92b178a8e2f fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_composition.py lines 270–310

    def test_composition_preserves_state_and_runtime(self) -> None:
        """Test that state and runtime are passed through composition."""

        class CustomState(AgentState[Any]):
            test: str

        class CustomContext(TypedDict):
            test: str

        state_values = []
        runtime_values = []

        def outer(
            request: ModelRequest, handler: Callable[[ModelRequest], ModelResponse]
        ) -> ModelResponse:
            state_values.append(("outer", request.state))
            runtime_values.append(("outer", request.runtime))
            return handler(request)

        def inner(
            request: ModelRequest, handler: Callable[[ModelRequest], ModelResponse]
        ) -> ModelResponse:
            state_values.append(("inner", request.state))
            runtime_values.append(("inner", request.runtime))
            return handler(request)

        composed = _chain_model_call_handlers([outer, inner])
        assert composed is not None

        test_state = CustomState(messages=[], test="state")
        test_runtime = Runtime(context=CustomContext(test="runtime"))

        # Create request with state and runtime
        test_request = create_test_request(state=test_state, runtime=test_runtime)
        result = composed(test_request, create_mock_base_handler())

        # Both handlers should see same state and runtime
        assert state_values == [("outer", test_state), ("inner", test_state)]
        assert runtime_values == [("outer", test_runtime), ("inner", test_runtime)]
        assert isinstance(result, _ComposedExtendedModelResponse)
        assert result.model_response.result[0].content == "test"

Domain

Subdomains

Defined In

Calls

Source

Frequently Asked Questions

What does test_composition_preserves_state_and_runtime() do?
test_composition_preserves_state_and_runtime() is a function in the langchain codebase, defined in libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_composition.py.
Where is test_composition_preserves_state_and_runtime() defined?
test_composition_preserves_state_and_runtime() is defined in libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_composition.py at line 270.
What does test_composition_preserves_state_and_runtime() call?
test_composition_preserves_state_and_runtime() calls 2 function(s): create_mock_base_handler, create_test_request.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free