Update agent.py

agent.py

@@ -1,5 +1,13 @@
-from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel 
-from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
+from smolagents import (
+    CodeAgent, 
+    InferenceClientModel, 
+    OpenAIServerModel, 
+    VisitWebpageTool, 
+    WebSearchTool, 
+    WikipediaSearchTool, 
+    PythonInterpreterTool,
+    FinalAnswerTool
+)
 import os
 from vision_tool import image_reasoning_tool
 
@@ -20,7 +28,7 @@ class GaiaAgent:
         #token = os.getenv("HF_API_TOKEN")
         #print("DEBUG: HF_API_TOKEN is", token[:6]+ "...")
         self.model = OpenAIServerModel(
-            max_tokens = 8096,
+            #max_tokens = 8096,
             temperature = 0.5, 
             model_id = 'qwen/qwen-2.5-coder-32b-instruct:free',
             #custom_role_conversions=None,      
@@ -29,7 +37,6 @@ class GaiaAgent:
             **common
         )
 
-
 #Creating the agent
         self.agent = CodeAgent(
             model=self.model,