AsyncMiddlewareOne Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  586fd389_254b_4efe_88b4_bc088fcdf935["AsyncMiddlewareOne"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  586fd389_254b_4efe_88b4_bc088fcdf935 -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  586fd389_254b_4efe_88b4_bc088fcdf935 -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  0243a80e_2368_783f_2053_6ba50e3df0df["abefore_model()"]
  586fd389_254b_4efe_88b4_bc088fcdf935 -->|method| 0243a80e_2368_783f_2053_6ba50e3df0df
  64ee8fe4_37ac_24fd_7c18_563c86e0ee35["awrap_model_call()"]
  586fd389_254b_4efe_88b4_bc088fcdf935 -->|method| 64ee8fe4_37ac_24fd_7c18_563c86e0ee35
  2ba261c7_077b_2c87_534a_59b560eda7b1["aafter_model()"]
  586fd389_254b_4efe_88b4_bc088fcdf935 -->|method| 2ba261c7_077b_2c87_534a_59b560eda7b1

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 616–629

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

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

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