remove now-unnecessary try block

langchain-ai · Nov 28, 2024 · c8e8f5d · c8e8f5d
1 parent eae1819
commit c8e8f5d
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/libs/checkpoint-postgres/langgraph/checkpoint/postgres/aio.py b/libs/checkpoint-postgres/langgraph/checkpoint/postgres/aio.py
@@ -84,17 +84,17 @@ async def setup(self) -> None:
             await cur.execute(
                 self.MIGRATIONS[0]
             )
-            try:
-                results = await cur.execute(
-                    "SELECT v FROM checkpoint_migrations ORDER BY v DESC LIMIT 1"
-                )
-                row = await results.fetchone()
-                if row is None:
-                    version = -1
-                else:
-                    version = row["v"]
-            except UndefinedTable:
+            await cur.execute(
+                self.MIGRATIONS[0]
+            )
+            results = await cur.execute(
+                "SELECT v FROM checkpoint_migrations ORDER BY v DESC LIMIT 1"
+            )
+            row = await results.fetchone()
+            if row is None:
                 version = -1
+            else:
+                version = row["v"]
             for v, migration in zip(
                 range(version + 1, len(self.MIGRATIONS)),
                 self.MIGRATIONS[version + 1 :],