export * from "./token";
export * from "./defi";
export * from "./your_protocol"; // Add your protocol here if it's not already in the list

export class SuiAgentKit {
  // ... existing code ...

  async customFunction(input: string): Promise<string> {
    // Implement your custom functionality
    return `Processed input: ${input}`;
  }
}

import { Tool } from "langchain/tools";
import { SuiAgentKit } from "../../agent";

export class CustomTool extends Tool {
 name = "custom_tool";
 description = "Description of what the custom tool does.";

 constructor(private SuiKit: SuiAgentKit) {
   super();
 }

 protected async _call(input: string): Promise<string> {
   try {
     const result = await this.customFunction(input);
     return JSON.stringify({
       status: "success",
       message: "Custom tool executed successfully",
       data: result,
     });
   } catch (error: any) {
     return JSON.stringify({
       status: "error",
       message: error.message,
       code: error.code || "UNKNOWN_ERROR",
     });
   }
 }
}

export * from "./custom_tool";

export * from "./tiplink";
export * from "./your_protocol"; // Add your protocol here if it's not already in the list

import { Action } from "../../types/action";
import { SuiAgentKit } from "../../agent";
import { z } from "zod";
import { custom_tool } from "../../tools";

const customAction: Action = {
 name: "CUSTOM_ACTION",
 similes: ["custom tool"],
 description: "Description of what the custom tool does.",
 examples: [
   {
     input: {},
     output: {
       status: "success",
       message: "Custom tool executed successfully",
       data: result,
     },
     explanation: "Custom tool executed successfully",
   },
 ],
 schema: z.object({
   input: z.string(),
 }),
 handler: async (agent: SuiAgentKit, input: Record<string, any>) => {
   const result = await agent.customFunction(input);
   return result;
 },
};

import customAction from "./your_protocol/custom_action";

export const ACTIONS = {
   // ... existing actions ...
 CUSTOM_ACTION: customAction,
}

import { SuiAgentKit, createSuiTools } from "sui-agent-kit";

// Initialize with private key and optional RPC URL
const agent = new SuiAgentKit(
 "your_sui_private_key",
 "https://fullnode.mainnet.sui.io:443",
 "your-openai-api-key"
);

const tools = createSuiTools(agent);
const customTool = tools.find(tool => tool.name === "custom_tool");

if (customTool) {
 const result = await customTool._call("your-input");
 console.log(result);
}

// or alternatively
const result = await agent.customFunction("your-input"); // assuming you have implemented `customFunction` method in SuiAgentKit
console.log(result);