From d04570f1784f28f82f3b2fb2af62b4d08bae30b4 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 5 May 2025 12:16:25 -0700 Subject: [PATCH] Lint --- libs/langgraph/langgraph/pregel/algo.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/libs/langgraph/langgraph/pregel/algo.py b/libs/langgraph/langgraph/pregel/algo.py index 7449cd01d..ff94ec0f6 100644 --- a/libs/langgraph/langgraph/pregel/algo.py +++ b/libs/langgraph/langgraph/pregel/algo.py @@ -387,6 +387,8 @@ def prepare_next_tasks( manager: Literal[None] = None, trigger_to_nodes: Optional[Mapping[str, Sequence[str]]] = None, updated_channels: Optional[set[str]] = None, + retry_policy: Sequence[RetryPolicy] = (), + cache_policy: Literal[None] = None, ) -> dict[str, PregelTask]: ... @@ -406,6 +408,8 @@ def prepare_next_tasks( manager: Union[None, ParentRunManager, AsyncParentRunManager], trigger_to_nodes: Optional[Mapping[str, Sequence[str]]] = None, updated_channels: Optional[set[str]] = None, + retry_policy: Sequence[RetryPolicy] = (), + cache_policy: Optional[CachePolicy] = None, ) -> dict[str, PregelExecutableTask]: ...