From 20909e495b02e892ab7bc32ab576ed322904e547 Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Tue, 18 Feb 2025 10:22:16 +0100
Subject: [PATCH] Added drop

---
 lightrag/kg/json_doc_status_impl.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lightrag/kg/json_doc_status_impl.py b/lightrag/kg/json_doc_status_impl.py
index e69352f3..871e7121 100644
--- a/lightrag/kg/json_doc_status_impl.py
+++ b/lightrag/kg/json_doc_status_impl.py
@@ -68,3 +68,8 @@ class JsonDocStatusStorage(DocStatusStorage):
         for doc_id in doc_ids:
             self._data.pop(doc_id, None)
         await self.index_done_callback()
+        
+    async def drop(self) -> None:
+        """Drop the storage"""
+        self._data.clear()
+