From a2a1a42c754c390f13e4b6c68abd0cbdc48d4eee Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 2 May 2025 10:41:53 -0700 Subject: [PATCH] Fix --- libs/langgraph/langgraph/pregel/loop.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py index 0bc710e04..95fd863b7 100644 --- a/libs/langgraph/langgraph/pregel/loop.py +++ b/libs/langgraph/langgraph/pregel/loop.py @@ -616,10 +616,10 @@ class PregelLoop(LoopProtocol): return True - def match_cached_writes(self) -> list[PregelExecutableTask]: + def match_cached_writes(self) -> Sequence[PregelExecutableTask]: raise NotImplementedError - async def amatch_cached_writes(self) -> list[PregelExecutableTask]: + async def amatch_cached_writes(self) -> Sequence[PregelExecutableTask]: raise NotImplementedError # private @@ -1048,9 +1048,9 @@ class SyncPregelLoop(PregelLoop, AbstractContextManager): return self.submit(cast(WritableManagedValue, managed_value).update, values) - def match_cached_writes(self) -> list[PregelExecutableTask]: + def match_cached_writes(self) -> Sequence[PregelExecutableTask]: if self.cache is None: - return + return () matched: list[PregelExecutableTask] = [] if cached := { t.cache_key.key: t @@ -1228,9 +1228,9 @@ class AsyncPregelLoop(PregelLoop, AbstractAsyncContextManager): return self.submit(cast(WritableManagedValue, managed_value).aupdate, values) - async def amatch_cached_writes(self) -> None: + async def amatch_cached_writes(self) -> Sequence[PregelExecutableTask]: if self.cache is None: - return + return [] matched: list[PregelExecutableTask] = [] if cached := { t.cache_key.key: t