checkpoint-postgres: fix setup for AsyncPostgresSaver (#1294)

langchain-ai · Aug 9, 2024 · 6595869 · 6595869
1 parent c303ef2
commit 6595869
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/libs/checkpoint-postgres/langgraph/checkpoint/postgres/aio.py b/libs/checkpoint-postgres/langgraph/checkpoint/postgres/aio.py
@@ -66,11 +66,10 @@ async def setup(self) -> None:
         async with self.lock:
             async with self.conn.cursor(binary=True) as cur:
                 try:
-                    version = (
-                        await cur.execute(
-                            "SELECT v FROM checkpoint_migrations ORDER BY v DESC LIMIT 1"
-                        )
-                    ).fetchone()["v"]
+                    results = await cur.execute(
+                        "SELECT v FROM checkpoint_migrations ORDER BY v DESC LIMIT 1"
+                    )
+                    version = (await results.fetchone())["v"]
                 except UndefinedTable:
                     version = -1
                 for v, migration in zip(