Spaces:
Sleeping
Sleeping
Explicitly pass API keys to models
Browse files
core_utils/core_model_loaders.py
CHANGED
|
@@ -18,4 +18,9 @@ def load_groq_llm():
|
|
| 18 |
|
| 19 |
def load_gemini_llm():
|
| 20 |
"""Loads the Gemini LLM without any Streamlit dependencies."""
|
| 21 |
-
return ChatGoogleGenerativeAI(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
def load_gemini_llm():
|
| 20 |
"""Loads the Gemini LLM without any Streamlit dependencies."""
|
| 21 |
+
return ChatGoogleGenerativeAI(
|
| 22 |
+
model="gemini-2.5-flash",
|
| 23 |
+
temperature=0,
|
| 24 |
+
max_retries=5,
|
| 25 |
+
google_api_key=os.getenv("GOOGLE_API_KEY")
|
| 26 |
+
)
|