Smolagents AI Agent Framework

Anwendungsbeispiel für SmolAgents

In Smolagents sind die Agenten in Klassen gekapselt, die es ihnen ermöglichen, bestimmte Aufgaben zu erfüllen. Zum Beispiel kann ein ManagedAgent erstellt werden, um verschiedene Aufgaben zu verwalten, indem verschiedene Agententypen verwendet werden. Im Folgenden finden Sie ein vereinfachtes Beispiel dafür, wie zwei Agenten - einer für die Websuche und ein anderer für die Bilderzeugung - implementiert werden können. Hier erfahren Sie, wie Sie ein benutzerdefiniertes Tool erstellen, das Reisezeiten aus Google Maps abruft, und wie Sie es in einem Reiseplanungsagenten verwenden:

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

Nach ein paar Schritten, in denen er Reisezeiten sammelt und Berechnungen durchführt, liefert der Agent diesen endgültigen Vorschlag: Eintägige Radtour nach Paris:

1. Start am Eiffelturm um 9:00 Uhr.
2. Sightseeing am Eiffelturm bis 10:30 Uhr.
3. Fahrt zur Kathedrale Notre-Dame um 10:46 Uhr.
4. Besichtigung der Kathedrale Notre-Dame bis 12:16 Uhr.
5. Fahrt zum Montmartre um 12:41 Uhr.
6. Sightseeing am Montmartre bis 14:11 Uhr.
7. Fahrt zum Jardin du Luxembourg um 14:33 Uhr.
8. Besichtigung des Jardin du Luxembourg bis 16:03 Uhr.
9. Fahrt zum Louvre-Museum um 16:12 Uhr.
10. Besichtigung des Louvre-Museums bis 17:42 Uhr.
11. Mittagspause bis 18:12 Uhr.
12. Geplante Endzeit: 18:12 Uhr.

Nachdem Sie ein Tool erstellt haben, können Sie es ganz einfach für den Hub freigeben:

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