Spaces:
Runtime error
Runtime error
File size: 3,426 Bytes
9b5b26a c19d193 6aae614 acd3601 8fe992b 9b5b26a acd3601 5df72d6 9b5b26a acd3601 9b5b26a acd3601 9b5b26a acd3601 9b5b26a 8c01ffb 6aae614 e121372 bf6d34c acd3601 fe328e0 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e 9b5b26a 8c01ffb 8fe992b acd3601 8c01ffb 861422e 8fe992b 9b5b26a 8c01ffb |
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 106 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
import random
from Gradio_UI import GradioUI
class Position:
def __init__(self, x: int, y: int):
self.x = x
self.y = y
class TreasureMapGame:
def __init__(self):
self.treasure_position = None
self.position = None
self.initialize_game()
def initialize_game(self):
self.treasure_position = Position(random.randint(-10, 10), random.randint(-10, 10))
self.position = Position(0, 0)
def check_distance(self) -> str:
distance = abs(self.treasure_position.x - self.position.x) + abs(self.treasure_position.y - self.position.y)
if distance == 0:
message = "You found the treasure!"
self.initialize_game()
else:
message = f"You are {distance} steps away from the treasure."
return message
def move(self, direction: str, steps: int):
if direction == "left":
self.position.x -= steps
if direction == "right":
self.position.x += steps
if direction == "down":
self.position.y -= steps
if direction == "up":
self.position.y += steps
game = TreasureMapGame()
# Below is an example of a tool that does nothing. Amaze us with your creativity !
@tool
def move(direction:str, steps:int) -> str: #it's import to specify the return type
"""A tool that moves your position to find the hidden treasure in the Treasure Map Game. Returns a message indicating the distance to the treasure.
Args:
direction: One of 'left', 'right', 'up' or 'down'
steps: Amount of steps to move in the specified direction
"""
game.move(direction, steps)
status = game.check_distance()
return status
@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='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
# model_id='mistralai/Mistral-7B-Instruct-v0.3',
model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
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, move, get_current_time_in_timezone], ## 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() |