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:16:00 +01:00
committed by GitHub
parent 941c89521c
commit a15d164d87
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lightrag/kg/postgres_impl.py
View File

@@ -499,7 +499,8 @@ class PGDocStatusStorage(DocStatusStorage):
sql = "select * from LIGHTRAG_DOC_STATUS where workspace=$1 and status=$2" sql = "select * from LIGHTRAG_DOC_STATUS where workspace=$1 and status=$2"
params = {"workspace": self.db.workspace, "status": status.value} params = {"workspace": self.db.workspace, "status": status.value}
result = await self.db.query(sql, params, True) result = await self.db.query(sql, params, True)
return { print("")
docs_by_status = {
element["id"]: DocProcessingStatus( element["id"]: DocProcessingStatus(
content=result[0]["content"], content=result[0]["content"],
content_summary=element["content_summary"], content_summary=element["content_summary"],
@@ -511,6 +512,8 @@ class PGDocStatusStorage(DocStatusStorage):
) )
for element in result for element in result
} }
print(f"Docs by status: {docs_by_status}")
return docs_by_status
async def index_done_callback(self) -> None: async def index_done_callback(self) -> None:
# PG handles persistence automatically # PG handles persistence automatically