Smolagent란?
AI 에이전트를 쉽고 효율적으로 개발할 수 있는 프레임워크이다. Smolagent는 작은 (small) 에이전트를 만드는 데 초점을 맞추고 있으며 복잡한 기능을 간결하게 구현할 수 있도록 설계되었다.
Smolagent로 AI 에이전트 만들어보기
(1) 템플릿 복제
허깅페이스에서 제공하는 템플릿을 복제한다. First_agent_template 링크로 이동하여, "Duplicate this Space" 버튼을 클릭하여 자신의 복사본을 만든다.
(2) API 토큰 설정
Hugging Face 토큰 설정 페이지에서 토큰을 생성한다. 토큰에 추론(inference) 권한이 있어야 한다.
복제한 Space로 이동하여 "Settings" 탭을 클릭한다. "Variables and Secrets" 섹션으로 스크롤하여 "New Secret" 버튼을 클릭, 이름으로 HF_TOKEN을 입력하고, 값에 생성한 토큰을 붙여 넣는다. "Save" 버튼을 클릭하여 토큰을 저장한다.
(3) app.py 파일 수정하기
이제 복제한 Space에서 app.py 파일을 수정한다.
기존에는 도구에 final_answer만 있어서 기능이 제한적이다. 더 유용한 에이전트를 만들기 위해 도구를 추가해보자. 에이전트에 도구를 추가하려면 tools 목록에 추가하면 된다.
위와 같이 tools 를 추가하면, 현재시간 확인, 텍스트 프롬프트로부터 이미지 생성, 웹 검색 수행을 할 수 있다.
(4) 에이전트 테스트하기
이제 에이전트를 다시 실행하고, 에이전트를 테스트할 수 있다. 예시로 아래 첨부한 사진과 같이 사용할 수 있다.
시간 질문에는 get_current_time_in_timezone 도구를, 이미지 생성 요청에는 image_generation_tool을 사용하게 될 것이다.
이렇게 직접 도구들을 만들고 추가해서 더 다양한 일을 수행하는 AI 에이전트를 만들 수 있다.
[Reference]
