from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool import datetime import requests import pytz import yaml import json import os from tools.final_answer import FinalAnswerTool from Gradio_UI import GradioUI @tool def task_manager(action: str, task_data: str = "") -> str: """A tool for managing daily tasks. Args: action: The operation to perform ("add", "list", "complete", "delete", or "help") task_data: Additional information related to the task (task description, task ID, etc.) Returns: A text containing information about the result of the operation """ import json import os from datetime import datetime # File to store task data TASKS_FILE = "tasks.json" # Create an empty task list if not os.path.exists(TASKS_FILE): with open(TASKS_FILE, "w") as f: json.dump({"tasks": []}, f) # Load tasks with open(TASKS_FILE, "r") as f: data = json.load(f) # Task list tasks = data["tasks"] # Help message if action == "help" or action == "": return """Task Manager Tool Usage: - add: Add a new task (e.g.: task_manager("add", "Buy milk from the store")) - list: List all tasks (e.g.: task_manager("list")) - complete: Mark a task as completed (e.g.: task_manager("complete", "1")) - delete: Delete a task (e.g.: task_manager("delete", "1")) - help: Show this help message""" # Add a new task elif action == "add": if not task_data: return "Error: Task description cannot be empty!" # Create a new task new_task = { "id": len(tasks) + 1, "description": task_data, "created_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "completed": False } # Add the task to the list tasks.append(new_task) # Save changes with open(TASKS_FILE, "w") as f: json.dump(data, f, indent=2) return f"Task added: {task_data} (ID: {new_task['id']})" # List tasks elif action == "list": if not tasks: return "Your task list is empty." result = "YOUR TASKS:\n" result += "-" * 50 + "\n" for task in tasks: status = "✓" if task["completed"] else "□" result += f"{task['id']}. [{status}] {task['description']} ({task['created_at']})\n" return result # Mark task as completed elif action == "complete": try: task_id = int(task_data) # Find task by ID for task in tasks: if task["id"] == task_id: task["completed"] = True # Save changes with open(TASKS_FILE, "w") as f: json.dump(data, f, indent=2) return f"Task completed: {task['description']}" return f"Error: Task with ID {task_id} not found." except ValueError: return "Error: You must specify a valid task ID." # Delete task elif action == "delete": try: task_id = int(task_data) # Find task by ID for i, task in enumerate(tasks): if task["id"] == task_id: # Remove task from the list deleted_task = tasks.pop(i) # Save changes with open(TASKS_FILE, "w") as f: json.dump(data, f, indent=2) return f"Task deleted: {deleted_task['description']}" return f"Error: Task with ID {task_id} not found." except ValueError: return "Error: You must specify a valid task ID." else: return f"Error: Unknown action '{action}'. Valid actions: add, list, complete, delete, help" @tool def get_current_time_in_timezone(timezone: str) -> str: """A tool that fetches the current local time in a specified timezone. Args: timezone: A string representing a valid timezone (e.g., 'America/New_York'). """ try: # Create timezone object tz = pytz.timezone(timezone) # Get current time in that timezone local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S") return f"The current local time in {timezone} is: {local_time}" except Exception as e: return f"Error fetching time for timezone '{timezone}': {str(e)}" final_answer = FinalAnswerTool() # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' model = HfApiModel( max_tokens=2096, temperature=0.5, model_id='Qwen/Qwen2.5-Coder-32B-Instruct', # it is possible that this model may be overloaded custom_role_conversions=None, ) # Import tool from Hub image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) with open("prompts.yaml", 'r') as stream: prompt_templates = yaml.safe_load(stream) agent = CodeAgent( model=model, tools=[final_answer, task_manager, get_current_time_in_timezone, image_generation_tool], # Eklediğimiz task_manager aracını dahil ettik max_steps=6, verbosity_level=1, grammar=None, planning_interval=None, name=None, description=None, prompt_templates=prompt_templates ) GradioUI(agent).launch()