From 93897c6df8483c6559ea5d19eeff7f96e38ea485 Mon Sep 17 00:00:00 2001 From: Andrew Nguonly Date: Fri, 17 May 2024 10:41:20 -0700 Subject: [PATCH] Enable write ahead log mode in SqliteSaver and AsyncSqliteSaver. --- langgraph/checkpoint/aiosqlite.py | 1 + langgraph/checkpoint/sqlite.py | 1 + 2 files changed, 2 insertions(+) diff --git a/langgraph/checkpoint/aiosqlite.py b/langgraph/checkpoint/aiosqlite.py index 0159645c2..1e9734e7a 100644 --- a/langgraph/checkpoint/aiosqlite.py +++ b/langgraph/checkpoint/aiosqlite.py @@ -191,6 +191,7 @@ class AsyncSqliteSaver(BaseCheckpointSaver, AbstractAsyncContextManager): await self.conn async with self.conn.executescript( """ + PRAGMA journal_mode=WAL; CREATE TABLE IF NOT EXISTS checkpoints ( thread_id TEXT NOT NULL, thread_ts TEXT NOT NULL, diff --git a/langgraph/checkpoint/sqlite.py b/langgraph/checkpoint/sqlite.py index f72a6f58f..23da9a7df 100644 --- a/langgraph/checkpoint/sqlite.py +++ b/langgraph/checkpoint/sqlite.py @@ -159,6 +159,7 @@ class SqliteSaver(BaseCheckpointSaver, AbstractContextManager): self.conn.executescript( """ + PRAGMA journal_mode=WAL; CREATE TABLE IF NOT EXISTS checkpoints ( thread_id TEXT NOT NULL, thread_ts TEXT NOT NULL,