Upload agent

agent.json

@@ -11,8 +11,8 @@
     "model": {
         "class": "HfApiModel",
         "data": {
-            "last_input_token_count": null,
-            "last_output_token_count": null,
+            "last_input_token_count": 15850,
+            "last_output_token_count": 110,
             "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct"
         }
     },

app.py

@@ -33,6 +33,7 @@ agent = CodeAgent(
     model=model,
     tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
     managed_agents=[],
+    class='CodeAgent',
     max_steps=10,
     verbosity_level=2,
     grammar=None,

tools/superhero_party_theme_generator.py

@@ -6,17 +6,17 @@ class SuperheroPartyThemeTool(Tool):
     description = """
     This tool suggests creative superhero-themed party ideas based on a category.
     It returns a unique party theme idea."""
-    inputs = {'category': {'type': 'string', 'description': "The type of superhero party (e.g., 'classic heroes', 'villain masquerade', 'futuristic gotham')."}}
+    inputs = {'category': {'type': 'string', 'description': "The type of superhero party (e.g., 'classic heroes', 'villain masquerade', 'futuristic Gotham')."}}
     output_type = "string"
 
     def forward(self, category: str):
         themes = {
             "classic heroes": "Justice League Gala: Guests come dressed as their favorite DC heroes with themed cocktails like 'The Kryptonite Punch'.",
             "villain masquerade": "Gotham Rogues' Ball: A mysterious masquerade where guests dress as classic Batman villains.",
-            "futuristic gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
+            "futuristic Gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
         }
 
-        return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic gotham'.")
+        return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic Gotham'.")
 
     def __init__(self, *args, **kwargs):
         self.is_initialized = False

tools/visit_webpage.py

@@ -1,9 +1,8 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import re
 import requests
+import re
 import markdownify
-import smolagents
 
 class VisitWebpageTool(Tool):
     name = "visit_webpage"
@@ -15,6 +14,15 @@ class VisitWebpageTool(Tool):
         super().__init__()
         self.max_output_length = max_output_length
 
+    def _truncate_content(self, content: str, max_length: int) -> str:
+        if len(content) <= max_length:
+            return content
+        return (
+            content[: max_length // 2]
+            + f"\n..._This content has been truncated to stay below {max_length} characters_...\n"
+            + content[-max_length // 2 :]
+        )
+
     def forward(self, url: str) -> str:
         try:
             import re
@@ -22,8 +30,6 @@ class VisitWebpageTool(Tool):
             import requests
             from markdownify import markdownify
             from requests.exceptions import RequestException
-
-            from smolagents.utils import truncate_content
         except ImportError as e:
             raise ImportError(
                 "You must install packages `markdownify` and `requests` to run this tool: for instance run `pip install markdownify requests`."
@@ -39,7 +45,7 @@ class VisitWebpageTool(Tool):
             # Remove multiple line breaks
             markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
 
-            return truncate_content(markdown_content, self.max_output_length)
+            return self._truncate_content(markdown_content, self.max_output_length)
 
         except requests.exceptions.Timeout:
             return "The request timed out. Please try again later or check the URL."