from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool import datetime import requests import pytz import yaml from tools.final_answer import FinalAnswerTool from Gradio_UI import GradioUI @tool def generate_blog_outline(topic: str) -> str: """Creates a structured outline for a blog post Args: topic: The main topic for the blog post """ return "Outline generated based on: " + topic @tool def suggest_blog_topics(main_theme: str) -> str: """Suggests related blog topics based on a theme Args: main_theme: The primary theme or area of interest """ return "Related topics for: " + main_theme @tool def get_current_time_in_timezone(timezone: str) -> str: """Fetches current local time in specified timezone Args: timezone: Valid timezone (e.g., 'America/New_York') """ try: tz = pytz.timezone(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() model = HfApiModel( max_tokens=2096, temperature=0.5, model_id='Qwen/Qwen2.5-Coder-32B-Instruct', custom_role_conversions=None, ) with open("prompts.yaml", 'r') as stream: prompt_templates = yaml.safe_load(stream) agent = CodeAgent( model=model, tools=[ final_answer, generate_blog_outline, suggest_blog_topics, get_current_time_in_timezone ], max_steps=6, verbosity_level=1, grammar=None, planning_interval=None, name="BlogAssistant", description="An agent that helps with blog writing and content planning" ) GradioUI(agent).launch()