From bbff3ed0abc26b5b403e75c432ca1fd95b6071c3 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Mon, 10 Mar 2025 17:30:40 +0800
Subject: [PATCH] Fix linting

---
 lightrag/lightrag.py | 12 +++++++++---
 lightrag/operate.py  |  2 +-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 5b42fa3d..3cd379b6 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -887,7 +887,9 @@ class LightRAG:
                                 self.chunks_vdb.upsert(chunks)
                             )
                             entity_relation_task = asyncio.create_task(
-                                self._process_entity_relation_graph(chunks, pipeline_status, pipeline_status_lock)
+                                self._process_entity_relation_graph(
+                                    chunks, pipeline_status, pipeline_status_lock
+                                )
                             )
                             full_docs_task = asyncio.create_task(
                                 self.full_docs.upsert(
@@ -1002,7 +1004,9 @@ class LightRAG:
                 pipeline_status["latest_message"] = log_message
                 pipeline_status["history_messages"].append(log_message)
 
-    async def _process_entity_relation_graph(self, chunk: dict[str, Any], pipeline_status=None, pipeline_status_lock=None) -> None:
+    async def _process_entity_relation_graph(
+        self, chunk: dict[str, Any], pipeline_status=None, pipeline_status_lock=None
+    ) -> None:
         try:
             await extract_entities(
                 chunk,
@@ -1018,7 +1022,9 @@ class LightRAG:
             logger.error("Failed to extract entities and relationships")
             raise e
 
-    async def _insert_done(self, pipeline_status=None, pipeline_status_lock=None) -> None:
+    async def _insert_done(
+        self, pipeline_status=None, pipeline_status_lock=None
+    ) -> None:
         tasks = [
             cast(StorageNameSpace, storage_inst).index_done_callback()
             for storage_inst in [  # type: ignore
diff --git a/lightrag/operate.py b/lightrag/operate.py
index 5d6b7c7d..e352ff79 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -341,7 +341,7 @@ async def extract_entities(
     relationships_vdb: BaseVectorStorage,
     global_config: dict[str, str],
     pipeline_status: dict = None,
-    pipeline_status_lock = None,
+    pipeline_status_lock=None,
     llm_response_cache: BaseKVStorage | None = None,
 ) -> None:
     use_llm_func: callable = global_config["llm_model_func"]