From 64f8cb7daee58bfa45d907f48bff2e1aea9becf4 Mon Sep 17 00:00:00 2001 From: Yannick Stephan Date: Mon, 17 Feb 2025 18:38:55 +0100 Subject: [PATCH] fixed lint --- lightrag/kg/neo4j_impl.py | 21 ++++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index e7a59b0c..b7b63580 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/lightrag/kg/neo4j_impl.py @@ -307,20 +307,35 @@ class Neo4JStorage(BaseGraphStorage): f"and {entity_name_label_target}: {str(e)}" ) # Return default edge properties on error - return {"weight": 0.0, "description": None, "keywords": None, "source_id": None} + return { + "weight": 0.0, + "description": None, + "keywords": None, + "source_id": None, + } logger.debug( f"{inspect.currentframe().f_code.co_name}: No edge found between {entity_name_label_source} and {entity_name_label_target}" ) # Return default edge properties when no edge found - return {"weight": 0.0, "description": None, "keywords": None, "source_id": None} + return { + "weight": 0.0, + "description": None, + "keywords": None, + "source_id": None, + } except Exception as e: logger.error( f"Error in get_edge between {source_node_id} and {target_node_id}: {str(e)}" ) # Return default edge properties on error - return {"weight": 0.0, "description": None, "keywords": None, "source_id": None} + return { + "weight": 0.0, + "description": None, + "keywords": None, + "source_id": None, + } async def get_node_edges(self, source_node_id: str) -> list[tuple[str, str]] | None: node_label = source_node_id.strip('"')