File size: 2,061 Bytes
265fd7f 9b5b26a c19d193 6aae614 a2ca67f 9b5b26a 265fd7f 9b5b26a 265fd7f 9b5b26a 265fd7f 9b5b26a 265fd7f 9b5b26a 265fd7f 9b5b26a 8c01ffb 6aae614 ae7a494 e121372 265fd7f 13d500a 8c01ffb 861422e 265fd7f 8c01ffb 8fe992b 265fd7f a2ca67f 265fd7f 8c01ffb 265fd7f 8fe992b 8c01ffb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from tools.blog_tools import generate_blog_section, improve_writing_style, check_readability, generate_seo_metadata
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,
generate_blog_section,
improve_writing_style,
check_readability,
generate_seo_metadata
],
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() |