Update agent.py

agent.py

@@ -1,21 +1,32 @@
-from smolagents import CodeAgent, InferenceClientModel
+from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel 
 from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
 import os
+from vision_tool import image_reasoning_tool
 
-token=os.getenv("HF_API_TOKEN")
+#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")
+
+common = dict(
+    api_base="https://openrouter.ai/api/v1",
+    api_key=OPENROUTER_API_KEY,
+    extra_body={"usage": {"include": True}}
+)
 class GaiaAgent:
     def __init__(self):
-        import os
-        token = os.getenv("HF_API_TOKEN")
-        print("DEBUG: HF_API_TOKEN is", token[:6]+ "...")
-        self.model = InferenceClientModel(
-            max_tokens = 1024,
+        #import os
+        #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 = 'meta-llama/Llama-3.1-8B-Instruct',
+            model_id = 'qwen/qwen-2.5-coder-32b-instruct:free',
             #custom_role_conversions=None,      
-            provider="hf-inference",
-            token=token
+            #provider="hf-inference",
+            token=token,
+            **common
         )
 
 
@@ -28,6 +39,7 @@ class GaiaAgent:
                 WikipediaSearchTool(),
                 PythonInterpreterTool(),
                 FinalAnswerTool(),
+                image_reasoning_tool,
             ], 
             max_steps=10,
             verbosity_level = 2,        #or other parameter