Spaces:
Sleeping
Sleeping
File size: 3,129 Bytes
e95d98f 9b5b26a e95d98f 9b5b26a c19d193 e95d98f 8fe992b 9b5b26a e95d98f 9b5b26a fcf5c16 9b5b26a fcf5c16 e95d98f 9b5b26a fcf5c16 9b5b26a e95d98f fcf5c16 e95d98f fcf5c16 9b5b26a 8c01ffb 6aae614 ae7a494 e121372 fcf5c16 2bc527b fcf5c16 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e e95d98f 8c01ffb 8fe992b e95d98f 8c01ffb 861422e 8fe992b e95d98f |
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 92 |
import asyncio
import datetime
import os
import dotenv
import python_weather
import pytz
import yaml
from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, HfApiModel, load_tool, tool
from Gradio_UI import GradioUI
from tools.final_answer import FinalAnswerTool
dotenv.load_dotenv(override=True)
@tool
def get_weather(location: str) -> str:
"""Tool for checking the full description of the weather for a specific location
Args:
location: name of the location for checking the weather, mandatory parameter
Return:
Description of the weather
"""
async def fetch_weather():
async with python_weather.Client(unit=python_weather.METRIC) as client:
# fetch a weather forecast from a city
weather = await client.get(location)
if weather is None:
return "Problem: unknown location or weather"
output = [f"Current temperature for {location} is {weather.temperature}"]
# get the weather forecast for a few days
for daily in weather:
desc = f"Weather for day {str(daily.date)}. Temperature high: {daily.highest_temperature}, low: {daily.lowest_temperature}. avg: {daily.temperature}."
output.append(desc)
return '\n'.join(output)
return asyncio.run(fetch_weather())
@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
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
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, get_current_time_in_timezone, get_weather, DuckDuckGoSearchTool(), VisitWebpageTool()],
## add your tools here (don't remove final answer)
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch()
|