nivakaran commited on
Commit
aa62cae
·
verified ·
1 Parent(s): 6e8abf9

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. src/storage/config.py +9 -4
src/storage/config.py CHANGED
@@ -44,11 +44,16 @@ class StorageConfig:
44
  "sentence-transformers/all-MiniLM-L6-v2"
45
  )
46
 
47
- # Neo4j Configuration
48
  NEO4J_URI: str = os.getenv("NEO4J_URI", "bolt://localhost:7687")
49
- NEO4J_USER: str = os.getenv("NEO4J_USER", "neo4j")
50
- NEO4J_PASSWORD: str = os.getenv("NEO4J_PASSWORD", "Roger2024")
51
- NEO4J_ENABLED: bool = os.getenv("NEO4J_ENABLED", "false").lower() == "true"
 
 
 
 
 
52
 
53
  # CSV Export Configuration
54
  CSV_EXPORT_DIR: str = os.getenv(
 
44
  "sentence-transformers/all-MiniLM-L6-v2"
45
  )
46
 
47
+ # Neo4j Configuration (supports both NEO4J_USER and NEO4J_USERNAME)
48
  NEO4J_URI: str = os.getenv("NEO4J_URI", "bolt://localhost:7687")
49
+ NEO4J_USER: str = os.getenv("NEO4J_USERNAME", os.getenv("NEO4J_USER", "neo4j"))
50
+ NEO4J_PASSWORD: str = os.getenv("NEO4J_PASSWORD", "")
51
+ NEO4J_DATABASE: str = os.getenv("NEO4J_DATABASE", "neo4j")
52
+ # Auto-enable if URI contains 'neo4j.io' (Aura) or explicitly set
53
+ NEO4J_ENABLED: bool = (
54
+ os.getenv("NEO4J_ENABLED", "").lower() == "true" or
55
+ "neo4j.io" in os.getenv("NEO4J_URI", "")
56
+ )
57
 
58
  # CSV Export Configuration
59
  CSV_EXPORT_DIR: str = os.getenv(