jashdoshi77 commited on
Commit
f475b4e
·
1 Parent(s): c095e08

Use environment variables for ChromaDB credentials

Browse files
Files changed (1) hide show
  1. src/config.py +8 -4
src/config.py CHANGED
@@ -40,12 +40,16 @@ SEASON_STRINGS = get_season_strings()
40
  # =============================================================================
41
  # CHROMADB CONFIGURATION
42
  # =============================================================================
 
 
43
  @dataclass
44
  class ChromaDBConfig:
45
- """Configuration for ChromaDB prediction tracking."""
46
- tenant: str = "70e82e68-9fa7-4224-9975-d49d355f6328"
47
- database: str = "NBA_ML"
48
- api_key: str = "ck-2bXunZK4X3BFSPHtwLG2Ki9xr5r6ZPxzADESDperHweT"
 
 
49
  collection_name: str = "predictions"
50
 
51
  CHROMADB_CONFIG = ChromaDBConfig()
 
40
  # =============================================================================
41
  # CHROMADB CONFIGURATION
42
  # =============================================================================
43
+ import os
44
+
45
  @dataclass
46
  class ChromaDBConfig:
47
+ """Configuration for ChromaDB prediction tracking.
48
+ Reads from environment variables for security, with fallback defaults.
49
+ """
50
+ tenant: str = os.environ.get("CHROMADB_TENANT", "70e82e68-9fa7-4224-9975-d49d355f6328")
51
+ database: str = os.environ.get("CHROMADB_DATABASE", "NBA_ML")
52
+ api_key: str = os.environ.get("CHROMADB_API_KEY", "ck-2bXunZK4X3BFSPHtwLG2Ki9xr5r6ZPxzADESDperHweT")
53
  collection_name: str = "predictions"
54
 
55
  CHROMADB_CONFIG = ChromaDBConfig()