bat-6 commited on
Commit
b79524e
·
1 Parent(s): caa78f1

feat: implement centralized configuration management and update default model to gemini-2.5-pro

Browse files
Files changed (2) hide show
  1. .env +1 -1
  2. 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-flash
 
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-3.1-flash-lite-preview"),
18
- os.getenv("FAST_MODEL", "gemini-2.5-flash-lite"),
19
- os.getenv("BALANCED_MODEL", "gemini-2.5-flash"),
20
- os.getenv("QUALITY_MODEL", "gemini-2.5-pro"),
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()