Reorganize env config sections, add data store config to env file.
• Add namespace prefix config option • Make AGE graph name optional • Update env variable requirements • Add comments for deprecated options
This commit is contained in:
@@ -75,8 +75,8 @@ class AGEStorage(BaseGraphStorage):
|
||||
.replace("'", "\\'")
|
||||
)
|
||||
HOST = os.environ["AGE_POSTGRES_HOST"].replace("\\", "\\\\").replace("'", "\\'")
|
||||
PORT = int(os.environ["AGE_POSTGRES_PORT"])
|
||||
self.graph_name = os.environ["AGE_GRAPH_NAME"]
|
||||
PORT = os.environ.get("AGE_POSTGRES_PORT", "8529")
|
||||
self.graph_name = namespace or os.environ.get("AGE_GRAPH_NAME", "lightrag")
|
||||
|
||||
connection_string = f"dbname='{DB}' user='{USER}' password='{PASSWORD}' host='{HOST}' port={PORT}"
|
||||
|
||||
|
@@ -107,10 +107,13 @@ STORAGE_ENV_REQUIREMENTS = {
|
||||
"AGE_POSTGRES_DB",
|
||||
"AGE_POSTGRES_USER",
|
||||
"AGE_POSTGRES_PASSWORD",
|
||||
"AGE_GRAPH_NAME",
|
||||
],
|
||||
"GremlinStorage": ["GREMLIN_HOST", "GREMLIN_PORT", "GREMLIN_GRAPH"],
|
||||
"PGGraphStorage": ["POSTGRES_USER", "POSTGRES_PASSWORD", "POSTGRES_DATABASE"],
|
||||
"PGGraphStorage": [
|
||||
"POSTGRES_USER",
|
||||
"POSTGRES_PASSWORD",
|
||||
"POSTGRES_DATABASE",
|
||||
],
|
||||
"OracleGraphStorage": [
|
||||
"ORACLE_DSN",
|
||||
"ORACLE_USER",
|
||||
|
Reference in New Issue
Block a user