Upload agent

agent.json

@@ -1,14 +1,20 @@
 {
     "class": "CodeAgent",
     "tools": [
+        "web_search",
+        "visit_webpage",
+        "suggest_menu",
+        "catering_service_tool",
+        "superhero_party_theme_generator",
         "final_answer"
     ],
     "model": {
-        "class": "HfApiModel",
+        "class": "LiteLLMModel",
         "data": {
-            "last_input_token_count": null,
-            "last_output_token_count": null,
-            "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct"
+            "last_input_token_count": 3845,
+            "last_output_token_count": 170,
+            "model_id": "gemini/gemini-2.0-flash-lite",
+            "api_base": null
         }
     },
     "managed_agents": {},
@@ -28,13 +34,16 @@
             "post_messages": "Based on the above, please provide an answer to the following user task:\n{{task}}"
         }
     },
-    "max_steps": 20,
-    "verbosity_level": 1,
+    "max_steps": 10,
+    "verbosity_level": 2,
     "grammar": null,
     "planning_interval": null,
     "name": null,
     "description": null,
     "requirements": [
+        "duckduckgo_search",
+        "markdownify",
+        "requests",
         "smolagents"
     ],
     "authorized_imports": [

app.py

@@ -1,18 +1,29 @@
 import yaml
 import os
-from smolagents import GradioUI, CodeAgent, HfApiModel
+from smolagents import GradioUI, CodeAgent, LiteLLMModel
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
+from tools.web_search import DuckDuckGoSearchTool as WebSearch
+from tools.visit_webpage import VisitWebpageTool as VisitWebpage
+from tools.suggest_menu import SimpleTool as SuggestMenu
+from tools.catering_service_tool import SimpleTool as CateringServiceTool
+from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
 
-model = HfApiModel(
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+model = LiteLLMModel(
+model_id='gemini/gemini-2.0-flash-lite',
+api_base=None,
 )
 
+web_search = WebSearch()
+visit_webpage = VisitWebpage()
+suggest_menu = SuggestMenu()
+catering_service_tool = CateringServiceTool()
+superhero_party_theme_generator = SuperheroPartyThemeGenerator()
 final_answer = FinalAnswer()
 
 
@@ -21,11 +32,11 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[],
+    tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
     managed_agents=[],
     class='CodeAgent',
-    max_steps=20,
-    verbosity_level=1,
+    max_steps=10,
+    verbosity_level=2,
     grammar=None,
     planning_interval=None,
     name=None,

requirements.txt

@@ -1 +1,4 @@
+duckduckgo_search
+markdownify
+requests
 smolagents

tools/visit_webpage.py

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