Spaces:
Sleeping
Sleeping
File size: 2,982 Bytes
9b5b26a 83f6647 9b5b26a c19d193 83f6647 8fe992b 9b5b26a 83f6647 9b5b26a 83f6647 9b5b26a 83f6647 9b5b26a 83f6647 9b5b26a 8c01ffb 83f6647 6aae614 ae7a494 8c01ffb 9b5b26a 8c01ffb 83f6647 861422e 83f6647 8c01ffb 8fe992b 83f6647 8c01ffb 83f6647 8fe992b 9b5b26a 83f6647 |
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 |
import datetime
import pytz
import yaml
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
from Gradio_UI import GradioUI
from tools.final_answer import 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
custom_role_conversions=None,
)
@tool
def analyze_sentiment(snippet: str) -> str:
"""A tool that analyzes if the sentiment of the snippet is positive, negative, or neutral.
Args:
snippet: A string representing the text to be analyzed.
"""
try:
output = model.create_client().chat.completions.create(
messages=[
{
"role": "system",
"content": "You are a sentiment analysis expert. Your task is to analyze the sentiment of the given text and respond with exactly one word: 'positive', 'negative', or 'neutral'. Do not include any other text in your response.",
},
{
"role": "user",
"content": f"Analyze the sentiment of this text: {snippet}",
},
],
stream=False,
max_tokens=5,
temperature=0.1,
)
sentiment = "Sentiment: " + output.choices[0].message.content.strip().lower()
return sentiment
except Exception as e:
return f"Error analyzing sentiment: {str(e)}"
@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)}"
web_search = DuckDuckGoSearchTool()
final_answer = FinalAnswerTool()
# 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=[
get_current_time_in_timezone,
web_search,
analyze_sentiment,
final_answer,
],
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates,
)
GradioUI(agent).launch()
|