AsyncCustomMiddleware Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  162efaa4_ef41_10af_b2b7_3ff2e1c0727c["AsyncCustomMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  162efaa4_ef41_10af_b2b7_3ff2e1c0727c -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  162efaa4_ef41_10af_b2b7_3ff2e1c0727c -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  dc05ca59_23b5_cbf7_ea94_ff774cf0fc5a["abefore_agent()"]
  162efaa4_ef41_10af_b2b7_3ff2e1c0727c -->|method| dc05ca59_23b5_cbf7_ea94_ff774cf0fc5a
  7322ead4_4fab_6293_fe16_c328ed9af04d["aafter_agent()"]
  162efaa4_ef41_10af_b2b7_3ff2e1c0727c -->|method| 7322ead4_4fab_6293_fe16_c328ed9af04d

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 836–849

        class AsyncCustomMiddleware(AgentMiddleware):
            async def abefore_agent(
                self, state: AgentState[Any], runtime: Runtime
            ) -> dict[str, Any] | None:
                if hook_type == "before":
                    execution_log.append("hook_called")
                return None

            async def aafter_agent(
                self, state: AgentState[Any], runtime: Runtime
            ) -> dict[str, Any] | None:
                if hook_type == "after":
                    execution_log.append("hook_called")
                return None