From 438caa776d89426d19bf8624b841057608fb1bc4 Mon Sep 17 00:00:00 2001 From: ultrageopro Date: Thu, 13 Feb 2025 15:26:45 +0300 Subject: [PATCH] fix: neo4j MAX_CONNECTION_POOL_SIZE impl --- lightrag/kg/neo4j_impl.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index 587a9167..731b7cc1 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/lightrag/kg/neo4j_impl.py @@ -55,9 +55,11 @@ class Neo4JStorage(BaseGraphStorage): PASSWORD = os.environ[ "NEO4J_PASSWORD", config.get("neo4j", "password", fallback=None) ] - MAX_CONNECTION_POOL_SIZE = os.environ.get( - "NEO4J_MAX_CONNECTION_POOL_SIZE", - config.get("neo4j", "connection_pool_size", fallback=800), + 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)