fallback default drops

2025-02-18 08:43:23 +01:00
parent 6eed29b1cb
commit fc4b830036
3 changed files with 16 additions and 0 deletions
--- a/lightrag/kg/mongo_impl.py
+++ b/lightrag/kg/mongo_impl.py
@@ -117,6 +117,10 @@ class MongoKVStorage(BaseKVStorage):
        # Mongo handles persistence automatically
        pass

+    async def drop(self) -> None:
+        """Drop the collection"""
+        await self._data.drop()
+

@final
@dataclass
@@ -198,6 +202,10 @@ class MongoDocStatusStorage(DocStatusStorage):
        # Mongo handles persistence automatically
        pass

+    async def drop(self) -> None:
+        """Drop the collection"""
+        await self._data.drop()
+

@final
@dataclass