Strumento codice personalizzato - Hello World

Questo codice di esempio mostra come utilizzare helpputils per testare lo strumento Codice personalizzato.

L'esempio di Hello World è il più semplice strumento di codice personalizzato possibile. Definisce una singola classe di strumenti che accetta un parametro name e restituisce un saluto. Utilizzalo come punto di partenza per il tuo strumento.

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

requisiti.txt

# no deps

Inserire i tre file nella radice di un archivio ZIP e caricare lo ZIP nella scheda Pacchetto. Una volta caricato, passare alla scheda Parametri, compilare la descrizione se si desidera sostituire l'impostazione predefinita e passare alla scheda Test per richiamare lo strumento. Con name="Alice", lo strumento restituisce:

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