From 2bc0e2df4299259fc7e3187c3c288a0c5145dfc1 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 16 Aug 2024 15:53:05 -0700 Subject: [PATCH] Fix --- libs/langgraph/langgraph/channels/base.py | 5 ----- libs/langgraph/langgraph/managed/scoped_value.py | 3 +-- 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/libs/langgraph/langgraph/channels/base.py b/libs/langgraph/langgraph/channels/base.py index 9c7794c46..fe47f0d8f 100644 --- a/libs/langgraph/langgraph/channels/base.py +++ b/libs/langgraph/langgraph/channels/base.py @@ -33,11 +33,6 @@ class BaseChannel(Generic[Value, Update, C], ABC): # serialize/deserialize methods - def tap(self) -> Optional[C]: - """Return the current checkpoint of the channel, without consuming it. - By default, it just calls checkpoint().""" - return self.checkpoint() - @abstractmethod def checkpoint(self) -> Optional[C]: """Return a serializable representation of the channel's current state. diff --git a/libs/langgraph/langgraph/managed/scoped_value.py b/libs/langgraph/langgraph/managed/scoped_value.py index adb16e1b4..997bc9469 100644 --- a/libs/langgraph/langgraph/managed/scoped_value.py +++ b/libs/langgraph/langgraph/managed/scoped_value.py @@ -18,7 +18,6 @@ from langgraph.managed.base import ( ConfiguredManagedValue, WritableManagedValue, ) -from langgraph.pregel.types import PregelTaskDescription V = dict[str, Any] @@ -66,7 +65,7 @@ class ScopedValue(WritableManagedValue[Value, Update]): f"Scope {scope} for shared state key not in config.configurable" ) - def __call__(self, step: int, task: PregelTaskDescription) -> Value: + def __call__(self, step: int) -> Value: return self.value.copy() def _process_update(