📔Creating Toolkits

Toolkits allow you to make tools available to all UnifAI agents as a service. This section shows you how to create, configure, and run your toolkits using both the JavaScript/TypeScript and Python SDK

Overview

A toolkit is a collection of tools that you register and serve to agents dynamically. It requires a Toolkit API Key — which you can get for free from UnifAI Console.

Toolkit Creation and Configuration

Let's break down the process of creating and configuring a toolkit:

Initialize the Toolkit

JavaScript/TypeScript

import { Toolkit } from 'unifai-sdk';

const toolkit = new Toolkit({ apiKey: 'YOUR_TOOLKIT_API_KEY' });

Python

Rust

Update Toolkit Details

You can optionally update the toolkit's name and description:

JavaScript/TypeScript

await toolkit.updateToolkit({ 
    name: "EchoChamber", 
    description: "What's in, what's out." 
});

Python

Rust

Register Action Handlers

Register actions that your toolkit will provide. The payloadDescription can be any string or dictionary that contains enough information for agents to understand the payload format. It acts like API documentation that agents can read to determine what parameters to use.

Note that payloadDescription doesn't have to be in a certain format, as long as agents can understand it as natural language and generate the correct payload. Think of it as the comments and docs for your API, agents read it and decide what parameters to use. In practice, using JSON schema is recommended to match the format of training data.

JavaScript/TypeScript

toolkit.action(
    {
        action: "echo",
        actionDescription: "Echo the message",
        payloadDescription: {
            content: {
              type: "string",
              description: "The message to echo"
            }
        }
    },
    async (ctx, payload) => {
        return ctx.result(`You are agent <${ctx.agentId}>, you said "${payload?.content}".`);
    }
);

Python

Rust

Start the Toolkit

Finally, start serving your toolkit:

JavaScript/TypeScript

await toolkit.run();

Python

Rust

Now your toolkit can be discovered and used by all agents connected to UnifAI.

Complete Example

Here's the complete code putting it all together:

JavaScript/TypeScript

import { Toolkit } from 'unifai-sdk';

async function main() {
  const toolkit = new Toolkit({ apiKey: 'YOUR_TOOLKIT_API_KEY' });

  // Optionally update the toolkit details
  await toolkit.updateToolkit({ 
    name: "EchoChamber", 
    description: "What's in, what's out." 
  });

  // Register an action handler
  toolkit.action(
    {
      action: "echo",
      actionDescription: "Echo the message",
      payloadDescription: {
        content: { type: "string" }
      }
    },
    async (ctx, payload) => {
      return ctx.result(`You are agent <${ctx.agentId}>, you said "${payload?.content}".`);
    }
  );

  // Start serving the toolkit
  await toolkit.run();
}

main().catch(console.error);

Python

Rust

Toolkit Examples

We open source our official toolkits at unifai-network/unifai-toolkits.