This commit is contained in:
Nuno Campos
2024-08-21 09:30:21 -07:00
parent 1dc09dc45f
commit 2bc0e2df42
2 changed files with 1 additions and 7 deletions
@@ -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.
@@ -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(