Spaces:
Sleeping
Sleeping
File size: 2,597 Bytes
9b5b26a c19d193 8fe992b 9b5b26a 9729baf 9b5b26a 9729baf 9b5b26a 9729baf 9b5b26a 9729baf 9b5b26a 8c01ffb 6aae614 ae7a494 9729baf ae7a494 e121372 9729baf 13d500a 8c01ffb 9b5b26a 9729baf 8c01ffb 861422e 9729baf 8c01ffb 8fe992b 9729baf c93f8f7 8c01ffb 861422e 8fe992b 9b5b26a 9729baf |
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 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import datetime
import pytz
import yaml
from Gradio_UI import GradioUI
from smolagents import CodeAgent, HfApiModel, load_tool, tool
from tools.final_answer import FinalAnswerTool
from tools.web_search import DuckDuckGoSearchTool
from tools.visit_webpage import VisitWebpageTool
@tool
def visit_webpage(url: str) -> str:
"""Visits a webpage at the given URL and reads its content as a markdown string
Args:
url: The URL of the webpage to visit (e.g., 'http://www.wikipedia.com'.).
"""
visit_webpage = VisitWebpageTool()
webpage_content = visit_webpage.forward(url)
return webpage_content
@tool
def duck_duck_go_search(query: str) -> str:
"""A tool that makes a web search request using DuckDuckGo.
Args:
query: a string with the user's web search query (e.g., 'When was the city of London founded?').
"""
search = DuckDuckGoSearchTool()
search_result = search.forward(query)
return search_result
@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:
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()
# 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,
# it is possible that this model may be overloaded
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
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,
# add your tools here (don't remove final answer)
tools=[final_answer, image_generation_tool,
duck_duck_go_search, visit_webpage, get_current_time_in_timezone],
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch()
|