사용자 정의 코드 도구 - Hello World

이 샘플 코드는 Aidputils를 사용하여 사용자 정의 코드 도구를 테스트하는 방법을 보여줍니다.

Hello World 예제는 가능한 가장 간단한 사용자 정의 코드 도구입니다. 이름 매개변수를 받아들이고 인사말을 반환하는 단일 도구 클래스를 정의합니다. 자신의 도구의 시작점으로 사용합니다.

도구_구현.py

from aidputils.agents.tools.custom_tools.base import CustomToolBase
 

@BaseTool.register
 class HelloTool(CustomToolBase):
     """A simple greeting tool."""
 
    @classmethod
     def _execute_tool(cls, conf, runtime_params, **context_vars):
         name = runtime_params.get("name", "World")
         return {"greeting": f"Hello, {name}!"}

도구_config.json

{
   "displayName": "Hello Tool",
   "description": "A simple hello world tool",
   "tools": [
     {
       "toolClassName": "HelloTool",
       "displayName": "Hello Tool",
       "description": "Returns a hello world greeting",
       "version": "1.0.0",
       "schema": [
         {
           "name": "name",
           "type": "string",
           "description": "Name to greet"
         }
       ],
       "conf": {}
     }
   ]
 }

요구 사항.txt

# no deps

ZIP 아카이브의 루트에 세 개의 파일을 패키지화하고 패키지 탭을 통해 ZIP을 업로드합니다. 업로드한 후 매개변수 탭으로 전환하고, 기본값을 무효화하려면 설명을 채우고, 도구를 호출하려면 테스트 탭으로 전환합니다. name="Alice"를 사용할 경우 도구는 다음을 반환합니다.

{"greeting": "Hello, Alice!"}