WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-11 12:17:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

Nicer NotImplementedError in sqlite checkpointer (#598)

Browse Source
This commit is contained in:
William FH
2024-06-05 06:35:21 -07:00
committed by GitHub
parent 683495aceb
commit eac9c600aa
2 changed files with 95 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/checkpoint/test_sqlite.py
+19 -1
View File
@@ -3,7 +3,12 @@ from langchain_core.runnables import RunnableConfig
from langgraph.channels.base import create_checkpoint
from langgraph.checkpoint.base import Checkpoint, CheckpointMetadata, empty_checkpoint
from langgraph.checkpoint.sqlite import SqliteSaver, _metadata_predicate, search_where
from langgraph.checkpoint.sqlite import (
_AIO_ERROR_MSG,
SqliteSaver,
_metadata_predicate,
search_where,
)
class TestSqliteSaver:
@@ -98,3 +103,16 @@ class TestSqliteSaver:
expected_predicate_3,
expected_param_values_3,
)
async def test_informative_async_errors(self):
# call method / assertions
with pytest.raises(NotImplementedError, match=_AIO_ERROR_MSG):
await self.sqlite_saver.aget(self.config_1)
with pytest.raises(NotImplementedError, match=_AIO_ERROR_MSG):
await self.sqlite_saver.aget_tuple(self.config_1)
with pytest.raises(NotImplementedError, match=_AIO_ERROR_MSG):
async for _ in self.sqlite_saver.alist(self.config_1):
pass
with pytest.raises(NotImplementedError, match=_AIO_ERROR_MSG):
async for _ in self.sqlite_saver.asearch(self.metadata_1):
pass