langgraph-checkpoint-postgres 2.0.0__tar.gz → 2.0.2__tar.gz

{langgraph_checkpoint_postgres-2.0.0 → langgraph_checkpoint_postgres-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-checkpoint-postgres
-Version: 2.0.0
+Version: 2.0.2
 Summary: Library with a Postgres implementation of LangGraph checkpoint saver.
 Home-page: https://www.github.com/langchain-ai/langgraph
 License: MIT
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: langgraph-checkpoint (>=2.0.0,<3.0.0)
+Requires-Dist: langgraph-checkpoint (>=2.0.2,<3.0.0)
 Requires-Dist: orjson (>=3.10.1)
 Requires-Dist: psycopg (>=3.0.0,<4.0.0)
 Requires-Dist: psycopg-pool (>=3.0.0,<4.0.0)

{langgraph_checkpoint_postgres-2.0.0 → langgraph_checkpoint_postgres-2.0.2}/langgraph/checkpoint/postgres/__init__.py

@@ -287,7 +287,7 @@ class PostgresSaver(BasePostgresSaver):
             >>> DB_URI = "postgres://postgres:postgres@localhost:5432/postgres?sslmode=disable"
             >>> with PostgresSaver.from_conn_string(DB_URI) as memory:
             >>>     config = {"configurable": {"thread_id": "1", "checkpoint_ns": ""}}
-            >>>     checkpoint = {"ts": "2024-05-04T06:32:42.235444+00:00", "id": "1ef4f797-8335-6428-8001-8a1503f9b875", "data": {"key": "value"}}
+            >>>     checkpoint = {"ts": "2024-05-04T06:32:42.235444+00:00", "id": "1ef4f797-8335-6428-8001-8a1503f9b875", "channel_values": {"key": "value"}}
             >>>     saved_config = memory.put(config, checkpoint, {"source": "input", "step": 1, "writes": {"key": "value"}}, {})
             >>> print(saved_config)
             {'configurable': {'thread_id': '1', 'checkpoint_ns': '', 'checkpoint_id': '1ef4f797-8335-6428-8001-8a1503f9b875'}}

{langgraph_checkpoint_postgres-2.0.0 → langgraph_checkpoint_postgres-2.0.2}/langgraph/checkpoint/postgres/base.py

@@ -249,7 +249,7 @@ class BasePostgresSaver(BaseCheckpointSaver[str]):
         filter: MetadataInput,
         before: Optional[RunnableConfig] = None,
     ) -> Tuple[str, List[Any]]:
-        """Return WHERE clause predicates for alist() given config, filter, cursor.
+        """Return WHERE clause predicates for alist() given config, filter, before.
 
         This method returns a tuple of a string and a tuple of values. The string
         is the parametered WHERE clause predicate (including the WHERE keyword):

{langgraph_checkpoint_postgres-2.0.0 → langgraph_checkpoint_postgres-2.0.2}/langgraph/store/postgres/aio.py

@@ -18,6 +18,7 @@ from psycopg.errors import UndefinedTable
 from psycopg.rows import dict_row
 
 from langgraph.store.base import GetOp, ListNamespacesOp, Op, PutOp, Result, SearchOp
+from langgraph.store.base.batch import AsyncBatchedBaseStore
 from langgraph.store.postgres.base import (
     BasePostgresStore,
     Row,
@@ -29,7 +30,9 @@ from langgraph.store.postgres.base import (
 logger = logging.getLogger(__name__)
 
 
-class AsyncPostgresStore(BasePostgresStore[AsyncConnection]):
+class AsyncPostgresStore(AsyncBatchedBaseStore, BasePostgresStore[AsyncConnection]):
+    __slots__ = ("_deserializer",)
+
     def __init__(
         self,
         conn: AsyncConnection[Any],
@@ -38,8 +41,8 @@ class AsyncPostgresStore(BasePostgresStore[AsyncConnection]):
             Callable[[Union[bytes, orjson.Fragment]], dict[str, Any]]
         ] = None,
     ) -> None:
-        super().__init__(deserializer=deserializer)
-        self.conn = conn
+        super().__init__()
+        self._deserializer = deserializer
         self.conn = conn
         self.loop = asyncio.get_running_loop()
 

{langgraph_checkpoint_postgres-2.0.0 → langgraph_checkpoint_postgres-2.0.2}/langgraph/store/postgres/base.py

@@ -59,20 +59,10 @@ CREATE INDEX IF NOT EXISTS store_prefix_idx ON store USING btree (prefix text_pa
 C = TypeVar("C", bound=BaseConnection)
 
 
-class BasePostgresStore(BaseStore, Generic[C]):
+class BasePostgresStore(Generic[C]):
     MIGRATIONS = MIGRATIONS
     conn: C
-    __slots__ = ("_deserializer",)
-
-    def __init__(
-        self,
-        *,
-        deserializer: Optional[
-            Callable[[Union[bytes, orjson.Fragment]], dict[str, Any]]
-        ] = None,
-    ) -> None:
-        super().__init__()
-        self._deserializer = deserializer
+    _deserializer: Optional[Callable[[Union[bytes, orjson.Fragment]], dict[str, Any]]]
 
     def _get_batch_GET_ops_queries(
         self,
@@ -166,7 +156,9 @@ class BasePostgresStore(BaseStore, Generic[C]):
                         params.extend([key, json.dumps(value)])
                 query += " AND " + " AND ".join(filter_conditions)
 
-            query += " LIMIT %s OFFSET %s"
+            # Note: we will need to not do this if sim/keyword search
+            # is used
+            query += " ORDER BY updated_at DESC LIMIT %s OFFSET %s"
             params.extend([op.limit, op.offset])
 
             queries.append((query, params))
@@ -227,7 +219,9 @@ class BasePostgresStore(BaseStore, Generic[C]):
         return queries
 
 
-class PostgresStore(BasePostgresStore[Connection]):
+class PostgresStore(BaseStore, BasePostgresStore[Connection]):
+    __slots__ = ("_deserializer",)
+
     def __init__(
         self,
         conn: Connection[Any],
@@ -236,7 +230,8 @@ class PostgresStore(BasePostgresStore[Connection]):
             Callable[[Union[bytes, orjson.Fragment]], dict[str, Any]]
         ] = None,
     ) -> None:
-        super().__init__(deserializer=deserializer)
+        super().__init__()
+        self._deserializer = deserializer
         self.conn = conn
 
     def batch(self, ops: Iterable[Op]) -> list[Result]:

{langgraph_checkpoint_postgres-2.0.0 → langgraph_checkpoint_postgres-2.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-checkpoint-postgres"
-version = "2.0.0"
+version = "2.0.2"
 description = "Library with a Postgres implementation of LangGraph checkpoint saver."
 authors = []
 license = "MIT"
@@ -10,7 +10,7 @@ packages = [{ include = "langgraph" }]
 
 [tool.poetry.dependencies]
 python = "^3.9.0,<4.0"
-langgraph-checkpoint = "^2.0.0"
+langgraph-checkpoint = "^2.0.2"
 orjson = ">=3.10.1"
 psycopg = "^3.0.0"
 psycopg-pool = "^3.0.0"