From 8916f8a912582725353b996ebbc595ab79e1cca2 Mon Sep 17 00:00:00 2001 From: Gardel Date: Thu, 22 May 2025 04:41:52 +0800 Subject: [PATCH] feat: add delete method for mongo storage implement --- lightrag/kg/mongo_impl.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index d49a36b7..6f7233a4 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -311,6 +311,17 @@ class MongoDocStatusStorage(DocStatusStorage): logger.error(f"Error dropping doc status {self._collection_name}: {e}") return {"status": "error", "message": str(e)} + async def delete(self, ids: list[str]) -> None: + try: + result = await self._data.delete_many({"_id": {"$in": ids}}) + deleted_count = result.deleted_count + + logger.info( + f"Dropped {deleted_count} documents from doc status {self._collection_name}" + ) + except PyMongoError as e: + logger.error(f"Error deleting doc status {self._collection_name}: {e}") + @final @dataclass