Spaces:
Running
Running
Commit
·
f475b4e
1
Parent(s):
c095e08
Use environment variables for ChromaDB credentials
Browse files- 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 |
-
|
| 47 |
-
|
| 48 |
-
|
|
|
|
|
|
|
| 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()
|