mirror/lume
1
0
Fork 0
mirror of https://github.com/trycua/lume.git synced 2026-05-24 23:09:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

Modify API key retrieval method

Browse Source 
Updated API key retrieval to allow for kwargs input.
This commit is contained in:
ddupont
2025-12-18 17:19:54 -05:00
committed by GitHub
parent 8e044585ea
commit 5e94349bb0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/python/agent/agent/loops/gemini.py
+2 -2
View File
@@ -630,7 +630,7 @@ class GeminiComputerUseConfig(AsyncAgentConfig):
# Authentication follows two modes based on environment variables:
# 1. Google AI Studio: Set GOOGLE_API_KEY
# 2. Vertex AI: Set GOOGLE_CLOUD_PROJECT, GOOGLE_CLOUD_LOCATION, GOOGLE_GENAI_USE_VERTEXAI=True
api_key = os.getenv("GOOGLE_API_KEY")
api_key = kwargs.get("api_key", os.getenv("GOOGLE_API_KEY"))
if api_key:
client = genai.Client(api_key=api_key)
@@ -847,7 +847,7 @@ class GeminiComputerUseConfig(AsyncAgentConfig):
import os
# Authentication: GOOGLE_API_KEY for AI Studio, or Vertex AI env vars
api_key = os.getenv("GOOGLE_API_KEY")
api_key = kwargs.get("api_key", os.getenv("GOOGLE_API_KEY"))
if api_key:
client = genai.Client(api_key=api_key)
else: