Benutzerdefiniertes Code-Tool - Hello World

Dieser Beispielcode zeigt, wie Sie mit Hilfe von Hilfsmitteln Ihr Custom Code-Tool testen können.

Das Hello World-Beispiel ist das einfachste Custom Code-Tool. Sie definiert eine einzelne Toolklasse, die einen Namensparameter akzeptiert und eine Begrüßung zurückgibt. Verwenden Sie es als Ausgangspunkt für Ihr eigenes Tool.

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

Verpacken Sie die drei Dateien in der Root eines ZIP-Archivs, und laden Sie die ZIP-Datei über die Registerkarte Package hoch. Wechseln Sie nach dem Hochladen zur Registerkarte Parameter, geben Sie die Beschreibung ein, wenn Sie den Standardwert außer Kraft setzen möchten, und wechseln Sie zur Registerkarte Test, um das Tool aufzurufen. Bei name="Alice" gibt das Tool Folgendes zurück:

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