From d14f98f01ba9bf1fd56804e837082af869450c3e Mon Sep 17 00:00:00 2001 From: vbarda Date: Tue, 15 Oct 2024 09:07:00 -0400 Subject: [PATCH] langgraph: fix type for RemainingSteps --- libs/langgraph/langgraph/managed/is_last_step.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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]