fix(plugin-evm): eth transfer action

elizaOS · Dec 10, 2024 · c66dbf5 · c66dbf5
1 parent 622000e
commit c66dbf5
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 11 deletions.
diff --git a/packages/plugin-evm/src/actions/transfer.ts b/packages/plugin-evm/src/actions/transfer.ts
@@ -2,7 +2,15 @@ import { ByteArray, parseEther, type Hex } from "viem";
 import { WalletProvider } from "../providers/wallet";
 import type { Transaction, TransferParams } from "../types";
 import { transferTemplate } from "../templates";
-import type { IAgentRuntime, Memory, State } from "@ai16z/eliza";
+import {
+    composeContext,
+    generateObjectDEPRECATED,
+    HandlerCallback,
+    ModelClass,
+    type IAgentRuntime,
+    type Memory,
+    type State,
+} from "@ai16z/eliza";
 
 export { transferTemplate };
 export class TransferAction {
@@ -15,6 +23,10 @@ export class TransferAction {
         const walletClient = this.walletProvider.getWalletClient();
         const [fromAddress] = await walletClient.getAddresses();
 
+        if (!params.data) {
+            params.data = "0x";
+        }
+
         await this.walletProvider.switchChain(runtime, params.fromChain);
 
         try {
@@ -57,11 +69,51 @@ export const transferAction = {
         runtime: IAgentRuntime,
         message: Memory,
         state: State,
-        options: any
+        options: any,
+        callback?: HandlerCallback
     ) => {
+        console.log("Transfer action handler called");
         const walletProvider = new WalletProvider(runtime);
         const action = new TransferAction(walletProvider);
-        return action.transfer(runtime, options);
+
+        // Compose transfer context
+        const transferContext = composeContext({
+            state,
+            template: transferTemplate,
+        });
+
+        // Generate transfer content
+        const content = await generateObjectDEPRECATED({
+            runtime,
+            context: transferContext,
+            modelClass: ModelClass.LARGE,
+        });
+
+        const paramOptions: TransferParams = {
+            fromChain: content.fromChain,
+            toAddress: content.toAddress,
+            amount: content.amount,
+            data: content.data,
+        };
+
+        try {
+            const transferResp = await action.transfer(runtime, paramOptions);
+            if (callback) {
+                callback({
+                    text: `Successfully transferred ${paramOptions.amount} tokens to ${paramOptions.toAddress}\nTransaction Hash: ${transferResp.hash}`,
+                });
+            }
+            return true;
+        } catch (error) {
+            console.error("Error during token transfer:", error);
+            if (callback) {
+                callback({
+                    text: `Error transferring tokens: ${error.message}`,
+                    content: { error: error.message },
+                });
+            }
+            return false;
+        }
     },
     template: transferTemplate,
     validate: async (runtime: IAgentRuntime) => {

diff --git a/packages/plugin-evm/src/templates/index.ts b/packages/plugin-evm/src/templates/index.ts
@@ -5,18 +5,18 @@ export const transferTemplate = `Given the recent messages and wallet informatio
 {{walletInfo}}
 
 Extract the following information about the requested transfer:
-- Chain to execute on (ethereum or base)
-- Amount to transfer
-- Recipient address
-- Token symbol or address (if not native token)
+- Chain to execute on: Must be one of ["ethereum", "base", "ethSepolia"]
+- Amount to transfer: Must be a string representing the amount in ETH (e.g., "0.1")
+- Recipient address: Must be a valid Ethereum address starting with "0x"
+- Token symbol or address (if not native token): Optional, leave as null for ETH transfers
 
-Respond with a JSON markdown block containing only the extracted values:
+Respond with a JSON markdown block containing only the extracted values. All fields except 'token' are required:
 
 \`\`\`json
 {
-    "chain": "ethereum" | "base" | null,
-    "amount": string | null,
-    "toAddress": string | null,
+    "fromChain": "ethereum" | "base",
+    "amount": string,
+    "toAddress": string,
     "token": string | null
 }
 \`\`\`