Herramienta de código personalizado - Hello World

Este código de ejemplo muestra cómo puede utilizar las ayudas para probar la herramienta de código personalizado.

El ejemplo de Hello World es la herramienta de código personalizado más sencilla posible. Define una sola clase de herramienta que acepta un parámetro name y devuelve un saludo. Úsalo como punto de partida para tu propia herramienta.

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

requirements.txt

# no deps

Empaquete los tres archivos en la raíz de un archivo ZIP y cargue el ZIP mediante el separador Paquete. Una vez cargado, cambie al separador Parámetros, rellene la descripción si desea sustituir el valor por defecto y cambie al separador Prueba para llamar a la herramienta. Con name="Alice", la herramienta devuelve:

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