feat: add tool suilend unstaking

Author: hkhangus

src/actions/agent/unstakeSuilend.ts

@@ -0,0 +1,50 @@
+import { Action } from "../../types/action";
+import { SuiAgentKit } from "../../agent";
+import { z } from "zod";
+import { unstaking_suilend } from "../../tools/suilend";
+import { IUnstakingParams } from "../../types/farming";
+
+const unstakeSuilend: Action = {
+  name: "UNSTAKE_SUILEND",
+  similes: ["unstake suilend", "unstake from suilend", "suilend unstake"],
+  description: "Unstake tokens from Suilend protocol",
+  examples: [
+    [
+      {
+        input: {
+          amount: 1,
+          symbol: "sSUI",
+        },
+        output: {
+          status: "success",
+          result: {
+            tx_hash: "5JvBtQveYFsZFYxXMfSxYzUJgzGfyRgkH9YuZxpvuR9Y",
+            tx_status: "success",
+          },
+        },
+        explanation: "Successfully unstaked 1 sSUI from Suilend protocol",
+      },
+    ],
+  ],
+  schema: z.object({
+    amount: z.number().positive(),
+    symbol: z.string(),
+  }),
+  handler: async (agent: SuiAgentKit, input: Record<string, any>) => {
+    const params: IUnstakingParams = {
+      type: "UNSTAKING",
+      amount: input.amount,
+      symbol: input.symbol,
+      poolId: "",
+    };
+
+    const result = await unstaking_suilend(agent, params);
+
+    return {
+      status: "success",
+      result,
+    };
+  },
+};
+
+export default unstakeSuilend;

src/langchain/agent/index.ts

@@ -10,3 +10,5 @@ export * from "./create_pool_cetus_clmm";
 export * from "./transfer_token";
 export * from "./register_sns";
 export * from "./get_sns_name_record";
+export * from "./stake_suilend";
+export * from "./unstake_suilend";

src/langchain/agent/stake_suilend.ts

@@ -7,8 +7,8 @@ export class SuiStakeSuilendTool extends Tool {
   description = `Stake tokens into Suilend protocol.
 
   Inputs (input is a JSON string):
-  amount: number - The amount of tokens to stake (must be positive and not specified minimum amount)
-  symbol: string - The token symbol to stake (e.g., "sSUI")
+  amount: number - The amount of tokens to stake (required, e.g 0.1, 0.01, 0.001, 1, ...)
+  symbol: string - The token symbol to stake (required, e.g., "sSUI")
   isStakeAndDeposit: boolean - When user want to deposit, set to true (optional)`;
 
   constructor(private suiKit: SuiAgentKit) {

src/langchain/agent/unstake_suilend.ts

@@ -0,0 +1,43 @@
+import { Tool } from "langchain/tools";
+import { SuiAgentKit } from "../../agent";
+import { IUnstakingParams } from "../../types/farming";
+
+export class SuiUnstakeSuilendTool extends Tool {
+  name = "sui_unstake_suilend";
+  description = `Unstake tokens from Suilend protocol.
+
+  Inputs (input is a JSON string):
+  amount: real number - The amount of tokens to unstake (required, e.g 0.1, 0.01, 0.001, 1, ...)
+  symbol: string - The token symbol to unstake (required, e.g., "sSUI")`;
+
+  constructor(private suiKit: SuiAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      const params: IUnstakingParams = {
+        type: "UNSTAKING",
+        amount: parsedInput.amount,
+        symbol: parsedInput.symbol,
+        poolId: "",
+      };
+
+      const result = await this.suiKit.unstakeSuilend(params);
+      return JSON.stringify({
+        status: "success",
+        result: {
+          tx_hash: result.tx_hash,
+          tx_status: result.tx_status,
+        },
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}

src/langchain/index.ts

@@ -14,8 +14,9 @@ import {
   SuiSwapTool,
   SuiRegisterSnsTool,
   SuiGetSnsNameRecordTool,
+  SuiStakeSuilendTool,
+  SuiUnstakeSuilendTool,
 } from "./agent";
-import { SuiStakeSuilendTool } from "./agent/stake_suilend";
 
 export function createSuiTools(suiKit: SuiAgentKit) {
   return [
@@ -32,5 +33,6 @@ export function createSuiTools(suiKit: SuiAgentKit) {
     new SuiRegisterSnsTool(suiKit),
     new SuiGetSnsNameRecordTool(suiKit),
     new SuiStakeSuilendTool(suiKit),
+    new SuiUnstakeSuilendTool(suiKit),
   ];
 }