Cadre d'agents d'intelligence artificielle Smolagents

Exemple d'application des SmolAgents

Dans Smolagents, les agents sont encapsulés dans des classes qui leur permettent d'effectuer des tâches spécifiques. Par exemple, un Agent géré peut être créé pour gérer différentes tâches en utilisant différents types d'agents. Voici un exemple simplifié de la mise en œuvre de deux agents, l'un pour la recherche sur le web et l'autre pour la génération d'images,Voici comment créer un outil personnalisé qui récupère les temps de trajet à partir de Google Maps, et comment l'utiliser dans un agent de planification de voyages :

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

Après quelques étapes de collecte des temps de parcours et de calculs, l'agent vous renvoie cette proposition finale : Itinéraire d'une journée à vélo à Paris :

1. Départ de la Tour Eiffel à 9h00.
2. Visite de la Tour Eiffel jusqu'à 10h30.
3. Se rendre à la cathédrale Notre-Dame à 10h46.
4. Visite de la cathédrale Notre-Dame jusqu'à 12h16.
5. Se rendre à Montmartre à 12h41.
6. Visite de Montmartre jusqu'à 14h11.
7. Voyage au Jardin du Luxembourg à 14h33.
8. Visite du Jardin du Luxembourg jusqu'à 16h03.
9. Voyage au musée du Louvre à 16h12.
10. Visite du musée du Louvre jusqu'à 17h42.
11. Pause déjeuner jusqu'à 18h12.
12. Heure de fin prévue : 18 h 12.

Après avoir créé un outil, il suffit de le partager avec le Hub :

get_travel_duration.push_to_hub("{votre_nom d'utilisateur}/get-travel-duration-tool")