Upload agent

agent.json

@@ -9,9 +9,10 @@
         "final_answer"
     ],
     "model": {
-        "class": "InferenceClientModel",
+        "class": "LiteLLMModel",
         "data": {
-            "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct"
+            "model_id": "gemini/gemini-2.0-flash",
+            "api_base": null
         }
     },
     "managed_agents": {},

app.py

@@ -1,6 +1,6 @@
 import yaml
 import os
-from smolagents import GradioUI, CodeAgent, InferenceClientModel
+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 = InferenceClientModel(
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+model = LiteLLMModel(
+model_id='gemini/gemini-2.0-flash',
+api_base=None,
 )
 
 web_search = WebSearch()

tools/web_search.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import duckduckgo_search
 import time
+import duckduckgo_search
 
 class DuckDuckGoSearchTool(Tool):
     """Web search tool that performs searches using the DuckDuckGo search engine.