WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-13 21:27:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

langgraph: fix type for RemainingSteps

Browse Source
This commit is contained in:
vbarda
2024-10-15 09:07:00 -04:00
parent c0b56bf60d
commit d14f98f01b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/managed/is_last_step.py
+3 -3
View File
@@ -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]