Update agent.py

agent.py

@@ -13,14 +13,19 @@ from vision_tool import image_reasoning_tool
 
 #token=os.getenv("HF_API_TOKEN")
 
-OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
-if not OPENROUTER_API_KEY:
-    raise EnvironmentError("OPENROUTER_API_KEY environment variable not set")
+#OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
+#if not OPENROUTER_API_KEY:
+    #raise EnvironmentError("OPENROUTER_API_KEY environment variable not set")
+
+#common = dict(
+    #api_base="https://openrouter.ai/api/v1",
+    #api_key=OPENROUTER_API_KEY,
+    #extra_body={"usage": {"include": True}}
+#)
 
 common = dict(
-    api_base="https://openrouter.ai/api/v1",
-    api_key=OPENROUTER_API_KEY,
-    extra_body={"usage": {"include": True}}
+    api_base = "https://generativelanguage.googleapis.com/v1beta/openai/",
+    api_key = os.getenv("GEMINI_API_KEY"),
 )
 class GaiaAgent:
     def __init__(self):
@@ -28,9 +33,10 @@ class GaiaAgent:
         #token = os.getenv("HF_API_TOKEN")
         #print("DEBUG: HF_API_TOKEN is", token[:6]+ "...")
         self.model = OpenAIServerModel(
-            #max_tokens = 8096,
-            temperature = 0.5, 
-            model_id = 'qwen/qwen-2.5-coder-32b-instruct:free',
+            max_tokens = 8096,
+            temperature = 0.5,
+            model_id = "gemini-2.0-flash", 
+            #model_id = 'qwen/qwen-2.5-coder-32b-instruct:free',
             #custom_role_conversions=None,      
             #provider="hf-inference",
             #token=token,