diff --git a/lightrag/kg/json_doc_status_impl.py b/lightrag/kg/json_doc_status_impl.py index 6c667891..050752eb 100644 --- a/lightrag/kg/json_doc_status_impl.py +++ b/lightrag/kg/json_doc_status_impl.py @@ -67,7 +67,4 @@ class JsonDocStatusStorage(DocStatusStorage): async def delete(self, doc_ids: list[str]): for doc_id in doc_ids: self._data.pop(doc_id, None) - await self.index_done_callback() - - async def drop(self) -> None: - raise NotImplementedError + await self.index_done_callback() \ No newline at end of file diff --git a/lightrag/kg/json_kv_impl.py b/lightrag/kg/json_kv_impl.py index 658e1239..7e13dea7 100644 --- a/lightrag/kg/json_kv_impl.py +++ b/lightrag/kg/json_kv_impl.py @@ -46,9 +46,6 @@ class JsonKVStorage(BaseKVStorage): left_data = {k: v for k, v in data.items() if k not in self._data} self._data.update(left_data) - async def drop(self) -> None: - self._data = {} - async def delete(self, ids: list[str]) -> None: for doc_id in ids: self._data.pop(doc_id, None) diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 4eb968cf..384bb604 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -117,11 +117,6 @@ class MongoKVStorage(BaseKVStorage): # Mongo handles persistence automatically pass - async def drop(self) -> None: - """Drop the collection""" - await self._data.drop() - - @final @dataclass class MongoDocStatusStorage(DocStatusStorage): @@ -169,10 +164,6 @@ class MongoDocStatusStorage(DocStatusStorage): ) await asyncio.gather(*update_tasks) - async def drop(self) -> None: - """Drop the collection""" - await self._data.drop() - async def get_status_counts(self) -> dict[str, int]: """Get counts of documents in each status""" pipeline = [{"$group": {"_id": "$status", "count": {"$sum": 1}}}] diff --git a/lightrag/kg/oracle_impl.py b/lightrag/kg/oracle_impl.py index c9d8d1b5..16013fe8 100644 --- a/lightrag/kg/oracle_impl.py +++ b/lightrag/kg/oracle_impl.py @@ -320,10 +320,6 @@ class OracleKVStorage(BaseKVStorage): # Oracle handles persistence automatically pass - async def drop(self) -> None: - raise NotImplementedError - - @final @dataclass class OracleVectorDBStorage(BaseVectorStorage): diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py index 5f845894..34c7c54d 100644 --- a/lightrag/kg/postgres_impl.py +++ b/lightrag/kg/postgres_impl.py @@ -300,11 +300,7 @@ class PGKVStorage(BaseKVStorage): async def index_done_callback(self) -> None: # PG handles persistence automatically pass - - async def drop(self) -> None: - raise NotImplementedError - - + @final @dataclass class PGVectorStorage(BaseVectorStorage): @@ -534,9 +530,6 @@ class PGDocStatusStorage(DocStatusStorage): ) return data - async def drop(self) -> None: - raise NotImplementedError - class PGGraphQueryException(Exception): """Exception for the AGE queries.""" diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py index 2d5c94ce..056fbc9e 100644 --- a/lightrag/kg/redis_impl.py +++ b/lightrag/kg/redis_impl.py @@ -58,11 +58,6 @@ class RedisKVStorage(BaseKVStorage): for k in data: data[k]["_id"] = k - async def drop(self) -> None: - keys = await self._redis.keys(f"{self.namespace}:*") - if keys: - await self._redis.delete(*keys) - async def index_done_callback(self) -> None: # Redis handles persistence automatically pass diff --git a/lightrag/kg/tidb_impl.py b/lightrag/kg/tidb_impl.py index 6dbfb934..1cff0bd9 100644 --- a/lightrag/kg/tidb_impl.py +++ b/lightrag/kg/tidb_impl.py @@ -214,10 +214,6 @@ class TiDBKVStorage(BaseKVStorage): # Ti handles persistence automatically pass - async def drop(self) -> None: - raise NotImplementedError - - @final @dataclass class TiDBVectorDBStorage(BaseVectorStorage):