Spaces:
Runtime error
Runtime error
feat: implement centralized configuration management and update default model to gemini-2.5-pro
Browse files- .env +1 -1
- src/recommendation_engine/config.py +6 -4
.env
CHANGED
|
@@ -1,2 +1,2 @@
|
|
| 1 |
GEMINI_API_KEY=AQ.Ab8RN6K-wFpuRR3mv7AUv7p6QqXf98Dlth0IRF3N42UIfjCEbw
|
| 2 |
-
GEMINI_MODEL_NAME=gemini-2.5-
|
|
|
|
| 1 |
GEMINI_API_KEY=AQ.Ab8RN6K-wFpuRR3mv7AUv7p6QqXf98Dlth0IRF3N42UIfjCEbw
|
| 2 |
+
GEMINI_MODEL_NAME=gemini-2.5-pro
|
src/recommendation_engine/config.py
CHANGED
|
@@ -13,11 +13,13 @@ DEBUG_MODE = os.getenv("DEBUG_MODE", "true").lower() == "true"
|
|
| 13 |
|
| 14 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
|
| 15 |
|
|
|
|
|
|
|
| 16 |
MODEL_CANDIDATES = [
|
| 17 |
-
os.getenv("PRIMARY_MODEL", "gemini-
|
| 18 |
-
os.getenv("FAST_MODEL",
|
| 19 |
-
os.getenv("BALANCED_MODEL", "gemini-2.5-flash"),
|
| 20 |
-
os.getenv("QUALITY_MODEL",
|
| 21 |
]
|
| 22 |
|
| 23 |
_seen = set()
|
|
|
|
| 13 |
|
| 14 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
|
| 15 |
|
| 16 |
+
# Primary model — reads GEMINI_MODEL_NAME from .env first, then PRIMARY_MODEL,
|
| 17 |
+
# then falls back to gemini-2.5-pro (smart: advanced reasoning, high quality output).
|
| 18 |
MODEL_CANDIDATES = [
|
| 19 |
+
os.getenv("GEMINI_MODEL_NAME") or os.getenv("PRIMARY_MODEL", "gemini-2.5-pro"),
|
| 20 |
+
os.getenv("FAST_MODEL", "gemini-2.5-flash"), # fast fallback
|
| 21 |
+
os.getenv("BALANCED_MODEL", "gemini-2.5-flash"), # same tier fallback
|
| 22 |
+
os.getenv("QUALITY_MODEL", "gemini-3.1-flash-lite"), # lightweight last resort
|
| 23 |
]
|
| 24 |
|
| 25 |
_seen = set()
|