Smolagents AI 代理框架

SmolAgents 應用範例

在 Smolagents 中，代理被封裝在允許它們執行特定任務的類別中。例如，一個 ManagedAgent 您可以利用不同的代理類型來管理各種任務。以下是一個簡化的範例，說明如何實作兩個代理程式，一個用於網路搜尋，另一個用於產生圖片，以下是如何製作自訂工具，從 Google 地圖取得旅行時間，以及如何將它用於旅行計畫代理程式：

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

經過收集旅行時間和運算的幾個步驟後，代理返回了這個最終建議：巴黎單車一日遊行程：

1.早上 9:00 從艾菲爾鐵塔開始。
2.在艾菲爾鐵塔觀光至上午 10:30。
3.上午 10:46 前往 Notre-Dame Cathedral。
4.在 Notre-Dame 大教堂觀光至中午 12:16。
5.中午 12:41 前往 Montmartre。
6.在 Montmartre 觀光至下午 2:11。
7.下午 2:33 前往盧森堡花園。
8.在盧森堡花園觀光至下午 4:03。
9.下午 4:12 前往羅浮宮博物館。
10.羅浮宮博物館觀光至下午 5:42。
11.午休至下午 6:12。
12.計劃結束時間：下午 6:12。

建立工具後，將它分享到 Hub 就這麼簡單：

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