Francisco Zanartu commited on
Commit
8aae565
·
1 Parent(s): 2c3a136

feat: add google_api_key parameter to ChatGoogleGenerativeAI instantiation

Browse files
Files changed (1) hide show
  1. src/llm/llms.py +2 -0
src/llm/llms.py CHANGED
@@ -12,6 +12,7 @@ google_llm = ChatGoogleGenerativeAI(
12
  model="gemini-2.5-flash",
13
  temperature=0,
14
  max_tokens=128000,
 
15
  )
16
  openai_llm = ChatOpenAI(model="gpt-5-mini-2025-08-07", temperature=0)
17
 
@@ -20,6 +21,7 @@ google_llm_pro = ChatGoogleGenerativeAI(
20
  model="gemini-2.5-pro",
21
  temperature=0.0,
22
  max_tokens=128000,
 
23
  )
24
 
25
  google_llm_with_search_tool = google_llm.bind_tools(
 
12
  model="gemini-2.5-flash",
13
  temperature=0,
14
  max_tokens=128000,
15
+ google_api_key=os.getenv("GEMINI_API_KEY"),
16
  )
17
  openai_llm = ChatOpenAI(model="gpt-5-mini-2025-08-07", temperature=0)
18
 
 
21
  model="gemini-2.5-pro",
22
  temperature=0.0,
23
  max_tokens=128000,
24
+ google_api_key=os.getenv("GEMINI_API_KEY"),
25
  )
26
 
27
  google_llm_with_search_tool = google_llm.bind_tools(