Home / Function/ test_runtime_injected_into_middleware() — langchain Function Reference

test_runtime_injected_into_middleware() — langchain Function Reference

Architecture documentation for the test_runtime_injected_into_middleware() function in test_framework.py from the langchain codebase.

Entity Profile

Dependency Diagram

graph TD
  4ba86910_398e_51dd_3b8a_1ce3327780c2["test_runtime_injected_into_middleware()"]
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  4ba86910_398e_51dd_3b8a_1ce3327780c2 -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  72ef93cf_d7c1_f508_47a7_a7badc3939ff["after_model()"]
  4ba86910_398e_51dd_3b8a_1ce3327780c2 -->|calls| 72ef93cf_d7c1_f508_47a7_a7badc3939ff
  style 4ba86910_398e_51dd_3b8a_1ce3327780c2 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 385–404

def test_runtime_injected_into_middleware() -> None:
    """Test that the runtime is injected into the middleware."""

    class CustomMiddleware(AgentMiddleware):
        def before_model(self, state: AgentState[Any], runtime: Runtime) -> None:
            assert runtime is not None

        def wrap_model_call(
            self,
            request: ModelRequest,
            handler: Callable[[ModelRequest], ModelResponse],
        ) -> ModelCallResult:
            assert request.runtime is not None
            return handler(request)

        def after_model(self, state: AgentState[Any], runtime: Runtime) -> None:
            assert runtime is not None

    agent = create_agent(model=FakeToolCallingModel(), middleware=[CustomMiddleware()])
    agent.invoke({"messages": [HumanMessage("Hello")]})

Domain

Subdomains

Frequently Asked Questions

What does test_runtime_injected_into_middleware() do?
test_runtime_injected_into_middleware() is a function in the langchain codebase, defined in libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py.
Where is test_runtime_injected_into_middleware() defined?
test_runtime_injected_into_middleware() is defined in libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py at line 385.
What does test_runtime_injected_into_middleware() call?
test_runtime_injected_into_middleware() calls 1 function(s): after_model.

Analyze Your Own Codebase

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

Try Supermodel Free