New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

checkpoint postgres: add a shallow checkpointer #2826

Merged

vbarda merged 14 commits into main from vb/add-shallow-checkpointer

Dec 20, 2024

-Original file line number
+Diff line change
@@ -1,4 +1,4 @@
     import threading
     from collections.abc import Iterator, Sequence
     from contextlib import contextmanager
     from typing import Any, Optional
@@ Expand All / @@ -19,6 +19,7 @@ @@
     )
     from langgraph.checkpoint.postgres import _internal
     from langgraph.checkpoint.postgres.base import BasePostgresSaver
+    from langgraph.checkpoint.postgres.shallow import ShallowPostgresSaver
     from langgraph.checkpoint.serde.base import SerializerProtocol
     Conn = _internal.Conn  # For backward compatibility
@@ Expand Down Expand Up / @@ -396,4 +397,4 @@ @@
                         yield cur
-    __all__ = ["PostgresSaver", "BasePostgresSaver", "Conn"]
+    __all__ = ["PostgresSaver", "BasePostgresSaver", "ShallowPostgresSaver", "Conn"]

-Original file line number
+Diff line change
@@ Expand Up / @@ -19,6 +19,7 @@ @@
     )
     from langgraph.checkpoint.postgres import _ainternal
     from langgraph.checkpoint.postgres.base import BasePostgresSaver
+    from langgraph.checkpoint.postgres.shallow import AsyncShallowPostgresSaver
     from langgraph.checkpoint.serde.base import SerializerProtocol
     Conn = _ainternal.Conn  # For backward compatibility
@@ Expand Down Expand Up / @@ -464,4 +465,4 @@ def put_writes( @@
             ).result()
-    __all__ = ["AsyncPostgresSaver", "Conn"]
+    __all__ = ["AsyncPostgresSaver", "AsyncShallowPostgresSaver", "Conn"]

Provide feedback