ModelSwappingMiddleware Class — langchain Architecture

Architecture documentation for the ModelSwappingMiddleware class in test_response_format.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  3af14c36_b09f_fc28_d887_95a53c65f883["ModelSwappingMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  3af14c36_b09f_fc28_d887_95a53c65f883 -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  2970c11e_b43a_df0c_1a52_30da10d3689c["test_response_format.py"]
  3af14c36_b09f_fc28_d887_95a53c65f883 -->|defined in| 2970c11e_b43a_df0c_1a52_30da10d3689c
  8bf7d888_c163_f6dc_ebf8_00b118b0434e["wrap_model_call()"]
  3af14c36_b09f_fc28_d887_95a53c65f883 -->|method| 8bf7d888_c163_f6dc_ebf8_00b118b0434e

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/test_response_format.py lines 825–832

        class ModelSwappingMiddleware(AgentMiddleware):
            def wrap_model_call(
                self,
                request: ModelRequest,
                handler: Callable[[ModelRequest], ModelResponse],
            ) -> ModelCallResult:
                # Replace the model with our custom test model
                return handler(request.override(model=model))