MixedMiddleware Class — langchain Architecture

Architecture documentation for the MixedMiddleware class in test_framework.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  cf73aa4b_70fa_11a3_b38b_90f5eba492dc["MixedMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  cf73aa4b_70fa_11a3_b38b_90f5eba492dc -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  cf73aa4b_70fa_11a3_b38b_90f5eba492dc -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  a41c6a20_8b97_0a4a_2b59_d2382a4f2d2f["before_model()"]
  cf73aa4b_70fa_11a3_b38b_90f5eba492dc -->|method| a41c6a20_8b97_0a4a_2b59_d2382a4f2d2f
  f207db9d_3f7a_9c6f_d85c_5e48d79f2dee["abefore_model()"]
  cf73aa4b_70fa_11a3_b38b_90f5eba492dc -->|method| f207db9d_3f7a_9c6f_d85c_5e48d79f2dee
  19d59b52_7973_0844_eb57_2971a2ebf227["wrap_model_call()"]
  cf73aa4b_70fa_11a3_b38b_90f5eba492dc -->|method| 19d59b52_7973_0844_eb57_2971a2ebf227
  10e1d990_cfdc_a164_2985_e0b72b2fdbb1["awrap_model_call()"]
  cf73aa4b_70fa_11a3_b38b_90f5eba492dc -->|method| 10e1d990_cfdc_a164_2985_e0b72b2fdbb1

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 507–528

    class MixedMiddleware(AgentMiddleware):
        def before_model(self, state: AgentState[Any], runtime: Runtime) -> None:
            calls.append("MixedMiddleware.before_model")

        async def abefore_model(self, state: AgentState[Any], runtime: Runtime) -> None:
            calls.append("MixedMiddleware.abefore_model")

        def wrap_model_call(
            self,
            request: ModelRequest,
            handler: Callable[[ModelRequest], ModelResponse],
        ) -> ModelCallResult:
            calls.append("MixedMiddleware.wrap_model_call")
            return handler(request)

        async def awrap_model_call(
            self,
            request: ModelRequest,
            handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
        ) -> ModelCallResult:
            calls.append("MixedMiddleware.awrap_model_call")
            return await handler(request)