ModifyMiddleware Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  378538a8_65f8_b6d7_ec5e_2b14261e804a["ModifyMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  378538a8_65f8_b6d7_ec5e_2b14261e804a -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  8310d669_2524_e019_e333_8473b50a4990["test_framework.py"]
  378538a8_65f8_b6d7_ec5e_2b14261e804a -->|defined in| 8310d669_2524_e019_e333_8473b50a4990
  79be80d5_fcbb_f961_c8c0_d0eb9e3d303b["wrap_model_call()"]
  378538a8_65f8_b6d7_ec5e_2b14261e804a -->|method| 79be80d5_fcbb_f961_c8c0_d0eb9e3d303b

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 251–258

    class ModifyMiddleware(AgentMiddleware):
        def wrap_model_call(
            self,
            request: ModelRequest,
            handler: Callable[[ModelRequest], ModelResponse],
        ) -> ModelCallResult:
            request.messages.append(HumanMessage("remember to be nice!"))
            return handler(request)