ToolFactory: LLM 기반 REST API 문서 자동화 툴 생성 파이프라인

브랜다이스 대학교 연구팀이 개발한 ToolFactory는 LLM을 기반으로 REST API 문서에서 AI 호환 툴을 자동 생성하는 오픈소스 파이프라인입니다. API 문서의 문제점을 해결하고, 오류 진단 및 검증된 툴 지식베이스를 활용하여 신뢰성을 높였으며, 당화학 연구 AI 에이전트 구축 사례를 통해 효과를 입증했습니다. 과학적 REST API와 AI 워크플로우의 원활한 통합에 기여할 것으로 기대됩니다.

브랜다이스 대학교 연구진의 혁신적인 ToolFactory 소개

최근 브랜다이스 대학교의 Xinyi Ni, Qiuyang Wang, Yukun Zhang, Pengyu Hong 연구팀이 LLM(Large Language Model)을 활용하여 REST API 문서로부터 AI 호환 툴을 자동으로 생성하는 혁신적인 오픈소스 파이프라인, ToolFactory를 개발했습니다. 이는 AI 에이전트 개발의 효율성을 극대화하고 사용자의 학습 곡선을 완화하는 데 크게 기여할 것으로 기대됩니다.

ToolFactory의 핵심 기능 및 강점

REST API는 AI 에이전트 구축에 유용한 자원이지만, AI 호환 툴로 변환하는 과정이 필요합니다. 기존 API 문서는 표준화 부족, 일관되지 않은 스키마, 불완전한 정보 등의 문제점을 가지고 있었습니다. ToolFactory는 이러한 문제를 해결하기 위해 다음과 같은 핵심 기능을 제공합니다.

• 자동화된 툴 생성: 비정형 API 문서를 분석하여 AI 호환 툴을 자동으로 생성합니다.
• 오류 진단 평가 방법: 생성된 툴의 신뢰성을 높이기 위해 오류를 진단하고 평가하는 방법을 구현했습니다.
• 검증된 툴의 지식베이스 활용: 불완전한 API 문서에서 누락된 정보를 추론하기 위해 검증된 툴의 지식베이스를 활용합니다.

실험 결과 및 활용 사례

연구팀은 167개의 API 문서와 744개의 엔드포인트를 포함하는 API 추출 벤치마크를 개발하고, JSON 스키마를 사용하여 주석을 달아 ToolFactory의 훈련과 검증에 활용했습니다. 실험 결과는 ToolFactory의 효과를 명확하게 보여주었습니다. 실제로 연구팀은 ToolFactory를 이용하여 당화학 연구를 위한 도메인 특정 AI 에이전트를 성공적으로 구축했습니다.

미래 전망

ToolFactory는 과학적 REST API를 AI 워크플로우에 원활하게 통합하는 데 중요한 역할을 할 것으로 예상됩니다. 이는 다양한 분야에서 AI 기술의 활용성을 높이고, AI 에이전트 개발의 문턱을 낮추는 데 크게 기여할 것으로 기대됩니다. 오픈소스로 공개된 ToolFactory는 앞으로 더 많은 연구자와 개발자들의 참여를 통해 지속적으로 발전하고 개선될 것으로 예상됩니다. 앞으로 ToolFactory가 AI 기술 발전에 어떤 영향을 미칠지 기대해 볼 만 합니다.