langgraph-checkpoint-postgres 3.2.0__tar.gz → 3.2.1__tar.gz

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: langgraph-checkpoint-postgres
-Version: 3.2.0
+Version: 3.2.1
 Summary: Library with a Postgres implementation of LangGraph checkpoint saver.
 Project-URL: Source, https://github.com/langchain-ai/langgraph/tree/main/libs/checkpoint-postgres
 Project-URL: Twitter, https://x.com/langchain_oss

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/langgraph/checkpoint/postgres/aio.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import asyncio
 from collections import defaultdict
 from collections.abc import AsyncIterator, Iterator, Mapping, Sequence
-from contextlib import asynccontextmanager
+from contextlib import asynccontextmanager, nullcontext
 from typing import Any, cast
 
 from langchain_core.runnables import RunnableConfig
@@ -425,7 +425,12 @@ class AsyncPostgresSaver(BasePostgresSaver):
                 Will be applied regardless of whether the AsyncPostgresSaver instance was initialized with a pipeline.
                 If pipeline mode is not supported, will fall back to using transaction context manager.
         """
-        async with self.lock, _ainternal.get_connection(self.conn) as conn:
+        # A pool hands each call its own connection, so only a shared one needs
+        # the lock.
+        lock = (
+            nullcontext() if isinstance(self.conn, AsyncConnectionPool) else self.lock
+        )
+        async with lock, _ainternal.get_connection(self.conn) as conn:
             if self.pipe:
                 # a connection in pipeline mode can be used concurrently
                 # in multiple threads/coroutines, but only one cursor can be
@@ -594,7 +599,11 @@ class AsyncPostgresSaver(BasePostgresSaver):
                 if value["parent_checkpoint_id"]
                 else None
             ),
-            await asyncio.to_thread(self._load_writes, value["pending_writes"]),
+            (
+                await asyncio.to_thread(self._load_writes, value["pending_writes"])
+                if value["pending_writes"]
+                else []
+            ),
         )
 
     def list(

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/langgraph/store/postgres/aio.py

@@ -413,7 +413,14 @@ class AsyncPostgresStore(AsyncBatchedBaseStore, BasePostgresStore[_ainternal.Con
     ) -> None:
         # Keep `conn` for compatibility with subclasses overriding this private hook.
         # All database I/O goes through `_cursor()`, which owns connection acquisition.
-        async with self._cursor(pipeline=True) as cur:
+        # A lone read is one statement: outside a pipeline it costs one round
+        # trip instead of two (its fetch, then the Sync on leaving the pipeline).
+        lone_read = (
+            not self.pipe
+            and PutOp not in grouped_ops
+            and sum(map(len, grouped_ops.values())) == 1
+        )
+        async with self._cursor(pipeline=not lone_read) as cur:
             if GetOp in grouped_ops:
                 await self._batch_get_ops(
                     cast(Sequence[tuple[int, GetOp]], grouped_ops[GetOp]),

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/langgraph/store/postgres/base.py

@@ -977,14 +977,19 @@ class PostgresStore(BaseStore, BasePostgresStore[_pg_internal.Conn]):
                     ):
                         yield cur
             else:
-                with conn.cursor(binary=True, row_factory=dict_row) as cur:
+                # Lone reads land here, so take the lock like the pipeline branch:
+                # another thread may be holding this connection in a pipeline.
+                with self.lock, conn.cursor(binary=True, row_factory=dict_row) as cur:
                     yield cur
 
     def batch(self, ops: Iterable[Op]) -> list[Result]:
         grouped_ops, num_ops = _group_ops(ops)
         results: list[Result] = [None] * num_ops
 
-        with self._cursor(pipeline=True) as cur:
+        # A lone read is one statement: outside a pipeline it costs one round
+        # trip instead of two (its fetch, then the Sync on leaving the pipeline).
+        lone_read = num_ops == 1 and PutOp not in grouped_ops and not self.pipe
+        with self._cursor(pipeline=not lone_read) as cur:
             if GetOp in grouped_ops:
                 self._batch_get_ops(
                     cast(Sequence[tuple[int, GetOp]], grouped_ops[GetOp]), results, cur

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-checkpoint-postgres"
-version = "3.2.0"
+version = "3.2.1"
 description = "Library with a Postgres implementation of LangGraph checkpoint saver."
 authors = []
 requires-python = ">=3.10"

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/tests/test_async.py

@@ -1,5 +1,6 @@
 # type: ignore
 
+import asyncio
 from contextlib import asynccontextmanager
 from typing import Any
 from uuid import uuid4
@@ -417,3 +418,33 @@ async def test_delta_channel_chain_reconstruction(saver_name: str) -> None:
         assert msgs[1].content == "reply-1"
         assert msgs[2].content == "there"
         assert msgs[3].content == "reply-3"
+
+
+async def test_pool_saver_reads_while_a_list_is_open(test_data) -> None:
+    async with _pool_saver() as saver:
+        config = await saver.aput(
+            test_data["configs"][0],
+            test_data["checkpoints"][0],
+            test_data["metadata"][0],
+            {},
+        )
+        listing = saver.alist(None)
+        try:
+            assert await anext(listing) is not None
+            assert await asyncio.wait_for(saver.aget_tuple(config), timeout=5)
+        finally:
+            await listing.aclose()
+
+
+async def test_base_saver_serializes_concurrent_calls(test_data) -> None:
+    async with _base_saver() as saver:
+        config = await saver.aput(
+            test_data["configs"][0],
+            test_data["checkpoints"][0],
+            test_data["metadata"][0],
+            {},
+        )
+        await asyncio.gather(
+            *(saver.aput_writes(config, [("x", i)], str(uuid4())) for i in range(20)),
+            *(saver.aget_tuple(config) for _ in range(20)),
+        )

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/tests/test_store.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import re
+import threading
 import time
 from contextlib import contextmanager
 from typing import Any
@@ -289,6 +290,30 @@ def test_basic_store_ops(store) -> None:
     assert deleted_item is None
 
 
+def test_lone_reads_and_puts_share_a_connection(store: PostgresStore) -> None:
+    errors: list[BaseException] = []
+
+    def run(op: Any) -> None:
+        try:
+            for i in range(100):
+                op(i)
+        except BaseException as e:
+            errors.append(e)
+
+    def read(i: int) -> None:
+        store.get(("race",), f"k{i % 10}")
+
+    def write(i: int) -> None:
+        store.put(("race",), f"k{i % 10}", {"i": i})
+
+    threads = [threading.Thread(target=run, args=(op,)) for op in [read, write] * 4]
+    for t in threads:
+        t.start()
+    for t in threads:
+        t.join()
+    assert errors == []
+
+
 def test_list_namespaces(store) -> None:
     # Create test data with various namespaces
     test_namespaces = [

{langgraph_checkpoint_postgres-3.2.0 → langgraph_checkpoint_postgres-3.2.1}/uv.lock

@@ -40,11 +40,11 @@ wheels = [
 
 [[package]]
 name = "certifi"
-version = "2026.1.4"
+version = "2026.10.19"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/e0/2d/a891ca51311197f6ad14a7ef42e2399f36cf2f9bd44752b3dc4eab60fdc5/certifi-2026.1.4.tar.gz", hash = "sha256:ac726dd470482006e014ad384921ed6438c457018f4b3d204aea4281258b2120", size = 154268, upload-time = "2026-01-04T02:42:41.825Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/a6/bd/1436c50cc46699de0230fbaacc1d1de7bfec413176099949e9d5112e88f2/certifi-2026.10.19.tar.gz", hash = "sha256:035c1e85572b141bf1c3a3eff0e33a8eb74917522f2559075ff8d2fea780a1ae", size = 137598, upload-time = "2026-10-20T02:24:16.604Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/e6/ad/3cc14f097111b4de0040c83a525973216457bbeeb63739ef1ed275c1c021/certifi-2026.1.4-py3-none-any.whl", hash = "sha256:9943707519e4add1115f44c2bc244f782c0249876bf51b6599fee1ffbedd685c", size = 152900, upload-time = "2026-01-04T02:42:40.15Z" },
+    { url = "https://files.pythonhosted.org/packages/c4/51/29046e48f693c54e5b7412216d447a1d8362c9653148d98c02cdf65fa733/certifi-2026.10.19-py3-none-any.whl", hash = "sha256:42d35340cc5b9ba78e818ddbff0c5b1c2b779de448a340fd1b989f5fa8670e1b", size = 136464, upload-time = "2026-10-20T02:24:15.072Z" },
 ]
 
 [[package]]
@@ -278,7 +278,7 @@ wheels = [
 
 [[package]]
 name = "langgraph-checkpoint"
-version = "4.3.0"
+version = "4.3.2"
 source = { editable = "../checkpoint" }
 dependencies = [
     { name = "langchain-core" },
@@ -353,7 +353,7 @@ test = [
 
 [[package]]
 name = "langgraph-checkpoint-postgres"
-version = "3.2.0"
+version = "3.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "langgraph-checkpoint" },