Struttura per agenti AI Smolagents

Esempio di applicazione degli SmolAgent

In Smolagents, gli agenti sono incapsulati in classi che consentono loro di eseguire compiti specifici. Ad esempio, un agente Agente gestito possono essere creati per gestire varie attività utilizzando diversi tipi di agenti. Ecco un esempio semplificato di come si possono implementare due agenti, uno per la ricerca sul Web e l'altro per la generazione di immagini. Ecco come creare uno strumento personalizzato che ottiene gli orari di viaggio da Google Maps e come utilizzarlo in un agente di pianificazione dei viaggi:

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.")

Dopo alcune fasi di raccolta dei tempi di viaggio e di esecuzione dei calcoli, l'agente restituisce questa proposta finale: Itinerario di un giorno per Parigi in bicicletta:

1. Partenza dalla Torre Eiffel alle 9:00.
2. Visita alla Torre Eiffel fino alle 10:30.
3. Viaggio verso la Cattedrale di Notre-Dame alle 10:46.
4. Visita alla Cattedrale di Notre-Dame fino alle 12:16.
5. Viaggio a Montmartre alle 12:41.
6. Visita di Montmartre fino alle 14:11.
7. Viaggio verso il Jardin du Luxembourg alle 14:33.
8. Visita al Jardin du Luxembourg fino alle 16.03.
9. Viaggio al Museo del Louvre alle 16:12.
10. Visita al Museo del Louvre fino alle 17:42.
11. Pausa pranzo fino alle 18:12.
12. Orario di fine previsto: 18:12.

Dopo aver costruito uno strumento, condividerlo nell'Hub è semplicissimo:

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