gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lightrag/kg/postgres_impl.py
View file

@ -301,7 +301,11 @@ class PGKVStorage(BaseKVStorage):
# PG handles persistence automatically
pass
async def drop(self) -> None:
"""Drop the storage"""
drop_sql = SQL_TEMPLATES["DROP_ALL"]
await self.db.execute(drop_sql)
@final
@dataclass
class PGVectorStorage(BaseVectorStorage):
@ -1194,4 +1198,12 @@ SQL_TEMPLATES = {
FROM LIGHTRAG_DOC_CHUNKS where workspace=$1)
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;
""",
}