gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #764 from ultrageopro/main

Browse Source 
fix: neo4j MAX_CONNECTION_POOL_SIZE impl
This commit is contained in:
zrguo
2025-02-14 09:36:37 +08:00
committed by GitHub
parent 5730987349 8773ad983f
commit 267fa7c0f5
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lightrag/api/lightrag_server.py
View File

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

18
lightrag/kg/neo4j_impl.py
View File

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