yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-11-02 18:59:32 +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 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: