First commit

tests/unit/test_langgraph_checkpoint_saver.py

@@ -0,0 +1,14 @@
+import pytest
+from types import SimpleNamespace
+from agent_framework.checkpoints.langgraph_saver import RepositoryCheckpointSaver
+
+@pytest.mark.asyncio
+async def test_repository_checkpoint_saver_put_get(tmp_path):
+    settings = SimpleNamespace(CHECKPOINT_REPOSITORY_PROVIDER='sqlite', SQLITE_DB_PATH=str(tmp_path/'db.sqlite'))
+    saver = RepositoryCheckpointSaver(settings)
+    config = {'configurable': {'thread_id': 't1'}}
+    next_config = await saver.aput(config, {'id': 'cp1', 'channel_values': {'x': 1}}, {'source': 'test'}, {})
+    assert next_config['configurable']['checkpoint_id'] == 'cp1'
+    tup = await saver.aget_tuple(config)
+    checkpoint = tup.checkpoint if hasattr(tup, 'checkpoint') else tup['checkpoint']
+    assert checkpoint['id'] == 'cp1'