Marco de agentes de IA Smolagents

Ejemplo de aplicación de SmolAgents

En Smolagents, los agentes se encapsulan en clases que les permiten realizar tareas específicas. Por ejemplo, un agente ManagedAgent se pueden crear para gestionar varias tareas utilizando diferentes tipos de agentes. A continuación se muestra un ejemplo simplificado de cómo se pueden implementar dos agentes: uno para la búsqueda web y otro para la generación de imágenes. A continuación se explica cómo crear una herramienta personalizada que obtenga los tiempos de viaje de Google Maps y cómo utilizarla en un agente planificador de viajes:

from typing import Optional
from smolagents import CodeAgent, HfApiModel, tool

@tool
def get_travel_duration(start_location: str, destination_location: str, transportation_mode: Optional[str] = None) -> str:
    """Gets the travel time between two places.

    Args:
        start_location: the place from which you start your ride
        destination_location: the place of arrival
        transportation_mode: The transportation mode, in 'driving', 'walking', 'bicycling', or 'transit'. Defaults to 'driving'.
    """
    import os   # All imports are placed within the function, to allow for sharing to Hub.
    import googlemaps
    from datetime import datetime

    gmaps = googlemaps.Client(os.getenv("GMAPS_API_KEY"))

    if transportation_mode is None:
        transportation_mode = "driving"
    try:
        directions_result = gmaps.directions(
            start_location,
            destination_location,
            mode=transportation_mode,
            departure_time=datetime(2025, 6, 6, 11, 0), # At 11, date far in the future
        )
        if len(directions_result) == 0:
            return "No way found between these places with the required transportation mode."
        return directions_result[0]["legs"][0]["duration"]["text"]
    except Exception as e:
        print(e)
        return e

agent = CodeAgent(tools=[get_travel_duration], model=HfApiModel(), additional_authorized_imports=["datetime"])

agent.run("Can you give me a nice one-day trip around Paris with a few locations and the times? Could be in the city or outside, but should fit in one day. I'm travelling only with a rented bicycle.")

Tras unos cuantos pasos de recopilación de tiempos de viaje y ejecución de cálculos, el agente devuelve esta propuesta final:Itinerario de un día en bicicleta por París:

1. Salida de la Torre Eiffel a las 9:00 h.
2. Visita a la Torre Eiffel hasta las 10:30.
3. Viaje a la catedral de Notre-Dame a las 10:46.
4. Visita a la Catedral de Notre-Dame hasta las 12:16 PM.
5. Viaje a Montmartre a las 12:41 PM.
6. Visita a Montmartre hasta las 14:11.
7. Viaje al Jardín de Luxemburgo a las 14:33.
8. Visita al Jardín de Luxemburgo hasta las 16:03.
9. Viaje al Museo del Louvre a las 16:12.
10. Visita al Museo del Louvre hasta las 17:42.
11. Pausa para comer hasta las 18:12.
12. Hora prevista de finalización: 18:12.

Después de crear una herramienta, compartirla con Hub es tan sencillo como:

get_travel_duration.push_to_hub("{su_nombre_de_usuario}/get-travel-duration-tool")