JumpMiddleware Class — langchain Architecture

Architecture documentation for the JumpMiddleware class in test_decorators.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  cd353dda_0cce_b520_4569_96a6f4d126f2["JumpMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  cd353dda_0cce_b520_4569_96a6f4d126f2 -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  85a104a3_11cf_6539_d0c0_ba69a17a41d4["test_decorators.py"]
  cd353dda_0cce_b520_4569_96a6f4d126f2 -->|defined in| 85a104a3_11cf_6539_d0c0_ba69a17a41d4
  68028dcd_ea5b_7d3a_0d75_c9ebd1cef92b["before_model()"]
  cd353dda_0cce_b520_4569_96a6f4d126f2 -->|method| 68028dcd_ea5b_7d3a_0d75_c9ebd1cef92b
  5db83528_8e08_d7fe_e2fc_f806b038e621["after_model()"]
  cd353dda_0cce_b520_4569_96a6f4d126f2 -->|method| 5db83528_8e08_d7fe_e2fc_f806b038e621

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_decorators.py lines 192–205

    class JumpMiddleware(AgentMiddleware):
        @hook_config(can_jump_to=["end", "model"])
        def before_model(
            self, state: AgentState[Any], runtime: Runtime[None]
        ) -> dict[str, Any] | None:
            if len(state["messages"]) > 5:
                return {"jump_to": "end"}
            return None

        @hook_config(can_jump_to=["tools"])
        def after_model(
            self, state: AgentState[Any], runtime: Runtime[None]
        ) -> dict[str, Any] | None:
            return {"jump_to": "tools"}