Added more drops

2025-02-18 10:01:21 +01:00
parent f02c881fd3
commit 8f6f4c249a
1 changed files with 7 additions and 1 deletions
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -1019,6 +1019,12 @@ class PGGraphStorage(BaseGraphStorage):
    ) -> KnowledgeGraph:
        raise NotImplementedError
 	    
+    async def drop(self) -> None:
+        """Drop the storage"""
+        drop_sql = SQL_TEMPLATES["drop_vdb_entity"]
+        await self.db.execute(drop_sql)
+        drop_sql = SQL_TEMPLATES["drop_vdb_relation"]
+        await self.db.execute(drop_sql)

 NAMESPACE_TABLE_MAP = {
    NameSpace.KV_STORE_FULL_DOCS: "LIGHTRAG_DOC_FULL",