Upload 6 files

app.py

@@ -1,36 +1,16 @@
-from smolagents import CodeAgent, InferenceClientModel, FinalAnswerTool
-from tools.party_song import suggest_party_music
-from tools.party_menu import suggest_menu
-from tools.party_time import estimate_preparation_time
-from smolagents import FinalAnswerTool
-
-#引入 yaml 库，用来加载系统提示（prompts.yaml）文件。
-import yaml 
-
-# 定义终极回答工具 FinalAnswerTool , smolagents 自带的工具，
-# 作用是告诉代理 “这是最后一步，把结果返回给用户”。
-# 没有这个工具，Agent 可能会一直循环思考，不知道什么时候结束。
-final_answer = FinalAnswerTool()
-
-#初始化模型
-model = InferenceClientModel(
-    model_id="Qwen/Qwen2.5-Coder-7B-Instruct",
-    max_tokens=1024,
-    temperature=0.5,
-    custom_role_conversions=None,
-)
-
-#加载系统提示
-with open("prompts.yaml", "r") as f:
-    prompt_templates = yaml.safe_load(f)
-
-#构建代理
-agent = CodeAgent(
-    model=model,
-    tools=[final_answer, suggest_party_music, suggest_menu, estimate_preparation_time],
-    prompt_templates=prompt_templates
-)
-
-if __name__ == "__main__":
-    from Gradio_UI import GradioUI
-    GradioUI(agent).launch()
+from smolagents import CodeAgent, InferenceClientModel
+from tools.party_menu import suggest_menu
+from tools.party_song import suggest_party_music
+from tools.party_time import estimate_preparation_time
+from tools.final_answer import FinalAnswerTool
+
+model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-7B-Instruct")
+
+agent = CodeAgent(
+    model=model,
+    tools=[FinalAnswerTool(), suggest_menu, suggest_party_music, estimate_preparation_time],
+)
+
+if __name__ == "__main__":
+    from Gradio_UI import GradioUI
+    GradioUI(agent).launch()

tools/party_menu.py

@@ -1,6 +1,5 @@
-from smolagents import CodeAgent, tool, InferenceClientModel
+from smolagents import tool
 
-# Tool to suggest a menu based on the occasion
 @tool
 def suggest_menu(occasion: str) -> str:
     """
@@ -20,9 +19,3 @@ def suggest_menu(occasion: str) -> str:
         return "Buffet with high-energy and healthy food."
     else:
         return "Custom menu for the butler."
-
-# Alfred, the butler, preparing the menu for the party
-agent = CodeAgent(tools=[suggest_menu], model=InferenceClientModel())
-
-# Preparing the menu for the party
-agent.run("Prepare a casual menu for the party.")

tools/party_song.py

@@ -1,14 +1,11 @@
-import os
-from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
-
-HF_TOKEN = os.getenv("HF_TOKEN")
-
-agent = CodeAgent(
-    tools=[DuckDuckGoSearchTool()],
-    model=InferenceClientModel(
-        model_id="Qwen/Qwen2.5-Coder-7B-Instruct",  # 或者你要用的模型
-        api_key=HF_TOKEN
-    )
-)
-
-agent.run("Search for the best music recommendations for a party at the Wayne's mansion.")
+from smolagents import tool, DuckDuckGoSearchTool
+
+@tool
+def suggest_party_music(query: str = "best music for a party") -> str:
+    """
+    Suggests music playlists or songs for a party by searching online.
+    Args:
+        query (str): Search query for music.
+    """
+    search_tool = DuckDuckGoSearchTool()
+    return search_tool(query)

tools/party_time.py

@@ -1,18 +1,23 @@
-from smolagents import CodeAgent, InferenceClientModel
-import numpy as np
-import time
+from smolagents import tool
 import datetime
 
-agent = CodeAgent(tools=[], model=InferenceClientModel(), additional_authorized_imports=['datetime'])
-
-agent.run(
+@tool
+def estimate_preparation_time(start_time: str = None) -> str:
     """
-    Alfred needs to prepare for the party. Here are the tasks:
-    1. Prepare the drinks - 30 minutes
-    2. Decorate the mansion - 60 minutes
-    3. Set up the menu - 45 minutes
-    4. Prepare the music and playlist - 45 minutes
-
-    If we start right now, at what time will the party be ready?
+    Estimates total preparation time for a party and returns when everything will be ready.
+    Args:
+        start_time (str, optional): Start time in format HH:MM. Defaults to current time.
     """
-)
+    tasks = {
+        "Prepare the drinks": 30,
+        "Decorate the mansion": 60,
+        "Set up the menu": 45,
+        "Prepare the music and playlist": 45,
+    }
+    total_minutes = sum(tasks.values())
+    if start_time:
+        now = datetime.datetime.strptime(start_time, "%H:%M")
+    else:
+        now = datetime.datetime.now()
+    ready_time = now + datetime.timedelta(minutes=total_minutes)
+    return f"Total prep time: {total_minutes} minutes. The party will be ready at {ready_time.strftime('%H:%M')}."