yangdx commited on
Commit
ceab23f
·
1 Parent(s): b89f76b

Use namespace for graph_name before falling back to env or default value

Browse files

- Update graph_name initialization
- Add namespace override support
- Maintain backward compatibility
- Prioritize namespace over env variable

Files changed (1) hide show
  1. lightrag/kg/postgres_impl.py +1 -1
lightrag/kg/postgres_impl.py CHANGED
@@ -587,7 +587,7 @@ class PGGraphStorage(BaseGraphStorage):
587
  print("no preloading of graph with AGE in production")
588
 
589
  def __post_init__(self):
590
- self.graph_name = os.environ.get("AGE_GRAPH_NAME", "lightrag")
591
  self._node_embed_algorithms = {
592
  "node2vec": self._node2vec_embed,
593
  }
 
587
  print("no preloading of graph with AGE in production")
588
 
589
  def __post_init__(self):
590
+ self.graph_name = self.namespace or os.environ.get("AGE_GRAPH_NAME", "lightrag")
591
  self._node_embed_algorithms = {
592
  "node2vec": self._node2vec_embed,
593
  }