yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-11-08 22:04:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove debug looging from Neo4j

Browse Source
This commit is contained in:
yangdx 2025-06-24 01:03:02 +08:00
parent 64d0df0caa
commit fd69c895cd
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lightrag/kg/neo4j_impl.py
View File

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