ML-Learner / server /src /checkpointer /__init__.py
VashuTheGreat2's picture
Upload folder using huggingface_hub
0225ffc verified
Raw History Blame Contribute Delete
403 Bytes
import aiosqlite
from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
from langgraph.checkpoint.memory import InMemorySaver
_checkpointer = None
async def get_checkpointer():
global _checkpointer
if _checkpointer is None:
conn = await aiosqlite.connect("db.sqlite")
_checkpointer = AsyncSqliteSaver(conn)
# _checkpointer=InMemorySaver()
return _checkpointer