Ferramenta de código personalizado - Hello World

Este exemplo de código demonstra como você pode usar aidputils para testar sua ferramenta Custom Code.

O exemplo do Hello World é a ferramenta de Código Personalizado mais simples possível. Define uma única classe de ferramenta que aceita um parâmetro de nome e retorna uma saudação. Use-o como ponto de partida para sua própria ferramenta.

tool_implementation.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}!"}

tool_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": {}
     }
   ]
 }

requerimentos.txt

# no deps

Empacote os três arquivos na raiz de um arquivo ZIP e faça upload do ZIP pela guia Pacote. Após o upload, alterne para a guia Parâmetros, preencha a Descrição se quiser substituir o padrão e alterne para a guia Testar para chamar a ferramenta. Com name="Alice", a ferramenta retorna:

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