Amodit commited on
Commit
1b770ce
·
1 Parent(s): 9a2f0cb

Explicitly pass API keys to models

Browse files
Files changed (1) hide show
  1. core_utils/core_model_loaders.py +6 -1
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(model="gemini-2.5-flash", temperature=0, max_retries=5)
 
 
 
 
 
 
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
+ )