From ad087073aa711cf98bc9725dee0c28f1f619a628 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Thu, 10 Apr 2025 01:06:46 +0800
Subject: [PATCH] Optimize logger for storage

---
 lightrag/kg/json_doc_status_impl.py | 2 +-
 lightrag/kg/json_kv_impl.py         | 2 +-
 lightrag/kg/nano_vector_db_impl.py  | 2 +-
 lightrag/kg/networkx_impl.py        | 2 +-
 lightrag/kg/postgres_impl.py        | 6 +++---
 5 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/lightrag/kg/json_doc_status_impl.py b/lightrag/kg/json_doc_status_impl.py
index 944d57d7..2611e78f 100644
--- a/lightrag/kg/json_doc_status_impl.py
+++ b/lightrag/kg/json_doc_status_impl.py
@@ -116,7 +116,7 @@ class JsonDocStatusStorage(DocStatusStorage):
         """
         if not data:
             return
-        logger.info(f"Inserting {len(data)} records to {self.namespace}")
+        logger.debug(f"Inserting {len(data)} records to {self.namespace}")
         async with self._storage_lock:
             self._data.update(data)
             await set_all_update_flags(self.namespace)
diff --git a/lightrag/kg/json_kv_impl.py b/lightrag/kg/json_kv_impl.py
index 82c18d95..d0cd64fb 100644
--- a/lightrag/kg/json_kv_impl.py
+++ b/lightrag/kg/json_kv_impl.py
@@ -121,7 +121,7 @@ class JsonKVStorage(BaseKVStorage):
         """
         if not data:
             return
-        logger.info(f"Inserting {len(data)} records to {self.namespace}")
+        logger.debug(f"Inserting {len(data)} records to {self.namespace}")
         async with self._storage_lock:
             self._data.update(data)
             await set_all_update_flags(self.namespace)
diff --git a/lightrag/kg/nano_vector_db_impl.py b/lightrag/kg/nano_vector_db_impl.py
index 56a52b92..3003eea0 100644
--- a/lightrag/kg/nano_vector_db_impl.py
+++ b/lightrag/kg/nano_vector_db_impl.py
@@ -85,7 +85,7 @@ class NanoVectorDBStorage(BaseVectorStorage):
            KG-storage-log should be used to avoid data corruption
         """
 
-        logger.info(f"Inserting {len(data)} to {self.namespace}")
+        logger.debug(f"Inserting {len(data)} to {self.namespace}")
         if not data:
             return
 
diff --git a/lightrag/kg/networkx_impl.py b/lightrag/kg/networkx_impl.py
index 2bea85a3..e746af2e 100644
--- a/lightrag/kg/networkx_impl.py
+++ b/lightrag/kg/networkx_impl.py
@@ -392,7 +392,7 @@ class NetworkXStorage(BaseGraphStorage):
             # Check if storage was updated by another process
             if self.storage_updated.value:
                 # Storage was updated by another process, reload data instead of saving
-                logger.warning(
+                logger.info(
                     f"Graph for {self.namespace} was updated by another process, reloading..."
                 )
                 self._graph = (
diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index c7947e43..1ba4ada5 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -361,7 +361,7 @@ class PGKVStorage(BaseKVStorage):
 
     ################ INSERT METHODS ################
     async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
-        logger.info(f"Inserting {len(data)} to {self.namespace}")
+        logger.debug(f"Inserting {len(data)} to {self.namespace}")
         if not data:
             return
 
@@ -560,7 +560,7 @@ class PGVectorStorage(BaseVectorStorage):
         return upsert_sql, data
 
     async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
-        logger.info(f"Inserting {len(data)} to {self.namespace}")
+        logger.debug(f"Inserting {len(data)} to {self.namespace}")
         if not data:
             return
 
@@ -949,7 +949,7 @@ class PGDocStatusStorage(DocStatusStorage):
         Args:
             data: dictionary of document IDs and their status data
         """
-        logger.info(f"Inserting {len(data)} to {self.namespace}")
+        logger.debug(f"Inserting {len(data)} to {self.namespace}")
         if not data:
             return