diff --git a/libs/langgraph/langgraph/managed/is_last_step.py b/libs/langgraph/langgraph/managed/is_last_step.py index a4ac277d3..ee7794802 100644 --- a/libs/langgraph/langgraph/managed/is_last_step.py +++ b/libs/langgraph/langgraph/managed/is_last_step.py @@ -11,9 +11,9 @@ class IsLastStepManager(ManagedValue[bool]): IsLastStep = Annotated[bool, IsLastStepManager] -class RemainingStepsManager(ManagedValue[bool]): - def __call__(self) -> bool: +class RemainingStepsManager(ManagedValue[int]): + def __call__(self) -> int: return self.loop.stop - self.loop.step -RemainingSteps = Annotated[bool, RemainingStepsManager] +RemainingSteps = Annotated[int, RemainingStepsManager]