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

env var naming conflict, qualified by prefix

Browse Source
This commit is contained in:
Ken Wiltshire
2024-11-01 14:24:17 -04:00
parent c86bc0edbe
commit caa24ee443
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/kg/neo4j_impl.py
View File

@@ -32,11 +32,14 @@ class GraphStorage(BaseGraphStorage):
credetial_parts = ['URI', 'USERNAME','PASSWORD'] credetial_parts = ['URI', 'USERNAME','PASSWORD']
credentials_set = all(x in os.environ for x in credetial_parts ) credentials_set = all(x in os.environ for x in credetial_parts )
if credentials_set: if credentials_set:
URI = os.environ["URI"] URI = os.environ["NEO4J_URI"]
USERNAME = os.environ["USERNAME"] USERNAME = os.environ["NEO4J_USERNAME"]
PASSWORD = os.environ["PASSWORD"] PASSWORD = os.environ["NEO4J_PASSWORD"]
else: else:
raise Exception (f"One or more Neo4J Credentials, {credetial_parts}, not found in the environment") raise Exception (f"One or more Neo4J Credentials, {credetial_parts}, not found in the environment")
print (f"uri: {URI}")
print (f"uri: {USERNAME}")
print (f"uri: {PASSWORD}")
self._driver = GraphDatabase.driver(URI, auth=(USERNAME, PASSWORD)) self._driver = GraphDatabase.driver(URI, auth=(USERNAME, PASSWORD))
self._node_embed_algorithms = { self._node_embed_algorithms = {