Smolagent AI 에이전트 프레임워크

스몰에이전트 적용 사례

스몰에이전트에서 에이전트는 특정 작업을 수행할 수 있는 클래스로 캡슐화됩니다. 예를 들어 관리 에이전트 를 만들어 다양한 에이전트 유형을 활용하여 다양한 작업을 관리할 수 있습니다. 다음은 웹 검색용 에이전트와 이미지 생성용 에이전트 두 가지를 구현하는 간단한 예시이며,다음은 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일 일정이라는 최종 제안을 내립니다:

1. 오전 9시에 에펠탑에서 출발합니다.
2. 오전 10시 30분까지 에펠탑 관광.
3. 오전 10시 46분에 노트르담 대성당으로 이동합니다.
4. 오후 12시 16분까지 노트르담 대성당 관광.
5. 오후 12시 41분에 몽마르트르로 이동합니다.
6. 오후 2시 11분까지 몽마르뜨에서 관광하기.
7. 오후 2시 33분에 자르뎅 뒤 룩셈부르크로 이동합니다.
8. 오후 4시 3분까지 자르뎅 뒤 룩셈부르크에서 관광.
9. 오후 4시 12분에 루브르 박물관으로 이동합니다.
10. 오후 5시 42분까지 루브르 박물관 관광.
11. 오후 6시 12분까지 점심시간.
12. 종료 예정 시간: 오후 6시 12분.

도구를 구축한 후 허브에 공유하는 것은 매우 간단합니다:

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