Spaces:
Sleeping
Sleeping
File size: 2,984 Bytes
70d4ff2 9b5b26a 21b2cb0 9b5b26a c19d193 70d4ff2 21b2cb0 6aae614 9b5b26a 70d4ff2 21b2cb0 9b5b26a 70d4ff2 9b5b26a 70d4ff2 9b5b26a 70d4ff2 9b5b26a 8c01ffb 6aae614 e121372 70d4ff2 21b2cb0 13d500a 8c01ffb 9b5b26a 8c01ffb 70d4ff2 861422e 70d4ff2 8c01ffb 8fe992b 70d4ff2 8c01ffb 70d4ff2 8fe992b 9b5b26a 70d4ff2 |
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 93 94 95 96 97 98 99 100 101 102 103 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import os
import pytz
import yaml
import pandas
from typing import List
from dotenv import load_dotenv
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
load_dotenv()
HF_TOKEN = os.getenv("HF_U1ACAPP_TOKEN")
@tool
def format_country_info(
population: int, languages: List[str], religions: List[str]
) -> str:
"""A tool that takes as argument the most recent info about a country and most specifically
the population number, the officially languages (accompanied and sorted by their respective percentage) used
as well as the common religions (accompanied and sorted by their respective percentage) found in the country.
It returns the same info formatted in markdown.
For any info field that is not available, its value is considered as N/A
Args:
population: the country's population number according to most recent info
languages: the languages used in the country
religions: the religions found in the country
"""
summary_info = {
"Population": population if population > 0 else "N/A",
"Languages": "<br>".join(
[f"- {lang}" for lang in languages] if len(languages) > 1 else languages
)
or "N/A",
"Religions": "<br>".join(
[f"- {rel}" for rel in religions] if len(religions) > 1 else religions
)
or "N/A",
}
df = pandas.DataFrame(summary_info, index=[1])
return df.to_markdown(index=False)
@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()
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
custom_role_conversions=None,
token=HF_TOKEN,
)
# 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,
DuckDuckGoSearchTool(),
format_country_info,
], ## 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()
|