Check if conn is alive (#381)

This commit is contained in:
William FH
2024-05-03 11:38:18 -07:00
committed by GitHub
parent d6091f5fa4
commit b185efe365
+2 -2
View File
@@ -52,8 +52,8 @@ class AsyncSqliteSaver(BaseCheckpointSaver, AbstractAsyncContextManager):
async def setup(self) -> None:
if self.is_setup:
return
await self.conn
if not self.conn.is_alive():
await self.conn
async with self.conn.executescript(
"""
CREATE TABLE IF NOT EXISTS checkpoints (