Upload agent

agent.json

@@ -8,11 +8,12 @@
         "final_answer"
     ],
     "model": {
-        "class": "AzureOpenAIServerModel",
+        "class": "LiteLLMModel",
         "data": {
-            "last_input_token_count": 2280,
-            "last_output_token_count": 764,
-            "model_id": "o3-mini"
+            "last_input_token_count": 14624,
+            "last_output_token_count": 139,
+            "model_id": "gemini/gemini-2.0-flash-001",
+            "api_base": null
         }
     },
     "managed_agents": {},
@@ -48,17 +49,17 @@
         "duckduckgo_search"
     ],
     "authorized_imports": [
-        "random",
-        "re",
-        "queue",
         "datetime",
         "itertools",
-        "math",
-        "time",
-        "collections",
+        "re",
+        "stat",
         "statistics",
+        "time",
+        "queue",
+        "random",
+        "math",
         "unicodedata",
-        "stat"
+        "collections"
     ],
     "executor_type": "local",
     "executor_kwargs": {},

app.py

@@ -1,6 +1,6 @@
 import yaml
 import os
-from smolagents import GradioUI, CodeAgent, AzureOpenAIServerModel
+from smolagents import GradioUI, CodeAgent, LiteLLMModel
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -14,8 +14,9 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
 
-model = AzureOpenAIServerModel(
-model_id='o3-mini',
+model = LiteLLMModel(
+model_id='gemini/gemini-2.0-flash-001',
+api_base=None,
 )
 
 web_search = WebSearch()

requirements.txt

@@ -1,4 +1,4 @@
 markdownify
 requests
-smolagents[azureopenai,openai]
+smolagents
 duckduckgo_search

tools/visit_webpage.py

@@ -1,9 +1,9 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import requests
-import smolagents
 import re
 import markdownify
+import requests
+import smolagents
 
 class VisitWebpageTool(Tool):
     name = "visit_webpage"