yangdx commited on
Commit
61ad9a7
·
1 Parent(s): 49166cd

Suppress Neo4j warning logs by setting logger level.

Browse files
Files changed (1) hide show
  1. lightrag/kg/neo4j_impl.py +3 -0
lightrag/kg/neo4j_impl.py CHANGED
@@ -15,6 +15,7 @@ from tenacity import (
15
  retry_if_exception_type,
16
  )
17
 
 
18
  from ..utils import logger
19
  from ..base import BaseGraphStorage
20
  from ..types import KnowledgeGraph, KnowledgeGraphNode, KnowledgeGraphEdge
@@ -37,6 +38,8 @@ config.read("config.ini", "utf-8")
37
  # Get maximum number of graph nodes from environment variable, default is 1000
38
  MAX_GRAPH_NODES = int(os.getenv("MAX_GRAPH_NODES", 1000))
39
 
 
 
40
 
41
  @final
42
  @dataclass
 
15
  retry_if_exception_type,
16
  )
17
 
18
+ import logging
19
  from ..utils import logger
20
  from ..base import BaseGraphStorage
21
  from ..types import KnowledgeGraph, KnowledgeGraphNode, KnowledgeGraphEdge
 
38
  # Get maximum number of graph nodes from environment variable, default is 1000
39
  MAX_GRAPH_NODES = int(os.getenv("MAX_GRAPH_NODES", 1000))
40
 
41
+ # Set neo4j logger level to ERROR to suppress warning logs
42
+ logging.getLogger("neo4j").setLevel(logging.ERROR)
43
 
44
  @final
45
  @dataclass