Merge pull request #764 from ultrageopro/main

fix: neo4j MAX_CONNECTION_POOL_SIZE impl

lightrag/api/lightrag_server.py

@@ -101,6 +101,7 @@ def estimate_tokens(text: str) -> int:
 
     return int(tokens)
 
+
 def get_default_host(binding_type: str) -> str:
     default_hosts = {
         "ollama": os.getenv("LLM_BINDING_HOST", "http://localhost:11434"),

lightrag/kg/neo4j_impl.py

@@ -48,17 +48,19 @@ class Neo4JStorage(BaseGraphStorage):
         self._driver = None
         self._driver_lock = asyncio.Lock()
 
-        URI = os.environ["NEO4J_URI", config.get("neo4j", "uri", fallback=None)]
-        USERNAME = os.environ[
+        URI = os.environ.get("NEO4J_URI", config.get("neo4j", "uri", fallback=None))
+        USERNAME = os.environ.get(
             "NEO4J_USERNAME", config.get("neo4j", "username", fallback=None)
-        ]
-        PASSWORD = os.environ[
+        )
+        PASSWORD = os.environ.get(
             "NEO4J_PASSWORD", config.get("neo4j", "password", fallback=None)
-        ]
-        MAX_CONNECTION_POOL_SIZE = os.environ.get(
+        )
+        MAX_CONNECTION_POOL_SIZE = int(
+            os.environ.get(
                 "NEO4J_MAX_CONNECTION_POOL_SIZE",
                 config.get("neo4j", "connection_pool_size", fallback=800),
             )
+        )
         DATABASE = os.environ.get(
             "NEO4J_DATABASE", re.sub(r"[^a-zA-Z0-9-]", "-", namespace)
         )