Upload agent

agent.json

@@ -3,6 +3,7 @@
     "tools": [
         "web_search",
         "suggest_menu",
+        "list_occasions",
         "final_answer"
     ],
     "model": {

app.py

@@ -7,6 +7,7 @@ CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
 from tools.web_search import DuckDuckGoSearchTool as WebSearch
 from tools.suggest_menu import SimpleTool as SuggestMenu
+from tools.list_occasions import SimpleTool as ListOccasions
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
@@ -18,6 +19,7 @@ api_base=None,
 
 web_search = WebSearch()
 suggest_menu = SuggestMenu()
+list_occasions = ListOccasions()
 final_answer = FinalAnswer()
 
 
@@ -26,7 +28,7 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[web_search, suggest_menu],
+    tools=[web_search, suggest_menu, list_occasions],
     managed_agents=[],
     class='CodeAgent',
     max_steps=20,

tools/list_occasions.py

@@ -0,0 +1,14 @@
+from smolagents import Tool
+from typing import Any, Optional
+
+class SimpleTool(Tool):
+    name = "list_occasions"
+    description = "Lists the available occasions."
+    inputs = {}
+    output_type = "string"
+
+    def forward(self) -> str:
+        """
+        Lists the available occasions.
+        """
+        return "casual, formal, superhero, custom"