Update agent.py

agent.py

@@ -17,15 +17,15 @@ from vision_tool import image_reasoning_tool
 #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://generativelanguage.googleapis.com/v1beta/openai/",
-    api_key = os.getenv("GEMINI_API_KEY"),
+    api_base="https://openrouter.ai/api/v1",
+    api_key=OPENROUTER_API_KEY,
+    extra_body={"usage": {"include": True}}
+)
+
+#common = dict(
+    #api_base = "https://generativelanguage.googleapis.com/v1beta/openai/",
+    #api_key = os.getenv("GEMINI_API_KEY"),
 )
 class GaiaAgent:
     def __init__(self):
@@ -35,8 +35,8 @@ class GaiaAgent:
         self.model = OpenAIServerModel(
             max_tokens = 8096,
             temperature = 0.5,
-            model_id = "gemini-2.0-flash", 
-            #model_id = 'qwen/qwen-2.5-coder-32b-instruct:free',
+            #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,