MyMiddleware Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  8c574d58_1f4a_c56d_49f8_177cce7f7bcc["MyMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  8c574d58_1f4a_c56d_49f8_177cce7f7bcc -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  8c574d58_1f4a_c56d_49f8_177cce7f7bcc -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  8d3bffd7_818e_92df_028f_7ddeaca51628["before_model()"]
  8c574d58_1f4a_c56d_49f8_177cce7f7bcc -->|method| 8d3bffd7_818e_92df_028f_7ddeaca51628
  c480d0fb_b331_c174_d91b_3b7ba5cbce89["after_model()"]
  8c574d58_1f4a_c56d_49f8_177cce7f7bcc -->|method| c480d0fb_b331_c174_d91b_3b7ba5cbce89

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 467–474

    class MyMiddleware(AgentMiddleware):
        def before_model(self, state: AgentState[Any], runtime: Runtime) -> dict[str, Any]:
            assert "jump_to" not in state
            return {"jump_to": "model"}

        def after_model(self, state: AgentState[Any], runtime: Runtime) -> dict[str, Any]:
            assert "jump_to" not in state
            return {"jump_to": "model"}