From 61fd3e6127eaf8f7755dc1ff482b7a7fd9dd34cf Mon Sep 17 00:00:00 2001
From: Yannick Stephan <stephan.yannick@me.com>
Date: Sun, 9 Feb 2025 11:21:32 +0100
Subject: [PATCH] cleaned code

---
 lightrag/lightrag.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index b2bdfaac..69dd85e9 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -19,7 +19,7 @@ from .utils import (
     convert_response_to_json,
     logger,
     set_logger,
-    statistic_data,
+    statistic_data
 )
 from .base import (
     BaseGraphStorage,
@@ -31,7 +31,6 @@ from .base import (
 from .namespace import NameSpace, make_namespace
 
 from .prompt import GRAPH_FIELD_SEP
-
 STORAGES = {
     "NetworkXStorage": ".kg.networkx_impl",
     "JsonKVStorage": ".kg.json_kv_impl",
@@ -560,8 +559,8 @@ class LightRAG:
                             "updated_at": datetime.now().isoformat(),
                         }
                     )
-                    await self.doc_status.upsert({doc_id: doc_status})
                     await self.chunks_vdb.upsert(chunks)
+                    await self.doc_status.upsert({doc_id: doc_status})
 
                 except Exception as e:
                     doc_status.update(
@@ -621,7 +620,7 @@ class LightRAG:
                         global_config=asdict(self),
                     )
                     if maybe_new_kg is None:
-                        logger.info("No entities or relationships extracted!")
+                        logger.warning("No entities or relationships extracted!")
                     # Update status to processed
                     await self.text_chunks.upsert(
                         {chunk_id: {"status": DocStatus.PROCESSED}}