diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 2fdbb270..109ba59d 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -159,6 +159,10 @@ class MongoKVStorage(BaseKVStorage): if not ids: return + # Convert to list if it's a set (MongoDB BSON cannot encode sets) + if isinstance(ids, set): + ids = list(ids) + try: result = await self._data.delete_many({"_id": {"$in": ids}}) logger.info( @@ -1044,6 +1048,10 @@ class MongoVectorDBStorage(BaseVectorStorage): if not ids: return + # Convert to list if it's a set (MongoDB BSON cannot encode sets) + if isinstance(ids, set): + ids = list(ids) + try: result = await self._data.delete_many({"_id": {"$in": ids}}) logger.debug(