Spaces:
Sleeping
Sleeping
File size: 2,877 Bytes
b174213 9b5b26a c19d193 6aae614 8fe992b 9b5b26a 8836d9e b174213 0b26969 b174213 9b5b26a 0b26969 9b5b26a 0b26969 9b5b26a 0b26969 8c01ffb 6aae614 ae7a494 e121372 bf6d34c 70fbfaa fe328e0 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e 9b5b26a 8c01ffb 8fe992b 0b26969 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 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
@tool
def get_list_of_people_on_iss() -> dict:
"""A tool that fetches the current number of people in space. When known it also fetches the names and spacecraft those people are on. Returns.
Returns:
dict: A JSON object containing the following keys:
- "message" (str): Status message (e.g., "success").
- "number" (int): The number of astronauts currently in space.
- "people" (list of dict): A list of astronauts, where each entry contains:
- "name" (str): The astronaut's name.
- "craft" (str): The name of the spacecraft they are on.
Args: None
"""
url = "http://api.open-notify.org/astros.json"
response = requests.get(url)
if response.status_code == 200:
return response.json()
else:
return f"Error fetching astronauts: {response.status_code}"
@tool
def get_iss_location() -> dict:
"""A tool that fetches the current location of the International Space Station (ISS).
Returns:
dict: A JSON object containing the following keys:
- "message" (str): Status message (e.g., "success").
- "timestamp" (int): The UNIX timestamp when the location was recorded.
- "iss_position" (dict): The latitude and longitude of the ISS.
- "latitude" (str): The current latitude of the ISS.
- "longitude" (str): The current longitude of the ISS.
Args: None
"""
url = "http://api.open-notify.org/iss-now.json"
response = requests.get(url)
if response.status_code == 200:
return response.json()
else:
return f"Error fetching ISS location: {response.status_code}"
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='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_list_of_people_on_iss, get_iss_location],
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch() |