diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py index b8c3fdc4e..c49d20697 100644 --- a/libs/langgraph/langgraph/pregel/loop.py +++ b/libs/langgraph/langgraph/pregel/loop.py @@ -73,10 +73,10 @@ class PregelLoop: checkpointer: Optional[BaseCheckpointSaver] checkpointer_get_next_version: Callable[[Optional[V]], V] checkpointer_put_writes: Optional[ - Callable[[RunnableConfig, Sequence[tuple[str, Any]], str], ...] + Callable[[RunnableConfig, Sequence[tuple[str, Any]], str], Any] ] checkpointer_put: Optional[ - Callable[[RunnableConfig, Checkpoint, CheckpointMetadata], ...] + Callable[[RunnableConfig, Checkpoint, CheckpointMetadata], Any] ] graph: "Pregel"