From d512db26e4969975e670964c2964d2d1cd15a923 Mon Sep 17 00:00:00 2001 From: yangdx Date: Thu, 26 Jun 2025 13:50:19 +0800 Subject: [PATCH] Fix MongoDB set handling in delete operations --- lightrag/kg/mongo_impl.py | 8 ++++++++ 1 file changed, 8 insertions(+) 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(