SuffixMiddleware Class — langchain Architecture

Architecture documentation for the SuffixMiddleware class in test_wrap_model_call.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  8ca35a63_a297_2520_de66_23864bf5b41c["SuffixMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  8ca35a63_a297_2520_de66_23864bf5b41c -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  573473e5_eb93_a7d7_84ba_c975071c09af["test_wrap_model_call.py"]
  8ca35a63_a297_2520_de66_23864bf5b41c -->|defined in| 573473e5_eb93_a7d7_84ba_c975071c09af
  08db9e62_87ed_ee01_e1ab_0bf7d07ce168["wrap_model_call()"]
  8ca35a63_a297_2520_de66_23864bf5b41c -->|method| 08db9e62_87ed_ee01_e1ab_0bf7d07ce168

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call.py lines 854–863

        class SuffixMiddleware(AgentMiddleware):
            def wrap_model_call(
                self,
                request: ModelRequest,
                handler: Callable[[ModelRequest], ModelResponse],
            ) -> ModelCallResult:
                result = handler(request)
                # result is ModelResponse, extract AIMessage from it
                ai_message = result.result[0]
                return AIMessage(content=f"{ai_message.content} [SUFFIX]")