Smolagents 人工智能代理框架

SmolAgents 应用实例

在 Smolagents 中，代理被封装在允许它们执行特定任务的类中。例如，一个 托管代理 通过使用不同的代理类型，可以创建管理各种任务的代理。下面是一个简化示例，说明如何实现两个代理--一个用于网络搜索，另一个用于生成图片。下面是如何制作一个自定义工具，从谷歌地图中获取旅行时间，以及如何将其用于旅行计划代理：

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 前往圣母大教堂。
4.游览巴黎圣母院大教堂至中午 12:16。
5.中午 12:41 前往蒙马特。
6.在蒙马特观光至下午 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")