AsyncMiddlewareTwo Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  e0a6b01f_f139_6ff0_b07a_68f2b9dc89a9["AsyncMiddlewareTwo"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  e0a6b01f_f139_6ff0_b07a_68f2b9dc89a9 -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  e0a6b01f_f139_6ff0_b07a_68f2b9dc89a9 -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  6d695cb8_ec5f_61b7_136d_5bd61717af57["abefore_model()"]
  e0a6b01f_f139_6ff0_b07a_68f2b9dc89a9 -->|method| 6d695cb8_ec5f_61b7_136d_5bd61717af57
  a5762091_b2c2_d2b8_9f9a_28ac8f8d0d1b["awrap_model_call()"]
  e0a6b01f_f139_6ff0_b07a_68f2b9dc89a9 -->|method| a5762091_b2c2_d2b8_9f9a_28ac8f8d0d1b
  d4d8331a_0b89_d0fd_9077_328fe501f864["aafter_model()"]
  e0a6b01f_f139_6ff0_b07a_68f2b9dc89a9 -->|method| d4d8331a_0b89_d0fd_9077_328fe501f864

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 631–644

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

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

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