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

Update postgres_impl.py

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-02-18 10:24:19 +01:00
committed by GitHub
parent 20909e495b
commit 9b9f989160
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/kg/postgres_impl.py
View File

@@ -307,7 +307,8 @@ class PGKVStorage(BaseKVStorage):
"""Drop the storage""" """Drop the storage"""
drop_sql = SQL_TEMPLATES["drop_all"] drop_sql = SQL_TEMPLATES["drop_all"]
await self.db.execute(drop_sql) await self.db.execute(drop_sql)
@final @final
@dataclass @dataclass
class PGVectorStorage(BaseVectorStorage): class PGVectorStorage(BaseVectorStorage):
@@ -549,11 +550,13 @@ class PGDocStatusStorage(DocStatusStorage):
}, },
) )
return data return data
async def drop(self) -> None: async def drop(self) -> None:
"""Drop the storage""" """Drop the storage"""
drop_sql = SQL_TEMPLATES["drop_doc_full"] drop_sql = SQL_TEMPLATES["drop_doc_full"]
await self.db.execute(drop_sql) await self.db.execute(drop_sql)
class PGGraphQueryException(Exception): class PGGraphQueryException(Exception):
"""Exception for the AGE queries.""" """Exception for the AGE queries."""
@@ -1033,7 +1036,7 @@ class PGGraphStorage(BaseGraphStorage):
self, node_label: str, max_depth: int = 5 self, node_label: str, max_depth: int = 5
) -> KnowledgeGraph: ) -> KnowledgeGraph:
raise NotImplementedError raise NotImplementedError
async def drop(self) -> None: async def drop(self) -> None:
"""Drop the storage""" """Drop the storage"""
drop_sql = SQL_TEMPLATES["drop_vdb_entity"] drop_sql = SQL_TEMPLATES["drop_vdb_entity"]
@@ -1041,6 +1044,7 @@ class PGGraphStorage(BaseGraphStorage):
drop_sql = SQL_TEMPLATES["drop_vdb_relation"] drop_sql = SQL_TEMPLATES["drop_vdb_relation"]
await self.db.execute(drop_sql) await self.db.execute(drop_sql)
NAMESPACE_TABLE_MAP = { NAMESPACE_TABLE_MAP = {
NameSpace.KV_STORE_FULL_DOCS: "LIGHTRAG_DOC_FULL", NameSpace.KV_STORE_FULL_DOCS: "LIGHTRAG_DOC_FULL",
NameSpace.KV_STORE_TEXT_CHUNKS: "LIGHTRAG_DOC_CHUNKS", NameSpace.KV_STORE_TEXT_CHUNKS: "LIGHTRAG_DOC_CHUNKS",
@@ -1245,5 +1249,4 @@ SQL_TEMPLATES = {
"drop_vdb_relation": """ "drop_vdb_relation": """
DROP TABLE IF EXISTS LIGHTRAG_VDB_RELATION CASCADE; DROP TABLE IF EXISTS LIGHTRAG_VDB_RELATION CASCADE;
""", """,
} }