Coool2 commited on
Commit
af77ca7
·
1 Parent(s): 931d897

Update agent2.py

Browse files
Files changed (1) hide show
  1. agent2.py +2 -2
agent2.py CHANGED
@@ -127,9 +127,9 @@ class GAIAAgent:
127
  """Initialize the agent with Gemini 2.0 Flash and tools"""
128
 
129
  # Get Gemini API key
130
- gemini_api_key = os.environ.get("GEMINI_API_KEY")
131
  if not gemini_api_key:
132
- raise ValueError("GEMINI_API_KEY environment variable not found")
133
 
134
  # Initialize Gemini 2.0 Flash model
135
  self.model = OpenAIServerModel(
 
127
  """Initialize the agent with Gemini 2.0 Flash and tools"""
128
 
129
  # Get Gemini API key
130
+ gemini_api_key = os.environ.get("GOOGLE_API_KEY")
131
  if not gemini_api_key:
132
+ raise ValueError("GOOGLE_API_KEY environment variable not found")
133
 
134
  # Initialize Gemini 2.0 Flash model
135
  self.model = OpenAIServerModel(