MixedMiddleware Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  784d6d35_d2e6_dbe1_b0cc_28d564aa53df["MixedMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  784d6d35_d2e6_dbe1_b0cc_28d564aa53df -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  573473e5_eb93_a7d7_84ba_c975071c09af["test_wrap_model_call.py"]
  784d6d35_d2e6_dbe1_b0cc_28d564aa53df -->|defined in| 573473e5_eb93_a7d7_84ba_c975071c09af
  8f5d866e_cfd6_528f_2636_be74f4ef1e28["before_model()"]
  784d6d35_d2e6_dbe1_b0cc_28d564aa53df -->|method| 8f5d866e_cfd6_528f_2636_be74f4ef1e28
  57eb737e_c465_44cb_0f93_a9f298204d77["abefore_model()"]
  784d6d35_d2e6_dbe1_b0cc_28d564aa53df -->|method| 57eb737e_c465_44cb_0f93_a9f298204d77
  bd0114be_48f5_b652_f082_992869b1e854["wrap_model_call()"]
  784d6d35_d2e6_dbe1_b0cc_28d564aa53df -->|method| bd0114be_48f5_b652_f082_992869b1e854
  3b68e785_0b40_0816_9ee0_6eab22acf6ac["awrap_model_call()"]
  784d6d35_d2e6_dbe1_b0cc_28d564aa53df -->|method| 3b68e785_0b40_0816_9ee0_6eab22acf6ac

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call.py lines 1494–1517

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

            @override
            async def abefore_model(self, state: AgentState[Any], runtime: Runtime[Any]) -> 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)