feat: enabled swap by token symbol + added erc20 permit and allowance

VisionaireLabs · Jan 28, 2025 · 7e5afdc · 7e5afdc
1 parent b9322f9
commit 7e5afdc
Show file tree

Hide file tree

Showing 8 changed files with 766 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -531,6 +531,21 @@ const inference = await agent.getInferenceByTopicId(42);
 console.log("Allora inference for topic 42:", inference);
 ```
 
+### Cross-Chain Swap
+
+```typescript
+import { PublicKey } from "@solana/web3.js";
+
+const signature = await agent.swap(
+  amount: "10",
+  fromChain: "bsc",
+  fromToken: "0x3c499c542cef5e3811e1192ce70d8cc03d5c3359",
+  toChain: "solana",
+  toToken: "0x0000000000000000000000000000000000000000",
+  dstAddr: "0xc2d3024d64f27d85e05c40056674Fd18772dd922",
+);
+```
+
 ## Examples
 
 ### LangGraph Multi-Agent System

diff --git a/package.json b/package.json
@@ -55,6 +55,7 @@
     "@meteora-ag/alpha-vault": "^1.1.7",
     "@meteora-ag/dlmm": "^1.3.0",
     "@onsol/tldparser": "^0.6.7",
+    "@openzeppelin/contracts": "^5.2.0",
     "@orca-so/common-sdk": "0.6.4",
     "@orca-so/whirlpools-sdk": "^0.13.12",
     "@pythnetwork/hermes-client": "^1.3.0",

diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/src/actions/mayan/swap.ts b/src/actions/mayan/swap.ts
@@ -1,4 +1,3 @@
-import { PublicKey } from "@solana/web3.js";
 import { Action } from "../../types/action";
 import { SolanaAgentKit } from "../../agent";
 import { z } from "zod";
@@ -27,8 +26,6 @@ const swapAction: Action = {
         explanation:
           "swap 0.02 0x0000000000000000000000000000000000000000 from solana to 0x0000000000000000000000000000000000000000 polygon destination 0x0cae42c0ce52e6e64c1e384ff98e686c6ee225f0",
       },
-    ],
-    [
       {
         input: {
           amount: "0.02",
@@ -47,6 +44,25 @@ const swapAction: Action = {
           "swap 0.02 0x0000000000000000000000000000000000000000 from solana to HZ1JovNiVvGrGNiiYvEozEVgZ58xaU3RKwX8eACQBCt3 solana destination 4ZgCP2idpqrxuQNfsjakJEm9nFyZ2xnT4CrDPKPULJPk",
       },
     ],
+    [
+      {
+        input: {
+          amount: "0.02",
+          fromChain: "solana",
+          fromToken: "sol",
+          toChain: "solana",
+          toToken: "HNT",
+          dstAddr: "4ZgCP2idpqrxuQNfsjakJEm9nFyZ2xnT4CrDPKPULJPk",
+        },
+        output: {
+          status: "success",
+          message: "Swap executed successfully",
+          url: "https://explorer.mayan.finance/swap/2GLNqs5gXCBSwRt6VjtfQRnLWYbcU1gzkgjWMWautv1RUj13Di4qJPjV29YRpoAdMYxgXj8ArMLzF3bCCZmVUXHz",
+        },
+        explanation:
+          "swap 0.02 sol from solana to hnt solana destination 4ZgCP2idpqrxuQNfsjakJEm9nFyZ2xnT4CrDPKPULJPk",
+      },
+    ],
   ],
   schema: z.object({
     amount: z
@@ -65,7 +81,7 @@ const swapAction: Action = {
       "optimism",
       "base",
     ]),
-    fromToken: z.string().min(32, "Invalid from mint address"),
+    fromToken: z.string(),
     toChain: z.enum([
       "solana",
       "ethereum",
@@ -76,7 +92,7 @@ const swapAction: Action = {
       "optimism",
       "base",
     ]),
-    toToken: z.string().min(32, "Invalid to mint address"),
+    toToken: z.string(),
     dstAddr: z.string().min(32, "Invalid destination address"),
     inputAmount: z.number().positive("Input amount must be positive"),
     slippageBps: z.number().min(0).max(10000).optional(),
@@ -87,7 +103,7 @@ const swapAction: Action = {
       throw new Error("one of the from or to chain should be solana.");
     }
 
-    const tx = await swap(
+    const url = await swap(
       agent,
       input.amount,
       input.fromChain,
@@ -101,10 +117,7 @@ const swapAction: Action = {
     return {
       status: "success",
       message: "Swap executed successfully",
-      transaction: tx,
-      inputAmount: input.inputAmount,
-      inputToken: input.inputMint || "SOL",
-      outputToken: input.outputMint,
+      url,
     };
   },
 };

diff --git a/src/langchain/mayan/swap.ts b/src/langchain/mayan/swap.ts
@@ -8,9 +8,9 @@ export class SolanaCrossChainSwapTool extends Tool {
   Inputs ( input is a JSON string):
   amount: string, eg "0.02" or "7"
   fromChain: string, eg "solana" or "ethereum"
-  fromToken: string, eg "0x0000000000000000000000000000000000000000" or "hntyVP6YFm1Hg25TN9WGLqM12b8TQmcknKrdu1oxWux"
+  fromToken: string, eg "0x0000000000000000000000000000000000000000" or "hntyVP6YFm1Hg25TN9WGLqM12b8TQmcknKrdu1oxWux" or "sol" or "Pol"
   toChain: string, eg "solana" or "ethereum"
-  toToken: string, eg "0x0000000000000000000000000000000000000000" or "0x6b175474e89094c44da98b954eedeac495271d0f"
+  toToken: string, eg "0x0000000000000000000000000000000000000000" or "0x6b175474e89094c44da98b954eedeac495271d0f" or "SOL" or "eth"
   dstAddr: string, eg "4ZgCP2idpqrxuQNfsjakJEm9nFyZ2xnT4CrDPKPULJPk" or "0x0cae42c0cE52E6E64C1e384fF98e686C6eE225f0"
   slippageBps: number, eg 10 (optional)`;