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

2026-09-12 20:57:52 +02:00 · 2024-08-09 13:39:30 +00:00
parent c303ef2a2b
commit 659586947d
1 changed files with 4 additions and 5 deletions
@@ -66,11 +66,10 @@ class AsyncPostgresSaver(BasePostgresSaver):
        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(