langchain Architecture Docs
Home / Class/ StateTrackingRetryMiddleware Class — langchain Architecture

StateTrackingRetryMiddleware Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  103bb31d_b66a_20da_7d3b_0391feec34a9["StateTrackingRetryMiddleware"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  103bb31d_b66a_20da_7d3b_0391feec34a9 -->|extends| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  573473e5_eb93_a7d7_84ba_c975071c09af["test_wrap_model_call.py"]
  103bb31d_b66a_20da_7d3b_0391feec34a9 -->|defined in| 573473e5_eb93_a7d7_84ba_c975071c09af
  7220784b_fa87_33e6_bd30_6aad9018dd85["wrap_model_call()"]
  103bb31d_b66a_20da_7d3b_0391feec34a9 -->|method| 7220784b_fa87_33e6_bd30_6aad9018dd85

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call.py lines 637–650

        class StateTrackingRetryMiddleware(AgentMiddleware):
            def wrap_model_call(
                self,
                request: ModelRequest,
                handler: Callable[[ModelRequest], ModelResponse],
            ) -> ModelCallResult:
                max_retries = 2
                for attempt in range(max_retries):
                    try:
                        return handler(request)
                    except Exception:
                        if attempt == max_retries - 1:
                            raise
                pytest.fail("Should have raised an exception")

Defined In

Extends

Source

Frequently Asked Questions

What is the StateTrackingRetryMiddleware class?
StateTrackingRetryMiddleware is a class in the langchain codebase, defined in libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call.py.
Where is StateTrackingRetryMiddleware defined?
StateTrackingRetryMiddleware is defined in libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call.py at line 637.
What does StateTrackingRetryMiddleware extend?
StateTrackingRetryMiddleware extends AgentMiddleware.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free