Outil de code personnalisé - Hello World

Cet exemple de code montre comment vous pouvez utiliser aidputils pour tester votre outil de code personnalisé.

L'exemple Hello World est l'outil de code personnalisé le plus simple possible. Elle définit une classe d'outil unique qui accepte un paramètre de nom et renvoie un message d'accueil. Utilisez-le comme point de départ pour votre propre outil.

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

Fichier_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

Packagez les trois fichiers à la racine d'une archive ZIP et chargez le fichier ZIP via l'onglet Package. Une fois chargé, passez à l'onglet Paramètres, renseignez la description si vous souhaitez remplacer la valeur par défaut, puis passez à l'onglet Test pour appeler l'outil. Avec name="Alice", l'outil renvoie :

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