Spaces:
Sleeping
Sleeping
File size: 3,429 Bytes
5ab5f5e 9b5b26a c19d193 6aae614 5ab5f5e 8fe992b 9b5b26a 5df72d6 5ab5f5e 9b5b26a 5ab5f5e 9b5b26a 5ab5f5e 9b5b26a 5ab5f5e 9b5b26a 8c01ffb 6aae614 5ab5f5e e121372 5ab5f5e 13d500a 8c01ffb 14bad24 8c01ffb 9b5b26a 5ab5f5e 8c01ffb 861422e 5ab5f5e 8c01ffb 8fe992b 5ab5f5e 8c01ffb 861422e 8fe992b 9b5b26a 5ab5f5e |
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 104 105 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
import os
import uuid
from dotenv import load_dotenv
from tools.image_caption import ImageCaptionTool
from tools.image_generation import ImageGenerationTool
from Gradio_UI import GradioUI
# Below is an example of a tool that does nothing. Amaze us with your creativity !
@tool
def basic_calculator(operation: str, num1: float, num2: float) -> str:
"""A tool that performs basic arithmetic operations.
Args:
operation: The arithmetic operation to perform ('add', 'subtract', 'multiply', 'divide').
num1: The first number.
num2: The second number.
"""
try:
if operation == 'add':
result = num1 + num2
elif operation == 'subtract':
result = num1 - num2
elif operation == 'multiply':
result = num1 * num2
elif operation == 'divide':
if num2 == 0:
return "Error: Division by zero is not allowed."
result = num1 / num2
else:
return "Error: Invalid operation. Please use 'add', 'subtract', 'multiply', or 'divide'."
return f"The result of {operation}ing {num1} and {num2} is: {result}"
except Exception as e:
return f"Error performing calculation: {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)}"
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,
)
# load_dotenv(override=True)
# token = os.getenv("HF_API_KEY")
# if not token:
# raise ValueError("HF_API_KEY environment variable not set")
# Import tool from Hub
# image_generation_tool = load_tool(
# "agents-course/text-to-image", trust_remote_code=True, token=token
# )
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, get_current_time_in_timezone, basic_calculator,
ImageGenerationTool(), ImageCaptionTool(),
DuckDuckGoSearchTool()
],
max_steps=15,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch()
|