Tim Luka Horstmann
commited on
Commit
·
8d04f0d
1
Parent(s):
d6f1c5d
Update to use gemini model via variables
Browse files
app.py
CHANGED
|
@@ -82,11 +82,11 @@ filename = "Qwen3-1.7B-Q4_K_M.gguf" # "deepcogito_cogito-v1-preview-llama-3B-Q4_
|
|
| 82 |
# Gemini Configuration
|
| 83 |
if USE_GEMINI:
|
| 84 |
gemini_api_key = os.getenv("GEMINI_API_KEY")
|
|
|
|
| 85 |
if not gemini_api_key:
|
| 86 |
logger.error("GEMINI_API_KEY environment variable not set but USE_GEMINI is true.")
|
| 87 |
raise ValueError("GEMINI_API_KEY not set")
|
| 88 |
gemini_client = genai.Client(api_key=gemini_api_key)
|
| 89 |
-
gemini_model = "gemini-2.5-flash-preview-05-20"
|
| 90 |
logger.info("Gemini API client initialized")
|
| 91 |
else:
|
| 92 |
gemini_client = None
|
|
|
|
| 82 |
# Gemini Configuration
|
| 83 |
if USE_GEMINI:
|
| 84 |
gemini_api_key = os.getenv("GEMINI_API_KEY")
|
| 85 |
+
gemini_model = os.getenv("GEMINI_MODEL")
|
| 86 |
if not gemini_api_key:
|
| 87 |
logger.error("GEMINI_API_KEY environment variable not set but USE_GEMINI is true.")
|
| 88 |
raise ValueError("GEMINI_API_KEY not set")
|
| 89 |
gemini_client = genai.Client(api_key=gemini_api_key)
|
|
|
|
| 90 |
logger.info("Gemini API client initialized")
|
| 91 |
else:
|
| 92 |
gemini_client = None
|