From 3df20ae78722ea9edc65a1219daf22fd75faa1b6 Mon Sep 17 00:00:00 2001
From: zrguo <zrguo.bupt@qq.com>
Date: Mon, 17 Mar 2025 15:59:54 +0800
Subject: [PATCH] fix lint

---
 lightrag/kg/postgres_impl.py | 32 ++++++++++++++++++++++++++------
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index 2b6f10b2..a2e9be38 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -437,7 +437,7 @@ class PGVectorStorage(BaseVectorStorage):
             chunk_ids = source_id.split("<SEP>")
         else:
             chunk_ids = [source_id]
-            
+
         data: dict[str, Any] = {
             "workspace": self.db.workspace,
             "id": item["__id__"],
@@ -456,7 +456,7 @@ class PGVectorStorage(BaseVectorStorage):
             chunk_ids = source_id.split("<SEP>")
         else:
             chunk_ids = [source_id]
-            
+
         data: dict[str, Any] = {
             "workspace": self.db.workspace,
             "id": item["__id__"],
@@ -962,7 +962,11 @@ class PGGraphStorage(BaseGraphStorage):
                             vertices.get(edge["end_id"], {}),
                         )
             else:
-                d[k] = json.loads(v) if isinstance(v, str) and ("{" in v or "[" in v) else v
+                d[k] = (
+                    json.loads(v)
+                    if isinstance(v, str) and ("{" in v or "[" in v)
+                    else v
+                )
 
         return d
 
@@ -1404,7 +1408,9 @@ class PGGraphStorage(BaseGraphStorage):
         embed_func = self._node_embed_algorithms[algorithm]
         return await embed_func()
 
-    async def get_knowledge_graph(self, node_label: str, max_depth: int = 5) -> KnowledgeGraph:
+    async def get_knowledge_graph(
+        self, node_label: str, max_depth: int = 5
+    ) -> KnowledgeGraph:
         """
         Retrieve a subgraph containing the specified node and its neighbors up to the specified depth.
 
@@ -1451,7 +1457,14 @@ class PGGraphStorage(BaseGraphStorage):
             edge_key = f"{src_id},{tgt_id}"
             if edge_key not in unique_edge_ids:
                 unique_edge_ids.add(edge_key)
-                edges.append((edge_key, src_id, tgt_id, {"source": edge_data[0], "target": edge_data[2]}))
+                edges.append(
+                    (
+                        edge_key,
+                        src_id,
+                        tgt_id,
+                        {"source": edge_data[0], "target": edge_data[2]},
+                    )
+                )
 
         # Process the query results.
         if node_label == "*":
@@ -1476,12 +1489,19 @@ class PGGraphStorage(BaseGraphStorage):
                 for node_id, node_data in nodes.items()
             ],
             edges=[
-                KnowledgeGraphEdge(id=edge_id, type="DIRECTED", source=src, target=tgt, properties=props)
+                KnowledgeGraphEdge(
+                    id=edge_id,
+                    type="DIRECTED",
+                    source=src,
+                    target=tgt,
+                    properties=props,
+                )
                 for edge_id, src, tgt, props in edges
             ],
         )
 
         return kg
+
     async def drop(self) -> None:
         """Drop the storage"""
         drop_sql = SQL_TEMPLATES["drop_vdb_entity"]