From a15d164d87fa2e201900f81cbcf69a549641cb73 Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Tue, 18 Feb 2025 10:16:00 +0100
Subject: [PATCH] Update postgres_impl.py

---
 lightrag/kg/postgres_impl.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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