Merge branch 'main' of github.com:3-land/solana-agent-kit into 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,38 +1,68 @@
-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,
+  TOKEN_BALANCES_ACTION: tokenBalancesAction,
   DEPLOY_TOKEN_ACTION: deployTokenAction,
   BALANCE_ACTION: balanceAction,
   TRANSFER_ACTION: transferAction,
@@ -65,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 { Action } from "../types/action";
-import { 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

@@ -0,0 +1,71 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { closeEmptyTokenAccounts } from "../../tools";
+
+const closeEmptyTokenAccountsAction: Action = {
+  name: "CLOSE_EMPTY_TOKEN_ACCOUNTS",
+  similes: [
+    "close token accounts",
+    "remove empty accounts",
+    "clean up token accounts",
+    "close SPL token accounts",
+    "clean wallet",
+  ],
+  description: `Close empty SPL Token accounts associated with your wallet to reclaim rent. 
+ This action will close both regular SPL Token accounts and Token-2022 accounts that have zero balance. `,
+  examples: [
+    [
+      {
+        input: {},
+        output: {
+          status: "success",
+          signature:
+            "3KmPyiZvJQk8CfBVVaz8nf3c2crb6iqjQVDqNxknnusyb1FTFpXqD8zVSCBAd1X3rUcD8WiG1bdSjFbeHsmcYGXY",
+          accountsClosed: 10,
+        },
+        explanation: "Closed 10 empty token accounts successfully.",
+      },
+    ],
+    [
+      {
+        input: {},
+        output: {
+          status: "success",
+          signature: "",
+          accountsClosed: 0,
+        },
+        explanation: "No empty token accounts were found to close.",
+      },
+    ],
+  ],
+  schema: z.object({}),
+  handler: async (agent: SolanaAgentKit) => {
+    try {
+      const result = await closeEmptyTokenAccounts(agent);
+
+      if (result.size === 0) {
+        return {
+          status: "success",
+          signature: "",
+          accountsClosed: 0,
+          message: "No empty token accounts found to close",
+        };
+      }
+
+      return {
+        status: "success",
+        signature: result.signature,
+        accountsClosed: result.size,
+        message: `Successfully closed ${result.size} empty token accounts`,
+      };
+    } catch (error: any) {
+      return {
+        status: "error",
+        message: `Failed to close empty token accounts: ${error.message}`,
+      };
+    }
+  },
+};
+
+export default closeEmptyTokenAccountsAction;

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

@@ -0,0 +1,80 @@
+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";
+
+const tokenBalancesAction: Action = {
+  name: "TOKEN_BALANCE_ACTION",
+  similes: [
+    "check token balances",
+    "get wallet token balances",
+    "view token balances",
+    "show token balances",
+    "check token balance",
+  ],
+  description: `Get the token balances of a Solana wallet.
+  If you want to get the balance of your wallet, you don't need to provide the wallet address.`,
+  examples: [
+    [
+      {
+        input: {},
+        output: {
+          status: "success",
+          balance: {
+            sol: 100,
+            tokens: [
+              {
+                tokenAddress: "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v",
+                name: "USD Coin",
+                symbol: "USDC",
+                balance: 100,
+                decimals: 9,
+              },
+            ],
+          },
+        },
+        explanation: "Get token balances of the wallet",
+      },
+    ],
+    [
+      {
+        input: {
+          walletAddress: "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v",
+        },
+        output: {
+          status: "success",
+          balance: {
+            sol: 100,
+            tokens: [
+              {
+                tokenAddress: "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v",
+                name: "USD Coin",
+                symbol: "USDC",
+                balance: 100,
+                decimals: 9,
+              },
+            ],
+          },
+        },
+        explanation: "Get address token balance",
+      },
+    ],
+  ],
+  schema: z.object({
+    walletAddress: z.string().optional(),
+  }),
+  handler: async (agent: SolanaAgentKit, input) => {
+    const balance = await get_token_balance(
+      agent,
+      input.tokenAddress && new PublicKey(input.tokenAddress),
+    );
+
+    return {
+      status: "success",
+      balance: balance,
+    };
+  },
+};
+
+export default tokenBalancesAction;

src/agent/index.ts

@@ -2,8 +2,12 @@ import { Connection, Keypair, PublicKey } from "@solana/web3.js";
 import { BN } from "@coral-xyz/anchor";
 import bs58 from "bs58";
 import Decimal from "decimal.js";
+import {
+  CreateCollectionOptions,
+  CreateSingleOptions,
+  StoreInitOptions,
+} from "@3land/listings-sdk/dist/types/implementation/implementationTypes";
 import { DEFAULT_OPTIONS } from "../constants";
-import { Config, TokenCheck } from "../types";
 import {
   deploy_collection,
   deploy_token,
@@ -56,14 +60,48 @@ import {
   create_TipLink,
   listNFTForSale,
   cancelListing,
+  closeEmptyTokenAccounts,
   fetchTokenReportSummary,
   fetchTokenDetailedReport,
   fetchPythPrice,
   fetchPythPriceFeedID,
   flashOpenTrade,
   flashCloseTrade,
+  createCollection,
+  createSingle,
+  multisig_transfer_from_treasury,
+  create_squads_multisig,
+  multisig_create_proposal,
+  multisig_deposit_to_treasury,
+  multisig_reject_proposal,
+  multisig_approve_proposal,
+  multisig_execute_proposal,
+  parseTransaction,
+  sendTransactionWithPriorityFee,
+  getAssetsByOwner,
+  getHeliusWebhook,
+  create_HeliusWebhook,
+  deleteHeliusWebhook,
+  createDriftUserAccount,
+  createVault,
+  depositIntoVault,
+  depositToDriftUserAccount,
+  getVaultAddress,
+  doesUserHaveDriftAccount,
+  driftUserAccountInfo,
+  requestWithdrawalFromVault,
+  tradeDriftVault,
+  driftPerpTrade,
+  updateVault,
+  getVaultInfo,
+  withdrawFromDriftUserAccount,
+  withdrawFromDriftVault,
+  updateVaultDelegate,
+  get_token_balance,
 } from "../tools";
 import {
+  Config,
+  TokenCheck,
   CollectionDeployment,
   CollectionOptions,
   GibworkCreateTaskReponse,
@@ -74,16 +112,9 @@ import {
   OrderParams,
   FlashTradeParams,
   FlashCloseTradeParams,
+  HeliusWebhookIdResponse,
+  HeliusWebhookResponse,
 } from "../types";
-import {
-  createCollection,
-  createSingle,
-} from "../tools/create_3land_collectible";
-import {
-  CreateCollectionOptions,
-  CreateSingleOptions,
-  StoreInitOptions,
-} from "@3land/listings-sdk/dist/types/implementation/implementationTypes";
 
 /**
  * Main class for interacting with Solana blockchain
@@ -159,6 +190,19 @@ export class SolanaAgentKit {
     return get_balance(this, token_address);
   }
 
+  async getTokenBalances(wallet_address?: PublicKey): Promise<{
+    sol: number;
+    tokens: Array<{
+      tokenAddress: string;
+      name: string;
+      symbol: string;
+      balance: number;
+      decimals: number;
+    }>;
+  }> {
+    return get_token_balance(this, wallet_address);
+  }
+
   async getBalanceOther(
     walletAddress: PublicKey,
     tokenAddress?: PublicKey,
@@ -547,6 +591,13 @@ export class SolanaAgentKit {
     return cancelListing(this, nftMint);
   }
 
+  async closeEmptyTokenAccounts(): Promise<{
+    signature: string;
+    size: number;
+  }> {
+    return closeEmptyTokenAccounts(this);
+  }
+
   async fetchTokenReportSummary(mint: string): Promise<TokenCheck> {
     return fetchTokenReportSummary(mint);
   }
@@ -572,6 +623,12 @@ export class SolanaAgentKit {
   async flashCloseTrade(params: FlashCloseTradeParams): Promise<string> {
     return flashCloseTrade(this, params);
   }
+  async heliusParseTransactions(transactionId: string): Promise<any> {
+    return parseTransaction(this, transactionId);
+  }
+  async getAllAssetsbyOwner(owner: PublicKey, limit: number): Promise<any> {
+    return getAssetsByOwner(this, owner, limit);
+  }
 
   async create3LandCollection(
     collectionOpts: CreateCollectionOptions,
@@ -612,4 +669,173 @@ export class SolanaAgentKit {
     );
     return `Transaction: ${tx}`;
   }
+  async sendTranctionWithPriority(
+    priorityLevel: string,
+    amount: number,
+    to: PublicKey,
+    splmintAddress?: PublicKey,
+  ): Promise<{ transactionId: string; fee: number }> {
+    return sendTransactionWithPriorityFee(
+      this,
+      priorityLevel,
+      amount,
+      to,
+      splmintAddress,
+    );
+  }
+
+  async createSquadsMultisig(creator: PublicKey): Promise<string> {
+    return create_squads_multisig(this, creator);
+  }
+
+  async depositToMultisig(
+    amount: number,
+    vaultIndex: number = 0,
+    mint?: PublicKey,
+  ): Promise<string> {
+    return multisig_deposit_to_treasury(this, amount, vaultIndex, mint);
+  }
+
+  async transferFromMultisig(
+    amount: number,
+    to: PublicKey,
+    vaultIndex: number = 0,
+    mint?: PublicKey,
+  ): Promise<string> {
+    return multisig_transfer_from_treasury(this, amount, to, vaultIndex, mint);
+  }
+
+  async createMultisigProposal(
+    transactionIndex?: number | bigint,
+  ): Promise<string> {
+    return multisig_create_proposal(this, transactionIndex);
+  }
+
+  async approveMultisigProposal(
+    transactionIndex?: number | bigint,
+  ): Promise<string> {
+    return multisig_approve_proposal(this, transactionIndex);
+  }
+
+  async rejectMultisigProposal(
+    transactionIndex?: number | bigint,
+  ): Promise<string> {
+    return multisig_reject_proposal(this, transactionIndex);
+  }
+
+  async executeMultisigTransaction(
+    transactionIndex?: number | bigint,
+  ): Promise<string> {
+    return multisig_execute_proposal(this, transactionIndex);
+  }
+  async CreateWebhook(
+    accountAddresses: string[],
+    webhookURL: string,
+  ): Promise<HeliusWebhookResponse> {
+    return create_HeliusWebhook(this, accountAddresses, webhookURL);
+  }
+  async getWebhook(id: string): Promise<HeliusWebhookIdResponse> {
+    return getHeliusWebhook(this, id);
+  }
+  async deleteWebhook(webhookID: string): Promise<any> {
+    return deleteHeliusWebhook(this, webhookID);
+  }
+
+  async createDriftUserAccount(depositAmount: number, depositSymbol: string) {
+    return await createDriftUserAccount(this, depositAmount, depositSymbol);
+  }
+  async createDriftVault(params: {
+    name: string;
+    marketName: `${string}-${string}`;
+    redeemPeriod: number;
+    maxTokens: number;
+    minDepositAmount: number;
+    managementFee: number;
+    profitShare: number;
+    hurdleRate?: number;
+    permissioned?: boolean;
+  }) {
+    return await createVault(this, params);
+  }
+  async depositIntoDriftVault(amount: number, vault: string) {
+    return await depositIntoVault(this, amount, vault);
+  }
+  async depositToDriftUserAccount(
+    amount: number,
+    symbol: string,
+    isRepayment?: boolean,
+  ) {
+    return await depositToDriftUserAccount(this, amount, symbol, isRepayment);
+  }
+  async deriveDriftVaultAddress(name: string) {
+    return await getVaultAddress(this, name);
+  }
+  async doesUserHaveDriftAccount() {
+    return await doesUserHaveDriftAccount(this);
+  }
+  async driftUserAccountInfo() {
+    return await driftUserAccountInfo(this);
+  }
+  async requestWithdrawalFromDriftVault(amount: number, vault: string) {
+    return await requestWithdrawalFromVault(this, amount, vault);
+  }
+  async tradeUsingDelegatedDriftVault(
+    vault: string,
+    amount: number,
+    symbol: string,
+    action: "long" | "short",
+    type: "market" | "limit",
+    price?: number,
+  ) {
+    return await tradeDriftVault(
+      this,
+      vault,
+      amount,
+      symbol,
+      action,
+      type,
+      price,
+    );
+  }
+  async tradeUsingDriftPerpAccount(
+    amount: number,
+    symbol: string,
+    action: "long" | "short",
+    type: "market" | "limit",
+    price?: number,
+  ) {
+    return await driftPerpTrade(this, { action, amount, symbol, type, price });
+  }
+  async updateDriftVault(
+    vaultAddress: string,
+    params: {
+      name: string;
+      marketName: `${string}-${string}`;
+      redeemPeriod: number;
+      maxTokens: number;
+      minDepositAmount: number;
+      managementFee: number;
+      profitShare: number;
+      hurdleRate?: number;
+      permissioned?: boolean;
+    },
+  ) {
+    return await updateVault(this, vaultAddress, params);
+  }
+  async getDriftVaultInfo(vaultName: string) {
+    return await getVaultInfo(this, vaultName);
+  }
+  async withdrawFromDriftAccount(
+    amount: number,
+    symbol: string,
+    isBorrow?: boolean,
+  ) {
+    return await withdrawFromDriftUserAccount(this, amount, symbol, isBorrow);
+  }
+  async withdrawFromDriftVault(vault: string) {
+    return await withdrawFromDriftVault(this, vault);
+  }
+  async updateDriftVaultDelegate(vaultAddress: string, delegate: string) {
+    return await updateVaultDelegate(this, vaultAddress, delegate);
+  }
 }

src/langchain/3land/create_collection.ts

@@ -0,0 +1,68 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+import {
+  CreateCollectionOptions,
+  StoreInitOptions,
+} from "@3land/listings-sdk/dist/types/implementation/implementationTypes";
+
+export class Solana3LandCreateCollection extends Tool {
+  name = "3land_minting_tool";
+  description = `Creates an NFT Collection that you can visit on 3.land's website (3.land/collection/{collectionAccount})
+  
+  Inputs:
+  privateKey (required): represents the privateKey of the wallet - can be an array of numbers, Uint8Array or base58 string
+  isMainnet (required): defines is the tx takes places in mainnet
+  collectionSymbol (required): the symbol of the collection
+  collectionName (required): the name of the collection
+  collectionDescription (required): the description of the collection
+  mainImageUrl (required): the image of the collection
+  coverImageUrl (optional): the cover image of the collection
+  `;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const inputFormat = JSON.parse(input);
+      const privateKey = inputFormat.privateKey;
+      const isMainnet = inputFormat.isMainnet;
+
+      const optionsWithBase58: StoreInitOptions = {
+        ...(privateKey && { privateKey }),
+        ...(isMainnet && { isMainnet }),
+      };
+
+      const collectionSymbol = inputFormat?.collectionSymbol;
+      const collectionName = inputFormat?.collectionName;
+      const collectionDescription = inputFormat?.collectionDescription;
+      const mainImageUrl = inputFormat?.mainImageUrl;
+      const coverImageUrl = inputFormat?.coverImageUrl;
+
+      const collectionOpts: CreateCollectionOptions = {
+        ...(collectionSymbol && { collectionSymbol }),
+        ...(collectionName && { collectionName }),
+        ...(collectionDescription && { collectionDescription }),
+        ...(mainImageUrl && { mainImageUrl }),
+        ...(coverImageUrl && { coverImageUrl }),
+      };
+
+      const tx = await this.solanaKit.create3LandCollection(
+        optionsWithBase58,
+        collectionOpts,
+      );
+      return JSON.stringify({
+        status: "success",
+        message: `Created Collection successfully ${tx}`,
+        transaction: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}

src/langchain/3land/create_single.ts

@@ -0,0 +1,91 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+import {
+  CreateSingleOptions,
+  StoreInitOptions,
+} from "@3land/listings-sdk/dist/types/implementation/implementationTypes";
+
+export class Solana3LandCreateSingle extends Tool {
+  name = "3land_minting_tool";
+  description = `Creates an NFT and lists it on 3.land's website
+
+  Inputs:
+  privateKey (required): represents the privateKey of the wallet - can be an array of numbers, Uint8Array or base58 string
+  collectionAccount (optional): represents the account for the nft collection
+  itemName (required): the name of the NFT
+  sellerFee (required): the fee of the seller
+  itemAmount (required): the amount of the NFTs that can be minted
+  itemDescription (required): the description of the NFT
+  traits (required): the traits of the NFT [{trait_type: string, value: string}]
+  price (required): the price of the item, if is 0 the listing will be free
+  mainImageUrl (required): the main image of the NFT
+  coverImageUrl (optional): the cover image of the NFT
+  splHash (optional): the hash of the spl token, if not provided listing will be in $SOL
+  isMainnet (required): defines is the tx takes places in mainnet
+  `;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const inputFormat = JSON.parse(input);
+      const privateKey = inputFormat.privateKey;
+      const isMainnet = inputFormat.isMainnet;
+
+      const optionsWithBase58: StoreInitOptions = {
+        ...(privateKey && { privateKey }),
+        ...(isMainnet && { isMainnet }),
+      };
+
+      const collectionAccount = inputFormat.collectionAccount;
+
+      const itemName = inputFormat?.itemName;
+      const sellerFee = inputFormat?.sellerFee;
+      const itemAmount = inputFormat?.itemAmount;
+      const itemSymbol = inputFormat?.itemSymbol;
+      const itemDescription = inputFormat?.itemDescription;
+      const traits = inputFormat?.traits;
+      const price = inputFormat?.price;
+      const mainImageUrl = inputFormat?.mainImageUrl;
+      const coverImageUrl = inputFormat?.coverImageUrl;
+      const splHash = inputFormat?.splHash;
+
+      const createItemOptions: CreateSingleOptions = {
+        ...(itemName && { itemName }),
+        ...(sellerFee && { sellerFee }),
+        ...(itemAmount && { itemAmount }),
+        ...(itemSymbol && { itemSymbol }),
+        ...(itemDescription && { itemDescription }),
+        ...(traits && { traits }),
+        ...(price && { price }),
+        ...(mainImageUrl && { mainImageUrl }),
+        ...(coverImageUrl && { coverImageUrl }),
+        ...(splHash && { splHash }),
+      };
+
+      if (!collectionAccount) {
+        throw new Error("Collection account is required");
+      }
+
+      const tx = await this.solanaKit.create3LandNft(
+        optionsWithBase58,
+        collectionAccount,
+        createItemOptions,
+        isMainnet,
+      );
+      return JSON.stringify({
+        status: "success",
+        message: `Created listing successfully ${tx}`,
+        transaction: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}

src/langchain/3land/index.ts

@@ -0,0 +1,2 @@
+export * from "./create_single";
+export * from "./create_collection";

src/langchain/adrena/close_trade.ts

@@ -0,0 +1,49 @@
+import { PublicKey } from "@solana/web3.js";
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaPerpCloseTradeTool extends Tool {
+  name = "solana_close_perp_trade";
+  description = `This tool can be used to close perpetuals trade ( It uses Adrena Protocol ).
+
+  Inputs ( input is a JSON string ):
+  tradeMint: string, eg "J1toso1uCk3RLmjorhTtrVwY9HJ7X8V9yYac6Y7kGCPn", "DezXAZ8z7PnrnRJjz3wXBoRgixCa6xjnB7YaB1pPB263" etc. (optional)
+  price?: number, eg 100 (optional)
+  side: string, eg: "long" or "short"`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+
+      const tx =
+        parsedInput.side === "long"
+          ? await this.solanaKit.closePerpTradeLong({
+              price: parsedInput.price,
+              tradeMint: new PublicKey(parsedInput.tradeMint),
+            })
+          : await this.solanaKit.closePerpTradeShort({
+              price: parsedInput.price,
+              tradeMint: new PublicKey(parsedInput.tradeMint),
+            });
+
+      return JSON.stringify({
+        status: "success",
+        message: "Perpetual trade closed successfully",
+        transaction: tx,
+        price: parsedInput.price,
+        tradeMint: new PublicKey(parsedInput.tradeMint),
+        side: parsedInput.side,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}

src/langchain/adrena/index.ts

@@ -0,0 +1,2 @@
+export * from "./open_trade";
+export * from "./close_trade";

src/langchain/adrena/open_trade.ts

@@ -0,0 +1,65 @@
+import { PublicKey } from "@solana/web3.js";
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaPerpOpenTradeTool extends Tool {
+  name = "solana_open_perp_trade";
+  description = `This tool can be used to open perpetuals trade ( It uses Adrena Protocol ).
+
+  Inputs ( input is a JSON string ):
+  collateralAmount: number, eg 1 or 0.01 (required)
+  collateralMint: string, eg "J1toso1uCk3RLmjorhTtrVwY9HJ7X8V9yYac6Y7kGCPn" or "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v" etc. (optional)
+  tradeMint: string, eg "J1toso1uCk3RLmjorhTtrVwY9HJ7X8V9yYac6Y7kGCPn", "DezXAZ8z7PnrnRJjz3wXBoRgixCa6xjnB7YaB1pPB263" etc. (optional)
+  leverage: number, eg 50000 = x5, 100000 = x10, 1000000 = x100 (optional)
+  price?: number, eg 100 (optional)
+  slippage?: number, eg 0.3 (optional)
+  side: string, eg: "long" or "short"`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+
+      const tx =
+        parsedInput.side === "long"
+          ? await this.solanaKit.openPerpTradeLong({
+              price: parsedInput.price,
+              collateralAmount: parsedInput.collateralAmount,
+              collateralMint: new PublicKey(parsedInput.collateralMint),
+              leverage: parsedInput.leverage,
+              tradeMint: new PublicKey(parsedInput.tradeMint),
+              slippage: parsedInput.slippage,
+            })
+          : await this.solanaKit.openPerpTradeLong({
+              price: parsedInput.price,
+              collateralAmount: parsedInput.collateralAmount,
+              collateralMint: new PublicKey(parsedInput.collateralMint),
+              leverage: parsedInput.leverage,
+              tradeMint: new PublicKey(parsedInput.tradeMint),
+              slippage: parsedInput.slippage,
+            });
+
+      return JSON.stringify({
+        status: "success",
+        message: "Perpetual trade opened successfully",
+        transaction: tx,
+        price: parsedInput.price,
+        collateralAmount: parsedInput.collateralAmount,
+        collateralMint: new PublicKey(parsedInput.collateralMint),
+        leverage: parsedInput.leverage,
+        tradeMint: new PublicKey(parsedInput.tradeMint),
+        slippage: parsedInput.slippage,
+        side: parsedInput.side,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}

src/langchain/agent/create_image.ts

@@ -0,0 +1,38 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+import { create_image } from "../../tools/agent";
+
+export class SolanaCreateImageTool extends Tool {
+  name = "solana_create_image";
+  description =
+    "Create an image using OpenAI's DALL-E. Input should be a string prompt for the image.";
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  private validateInput(input: string): void {
+    if (typeof input !== "string" || input.trim().length === 0) {
+      throw new Error("Input must be a non-empty string prompt");
+    }
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      this.validateInput(input);
+      const result = await create_image(this.solanaKit, input.trim());
+
+      return JSON.stringify({
+        status: "success",
+        message: "Image created successfully",
+        ...result,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}

src/langchain/agent/index.ts

@@ -0,0 +1,2 @@
+export * from "./create_image";
+export * from "./wallet_address";

src/langchain/agent/wallet_address.ts

@@ -0,0 +1,15 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaGetWalletAddressTool extends Tool {
+  name = "solana_get_wallet_address";
+  description = `Get the wallet address of the agent`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(_input: string): Promise<string> {
+    return this.solanaKit.wallet_address.toString();
+  }
+}

src/langchain/alldomains/get_all_tld.ts

@@ -0,0 +1,29 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaGetAllTlds extends Tool {
+  name = "solana_get_all_tlds";
+  description = `Get all active top-level domains (TLDs) in the AllDomains Name Service`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(): Promise<string> {
+    try {
+      const tlds = await this.solanaKit.getAllDomainsTLDs();
+
+      return JSON.stringify({
+        status: "success",
+        message: "TLDs fetched successfully",
+        tlds,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "FETCH_TLDS_ERROR",
+      });
+    }
+  }
+}

src/langchain/alldomains/index.ts

@@ -0,0 +1,4 @@
+export * from "./resolve_all_domains";
+export * from "./owned_domains";
+export * from "./tld_domains";
+export * from "./get_all_tld";

src/langchain/alldomains/owned_domains.ts

@@ -0,0 +1,34 @@
+import { PublicKey } from "@solana/web3.js";
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaGetOwnedDomains extends Tool {
+  name = "solana_get_owned_domains";
+  description = `Get all domains owned by a specific wallet address.
+
+  Inputs:
+  owner: string, eg "4Be9CvxqHW6BYiRAxW9Q3xu1ycTMWaL5z8NX4HR3ha7t" (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      const ownerPubkey = new PublicKey(input.trim());
+      const domains = await this.solanaKit.getOwnedAllDomains(ownerPubkey);
+
+      return JSON.stringify({
+        status: "success",
+        message: "Owned domains fetched successfully",
+        domains,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "FETCH_OWNED_DOMAINS_ERROR",
+      });
+    }
+  }
+}

src/langchain/alldomains/resolve_all_domains.ts

@@ -0,0 +1,42 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaResolveAllDomainsTool extends Tool {
+  name = "solana_resolve_all_domains";
+  description = `Resolve domain names to a public key for ALL domain types EXCEPT .sol domains.
+  Use this for domains like .blink, .bonk, etc.
+  DO NOT use this for .sol domains (use solana_resolve_domain instead).
+
+  Input:
+  domain: string, eg "mydomain.blink" or "mydomain.bonk" (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      const owner = await this.solanaKit.resolveAllDomains(input);
+
+      if (!owner) {
+        return JSON.stringify({
+          status: "error",
+          message: "Domain not found",
+          code: "DOMAIN_NOT_FOUND",
+        });
+      }
+
+      return JSON.stringify({
+        status: "success",
+        message: "Domain resolved successfully",
+        owner: owner?.toString(),
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "DOMAIN_RESOLUTION_ERROR",
+      });
+    }
+  }
+}

src/langchain/alldomains/tld_domains.ts

@@ -0,0 +1,32 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaGetOwnedTldDomains extends Tool {
+  name = "solana_get_owned_tld_domains";
+  description = `Get all domains owned by the agent's wallet for a specific TLD.
+
+  Inputs:
+  tld: string, eg "bonk" (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      const domains = await this.solanaKit.getOwnedDomainsForTLD(input);
+
+      return JSON.stringify({
+        status: "success",
+        message: "TLD domains fetched successfully",
+        domains,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "FETCH_TLD_DOMAINS_ERROR",
+      });
+    }
+  }
+}

src/langchain/dexscreener/index.ts

@@ -0,0 +1 @@
+export * from "./token_data_ticker";

src/langchain/dexscreener/token_data_ticker.ts

@@ -0,0 +1,31 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaTokenDataByTickerTool extends Tool {
+  name = "solana_token_data_by_ticker";
+  description = `Get the token data for a given token ticker
+
+  Inputs: ticker is required.
+  ticker: string, eg "USDC" (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const ticker = input.trim();
+      const tokenData = await this.solanaKit.getTokenDataByTicker(ticker);
+      return JSON.stringify({
+        status: "success",
+        tokenData,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/create_user_account.ts

@@ -0,0 +1,38 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaCreateDriftUserAccountTool extends Tool {
+  name = "create_drift_user_account";
+  description = `Create a new user account with a deposit on Drift protocol.
+  
+  Inputs (JSON string):
+  - amount: number, amount of the token to deposit (required)
+  - symbol: string, symbol of the token to deposit (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const res = await this.solanaKit.createDriftUserAccount(
+        parsedInput.amount,
+        parsedInput.symbol,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: `User account created with ${parsedInput.amount} ${parsedInput.symbol} successfully deposited`,
+        account: res.account,
+        signature: res.txSignature,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "CREATE_DRIFT_USER_ACCOUNT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/create_vault.ts

@@ -0,0 +1,42 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaCreateDriftVaultTool extends Tool {
+  name = "create_drift_vault";
+  description = `Create a new drift vault delegating the agents address as the owner.
+  
+  Inputs (JSON string):
+  - name: string, unique vault name (min 5 chars)
+  - marketName: string, market name in TOKEN-SPOT format
+  - redeemPeriod: number, days to wait before funds can be redeemed (min 1)
+  - maxTokens: number, maximum tokens vault can accommodate (min 100)
+  - minDepositAmount: number, minimum deposit amount
+  - managementFee: number, fee percentage for managing funds (max 20)
+  - profitShare: number, profit sharing percentage (max 90, default 5)
+  - hurdleRate: number, optional hurdle rate
+  - permissioned: boolean, whether vault has whitelist`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.createDriftVault(parsedInput);
+
+      return JSON.stringify({
+        status: "success",
+        message: "Drift vault created successfully",
+        vaultName: parsedInput.name,
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "CREATE_DRIFT_VAULT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/deposit_into_vault.ts

@@ -0,0 +1,37 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaDepositIntoDriftVaultTool extends Tool {
+  name = "deposit_into_drift_vault";
+  description = `Deposit funds into an existing drift vault.
+  
+  Inputs (JSON string):
+  - vaultAddress: string, address of the vault (required)
+  - amount: number, amount to deposit (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.depositIntoDriftVault(
+        parsedInput.amount,
+        parsedInput.vaultAddress,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Funds deposited successfully",
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "DEPOSIT_INTO_VAULT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/deposit_to_user_account.ts

@@ -0,0 +1,39 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaDepositToDriftUserAccountTool extends Tool {
+  name = "deposit_to_drift_user_account";
+  description = `Deposit funds into your drift user account.
+  
+  Inputs (JSON string):
+  - amount: number, amount to deposit (required)
+  - symbol: string, token symbol (required)
+  - repay: boolean, whether to repay borrowed funds (optional, default: false)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.depositToDriftUserAccount(
+        parsedInput.amount,
+        parsedInput.symbol,
+        parsedInput.repay,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Funds deposited successfully",
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "DEPOSIT_TO_DRIFT_ACCOUNT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/derive_vault_address.ts

@@ -0,0 +1,32 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaDeriveVaultAddressTool extends Tool {
+  name = "derive_drift_vault_address";
+  description = `Derive a drift vault address from the vault's name.
+  
+  Inputs (JSON string):
+  - name: string, name of the vault to derive the address of (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const address = await this.solanaKit.deriveDriftVaultAddress(input);
+
+      return JSON.stringify({
+        status: "success",
+        message: "Vault address derived successfully",
+        address,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "DERIVE_VAULT_ADDRESS_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/does_user_have_drift_account.ts

@@ -0,0 +1,38 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaCheckDriftAccountTool extends Tool {
+  name = "does_user_have_drift_account";
+  description = `Check if a user has a Drift account.
+  
+  Inputs: No inputs required - checks the current user's account`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(_input: string): Promise<string> {
+    try {
+      const res = await this.solanaKit.doesUserHaveDriftAccount();
+
+      if (!res.hasAccount) {
+        return JSON.stringify({
+          status: "error",
+          message: "You do not have a Drift account",
+        });
+      }
+
+      return JSON.stringify({
+        status: "success",
+        message: "Nice! You have a Drift account",
+        account: res.account,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "CHECK_DRIFT_ACCOUNT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/drift_user_account_info.ts

@@ -0,0 +1,29 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaDriftUserAccountInfoTool extends Tool {
+  name = "drift_user_account_info";
+  description = `Get information about your drift account.
+  
+  Inputs: No inputs required - retrieves current user's account info`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(_input: string): Promise<string> {
+    try {
+      const accountInfo = await this.solanaKit.driftUserAccountInfo();
+      return JSON.stringify({
+        status: "success",
+        data: accountInfo,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "DRIFT_ACCOUNT_INFO_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/index.ts

@@ -0,0 +1,15 @@
+export * from "./create_user_account";
+export * from "./create_vault";
+export * from "./deposit_into_vault";
+export * from "./deposit_to_user_account";
+export * from "./derive_vault_address";
+export * from "./does_user_have_drift_account";
+export * from "./drift_user_account_info";
+export * from "./request_withdrawal";
+export * from "./trade_delegated_vault";
+export * from "./trade_perp_account";
+export * from "./update_drift_vault_delegate";
+export * from "./update_vault";
+export * from "./vault_info";
+export * from "./withdraw_from_account";
+export * from "./withdraw_from_vault";

src/langchain/drift/request_withdrawal.ts

@@ -0,0 +1,37 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaRequestDriftWithdrawalTool extends Tool {
+  name = "request_withdrawal_from_drift_vault";
+  description = `Request a withdrawal from an existing drift vault.
+  
+  Inputs (JSON string):
+  - vaultAddress: string, vault address (required)
+  - amount: number, amount of shares to withdraw (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.requestWithdrawalFromDriftVault(
+        parsedInput.amount,
+        parsedInput.vaultAddress,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Withdrawal request successful",
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "REQUEST_DRIFT_WITHDRAWAL_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/trade_delegated_vault.ts

@@ -0,0 +1,49 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaTradeDelegatedDriftVaultTool extends Tool {
+  name = "trade_delegated_drift_vault";
+  description = `Carry out trades in a Drift vault.
+  
+  Inputs (JSON string):
+  - vaultAddress: string, address of the Drift vault
+  - amount: number, amount to trade
+  - symbol: string, symbol of the token to trade
+  - action: "long" | "short", trade direction
+  - type: "market" | "limit", order type
+  - price: number, optional limit price`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.tradeUsingDelegatedDriftVault(
+        parsedInput.vaultAddress,
+        parsedInput.amount,
+        parsedInput.symbol,
+        parsedInput.action,
+        parsedInput.type,
+        parsedInput.price,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message:
+          parsedInput.type === "limit"
+            ? "Order placed successfully"
+            : "Trade successful",
+        transactionId: tx,
+        ...parsedInput,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "TRADE_DRIFT_VAULT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/trade_perp_account.ts

@@ -0,0 +1,42 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaTradeDriftPerpAccountTool extends Tool {
+  name = "trade_drift_perp_account";
+  description = `Trade a perpetual account on Drift protocol.
+  
+  Inputs (JSON string):
+  - amount: number, amount to trade (required)
+  - symbol: string, token symbol (required)
+  - action: "long" | "short", trade direction (required)
+  - type: "market" | "limit", order type (required)
+  - price: number, required for limit orders`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const signature = await this.solanaKit.tradeUsingDriftPerpAccount(
+        parsedInput.amount,
+        parsedInput.symbol,
+        parsedInput.action,
+        parsedInput.type,
+        parsedInput.price,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        signature,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "TRADE_PERP_ACCOUNT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/update_drift_vault_delegate.ts

@@ -0,0 +1,37 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaUpdateDriftVaultDelegateTool extends Tool {
+  name = "update_drift_vault_delegate";
+  description = `Update the delegate of a drift vault.
+  
+  Inputs (JSON string):
+  - vaultAddress: string, address of the vault (required)
+  - newDelegate: string, address of the new delegate (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.updateDriftVaultDelegate(
+        parsedInput.vaultAddress,
+        parsedInput.newDelegate,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Vault delegate updated successfully",
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UPDATE_DRIFT_VAULT_DELEGATE_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/update_vault.ts

@@ -0,0 +1,52 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaUpdateDriftVaultTool extends Tool {
+  name = "update_drift_vault";
+  description = `Update an existing drift vault with new settings.
+  
+  Inputs (JSON string):
+  - vaultAddress: string, vault address (required)
+  - redeemPeriod: number, days until redemption (optional)
+  - maxTokens: number, maximum tokens allowed (optional)
+  - minDepositAmount: number, minimum deposit amount (optional)
+  - managementFee: number, management fee percentage (optional)
+  - profitShare: number, profit sharing percentage (optional)
+  - hurdleRate: number, hurdle rate (optional)
+  - permissioned: boolean, whitelist requirement (optional)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.updateDriftVault(
+        parsedInput.vaultAddress,
+        // @ts-expect-error - type mismatch
+        {
+          hurdleRate: parsedInput.hurdleRate,
+          maxTokens: parsedInput.maxTokens,
+          minDepositAmount: parsedInput.minDepositAmount,
+          profitShare: parsedInput.profitShare,
+          managementFee: parsedInput.managementFee,
+          permissioned: parsedInput.permissioned,
+          redeemPeriod: parsedInput.redeemPeriod,
+        },
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Drift vault parameters updated successfully",
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UPDATE_DRIFT_VAULT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/vault_info.ts

@@ -0,0 +1,32 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaDriftVaultInfoTool extends Tool {
+  name = "drift_vault_info";
+  description = `Get information about a drift vault.
+  
+  Inputs (JSON string):
+  - vaultNameOrAddress: string, name or address of the vault (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const vaultInfo = await this.solanaKit.getDriftVaultInfo(input);
+
+      return JSON.stringify({
+        status: "success",
+        message: "Vault info retrieved successfully",
+        data: vaultInfo,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "DRIFT_VAULT_INFO_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/withdraw_from_account.ts

@@ -0,0 +1,39 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaWithdrawFromDriftAccountTool extends Tool {
+  name = "withdraw_from_drift_account";
+  description = `Withdraw or borrow funds from your drift account.
+  
+  Inputs (JSON string):
+  - amount: number, amount to withdraw (required)
+  - symbol: string, token symbol (required)
+  - isBorrow: boolean, whether to borrow funds instead of withdrawing (optional, default: false)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const tx = await this.solanaKit.withdrawFromDriftAccount(
+        parsedInput.amount,
+        parsedInput.symbol,
+        parsedInput.isBorrow,
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Funds withdrawn successfully",
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "WITHDRAW_FROM_DRIFT_ACCOUNT_ERROR",
+      });
+    }
+  }
+}

src/langchain/drift/withdraw_from_vault.ts

@@ -0,0 +1,32 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+
+export class SolanaWithdrawFromDriftVaultTool extends Tool {
+  name = "withdraw_from_drift_vault";
+  description = `Withdraw funds from a vault given the redemption time has elapsed.
+  
+  Inputs (JSON string):
+  - vaultAddress: string, vault address (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const tx = await this.solanaKit.withdrawFromDriftVault(input);
+
+      return JSON.stringify({
+        status: "success",
+        message: "Withdrawal successful",
+        signature: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "WITHDRAW_FROM_DRIFT_VAULT_ERROR",
+      });
+    }
+  }
+}