Francisco Zanartu commited on
Commit ·
8aae565
1
Parent(s): 2c3a136
feat: add google_api_key parameter to ChatGoogleGenerativeAI instantiation
Browse files- 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(
|