diff --git a/libs/checkpoint/langgraph/store/base/__init__.py b/libs/checkpoint/langgraph/store/base/__init__.py index ab3d5d89e..dea018551 100644 --- a/libs/checkpoint/langgraph/store/base/__init__.py +++ b/libs/checkpoint/langgraph/store/base/__init__.py @@ -682,7 +682,7 @@ class BaseStore(ABC): Returns: The retrieved item or None if not found. """ - return self.batch([GetOp(namespace, key, refresh_ttl)])[0] + return self.batch([GetOp(namespace, str(key), refresh_ttl)])[0] def search( self, @@ -811,7 +811,7 @@ class BaseStore(ABC): f"TTL is not supported by {self.__class__.__name__}. " f"Use a store implementation that supports TTL or set ttl=None." ) - self.batch([PutOp(namespace, key, value, index=index, ttl=ttl)]) + self.batch([PutOp(namespace, str(key), value, index=index, ttl=ttl)]) def delete(self, namespace: tuple[str, ...], key: str) -> None: """Delete an item. @@ -820,7 +820,7 @@ class BaseStore(ABC): namespace: Hierarchical path for the item. key: Unique identifier within the namespace. """ - self.batch([PutOp(namespace, key, None, ttl=None)]) + self.batch([PutOp(namespace, str(key), None, ttl=None)]) def list_namespaces( self, @@ -887,7 +887,7 @@ class BaseStore(ABC): Returns: The retrieved item or None if not found. """ - return (await self.abatch([GetOp(namespace, key, refresh_ttl)]))[0] + return (await self.abatch([GetOp(namespace, str(key), refresh_ttl)]))[0] async def asearch( self, @@ -1027,7 +1027,7 @@ class BaseStore(ABC): f"TTL is not supported by {self.__class__.__name__}. " f"Use a store implementation that supports TTL or set ttl=None." ) - await self.abatch([PutOp(namespace, key, value, index=index, ttl=ttl)]) + await self.abatch([PutOp(namespace, str(key), value, index=index, ttl=ttl)]) async def adelete(self, namespace: tuple[str, ...], key: str) -> None: """Asynchronously delete an item. @@ -1036,7 +1036,7 @@ class BaseStore(ABC): namespace: Hierarchical path for the item. key: Unique identifier within the namespace. """ - await self.abatch([PutOp(namespace, key, None)]) + await self.abatch([PutOp(namespace, str(key), None)]) async def alist_namespaces( self, diff --git a/libs/checkpoint/pyproject.toml b/libs/checkpoint/pyproject.toml index 07a092f2c..249bf3728 100644 --- a/libs/checkpoint/pyproject.toml +++ b/libs/checkpoint/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "langgraph-checkpoint" -version = "2.0.17" +version = "2.0.18" description = "Library with base interfaces for LangGraph checkpoint savers." authors = [] license = "MIT"