TestMiddleware Class — langchain Architecture

Architecture documentation for the TestMiddleware class in test_injected_runtime_create_agent.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  b33b7e95_8cbe_3b00_6fa1_08add218b8fb["TestMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  b33b7e95_8cbe_3b00_6fa1_08add218b8fb -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  a4d83fe9_0a39_5761_b1a5_de54f81974c9["test_injected_runtime_create_agent.py"]
  b33b7e95_8cbe_3b00_6fa1_08add218b8fb -->|defined in| a4d83fe9_0a39_5761_b1a5_de54f81974c9
  a7926c87_146a_18d7_fbc4_3911991d3131["before_model()"]
  b33b7e95_8cbe_3b00_6fa1_08add218b8fb -->|method| a7926c87_146a_18d7_fbc4_3911991d3131
  e50fdb36_ee70_53c9_2dea_eac2952227d3["after_model()"]
  b33b7e95_8cbe_3b00_6fa1_08add218b8fb -->|method| e50fdb36_ee70_53c9_2dea_eac2952227d3

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/test_injected_runtime_create_agent.py lines 469–476

    class TestMiddleware(AgentMiddleware):
        def before_model(self, state: AgentState[Any], runtime: Runtime) -> dict[str, Any]:
            middleware_calls.append("before_model")
            return {}

        def after_model(self, state: AgentState[Any], runtime: Runtime) -> dict[str, Any]:
            middleware_calls.append("after_model")
            return {}