gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Addded drop to postgresql

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-02-18 09:10:50 +01:00
committed by GitHub
parent a191dfad61
commit f11900451a
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/kg/postgres_impl.py
View File

@@ -301,6 +301,10 @@ class PGKVStorage(BaseKVStorage):
# PG handles persistence automatically # PG handles persistence automatically
pass pass
async def drop(self) -> None:
"""Drop the storage"""
drop_sql = SQL_TEMPLATES["DROP_ALL"]
await self.db.execute(drop_sql)
@final @final
@dataclass @dataclass
@@ -1194,4 +1198,12 @@ SQL_TEMPLATES = {
FROM LIGHTRAG_DOC_CHUNKS where workspace=$1) FROM LIGHTRAG_DOC_CHUNKS where workspace=$1)
WHERE distance>$2 ORDER BY distance DESC LIMIT $3 WHERE distance>$2 ORDER BY distance DESC LIMIT $3
""", """,
#DROP everything
"drop": """
DROP TABLE IF EXISTS LIGHTRAG_DOC_FULL CASCADE;
DROP TABLE IF EXISTS LIGHTRAG_DOC_CHUNKS CASCADE;
DROP TABLE IF EXISTS LIGHTRAG_LLM_CACHE CASCADE;
DROP TABLE IF EXISTS LIGHTRAG_VDB_ENTITY CASCADE;
DROP TABLE IF EXISTS LIGHTRAG_VDB_RELATION CASCADE;
""",
} }