fix(graphgen): fix clear method

Author: ChenZiHong-Gavin

graphgen/graphgen.py

@@ -212,11 +212,11 @@ def clear(self):
         loop.run_until_complete(self.async_clear())
 
     async def async_clear(self):
-        await self.full_docs_storage.clear()
-        await self.text_chunks_storage.clear()
-        await self.wiki_storage.clear()
+        await self.full_docs_storage.drop()
+        await self.text_chunks_storage.drop()
+        await self.wiki_storage.drop()
         await self.graph_storage.clear()
-        await self.rephrase_storage.clear()
-        await self.qa_storage.clear()
+        await self.rephrase_storage.drop()
+        await self.qa_storage.drop()
 
-        logger.info("All caches are cleared")
+        logger.info("All caches are cleared")

graphgen/operators/extract_kg.py

@@ -21,7 +21,6 @@ async def extract_kg(
         max_concurrent: int = 1000
 ):
     """
-
     :param llm_client: Synthesizer LLM model to extract entities and relationships
     :param kg_instance
     :param tokenizer_instance

models/storage/json_storage.py

@@ -45,7 +45,3 @@ async def upsert(self, data: dict):
 
     async def drop(self):
         self._data = {}
-
-    async def clear(self):
-        if os.path.exists(self._file_name):
-            os.remove(self._file_name)