Add: tensor trade support

src/langchain/index.ts

@@ -1326,6 +1326,133 @@ export class SolanaTipLinkTool extends Tool {
   }
 }
 
+export class SolanaListNFTForSaleTool extends Tool {
+  name = "solana_list_nft_for_sale";
+  description = `List an NFT for sale on Tensor Trade.
+
+  Inputs (input is a JSON string):
+  nftMint: string, the mint address of the NFT (required)
+  price: number, price in SOL (required)
+  expirySeconds: number, expiry time in seconds (optional)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      
+      // Validate NFT ownership first
+      const nftAccount = await this.solanaKit.connection.getTokenAccountsByOwner(
+        this.solanaKit.wallet_address,
+        { mint: new PublicKey(parsedInput.nftMint) }
+      );
+      
+      if (nftAccount.value.length === 0) {
+        return JSON.stringify({
+          status: "error",
+          message: "NFT not found in wallet. Please make sure you own this NFT.",
+          code: "NFT_NOT_FOUND"
+        });
+      }
+
+      const tx = await this.solanaKit.tensorListNFT(
+        new PublicKey(parsedInput.nftMint),
+        parsedInput.price,
+        parsedInput.expirySeconds
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "NFT listed for sale successfully",
+        transaction: tx,
+        price: parsedInput.price,
+        nftMint: parsedInput.nftMint
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR"
+      });
+    }
+  }
+}
+
+export class SolanaBuyNFTTool extends Tool {
+  name = "solana_buy_nft";
+  description = `Buy an NFT listed on Tensor Trade.
+
+  Inputs (input is a JSON string):
+  nftMint: string, the mint address of the NFT (required)
+  maxPrice: number, maximum price willing to pay in SOL (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      
+      const tx = await this.solanaKit.tensorBuyNFT(
+        new PublicKey(parsedInput.nftMint),
+        parsedInput.maxPrice
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "NFT purchased successfully",
+        transaction: tx,
+        maxPrice: parsedInput.maxPrice,
+        nftMint: parsedInput.nftMint
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR"
+      });
+    }
+  }
+}
+
+export class SolanaCancelNFTListingTool extends Tool {
+  name = "solana_cancel_nft_listing";
+  description = `Cancel an NFT listing on Tensor Trade.
+
+  Inputs (input is a JSON string):
+  nftMint: string, the mint address of the NFT (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = JSON.parse(input);
+      
+      const tx = await this.solanaKit.tensorCancelListing(
+        new PublicKey(parsedInput.nftMint)
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "NFT listing cancelled successfully",
+        transaction: tx,
+        nftMint: parsedInput.nftMint
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR"
+      });
+    }
+  }
+}
+
 export function createSolanaTools(solanaKit: SolanaAgentKit) {
   return [
     new SolanaBalanceTool(solanaKit),
@@ -1362,5 +1489,8 @@ export function createSolanaTools(solanaKit: SolanaAgentKit) {
     new SolanaCreateGibworkTask(solanaKit),
     new SolanaRockPaperScissorsTool(solanaKit),
     new SolanaTipLinkTool(solanaKit),
+    new SolanaListNFTForSaleTool(solanaKit),
+    new SolanaBuyNFTTool(solanaKit),
+    new SolanaCancelNFTListingTool(solanaKit),
   ];
 }