diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index 06315d76..7fe3da15 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/lightrag/kg/neo4j_impl.py @@ -1,4 +1,3 @@ -import inspect import os import re from dataclasses import dataclass @@ -307,7 +306,7 @@ class Neo4JStorage(BaseGraphStorage): for label in node_dict["labels"] if label != "base" ] - logger.debug(f"Neo4j query node {query} return: {node_dict}") + # logger.debug(f"Neo4j query node {query} return: {node_dict}") return node_dict return None finally: @@ -382,9 +381,9 @@ class Neo4JStorage(BaseGraphStorage): return 0 degree = record["degree"] - logger.debug( - f"Neo4j query node degree for {node_id} return: {degree}" - ) + # logger.debug( + # f"Neo4j query node degree for {node_id} return: {degree}" + # ) return degree finally: await result.consume() # Ensure result is fully consumed @@ -424,7 +423,7 @@ class Neo4JStorage(BaseGraphStorage): logger.warning(f"No node found with label '{nid}'") degrees[nid] = 0 - logger.debug(f"Neo4j batch node degree query returned: {degrees}") + # logger.debug(f"Neo4j batch node degree query returned: {degrees}") return degrees async def edge_degree(self, src_id: str, tgt_id: str) -> int: @@ -512,7 +511,7 @@ class Neo4JStorage(BaseGraphStorage): if records: try: edge_result = dict(records[0]["edge_properties"]) - logger.debug(f"Result: {edge_result}") + # logger.debug(f"Result: {edge_result}") # Ensure required keys exist with defaults required_keys = { "weight": 0.0, @@ -528,9 +527,9 @@ class Neo4JStorage(BaseGraphStorage): f"missing {key}, using default: {default_value}" ) - logger.debug( - f"{inspect.currentframe().f_code.co_name}:query:{query}:result:{edge_result}" - ) + # logger.debug( + # f"{inspect.currentframe().f_code.co_name}:query:{query}:result:{edge_result}" + # ) return edge_result except (KeyError, TypeError, ValueError) as e: logger.error( @@ -545,9 +544,9 @@ class Neo4JStorage(BaseGraphStorage): "keywords": None, } - logger.debug( - f"{inspect.currentframe().f_code.co_name}: No edge found between {source_node_id} and {target_node_id}" - ) + # logger.debug( + # f"{inspect.currentframe().f_code.co_name}: No edge found between {source_node_id} and {target_node_id}" + # ) # Return None when no edge found return None finally: