Remove option to only checkpoint at end of run

- Now that checkpoint at end of each step adds no latency there is not point to keep this
- This will make it easier to add future features
This commit is contained in:
Nuno Campos
2024-05-06 11:52:49 -07:00
parent f304908102
commit 3ff3def62b
12 changed files with 303 additions and 1327 deletions
+1 -5
View File
@@ -3,7 +3,6 @@ from typing import Any, Optional
from langgraph.checkpoint.base import (
Checkpoint,
CheckpointAt,
SerializerProtocol,
copy_checkpoint,
)
@@ -21,17 +20,14 @@ class NoopSerializer(SerializerProtocol):
class MemorySaverAssertImmutable(MemorySaver):
serde = NoopSerializer()
at = CheckpointAt.END_OF_STEP
storage_for_copies: defaultdict[str, dict[str, Checkpoint]]
def __init__(
self,
*,
serde: Optional[SerializerProtocol] = None,
at: Optional[CheckpointAt] = None,
) -> None:
super().__init__(serde=serde, at=at)
super().__init__(serde=serde)
self.storage_for_copies = defaultdict(dict)
def put(self, config: dict, checkpoint: Checkpoint) -> None: