feat: update implementation

Author: Resister-boy

src/actions/index.ts

@@ -129,6 +129,8 @@ import getSanctumApyAction from "./sanctum/getApy";
 import getSanctumTvlAction from "./sanctum/getTvl";
 import sanctumAddLiquidityAction from "./sanctum/addLiquidity";
 import sanctumRemoveLiquidityAction from "./sanctum/removeLiquidity";
+import sanctumGetOwnedLstAction from "./sanctum/getOwnedLst";
+import sanctumSwapLstAction from "./sanctum/swapLst";
 
 export const ACTIONS = {
   GET_INFO_ACTION: getInfoAction,
@@ -266,6 +268,8 @@ export const ACTIONS = {
   SANCTUM_GET_TVL_ACTION: getSanctumTvlAction,
   SANCTUM_ADD_LIQUIDITY_ACTION: sanctumAddLiquidityAction,
   SANCTUM_REMOVE_LIQUIDITY_ACTION: sanctumRemoveLiquidityAction,
+  SANCTUM_GET_OWNED_LST_ACTION: sanctumGetOwnedLstAction,
+  SANCTUM_SWAP_LST_ACTION: sanctumSwapLstAction,
 };
 
 export type { Action, ActionExample, Handler } from "../types/action";

src/agent/index.ts

@@ -153,6 +153,8 @@ import {
   get_tvl as getSanctumLSTTvl,
   add_liquidity as addSanctumLSTLiquidity,
   remove_liquidity as removeSanctumLSTLiquidity,
+  get_owned_lst as getSanctumOwnedLst,
+  swap_lst as swapLst,
 } from "../tools";
 import {
   Config,
@@ -1478,6 +1480,10 @@ export class SolanaAgentKit {
     return getSanctumLSTTvl(inputs);
   }
 
+  async getSanctumOwnedLst() {
+    return getSanctumOwnedLst(this);
+  }
+
   async addSanctumLiquidity(
     lstMint: string,
     amount: string,
@@ -1507,4 +1513,21 @@ export class SolanaAgentKit {
       priorityFee,
     );
   }
+
+  async swapSanctumLst(
+    inputLstMint: string,
+    amount: string,
+    quotedAmount: string,
+    priorityFee: number,
+    outputLstMint: string,
+  ) {
+    return swapLst(
+      this,
+      inputLstMint,
+      amount,
+      quotedAmount,
+      priorityFee,
+      outputLstMint,
+    );
+  }
 }

src/langchain/index.ts

@@ -174,6 +174,8 @@ import {
   SanctumGetTvlTool,
   SanctumAddLiquidityTool,
   SanctumRemoveLiquidityTool,
+  SanctumGetOwnedLstTool,
+  SanctumSwapLstTool,
 } from "./index";
 import {
   SolanaCancelLimitOrdersTool,
@@ -320,5 +322,7 @@ export function createSolanaTools(solanaKit: SolanaAgentKit) {
     new SanctumGetTvlTool(solanaKit),
     new SanctumAddLiquidityTool(solanaKit),
     new SanctumRemoveLiquidityTool(solanaKit),
+    new SanctumGetOwnedLstTool(solanaKit),
+    new SanctumSwapLstTool(solanaKit),
   ];
 }

src/langchain/sanctum/index.ts

@@ -3,3 +3,5 @@ export * from "./get_apy";
 export * from "./get_tvl";
 export * from "./add_liquidity";
 export * from "./remove_liquidity";
+export * from "./get_owned_lst";
+export * from "./swap_lst";

src/tools/sanctum/index.ts

@@ -3,3 +3,5 @@ export * from "./get_apy";
 export * from "./get_tvl";
 export * from "./add_liquidity";
 export * from "./remove_liquidity";
+export * from "./get_owned_lst";
+export * from "./swap_lst";