Added more drops

This commit is contained in:
Saifeddine ALOUI 2025-02-18 10:01:21 +01:00 committed by GitHub
parent f02c881fd3
commit 8f6f4c249a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1018,7 +1018,13 @@ class PGGraphStorage(BaseGraphStorage):
self, node_label: str, max_depth: int = 5
) -> 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",