RetryMiddleware Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  0bea8529_6c7b_cff0_33a7_0b0979b10dd9["RetryMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  0bea8529_6c7b_cff0_33a7_0b0979b10dd9 -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  573473e5_eb93_a7d7_84ba_c975071c09af["test_wrap_model_call.py"]
  0bea8529_6c7b_cff0_33a7_0b0979b10dd9 -->|defined in| 573473e5_eb93_a7d7_84ba_c975071c09af
  3f15e359_06d9_adf5_5127_6f7a71c9513d["wrap_model_call()"]
  0bea8529_6c7b_cff0_33a7_0b0979b10dd9 -->|method| 3f15e359_06d9_adf5_5127_6f7a71c9513d
  bd51b4b6_85b2_b55f_719b_5240c1a33cae["awrap_model_call()"]
  0bea8529_6c7b_cff0_33a7_0b0979b10dd9 -->|method| bd51b4b6_85b2_b55f_719b_5240c1a33cae

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call.py lines 807–822

        class RetryMiddleware(AgentMiddleware):
            def wrap_model_call(
                self,
                request: ModelRequest,
                handler: Callable[[ModelRequest], ModelResponse],
            ) -> ModelCallResult:
                log.append("retry-before")
                try:
                    result = handler(request)
                    log.append("retry-after")
                except Exception:
                    log.append("retry-retrying")
                    result = handler(request)
                    log.append("retry-after")

                return result