yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-05 03:18:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #805 from ultrageopro/main

Browse Source 
feat: neo4j custom timeouts & fix: neo4j max_pool_size
This commit is contained in:
Yannick Stephan 2025-02-17 18:42:29 +01:00 committed by GitHub
commit 442838d0f1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lightrag/kg/neo4j_impl.py
View File

@ -69,12 +69,24 @@ class Neo4JStorage(BaseGraphStorage):
config.get("neo4j", "connection_pool_size", fallback=800),
)
)
CONNECTION_TIMEOUT = float(
os.environ.get("NEO4J_CONNECTION_TIMEOUT"),
config.get("neo4j", "connection_timeout", fallback=60.0),
)
CONNECTION_ACQUISITION_TIMEOUT = float(
os.environ.get("NEO4J_CONNECTION_ACQUISITION_TIMEOUT"),
config.get("neo4j", "connection_acquisition_timeout", fallback=60.0),
)
DATABASE = os.environ.get(
"NEO4J_DATABASE", re.sub(r"[^a-zA-Z0-9-]", "-", namespace)
)
self._driver: AsyncDriver = AsyncGraphDatabase.driver(
URI, auth=(USERNAME, PASSWORD)
URI,
auth=(USERNAME, PASSWORD),
max_connection_pool_size=MAX_CONNECTION_POOL_SIZE,
connection_timeout=CONNECTION_TIMEOUT,
connection_acquisition_timeout=CONNECTION_ACQUISITION_TIMEOUT,
)
# Try to connect to the database
@ -82,6 +94,8 @@ class Neo4JStorage(BaseGraphStorage):
URI,
auth=(USERNAME, PASSWORD),
max_connection_pool_size=MAX_CONNECTION_POOL_SIZE,
connection_timeout=CONNECTION_TIMEOUT,
connection_acquisition_timeout=CONNECTION_ACQUISITION_TIMEOUT,
) as _sync_driver:
for database in (DATABASE, None):
self._DATABASE = database