merge: main

src/actions/agent/createImage.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { create_image } from "../tools/create_image";
+import { create_image } from "../../tools/agent";
 
 const createImageAction: Action = {
   name: "CREATE_IMAGE",

src/actions/agent/getWalletAddress.ts

@@ -1,7 +1,7 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { SolanaAgentKit } from "..";
-import { get_wallet_address } from "../tools";
-import { Action } from "../types/action";
+import { get_wallet_address } from "../../tools/agent";
 
 const getWalletAddressAction: Action = {
   name: "GET_WALLET_ADDRESS",

src/actions/alldomains/getAllDomainsTLDs.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { getAllDomainsTLDs } from "../tools";
+import { getAllDomainsTLDs } from "../../tools";
 
 const getAllDomainsTLDsAction: Action = {
   name: "GET_ALL_TLDS",

src/actions/alldomains/getOwnedAllDomains.ts

@@ -1,8 +1,8 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
-import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
-import { getOwnedAllDomains } from "../tools";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getOwnedAllDomains } from "../../tools";
 
 const getOwnedAllDomainsAction: Action = {
   name: "GET_OWNED_ALL_DOMAINS",

src/actions/alldomains/getOwnedDomainsForTLD.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { getOwnedDomainsForTLD } from "../tools";
+import { getOwnedDomainsForTLD } from "../../tools";
 
 const getOwnedDomainsForTLDAction: Action = {
   name: "GET_OWNED_DOMAINS_FOR_TLD",

src/actions/alldomains/resolveDomain.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { resolveAllDomains } from "../tools";
+import { resolveAllDomains } from "../../tools";
 
 const resolveDomainAction: Action = {
   name: "RESOLVE_ALL_DOMAINS",

src/actions/dexscreener/tokenDataByTicker.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { getTokenDataByTicker } from "../tools";
+import { getTokenDataByTicker } from "../../tools/dexscreener";
 
 const tokenDataByTickerAction: Action = {
   name: "GET_TOKEN_DATA_BY_TICKER",

src/actions/drift/createDriftUserAccount.ts

@@ -0,0 +1,59 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { createDriftUserAccount } from "../../tools";
+
+const createDriftUserAccountAction: Action = {
+  name: "CREATE_DRIFT_USER_ACCOUNT",
+  similes: [
+    "create drift account",
+    "create drift user account",
+    "create user account on drift",
+  ],
+  description: "Create a new user account on Drift protocol",
+  examples: [
+    [
+      {
+        input: {
+          amount: 100,
+          symbol: "SOL",
+        },
+        output: {
+          status: "success",
+          message: "User account created with 100 SOL successfully deposited",
+          account: "4xKpN2...",
+        },
+        explanation: "Create a new user account with 100 SOL",
+      },
+    ],
+  ],
+  schema: z.object({
+    amount: z.number().positive().describe("Amount of the token to deposit"),
+    symbol: z.string().describe("Symbol of the token to deposit"),
+  }),
+  handler: async (agent, input) => {
+    try {
+      const res = await createDriftUserAccount(
+        agent,
+        input.amount,
+        input.symbol,
+      );
+
+      return {
+        status: "success",
+        message:
+          res.message ??
+          `User account created with ${input.amount} ${input.symobl} successfully deposited.`,
+        account: res.account,
+        signature: res.txSignature,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message is a string
+        message: `Failed to create user account: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default createDriftUserAccountAction;

src/actions/drift/createVault.ts

@@ -0,0 +1,108 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import type { SolanaAgentKit } from "../..";
+import { createVault } from "../../tools";
+
+const createDriftVaultAction: Action = {
+  name: "CREATE_DRIFT_VAULT",
+  similes: ["create a drift vault", "open a drift vault", "create vault"],
+  description:
+    "Create a new drift vault delegating the agents address as the owner.",
+  examples: [
+    [
+      {
+        input: {
+          name: "My Drift Vault",
+          marketName: "SOL-SPOT",
+          redeemPeriod: 30,
+          maxTokens: 1000,
+          minDepositAmount: 100,
+          managementFee: 10,
+          profitShare: 5,
+          hurdleRate: 0.1,
+          permissioned: false,
+        },
+        output: {
+          status: "success",
+          message: "Drift vault created successfully",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Create a drift vault",
+      },
+    ],
+  ],
+  schema: z.object({
+    name: z
+      .string()
+      .min(5, "Name must be at least 5 characters")
+      .describe("Has to be unique. 2 Vaults can not have the same name."),
+    // regex matches SOL-SPOT
+    marketName: z
+      .string()
+      .describe('Market name must be in the format "TOKEN-SPOT"'),
+    redeemPeriod: z
+      .number()
+      .int()
+      .min(1, "Redeem period must be at least 1")
+      .describe(
+        "Number of days to wait before funds deposited in a vault can be redeemed ",
+      ),
+    maxTokens: z
+      .number()
+      .int()
+      .min(100, "Max tokens must be at least 100")
+      .describe(
+        "The maximum amount of tokens the vault will be accomodating. For example some vaults have a cap at 10 million USDC",
+      ),
+    minDepositAmount: z.number().positive().describe("Minimum deposit amount"),
+    managementFee: z
+      .number()
+      .positive()
+      .max(20)
+      .describe(
+        "How much of a fee you'll be taking to manage depositors funds. This is in percentage e.g 2 for 2%",
+      ),
+    profitShare: z
+      .number()
+      .positive()
+      .max(90)
+      .optional()
+      .default(5)
+      .describe(
+        "How much of the profit you'll be sharing with depositors. This is in percentage e.g 2 for 2%. Defaults to 5%",
+      ),
+    hurdleRate: z.number().optional(),
+    permissioned: z
+      .boolean()
+      .optional()
+      .describe("Should the vault have a whitelist of not"),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    try {
+      const tx = await createVault(
+        agent,
+        // @ts-expect-error - zod schema validation
+        {
+          ...input,
+        },
+      );
+
+      return {
+        status: "success",
+        message:
+          "Drift vault created successfully. Please note down the name of your vault as it is unique and was used to derive your vault address",
+        vaultName: input.name,
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - e is not a string
+        message: `Failed to create drift vault: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default createDriftVaultAction;

src/actions/drift/depositIntoVault.ts

@@ -0,0 +1,56 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { depositIntoVault } from "../../tools";
+
+const depositIntoDriftVaultAction: Action = {
+  name: "DEPOSIT_INTO_DRIFT_VAULT",
+  description: "Deposit funds into an existing drift vault",
+  similes: ["deposit into drift vault", "add funds to drift vault"],
+  examples: [
+    [
+      {
+        input: {
+          amount: 100,
+          vaultAddress: "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBD",
+        },
+        output: {
+          status: "success",
+          message: "Funds deposited successfully",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Deposit 100 USDC into a drift vault",
+      },
+    ],
+  ],
+  schema: z.object({
+    vaultAddress: z.string(),
+    amount: z
+      .number()
+      .positive()
+      .describe("The amount in tokens you'd like to deposit into the vault"),
+  }),
+  handler: async (agent, input) => {
+    try {
+      const tx = await depositIntoVault(
+        agent,
+        input.amount as number,
+        input.vaultAddress as string,
+      );
+
+      return {
+        status: "success",
+        message: "Funds deposited successfully",
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to deposit funds: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default depositIntoDriftVaultAction;

src/actions/drift/depositToDriftUserAccount.ts

@@ -0,0 +1,73 @@
+import { z } from "zod";
+import type { SolanaAgentKit } from "../../agent";
+import type { Action } from "../../types";
+import { depositToDriftUserAccount } from "../../tools";
+
+const depositToDriftUserAccountAction: Action = {
+  name: "DEPOSIT_TO_DRIFT_USER_ACCOUNT",
+  description: "Deposit funds into your drift user account",
+  similes: [
+    "deposit into drift user account",
+    "add funds to drift user account",
+    "add funds to my drift account",
+    "deposit collateral into drift account",
+  ],
+  examples: [
+    [
+      {
+        input: {
+          amount: 100,
+          symbol: "usdc",
+        },
+        output: {
+          status: "success",
+          message: "Funds deposited successfully",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Deposit 100 USDC into your drift user account",
+      },
+    ],
+  ],
+  schema: z.object({
+    amount: z
+      .number()
+      .positive()
+      .describe(
+        "The amount in tokens you'd like to deposit into your drift user account",
+      ),
+    symbol: z
+      .string()
+      .toUpperCase()
+      .describe("The symbol of the token you'd like to deposit"),
+    repay: z
+      .boolean()
+      .optional()
+      .default(false)
+      .describe("Whether or not to repay the borrowed funds in the account"),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    try {
+      const tx = await depositToDriftUserAccount(
+        agent,
+        input.amount as number,
+        input.symbol as string,
+        input.repay as boolean,
+      );
+
+      return {
+        status: "success",
+        message: "Funds deposited successfully",
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to deposit funds: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default depositToDriftUserAccountAction;

src/actions/drift/deriveVaultAddress.ts

@@ -0,0 +1,46 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { getVaultAddress } from "../../tools";
+
+const deriveDriftVaultAddressAction: Action = {
+  name: "DERIVE_DRIFT_VAULT_ADDRESS_ACTION",
+  similes: ["derive drift vault address", "get drift vault address"],
+  description: "Derive a drift vault address from the vaults name",
+  examples: [
+    [
+      {
+        input: {
+          name: "My Drift Vault",
+        },
+        output: {
+          status: "success",
+          message: "Vault address derived successfully",
+          address: "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBD",
+        },
+        explanation: "Derive a drift vault address",
+      },
+    ],
+  ],
+  schema: z.object({
+    name: z.string().describe("The name of the vault to derive the address of"),
+  }),
+  handler: async (agent, input) => {
+    try {
+      const address = await getVaultAddress(agent, input.name as string);
+
+      return {
+        status: "success",
+        message: "Vault address derived successfully",
+        address,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to derive vault address: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default deriveDriftVaultAddressAction;

src/actions/drift/doesUserHaveDriftAccount.ts

@@ -0,0 +1,53 @@
+import { z } from "zod";
+import { doesUserHaveDriftAccount } from "../../tools";
+import type { Action } from "../../types";
+
+export const doesUserHaveDriftAccountAction: Action = {
+  name: "DOES_USER_HAVE_DRIFT_ACCOUNT",
+  description: "Check if a user has a Drift account",
+  similes: [
+    "check if user has drift account",
+    "check if user has account on drift",
+    "do I have an account on drift",
+  ],
+  examples: [
+    [
+      {
+        input: {},
+        output: {
+          status: "success",
+          message: "Nice! You have a Drift account",
+          account: "4xKpN2...",
+        },
+        explanation: "Check if a user has a Drift account",
+      },
+    ],
+  ],
+  schema: z.object({}),
+  handler: async (agent) => {
+    try {
+      const res = await doesUserHaveDriftAccount(agent);
+
+      if (!res.hasAccount) {
+        return {
+          status: "error",
+          message: "You do not have a Drift account",
+        };
+      }
+
+      return {
+        status: "success",
+        message: "Nice! You have a Drift account",
+        account: res.account,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message is a string
+        message: `Failed to check if you have a Drift account: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default doesUserHaveDriftAccountAction;

src/actions/drift/driftUserAccountInfo.ts

@@ -0,0 +1,39 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { driftUserAccountInfo } from "../../tools";
+
+const driftUserAccountInfoAction: Action = {
+  name: "DRIFT_USER_ACCOUNT_INFO",
+  similes: ["get drift user account info", "get drift account info"],
+  description: "Get information about your drift account",
+  examples: [
+    [
+      {
+        input: {},
+        explanation: "Get information about your drift account",
+        output: {
+          status: "success",
+          data: {},
+        },
+      },
+    ],
+  ],
+  schema: z.object({}),
+  handler: async (agent) => {
+    try {
+      const accountInfo = await driftUserAccountInfo(agent);
+      return {
+        status: "success",
+        data: accountInfo,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message is a string
+        message: `Failed to get drift account info: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default driftUserAccountInfoAction;

src/actions/drift/requestWithdrawalFromVault.ts

@@ -0,0 +1,57 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import type { SolanaAgentKit } from "../../agent";
+import { requestWithdrawalFromVault } from "../../tools";
+
+const requestWithdrawalFromVaultAction: Action = {
+  name: "REQUEST_WITHDRAWAL_FROM_DRIFT_VAULT",
+  description: "Request a withdrawal from an existing drift vault",
+  similes: ["withdraw from drift vault", "request withdrawal from vault"],
+  examples: [
+    [
+      {
+        input: {
+          amount: 100,
+          vaultAddress: "2nFeP7taii",
+        },
+        output: {
+          status: "success",
+          message: "Withdrawal request successful",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Request a withdrawal of 100 USDC from a drift vault",
+      },
+    ],
+  ],
+  schema: z.object({
+    vaultAddress: z.string(),
+    amount: z
+      .number()
+      .positive()
+      .describe("Amount of shares you would like to withdraw from the vault"),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    try {
+      const tx = await requestWithdrawalFromVault(
+        agent,
+        input.amount as number,
+        input.vaultAddress as string,
+      );
+
+      return {
+        status: "success",
+        message: "Withdrawal request successful",
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to request withdrawal: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default requestWithdrawalFromVaultAction;

src/actions/drift/tradeDelegatedDriftVault.ts

@@ -0,0 +1,114 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import type { SolanaAgentKit } from "../../agent";
+import { tradeDriftVault } from "../../tools";
+
+const tradeDelegatedDriftVaultAction: Action = {
+  name: "TRADE_DELEGATED_DRIFT_VAULT",
+  similes: [
+    "trade delegated drift vault",
+    "trade delegated vault",
+    "trade vault",
+    "trade drift vault",
+    "trade delegated vault",
+    "trade vault",
+    "trade drift vault",
+    "open drift vault trade",
+  ],
+  description: "Carry out trades in a Drift vault.",
+  examples: [
+    [
+      {
+        input: {
+          vaultAddress: "J1S9H3QjnRtBbbuD4HjPV6RpRhwuk4zKbxsnCHuTgh9w",
+          amount: 100,
+          symbol: "SOL",
+          action: "buy",
+          type: "market",
+        },
+        output: {
+          status: "success",
+          message: "Trade successful",
+          transactionId: "7nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkN",
+          amount: 100,
+          symbol: "SOL",
+          action: "buy",
+          type: "market",
+        },
+        explanation: "Buy 100 SOL in the vault",
+      },
+    ],
+    [
+      {
+        input: {
+          vaultAddress: "J1S9H3QjnRtBbbuD4HjPV6RpRhwuk4zKbxsnCHuTgh9w",
+          amount: 50,
+          symbol: "SOL",
+          action: "sell",
+          type: "limit",
+          price: 200,
+        },
+        output: {
+          status: "success",
+          message: "Order placed successful",
+          transactionId: "8nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkM",
+          amount: 50,
+          symbol: "SOL",
+          action: "sell",
+          type: "limit",
+          price: 200,
+        },
+        explanation: "Sell 50 SOL in the vault at $200",
+      },
+    ],
+  ],
+  schema: z.object({
+    vaultAddress: z.string().describe("Address of the Drift vault to trade in"),
+    amount: z.number().positive().describe("Amount to trade"),
+    symbol: z.string().describe("Symbol of the token to trade"),
+    action: z.enum(["long", "short"]).describe("Trade action - long or short"),
+    type: z.enum(["market", "limit"]).describe("Trade type - market or limit"),
+    price: z.number().positive().optional().describe("Price for limit order"),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    try {
+      const params = {
+        vaultAddress: input.vaultAddress as string,
+        amount: input.amount as number,
+        symbol: input.symbol as string,
+        action: input.action as "long" | "short",
+        type: input.type as "market" | "limit",
+        price: input.price as number | undefined,
+      };
+
+      // Carry out the trade
+      const transactionId = await tradeDriftVault(
+        agent,
+        params.vaultAddress,
+        params.amount,
+        params.symbol,
+        params.action,
+        params.type,
+        params.price,
+      );
+
+      return {
+        status: "success",
+        message:
+          params.type === "limit"
+            ? "Order placed successfully"
+            : "Trade successful",
+        transactionId,
+        ...params,
+      };
+    } catch (error) {
+      return {
+        status: "error",
+        // @ts-expect-error error is not a string
+        message: error.message,
+      };
+    }
+  },
+};
+
+export default tradeDelegatedDriftVaultAction;

src/actions/drift/tradePerpAccount.ts

@@ -0,0 +1,82 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { driftPerpTrade } from "../../tools";
+
+export const tradeDriftPerpAccountAction: Action = {
+  name: "TRADE_DRIFT_PERP_ACCOUNT",
+  similes: [
+    "trade drift perp account",
+    "trade drift perp",
+    "trade drift perpetual account",
+    "trade perp account",
+    "trade account",
+  ],
+  description: "Trade a perpetual account on Drift protocol",
+  examples: [
+    [
+      {
+        input: {
+          amount: 100,
+          symbol: "SOL",
+          action: "long",
+          type: "market",
+        },
+        output: {
+          status: "success",
+          message: "Trade successful",
+        },
+        explanation: "Open a $100 long position on SOL.",
+      },
+    ],
+    [
+      {
+        input: {
+          amount: 50,
+          symbol: "BTC",
+          action: "short",
+          type: "limit",
+          price: 50000,
+        },
+        output: {
+          status: "success",
+          message: "Trade successful",
+        },
+        explanation: "$50 short position on BTC at $50,000.",
+      },
+    ],
+  ],
+  schema: z.object({
+    amount: z.number().positive(),
+    symbol: z
+      .string()
+      .toUpperCase()
+      .describe("Symbol of the token to open a position on "),
+    action: z.enum(["long", "short"]),
+    type: z.enum(["market", "limit"]),
+    price: z.number().positive().optional(),
+  }),
+  handler: async (agent, input) => {
+    try {
+      const signature = await driftPerpTrade(agent, {
+        action: input.action,
+        amount: input.amount,
+        symbol: input.symbol,
+        type: input.type,
+        price: input.price,
+      });
+
+      return {
+        status: "success",
+        signature: signature,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message is a string
+        message: `Failed to trade perp account: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default tradeDriftPerpAccountAction;

src/actions/drift/updateDriftVaultDelegate.ts

@@ -0,0 +1,53 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { updateVaultDelegate } from "../../tools";
+
+const updateDriftVaultDelegateAction: Action = {
+  name: "UPDATE_DRIFT_VAULT_DELEGATE_ACTION",
+  similes: ["update drift vault delegate", "change drift vault delegate"],
+  description: "Update the delegate of a drift vault",
+  examples: [
+    [
+      {
+        input: {
+          vaultAddress: "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBD",
+          newDelegate: "2nFeP7tai",
+        },
+        output: {
+          status: "success",
+          message: "Vault delegate updated successfully",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Update the delegate of a drift vault to another address",
+      },
+    ],
+  ],
+  schema: z.object({
+    vaultAddress: z.string(),
+    newDelegate: z.string(),
+  }),
+  handler: async (agent, input) => {
+    try {
+      const tx = await updateVaultDelegate(
+        agent,
+        input.vaultAddress as string,
+        input.newDelegate as string,
+      );
+
+      return {
+        status: "success",
+        message: "Vault delegate updated successfully",
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to update vault delegate: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default updateDriftVaultDelegateAction;

src/actions/drift/updateVault.ts

@@ -0,0 +1,87 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import type { SolanaAgentKit } from "../../agent";
+import { updateVault } from "../../tools";
+
+const updateDriftVaultAction: Action = {
+  name: "UPDATE_DRIFT_VAULT",
+  similes: ["update a drift vault", "modify a drift vault", "update vault"],
+  description: "Update an existing drift vault with new settings.",
+  examples: [
+    [
+      {
+        input: {
+          redeemPeriod: 30,
+          maxTokens: 10000,
+          minDepositAmount: 10,
+          managementFee: 5,
+          profitShare: 10,
+          handleRate: 0.1,
+          permissioned: false,
+          vaultAddress: "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBD",
+        },
+        output: {
+          status: "success",
+          message: "Drift vault updated successfully",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Update a drift vault",
+      },
+    ],
+  ],
+  schema: z.object({
+    vaultAddress: z.string(),
+    name: z.string().min(5, "Name must be at least 5 characters").optional(),
+    // regex matches SOL-SPOT
+    marketName: z
+      .string()
+      .regex(/^([A-Za-z0-9]{2,7})-SPOT$/)
+      .optional(),
+    redeemPeriod: z
+      .number()
+      .int()
+      .min(1, "Redeem period must be at least 1")
+      .optional(),
+    maxTokens: z
+      .number()
+      .int()
+      .min(100, "Max tokens must be at least 100")
+      .optional(),
+    minDepositAmount: z.number().positive().optional(),
+    managementFee: z.number().positive().max(20).optional(),
+    profitShare: z.number().positive().max(90).optional(),
+    handleRate: z.number().optional(),
+    permissioned: z
+      .boolean()
+      .optional()
+      .describe("Should the vault have a whitelist of not"),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    try {
+      const tx = await updateVault(agent, input.vaultAddress, {
+        hurdleRate: input.hurdleRate,
+        maxTokens: input.maxTokens,
+        minDepositAmount: input.minDepositAmount,
+        profitShare: input.profitShare,
+        managementFee: input.managementFee,
+        permissioned: input.permissioned,
+        redeemPeriod: input.redeemPeriod,
+      });
+
+      return {
+        status: "success",
+        message: "Drift vault parameters updated successfully",
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to update drift vault: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default updateDriftVaultAction;

src/actions/drift/vaultInfo.ts

@@ -0,0 +1,57 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { getVaultInfo } from "../../tools";
+import type { SolanaAgentKit } from "../../agent";
+
+const vaultInfoAction: Action = {
+  name: "DRIFT_VAULT_INFO",
+  similes: ["get drift vault info", "vault info", "vault information"],
+  description: "Get information about a drift vault",
+  examples: [
+    [
+      {
+        input: {
+          vaultNameOrAddress: "test-vault",
+        },
+        output: {
+          status: "success",
+          message: "Vault info retrieved successfully",
+          data: {
+            name: "My Drift Vault",
+            marketName: "SOL-SPOT",
+            redeemPeriod: 30,
+            maxTokens: 1000,
+            minDepositAmount: 100,
+            managementFee: 10,
+            profitShare: 5,
+            hurdleRate: 0.1,
+            permissioned: false,
+          },
+        },
+        explanation: "Get information about a drift vault",
+      },
+    ],
+  ],
+  schema: z.object({
+    vaultNameOrAddress: z.string().describe("Name or address of the vault"),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    try {
+      const vaultInfo = await getVaultInfo(agent, input.vaultNameOrAddress);
+
+      return {
+        status: "success",
+        message: "Vault info retrieved successfully",
+        data: vaultInfo,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to retrieve vault info: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default vaultInfoAction;

src/actions/drift/withdrawFromDriftAccount.ts

@@ -0,0 +1,77 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import { withdrawFromDriftUserAccount } from "../../tools";
+
+const withdrawFromDriftAccountAction: Action = {
+  name: "WITHDRAW_OR_BORROW_FROM_DRIFT_ACCOUNT",
+  description: "Withdraw funds from your drift account",
+  similes: [
+    "withdraw from drift account",
+    "withdraw funds from drift account",
+    "withdraw funds from my drift account",
+    "borrow from drift account",
+    "borrow funds from my drift account",
+    "borrow from drift",
+    "withdraw from drift",
+  ],
+  examples: [
+    [
+      {
+        input: {
+          amount: 100,
+          symbol: "usdc",
+        },
+        output: {
+          status: "success",
+          message: "Funds withdrawn successfully",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Withdraw 100 USDC from your drift account",
+      },
+    ],
+  ],
+  schema: z.object({
+    amount: z
+      .number()
+      .positive()
+      .describe(
+        "The amount in tokens you'd like to withdraw from your drift account",
+      ),
+    symbol: z
+      .string()
+      .toUpperCase()
+      .describe("The symbol of the token you'd like to withdraw"),
+    isBorrow: z
+      .boolean()
+      .optional()
+      .default(false)
+      .describe(
+        "Whether or not to borrow funds based on collateral provided instead of withdrawing",
+      ),
+  }),
+  handler: async (agent, input) => {
+    try {
+      const tx = await withdrawFromDriftUserAccount(
+        agent,
+        input.amount,
+        input.symbol,
+        input.isBorrow,
+      );
+
+      return {
+        status: "success",
+        message: "Funds withdrawn successfully",
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message is a string
+        message: `Failed to withdraw funds: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default withdrawFromDriftAccountAction;

src/actions/drift/withdrawFromVault.ts

@@ -0,0 +1,52 @@
+import { z } from "zod";
+import type { Action } from "../../types";
+import type { SolanaAgentKit } from "../../agent";
+import { withdrawFromDriftVault } from "../../tools";
+
+const withdrawFromVaultAction: Action = {
+  name: "WITHDRAW_FROM_DRIFT_VAULT",
+  description:
+    "Withdraw funds from a vault given the redemption time has elapsed.",
+  similes: ["withdraw from drift vault", "redeem funds from vault"],
+  examples: [
+    [
+      {
+        input: {
+          vaultAddress: "2nFeP7taii",
+        },
+        output: {
+          status: "success",
+          message: "Withdrawal successful",
+          signature:
+            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
+        },
+        explanation: "Withdraw funds from a drift vault",
+      },
+    ],
+  ],
+  schema: z.object({
+    vaultAddress: z.string(),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    try {
+      const tx = await withdrawFromDriftVault(
+        agent,
+        input.vaultAddress as string,
+      );
+
+      return {
+        status: "success",
+        message: "Withdrawal successful",
+        signature: tx,
+      };
+    } catch (e) {
+      return {
+        status: "error",
+        // @ts-expect-error - error message
+        message: `Failed to withdraw funds: ${e.message}`,
+      };
+    }
+  },
+};
+
+export default withdrawFromVaultAction;

src/actions/flash/flashCloseTrade.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { flashCloseTrade } from "../tools";
+import { flashCloseTrade } from "../../tools/flash";
 
 const flashCloseTradeAction: Action = {
   name: "FLASH_CLOSE_TRADE",

src/actions/flash/flashOpenTrade.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { flashOpenTrade } from "../tools";
+import { flashOpenTrade } from "../../tools/flash";
 
 const flashOpenTradeAction: Action = {
   name: "FLASH_OPEN_TRADE",

src/actions/gibwork/createGibworkTask.ts

@@ -1,8 +1,8 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
-import { create_gibwork_task } from "../tools";
+import { create_gibwork_task } from "../../tools/gibwork";
 
 const createGibworkTaskAction: Action = {
   name: "CREATE_GIBWORK_TASK",

src/actions/helius/createWebhook.ts

@@ -0,0 +1,57 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { create_HeliusWebhook } from "../../tools/helius";
+
+const createWebhookAction: Action = {
+  name: "CREATE_HELIOUS_WEBHOOK",
+  similes: ["setup webhook", "register webhook", "initiate webhook"],
+  description:
+    "Creates a new webhook in the Helius system to monitor transactions for specified account addresses",
+  examples: [
+    [
+      {
+        input: {
+          accountAddresses: [
+            "BVdNLvyG2DNiWAXBE9qAmc4MTQXymd5Bzfo9xrQSUzVP",
+            "Eo2ciguhMLmcTWXELuEQPdu7DWZt67LHXb2rdHZUbot7",
+          ],
+          webhookURL: "https://yourdomain.com/webhook",
+        },
+        output: {
+          status: "success",
+          webhookURL: "https://yourdomain.com/webhook",
+          webhookID: "webhook_123",
+          message: "Webhook created successfully.",
+        },
+        explanation:
+          "Creates a Webhook to send live notifications on the given Url with the wallet Addresses.",
+      },
+    ],
+  ],
+  schema: z.object({
+    accountAddresses: z
+      .array(z.string())
+      .min(1)
+      .describe("List of Solana account public keys to monitor"),
+    webhookURL: z
+      .string()
+      .url()
+      .describe("The URL where Helius will send webhook notifications"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const response = await create_HeliusWebhook(
+      agent,
+      input.accountAddresses,
+      input.webhookURL,
+    );
+
+    return {
+      status: "success",
+      ...response,
+      message: "Webhook created successfully.",
+    };
+  },
+};
+
+export default createWebhookAction;

src/actions/helius/deleteWebhook.ts

@@ -0,0 +1,40 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { deleteHeliusWebhook } from "../../tools/helius";
+
+const deleteWebhookAction: Action = {
+  name: "DELETE_HELIOUS_WEBHOOK",
+  similes: ["remove webhook", "unregister webhook", "delete webhook"],
+  description: "Deletes a Helius webhook by its unique ID",
+  examples: [
+    [
+      {
+        input: {
+          webhookID: "webhook_123",
+        },
+        output: {
+          status: "success",
+          message: "Webhook deleted successfully.",
+        },
+        explanation: "Permanently removes a Helius webhook.",
+      },
+    ],
+  ],
+  schema: z.object({
+    webhookID: z
+      .string()
+      .min(1)
+      .describe("The unique identifier of the Helius webhook to delete"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const result = await deleteHeliusWebhook(agent, input.webhookID);
+
+    return {
+      status: "success",
+      message: result.message || "Webhook deleted successfully.",
+    };
+  },
+};
+
+export default deleteWebhookAction;

src/actions/helius/getAssetsbyOwner.ts

@@ -0,0 +1,75 @@
+import { Action } from "../../types/action";
+import { PublicKey } from "@solana/web3.js";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getAssetsByOwner } from "../../tools/helius";
+
+const getAssetsByOwnerAction: Action = {
+  name: "FETCH_ASSETS_BY_OWNER",
+  similes: [
+    "fetch assets",
+    "get assets",
+    "retrieve assets",
+    "list assets",
+    "assets by owner",
+  ],
+  description:
+    "Fetch assets owned by a specific Solana wallet address using the Helius Digital Asset Standard API",
+  examples: [
+    [
+      {
+        input: {
+          ownerPublicKey: "4Pf8q3mHGLdkoc1M8xWZwW5q32gYmdhwC2gJ8K9EAGDX",
+          limit: 10,
+        },
+        output: {
+          status: "success",
+          assets: [
+            {
+              name: "Helius NFT #1",
+              type: "NFT",
+              owner: "4Pf8q3mHGLdkoc1M8xWZwW5q32gYmdhwC2gJ8K9EAGDX",
+            },
+            {
+              name: "Helius Token #10",
+              type: "Token",
+              owner: "4Pf8q3mHGLdkoc1M8xWZwW5q32gYmdhwC2gJ8K9EAGDX",
+            },
+          ],
+          message: "Successfully fetched assets for the wallet address",
+        },
+        explanation:
+          "Fetches a list of assets from the for the given wallet address with a limit of 10 items.",
+      },
+    ],
+  ],
+  schema: z.object({
+    ownerPublicKey: z.string().describe("Owner's Solana wallet PublicKey"),
+    limit: z
+      .number()
+      .positive()
+      .describe("Number of assets to retrieve per request"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    try {
+      const assets = await getAssetsByOwner(
+        agent,
+        new PublicKey(input.ownerPublicKey),
+        input.limit,
+      );
+
+      return {
+        status: "success",
+        assets: assets,
+        message: `Successfully fetched assets for the wallet address: ${input.ownerPublicKey}`,
+      };
+    } catch (error: any) {
+      return {
+        status: "error",
+        message: `Failed to fetch assets: ${error.message}`,
+      };
+    }
+  },
+};
+
+export default getAssetsByOwnerAction;

src/actions/helius/getWebhook.ts

@@ -0,0 +1,47 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getHeliusWebhook } from "../../tools/helius";
+
+const getWebhookAction: Action = {
+  name: "GET_HELIOUS_WEBHOOK",
+  similes: ["fetch webhook details", "retrieve webhook", "get webhook info"],
+  description: "Retrieves details of a Helius webhook by its unique ID",
+  examples: [
+    [
+      {
+        input: {
+          webhookID: "webhook_123",
+        },
+        output: {
+          status: "success",
+          wallet: "WalletPublicKey",
+          webhookURL: "https://yourdomain.com/webhook",
+          transactionTypes: ["Any"],
+          accountAddresses: ["SomePublicKey", "AnotherPublicKey"],
+          webhookType: "enhanced",
+          message: "Webhook details retrieved successfully.",
+        },
+        explanation:
+          "Retrieves detailed information about an existing Helius webhook, including the wallet address it monitors, the types of transactions it tracks, and the specific webhook URL.",
+      },
+    ],
+  ],
+  schema: z.object({
+    webhookID: z
+      .string()
+      .min(1)
+      .describe("The unique identifier of the Helius webhook to retrieve"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const webhookDetails = await getHeliusWebhook(agent, input.webhookID);
+
+    return {
+      status: "success",
+      ...webhookDetails,
+      message: "Webhook details retrieved successfully.",
+    };
+  },
+};
+
+export default getWebhookAction;

src/actions/helius/parseTransaction.ts

@@ -0,0 +1,65 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { parseTransaction } from "../../tools/helius";
+
+const parseSolanaTransactionAction: Action = {
+  name: "PARSE_SOLANA_TRANSACTION",
+  similes: [
+    "parse transaction",
+    "analyze transaction",
+    "inspect transaction",
+    "decode transaction",
+  ],
+  description:
+    "Parse a Solana transaction to retrieve detailed information using the Helius Enhanced Transactions API",
+  examples: [
+    [
+      {
+        input: {
+          transactionId:
+            "4zZVvbgzcriyjAeEiK1w7CeDCt7gYThUCZat3ULTNerzKHF4WLfRG2YUjbRovfFJ639TAyARB4oyRDcLVUvrakq7",
+        },
+        output: {
+          status: "success",
+          transaction: {
+            details: "Transaction details...",
+            involvedAccounts: ["Account1", "Account2"],
+            executedOperations: [{ operation: "Transfer", amount: "1000 SOL" }],
+          },
+          message:
+            "Successfully parsed transaction: 4zZVvbgzcriyjAeEiK1w7CeDCt7gYThUCZat3ULTNerzKHF4WLfRG2YUjbRovfFJ639TAyARB4oyRDcLVUvrakq7",
+        },
+        explanation:
+          "Parse a Transaction to transform it into human readable format.",
+      },
+    ],
+  ],
+  schema: z.object({
+    transactionId: z
+      .string()
+      .min(1)
+      .describe("The Solana transaction ID to parse"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    try {
+      const parsedTransactionData = await parseTransaction(
+        agent,
+        input.transactionId,
+      );
+
+      return {
+        status: "success",
+        transaction: parsedTransactionData,
+        message: `Successfully parsed transaction: ${input.transactionId}`,
+      };
+    } catch (error: any) {
+      return {
+        status: "error",
+        message: `Failed to parse transaction: ${error.message}`,
+      };
+    }
+  },
+};
+
+export default parseSolanaTransactionAction;

src/actions/helius/sendTransactionWithPriority.ts

@@ -0,0 +1,76 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { sendTransactionWithPriorityFee } from "../../tools/helius";
+import { PublicKey } from "@solana/web3.js";
+
+const sendTransactionWithPriorityFeeAction: Action = {
+  name: "SEND_TRANSACTION_WITH_PRIORITY_FEE",
+  similes: [
+    "send SOL with fee",
+    "transfer tokens with priority",
+    "execute priority transaction",
+  ],
+  description:
+    "Sends SOL or SPL tokens from a wallet with an estimated priority fee, ensuring faster processing on the Solana network.",
+  examples: [
+    [
+      {
+        input: {
+          priorityLevel: "High",
+          amount: 2,
+          to: "BVdNLvyG2DNiWAXBE9qAmc4MTQXymd5Bzfo9xrQSUzVP",
+          splmintAddress: "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v",
+        },
+        output: {
+          status: "success",
+          transactionId: "5Xgq9xVABhwXpNStWpfqxS6Vm5Eau91pjfeHNwJbRgis",
+          fee: 5000,
+          message: "Transaction sent with priority fee successfully.",
+        },
+        explanation:
+          "Sends 2 USDC to BVdNLvyG2DNiWAXBE9qAmc4MTQXymd5Bzfo9xrQSUzVP with High priority fee option.",
+      },
+    ],
+  ],
+  schema: z.object({
+    priorityLevel: z
+      .enum(["Min", "Low", "Medium", "High", "VeryHigh", "UnsafeMax"])
+      .describe("Priority level to determine the urgency of the transaction."),
+    amount: z
+      .number()
+      .positive()
+      .describe("Amount of SOL or SPL tokens to send."),
+    to: z.string().describe("Recipient's PublicKey."),
+    splmintAddress: z
+      .string()
+      .optional()
+      .describe(
+        "Optional SPL token address, if transferring tokens other than SOL.",
+      ),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const { priorityLevel, amount, to, splmintAddress } = input;
+    const toPublicKey = new PublicKey(to);
+    const splmintPublicKey = splmintAddress
+      ? new PublicKey(splmintAddress)
+      : undefined;
+
+    const result = await sendTransactionWithPriorityFee(
+      agent,
+      priorityLevel,
+      amount,
+      toPublicKey,
+      splmintPublicKey,
+    );
+
+    return {
+      status: "success",
+      transactionId: result.transactionId,
+      fee: result.fee,
+      message: "Transaction sent with priority fee successfully.",
+    };
+  },
+};
+
+export default sendTransactionWithPriorityFeeAction;

src/actions/index.ts

@@ -1,36 +1,64 @@
-import deployTokenAction from "./deployToken";
-import balanceAction from "./balance";
-import transferAction from "./transfer";
-import deployCollectionAction from "./deployCollection";
-import mintNFTAction from "./mintNFT";
-import tradeAction from "./trade";
-import requestFundsAction from "./requestFunds";
-import resolveDomainAction from "./resolveDomain";
-import getTokenDataAction from "./getTokenData";
-import getTPSAction from "./getTPS";
-import fetchPriceAction from "./fetchPrice";
-import stakeWithJupAction from "./stakeWithJup";
-import stakeWithSolayerAction from "./stakeWithSolayer";
-import registerDomainAction from "./registerDomain";
-import lendAssetAction from "./lendAsset";
-import createGibworkTaskAction from "./createGibworkTask";
-import resolveSolDomainAction from "./resolveSolDomain";
-import pythFetchPriceAction from "./pythFetchPrice";
-import getOwnedDomainsForTLDAction from "./getOwnedDomainsForTLD";
-import getPrimaryDomainAction from "./getPrimaryDomain";
-import getAllDomainsTLDsAction from "./getAllDomainsTLDs";
-import getOwnedAllDomainsAction from "./getOwnedAllDomains";
-import createImageAction from "./createImage";
-import getMainAllDomainsDomainAction from "./getMainAllDomainsDomain";
-import getAllRegisteredAllDomainsAction from "./getAllRegisteredAllDomains";
-import raydiumCreateCpmmAction from "./raydiumCreateCpmm";
-import raydiumCreateAmmV4Action from "./raydiumCreateAmmV4";
-import createOrcaSingleSidedWhirlpoolAction from "./createOrcaSingleSidedWhirlpool";
-import launchPumpfunTokenAction from "./launchPumpfunToken";
-import getWalletAddressAction from "./getWalletAddress";
-import flashOpenTradeAction from "./flashOpenTrade";
-import flashCloseTradeAction from "./flashCloseTrade";
 import tokenBalancesAction from "./tokenBalances";
+import deployTokenAction from "./metaplex/deployToken";
+import balanceAction from "./solana/balance";
+import transferAction from "./solana/transfer";
+import deployCollectionAction from "./metaplex/deployCollection";
+import mintNFTAction from "./metaplex/mintNFT";
+import tradeAction from "./jupiter/trade";
+import requestFundsAction from "./solana/requestFunds";
+import resolveDomainAction from "./sns/registerDomain";
+import getTokenDataAction from "./jupiter/getTokenData";
+import getTPSAction from "./solana/getTPS";
+import fetchPriceAction from "./jupiter/fetchPrice";
+import stakeWithJupAction from "./jupiter/stakeWithJup";
+import stakeWithSolayerAction from "./solayer/stakeWithSolayer";
+import registerDomainAction from "./sns/registerDomain";
+import lendAssetAction from "./lulo/lendAsset";
+import createGibworkTaskAction from "./gibwork/createGibworkTask";
+import resolveSolDomainAction from "./sns/resolveSolDomain";
+import pythFetchPriceAction from "./pyth/pythFetchPrice";
+import getOwnedDomainsForTLDAction from "./alldomains/getOwnedDomainsForTLD";
+import getPrimaryDomainAction from "./sns/getPrimaryDomain";
+import getAllDomainsTLDsAction from "./alldomains/getAllDomainsTLDs";
+import getOwnedAllDomainsAction from "./alldomains/getOwnedAllDomains";
+import createImageAction from "./agent/createImage";
+import getMainAllDomainsDomainAction from "./sns/getMainAllDomainsDomain";
+import getAllRegisteredAllDomainsAction from "./sns/getAllRegisteredAllDomains";
+import raydiumCreateCpmmAction from "./raydium/raydiumCreateCpmm";
+import raydiumCreateAmmV4Action from "./raydium/raydiumCreateAmmV4";
+import createOrcaSingleSidedWhirlpoolAction from "./orca/createOrcaSingleSidedWhirlpool";
+import launchPumpfunTokenAction from "./pumpfun/launchPumpfunToken";
+import getWalletAddressAction from "./agent/getWalletAddress";
+import flashOpenTradeAction from "./flash/flashOpenTrade";
+import flashCloseTradeAction from "./flash/flashCloseTrade";
+import createMultisigAction from "./squads/createMultisig";
+import approveMultisigProposalAction from "./squads/approveMultisigProposal";
+import createMultisigProposalAction from "./squads/createMultisigProposal";
+import depositToMultisigAction from "./squads/depositToMultisigTreasury";
+import executeMultisigProposalAction from "./squads/executeMultisigProposal";
+import rejectMultisigProposalAction from "./squads/rejectMultisigProposal";
+import transferFromMultisigAction from "./squads/transferFromMultisigTreasury";
+import createWebhookAction from "./helius/createWebhook";
+import deleteWebhookAction from "./helius/deleteWebhook";
+import getAssetsByOwnerAction from "./helius/getAssetsbyOwner";
+import getWebhookAction from "./helius/getWebhook";
+import parseSolanaTransactionAction from "./helius/parseTransaction";
+import sendTransactionWithPriorityFeeAction from "./helius/sendTransactionWithPriority";
+import createDriftVaultAction from "./drift/createVault";
+import updateDriftVaultAction from "./drift/updateVault";
+import depositIntoDriftVaultAction from "./drift/depositIntoVault";
+import requestWithdrawalFromVaultAction from "./drift/requestWithdrawalFromVault";
+import withdrawFromVaultAction from "./drift/withdrawFromVault";
+import tradeDelegatedDriftVaultAction from "./drift/tradeDelegatedDriftVault";
+import vaultInfoAction from "./drift/vaultInfo";
+import createDriftUserAccountAction from "./drift/createDriftUserAccount";
+import tradeDriftPerpAccountAction from "./drift/tradePerpAccount";
+import doesUserHaveDriftAccountAction from "./drift/doesUserHaveDriftAccount";
+import depositToDriftUserAccountAction from "./drift/depositToDriftUserAccount";
+import withdrawFromDriftAccountAction from "./drift/withdrawFromDriftAccount";
+import driftUserAccountInfoAction from "./drift/driftUserAccountInfo";
+import deriveDriftVaultAddressAction from "./drift/deriveVaultAddress";
+import updateDriftVaultDelegateAction from "./drift/updateDriftVaultDelegate";
 
 export const ACTIONS = {
   WALLET_ADDRESS_ACTION: getWalletAddressAction,
@@ -67,6 +95,34 @@ export const ACTIONS = {
   LAUNCH_PUMPFUN_TOKEN_ACTION: launchPumpfunTokenAction,
   FLASH_OPEN_TRADE_ACTION: flashOpenTradeAction,
   FLASH_CLOSE_TRADE_ACTION: flashCloseTradeAction,
+  CREATE_MULTISIG_ACTION: createMultisigAction,
+  DEPOSIT_TO_MULTISIG_ACTION: depositToMultisigAction,
+  TRANSFER_FROM_MULTISIG_ACTION: transferFromMultisigAction,
+  CREATE_MULTISIG_PROPOSAL_ACTION: createMultisigProposalAction,
+  APPROVE_MULTISIG_PROPOSAL_ACTION: approveMultisigProposalAction,
+  REJECT_MULTISIG_PROPOSAL_ACTION: rejectMultisigProposalAction,
+  EXECUTE_MULTISIG_PROPOSAL_ACTION: executeMultisigProposalAction,
+  CREATE_WEBHOOK_ACTION: createWebhookAction,
+  DELETE_WEBHOOK_ACTION: deleteWebhookAction,
+  GET_ASSETS_BY_OWNER_ACTION: getAssetsByOwnerAction,
+  GET_WEBHOOK_ACTION: getWebhookAction,
+  PARSE_TRANSACTION_ACTION: parseSolanaTransactionAction,
+  SEND_TRANSACTION_WITH_PRIORITY_ACTION: sendTransactionWithPriorityFeeAction,
+  CREATE_DRIFT_VAULT_ACTION: createDriftVaultAction,
+  UPDATE_DRIFT_VAULT_ACTION: updateDriftVaultAction,
+  DEPOSIT_INTO_DRIFT_VAULT_ACTION: depositIntoDriftVaultAction,
+  REQUEST_WITHDRAWAL_FROM_DRIFT_VAULT_ACTION: requestWithdrawalFromVaultAction,
+  WITHDRAW_FROM_DRIFT_VAULT_ACTION: withdrawFromVaultAction,
+  TRADE_DELEGATED_DRIFT_VAULT_ACTION: tradeDelegatedDriftVaultAction,
+  DRIFT_VAULT_INFO_ACTION: vaultInfoAction,
+  CREATE_DRIFT_USER_ACCOUNT_ACTION: createDriftUserAccountAction,
+  TRADE_DRIFT_PERP_ACCOUNT_ACTION: tradeDriftPerpAccountAction,
+  DOES_USER_HAVE_DRIFT_ACCOUNT_ACTION: doesUserHaveDriftAccountAction,
+  DEPOSIT_TO_DRIFT_USER_ACCOUNT_ACTION: depositToDriftUserAccountAction,
+  WITHDRAW_OR_BORROW_FROM_DRIFT_ACCOUNT_ACTION: withdrawFromDriftAccountAction,
+  DRIFT_USER_ACCOUNT_INFO_ACTION: driftUserAccountInfoAction,
+  DERIVE_DRIFT_VAULT_ADDRESS_ACTION: deriveDriftVaultAddressAction,
+  UPDATE_DRIFT_VAULT_DELEGATE_ACTION: updateDriftVaultDelegateAction,
 };
 
 export type { Action, ActionExample, Handler } from "../types/action";

src/actions/jupiter/fetchPrice.ts

@@ -1,8 +1,8 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
-import { fetchPrice } from "../tools";
+import { fetchPrice } from "../../tools/jupiter";
 
 const fetchPriceAction: Action = {
   name: "FETCH_PRICE",

src/actions/jupiter/getTokenData.ts

@@ -1,9 +1,9 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
-import { JupiterTokenData } from "../types";
-import { getTokenAddressFromTicker, getTokenDataByAddress } from "../tools";
+import { JupiterTokenData } from "../../types";
+import { getTokenAddressFromTicker, getTokenDataByAddress } from "../../tools";
 
 const getTokenDataAction: Action = {
   name: "GET_TOKEN_DATA",

src/actions/jupiter/stakeWithJup.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { stakeWithJup } from "../tools";
+import { stakeWithJup } from "../../tools";
 
 const stakeWithJupAction: Action = {
   name: "STAKE_WITH_JUPITER",

src/actions/jupiter/trade.ts

@@ -1,8 +1,8 @@
 import { PublicKey } from "@solana/web3.js";
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { trade } from "../tools";
+import { trade } from "../../tools";
 
 const tradeAction: Action = {
   name: "TRADE",

src/actions/lightprotocol/compressedAirdrop.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { sendCompressedAirdrop } from "../tools";
+import { sendCompressedAirdrop } from "../../tools";
 
 const compressedAirdropAction: Action = {
   name: "COMPRESSED_AIRDROP",

src/actions/lulo/lendAsset.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { lendAsset } from "../tools";
+import { lendAsset } from "../../tools/lulo";
 
 const lendAssetAction: Action = {
   name: "LEND_ASSET",

src/actions/metaplex/deployCollection.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { deploy_collection } from "../tools";
+import { deploy_collection } from "../../tools/metaplex";
 
 interface CollectionOptions {
   name: string;

src/actions/metaplex/deployToken.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { deploy_token } from "../tools";
+import { deploy_token } from "../../tools";
 
 const deployTokenAction: Action = {
   name: "DEPLOY_TOKEN",

src/actions/metaplex/mintNFT.ts

@@ -1,8 +1,8 @@
 import { PublicKey } from "@solana/web3.js";
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { mintCollectionNFT } from "../tools";
+import { mintCollectionNFT } from "../../tools/metaplex";
 
 const mintNFTAction: Action = {
   name: "MINT_NFT",
@@ -13,7 +13,7 @@ const mintNFTAction: Action = {
     "create token",
     "add nft to collection",
   ],
-  description: `Mint a new NFT in a collection on Solana blockchain.`,
+  description: "Mint a new NFT in a collection on Solana blockchain.",
   examples: [
     [
       {

src/actions/openbook/createOpenbookMarket.ts

@@ -1,8 +1,8 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
-import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
-import { openbookCreateMarket } from "../tools";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { openbookCreateMarket } from "../../tools/openbook";
 
 const createOpenbookMarketAction: Action = {
   name: "CREATE_OPENBOOK_MARKET",

src/actions/orca/createOrcaSingleSidedWhirlpool.ts

@@ -1,9 +1,9 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
 import { Decimal } from "decimal.js";
-import { orcaCreateSingleSidedLiquidityPool } from "../tools";
+import { orcaCreateSingleSidedLiquidityPool } from "../../tools";
 
 // Fee tiers mapping from the original tool
 const FEE_TIERS = {

src/actions/pumpfun/launchPumpfunToken.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { launchPumpFunToken } from "../tools";
+import { launchPumpFunToken } from "../../tools";
 
 const launchPumpfunTokenAction: Action = {
   name: "LAUNCH_PUMPFUN_TOKEN",

src/actions/pyth/pythFetchPrice.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { fetchPythPrice, fetchPythPriceFeedID } from "../tools";
+import { fetchPythPrice, fetchPythPriceFeedID } from "../../tools";
 
 const pythFetchPriceAction: Action = {
   name: "PYTH_FETCH_PRICE",

src/actions/raydium/raydiumCreateAmmV4.ts

@@ -1,9 +1,9 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
 import BN from "bn.js";
-import { raydiumCreateAmmV4 } from "../tools";
+import { raydiumCreateAmmV4 } from "../../tools";
 
 const raydiumCreateAmmV4Action: Action = {
   name: "RAYDIUM_CREATE_AMM_V4",

src/actions/raydium/raydiumCreateClmm.ts

@@ -1,10 +1,10 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
 import { BN } from "@coral-xyz/anchor";
 import Decimal from "decimal.js";
-import { raydiumCreateClmm } from "../tools";
+import { raydiumCreateClmm } from "../../tools";
 
 const raydiumCreateClmmAction: Action = {
   name: "RAYDIUM_CREATE_CLMM",

src/actions/raydium/raydiumCreateCpmm.ts

@@ -1,9 +1,9 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
 import BN from "bn.js";
-import { raydiumCreateCpmm } from "../tools";
+import { raydiumCreateCpmm } from "../../tools";
 
 const raydiumCreateCpmmAction: Action = {
   name: "RAYDIUM_CREATE_CPMM",

src/actions/sns/getAllRegisteredAllDomains.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { getAllRegisteredAllDomains } from "../tools";
+import { getAllRegisteredAllDomains } from "../../tools";
 
 const getAllRegisteredAllDomainsAction: Action = {
   name: "GET_ALL_REGISTERED_ALL_DOMAINS",

src/actions/sns/getMainAllDomainsDomain.ts

@@ -1,8 +1,8 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
-import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
-import { getMainAllDomainsDomain } from "../tools";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getMainAllDomainsDomain } from "../../tools";
 
 const getMainAllDomainsDomainAction: Action = {
   name: "GET_MAIN_ALL_DOMAINS_DOMAIN",

src/actions/sns/getPrimaryDomain.ts

@@ -1,8 +1,8 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
-import { z } from "zod";
 import { PublicKey } from "@solana/web3.js";
-import { getPrimaryDomain } from "../tools";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getPrimaryDomain } from "../../tools";
 
 const getPrimaryDomainAction: Action = {
   name: "GET_PRIMARY_DOMAIN",

src/actions/sns/registerDomain.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { registerDomain } from "../tools";
+import { registerDomain } from "../../tools";
 
 const registerDomainAction: Action = {
   name: "REGISTER_DOMAIN",

src/actions/sns/resolveSolDomain.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { resolveSolDomain } from "../tools";
+import { resolveSolDomain } from "../../tools/";
 
 const resolveSolDomainAction: Action = {
   name: "RESOLVE_SOL_DOMAIN",

src/actions/solana/balance.ts

@@ -1,8 +1,8 @@
 import { PublicKey } from "@solana/web3.js";
-import type { Action } from "../types/action";
-import type { SolanaAgentKit } from "../agent";
+import type { Action } from "../../types/action";
+import type { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { get_balance } from "../tools";
+import { get_balance } from "../../tools";
 
 const balanceAction: Action = {
   name: "BALANCE_ACTION",

src/actions/solana/closeEmptyTokenAccounts.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { closeEmptyTokenAccounts } from "../tools";
+import { closeEmptyTokenAccounts } from "../../tools";
 
 const closeEmptyTokenAccountsAction: Action = {
   name: "CLOSE_EMPTY_TOKEN_ACCOUNTS",

src/actions/solana/getTPS.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { getTPS } from "../tools";
+import { getTPS } from "../../tools/solana";
 
 const getTPSAction: Action = {
   name: "GET_TPS",

src/actions/solana/requestFunds.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { request_faucet_funds } from "../tools";
+import { request_faucet_funds } from "../../tools/solana";
 
 const requestFundsAction: Action = {
   name: "REQUEST_FUNDS",

src/actions/solana/transfer.ts

@@ -1,8 +1,8 @@
 import { PublicKey } from "@solana/web3.js";
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { transfer } from "../tools";
+import { transfer } from "../../tools";
 
 const transferAction: Action = {
   name: "TRANSFER",

src/actions/solayer/stakeWithSolayer.ts

@@ -1,7 +1,7 @@
-import { Action } from "../types/action";
-import { SolanaAgentKit } from "../agent";
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
-import { stakeWithSolayer } from "../tools";
+import { stakeWithSolayer } from "../../tools";
 
 const stakeWithSolayerAction: Action = {
   name: "STAKE_WITH_SOLAYER",

src/actions/squads/approveMultisigProposal.ts

@@ -0,0 +1,50 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { multisig_approve_proposal } from "../../tools";
+
+const approveMultisigProposalAction: Action = {
+  name: "APPROVE_MULTISIG_PROPOSAL_ACTION",
+  similes: [
+    "approve proposal",
+    "approve proposal to transfer funds",
+    "approve proposal to transfer funds from 2-of-2 multisig",
+    "approve proposal to transfer funds from 2-of-2 multisig account",
+    "approve proposal to transfer funds from 2-of-2 multisig account on Solana",
+  ],
+  description: `Approve a proposal to transfer funds from a 2-of-2 multisig account on Solana with the user and the agent, where both approvals will be required to run the transactions.
+  
+  Note: For one AI agent, only one 2-by-2 multisig can be created as it is pair-wise.`,
+  examples: [
+    [
+      {
+        input: {
+          transactionIndex: 0,
+        },
+        output: {
+          status: "success",
+          message: "Proposal approved successfully",
+          transaction: "4xKpN2...",
+          transactionIndex: "0",
+        },
+        explanation:
+          "Approve a proposal to transfer 1 SOL from 2-of-2 multisig account on Solana",
+      },
+    ],
+  ],
+  schema: z.object({
+    transactionIndex: z.number().optional(),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const tx = await multisig_approve_proposal(agent, input.transactionIndex);
+
+    return {
+      status: "success",
+      message: "Proposal approved successfully",
+      transaction: tx,
+      transactionIndex: input.transactionIndex.toString(),
+    };
+  },
+};
+
+export default approveMultisigProposalAction;

src/actions/squads/createMultisig.ts

@@ -0,0 +1,52 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { create_squads_multisig } from "../../tools";
+import { PublicKey } from "@solana/web3.js";
+
+const createMultisigAction: Action = {
+  name: "CREATE_MULTISIG_ACTION",
+  similes: [
+    "create multisig",
+    "create squads multisig",
+    "create 2-by-2 multisig",
+    "create 2-of-2 multisig",
+    "create 2-of-2 multisig account",
+    "create 2-of-2 multisig account on Solana",
+  ],
+  description: `Create a 2-of-2 multisig account on Solana using Squads with the user and the agent, where both approvals will be required to run the transactions.
+  
+  Note: For one AI agent, only one 2-by-2 multisig can be created as it is pair-wise.`,
+  examples: [
+    [
+      {
+        input: {
+          creator: "7nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkN",
+        },
+        output: {
+          status: "success",
+          message: "2-by-2 multisig account created successfully",
+          signature: "4xKpN2...",
+        },
+        explanation: "Create a 2-of-2 multisig account on Solana",
+      },
+    ],
+  ],
+  schema: z.object({
+    creator: z.string(),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const multisig = await create_squads_multisig(
+      agent,
+      new PublicKey(input.creator as string),
+    );
+
+    return {
+      status: "success",
+      message: "2-by-2 multisig account created successfully",
+      signature: multisig,
+    };
+  },
+};
+
+export default createMultisigAction;

src/actions/squads/createMultisigProposal.ts

@@ -0,0 +1,55 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { multisig_create_proposal } from "../../tools";
+
+const createMultisigProposalAction: Action = {
+  name: "CREATE_MULTISIG_PROPOSAL_ACTION",
+  similes: [
+    "create proposal",
+    "create proposal to transfer funds",
+    "create proposal to transfer funds from 2-of-2 multisig",
+    "create proposal to transfer funds from 2-of-2 multisig account",
+    "create proposal to transfer funds from 2-of-2 multisig account on Solana",
+  ],
+  description: `Create a proposal to transfer funds from a 2-of-2 multisig account on Solana with the user and the agent, where both approvals will be required to run the transactions.
+  
+  If transactionIndex is not provided, the latest index will automatically be fetched and used.`,
+  examples: [
+    [
+      {
+        input: {
+          transactionIndex: 0,
+        },
+        output: {
+          status: "success",
+          message: "Proposal created successfully",
+          transaction: "4xKpN2...",
+          transactionIndex: "0",
+        },
+        explanation:
+          "Create a proposal to transfer 1 SOL from 2-of-2 multisig account on Solana",
+      },
+    ],
+  ],
+  schema: z.object({
+    transactionIndex: z.number().optional(),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const transactionIndex =
+      input.transactionIndex !== undefined
+        ? Number(input.transactionIndex)
+        : undefined;
+
+    const multisig = await multisig_create_proposal(agent, transactionIndex);
+
+    return {
+      status: "success",
+      message: "Proposal created successfully",
+      transaction: multisig,
+      transactionIndex: transactionIndex,
+    };
+  },
+};
+
+export default createMultisigProposalAction;

src/actions/squads/depositToMultisigTreasury.ts

@@ -0,0 +1,49 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { multisig_deposit_to_treasury } from "../../tools";
+
+const depositToMultisigAction: Action = {
+  name: "DEPOSIT_TO_MULTISIG_ACTION",
+  similes: [
+    "deposit to multisig",
+    "deposit to squads multisig",
+    "deposit to 2-of-2 multisig account",
+    "deposit to 2-of-2 multisig account on Solana",
+    "deposit SOL to 2-of-2 multisig",
+    "deposit SPL tokens to 2-of-2 multisig",
+  ],
+  description: `Deposit funds to a 2-of-2 multisig account on Solana with the user and the agent, where both approvals will be required to run the transactions.`,
+  examples: [
+    [
+      {
+        input: {
+          amount: 1,
+        },
+        output: {
+          status: "success",
+          message: "Funds deposited to 2-by-2 multisig account successfully",
+          signature: "4xKpN2...",
+        },
+        explanation: "Deposit 1 SOL to 2-of-2 multisig account on Solana",
+      },
+    ],
+  ],
+  schema: z.object({
+    amount: z.number().min(0, "Amount must be greater than 0"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const multisig = await multisig_deposit_to_treasury(
+      agent,
+      input.amount as number,
+    );
+
+    return {
+      status: "success",
+      message: "Funds deposited to 2-by-2 multisig account successfully",
+      signature: multisig,
+    };
+  },
+};
+
+export default depositToMultisigAction;

src/actions/squads/executeMultisigProposal.ts

@@ -0,0 +1,53 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { multisig_execute_proposal } from "../../tools";
+
+const executeMultisigProposalAction: Action = {
+  name: "EXECUTE_MULTISIG_PROPOSAL_ACTION",
+  similes: [
+    "execute proposal",
+    "execute proposal to transfer funds",
+    "execute proposal to transfer funds from 2-of-2 multisig",
+    "execute proposal to transfer funds from 2-of-2 multisig account",
+    "execute proposal to transfer funds from 2-of-2 multisig account on Solana",
+  ],
+  description: `Execute a proposal to transfer funds from a 2-of-2 multisig account on Solana with the user and the agent, where both approvals will be required to run the transactions.`,
+  examples: [
+    [
+      {
+        input: {
+          proposalIndex: 0,
+        },
+        output: {
+          status: "success",
+          message: "Proposal executed successfully",
+          transaction: "4xKpN2...",
+          proposalIndex: "0",
+        },
+        explanation:
+          "Execute a proposal to transfer 1 SOL from 2-of-2 multisig account on Solana",
+      },
+    ],
+  ],
+  schema: z.object({
+    proposalIndex: z.number().optional(),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const proposalIndex =
+      input.proposalIndex !== undefined
+        ? Number(input.proposalIndex)
+        : undefined;
+
+    const multisig = await multisig_execute_proposal(agent, proposalIndex);
+
+    return {
+      status: "success",
+      message: "Proposal executed successfully",
+      transaction: multisig,
+      proposalIndex,
+    };
+  },
+};
+
+export default executeMultisigProposalAction;

src/actions/squads/rejectMultisigProposal.ts

@@ -0,0 +1,53 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { multisig_reject_proposal } from "../../tools";
+
+const rejectMultisigProposalAction: Action = {
+  name: "REJECT_MULTISIG_PROPOSAL_ACTION",
+  similes: [
+    "reject proposal",
+    "reject proposal to transfer funds",
+    "reject proposal to transfer funds from 2-of-2 multisig",
+    "reject proposal to transfer funds from 2-of-2 multisig account",
+    "reject proposal to transfer funds from 2-of-2 multisig account on Solana",
+  ],
+  description: `Reject a proposal to transfer funds from a 2-of-2 multisig account on Solana with the user and the agent, where both approvals will be required to run the transactions.`,
+  examples: [
+    [
+      {
+        input: {
+          proposalIndex: 0,
+        },
+        output: {
+          status: "success",
+          message: "Proposal rejected successfully",
+          transaction: "4xKpN2...",
+          proposalIndex: "0",
+        },
+        explanation:
+          "Reject a proposal to transfer 1 SOL from 2-of-2 multisig account on Solana",
+      },
+    ],
+  ],
+  schema: z.object({
+    proposalIndex: z.number().optional(),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const proposalIndex =
+      input.proposalIndex !== undefined
+        ? Number(input.proposalIndex)
+        : undefined;
+
+    const tx = await multisig_reject_proposal(agent, proposalIndex);
+
+    return {
+      status: "success",
+      message: "Proposal rejected successfully",
+      transaction: tx,
+      proposalIndex,
+    };
+  },
+};
+
+export default rejectMultisigProposalAction;

src/actions/squads/transferFromMultisigTreasury.ts

@@ -0,0 +1,56 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { multisig_transfer_from_treasury } from "../../tools";
+import { PublicKey } from "@solana/web3.js";
+
+const transferFromMultisigAction: Action = {
+  name: "TRANSFER_FROM_MULTISIG_ACTION",
+  similes: [
+    "transfer from multisig",
+    "transfer from squads multisig",
+    "transfer SOL from 2-by-2 multisig",
+    "transfer from 2-of-2 multisig account",
+    "transfer from 2-of-2 multisig account on Solana",
+  ],
+  description: `Create a transaction to transfer funds from a 2-of-2 multisig account on Solana using Squads with the user and the agent, where both approvals will be required to run the transactions.`,
+  examples: [
+    [
+      {
+        input: {
+          amount: 1,
+          recipient: "7nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkN",
+        },
+        output: {
+          status: "success",
+          message: "Transaction added to 2-by-2 multisig account successfully",
+          transaction: "4xKpN2...",
+          amount: "1",
+          recipient: "7nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkN",
+        },
+        explanation: "Transfer 1 SOL from 2-of-2 multisig account on Solana",
+      },
+    ],
+  ],
+  schema: z.object({
+    amount: z.number().min(0, "Amount must be greater than 0"),
+    recipient: z.string(),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    const multisig = await multisig_transfer_from_treasury(
+      agent,
+      input.amount as number,
+      new PublicKey(input.recipient as string),
+    );
+
+    return {
+      status: "success",
+      message: "Transaction added to 2-by-2 multisig account successfully",
+      transaction: multisig,
+      amount: input.amount,
+      recipient: input.recipient,
+    };
+  },
+};
+
+export default transferFromMultisigAction;

src/actions/tokenBalances.ts

@@ -2,7 +2,7 @@ import { PublicKey } from "@solana/web3.js";
 import type { Action } from "../types/action";
 import type { SolanaAgentKit } from "../agent";
 import { z } from "zod";
-import { get_token_balance } from "../tools/get_token_balances";
+import { get_token_balance } from "../tools";
 
 const tokenBalancesAction: Action = {
   name: "TOKEN_BALANCE_ACTION",
@@ -64,7 +64,7 @@ const tokenBalancesAction: Action = {
   schema: z.object({
     walletAddress: z.string().optional(),
   }),
-  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+  handler: async (agent: SolanaAgentKit, input) => {
     const balance = await get_token_balance(
       agent,
       input.tokenAddress && new PublicKey(input.tokenAddress),