MostlySyncMiddleware Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  90ee81dd_26cf_08d8_d185_e5ea9052946b["MostlySyncMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  90ee81dd_26cf_08d8_d185_e5ea9052946b -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  90ee81dd_26cf_08d8_d185_e5ea9052946b -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  79d43dca_0561_c279_c472_317e6bd19324["before_model()"]
  90ee81dd_26cf_08d8_d185_e5ea9052946b -->|method| 79d43dca_0561_c279_c472_317e6bd19324
  52a9340b_bf14_42cd_68f6_292be68d8d1e["wrap_model_call()"]
  90ee81dd_26cf_08d8_d185_e5ea9052946b -->|method| 52a9340b_bf14_42cd_68f6_292be68d8d1e
  239f6e31_4925_40f9_53aa_7104bf495d5c["awrap_model_call()"]
  90ee81dd_26cf_08d8_d185_e5ea9052946b -->|method| 239f6e31_4925_40f9_53aa_7104bf495d5c
  c3635e27_cc14_1814_2662_bc84be46e105["after_model()"]
  90ee81dd_26cf_08d8_d185_e5ea9052946b -->|method| c3635e27_cc14_1814_2662_bc84be46e105

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 704–725

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

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

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

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