feat: add drift functions to agent class

src/actions/drift/createVault.ts

@@ -90,7 +90,9 @@ const createDriftVaultAction: Action = {
 
       return {
         status: "success",
-        message: "Drift vault created successfully",
+        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) {

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/tradeDelegatedDriftVault.ts

@@ -66,7 +66,7 @@ const tradeDelegatedDriftVaultAction: Action = {
     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(["buy", "sell"]).describe("Trade action - buy or sell"),
+    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"),
   }),
@@ -76,7 +76,7 @@ const tradeDelegatedDriftVaultAction: Action = {
         vaultAddress: input.vaultAddress as string,
         amount: input.amount as number,
         symbol: input.symbol as string,
-        action: input.action as "buy" | "sell",
+        action: input.action as "long" | "short",
         type: input.type as "market" | "limit",
         price: input.price as number | undefined,
       };

src/actions/drift/vaultInfo.ts

@@ -13,7 +13,7 @@ const vaultInfoAction: Action = {
     [
       {
         input: {
-          vaultAddress: "2nFeP7taii",
+          vaultName: "test-vault",
         },
         output: {
           status: "success",
@@ -35,35 +35,16 @@ const vaultInfoAction: Action = {
     ],
   ],
   schema: z.object({
-    vaultAddress: z.string(),
+    vaultName: z.string(),
   }),
   handler: async (agent: SolanaAgentKit, input) => {
     try {
-      const vaultInfo = await getVaultInfo(agent, input.vaultAddress as string);
-      const spotToken = MainnetSpotMarkets[vaultInfo.spotMarketIndex];
-      const data = {
-        name: decodeName(vaultInfo.name),
-        marketName: `${spotToken.symbol}-SPOT`,
-        redeemPeriod: vaultInfo.redeemPeriod.toNumber(),
-        maxTokens: vaultInfo.maxTokens.div(spotToken.precision).toNumber(),
-        minDepositAmount: vaultInfo.minDepositAmount
-          .div(spotToken.precision)
-          .toNumber(),
-        managementFee:
-          (vaultInfo.managementFee.toNumber() /
-            PERCENTAGE_PRECISION.toNumber()) *
-          100,
-        profitShare:
-          (vaultInfo.profitShare / PERCENTAGE_PRECISION.toNumber()) * 100,
-        hurdleRate:
-          (vaultInfo.hurdleRate / PERCENTAGE_PRECISION.toNumber()) * 100,
-        permissioned: vaultInfo.permissioned,
-      };
+      const vaultInfo = await getVaultInfo(agent, input.vaultName as string);
 
       return {
         status: "success",
         message: "Vault info retrieved successfully",
-        data,
+        data: vaultInfo,
       };
     } catch (e) {
       return {

src/actions/index.ts

@@ -56,6 +56,7 @@ 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";
 
 export const ACTIONS = {
   WALLET_ADDRESS_ACTION: getWalletAddressAction,
@@ -117,6 +118,7 @@ export const ACTIONS = {
   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,
 };
 
 export type { Action, ActionExample, Handler } from "../types/action";