add callback

IkigaiLabsETH · Jan 12, 2025 · ea0a9b7 · ea0a9b7
1 parent 96b5d09
commit ea0a9b7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/packages/plugin-nft-collections/src/actions/list-nft.ts b/packages/plugin-nft-collections/src/actions/list-nft.ts
@@ -1,5 +1,6 @@
 import { Action, IAgentRuntime, Memory, State } from "@elizaos/core";
 import { ReservoirService } from "../services/reservoir";
+import { HandlerCallback } from "@elizaos/core";
 
 // Helper function to extract NFT listing details from the message
 function extractListingDetails(text: string): {
@@ -39,7 +40,9 @@ export const listNFTAction = (nftService: ReservoirService): Action => {
         handler: async (
             runtime: IAgentRuntime,
             message: Memory,
-            state?: State
+            state: State,
+            options: any,
+            callback: HandlerCallback
         ) => {
             try {
                 const {
@@ -92,6 +95,10 @@ export const listNFTAction = (nftService: ReservoirService): Action => {
                         ? `• Transaction: ${listing.transactionHash}\n`
                         : "");
 
+                callback({
+                    text: response,
+                });
+
                 await runtime.messageManager.createMemory({
                     id: message.id,
                     content: { text: response },

diff --git a/packages/plugin-nft-collections/src/actions/sweep-floor.ts b/packages/plugin-nft-collections/src/actions/sweep-floor.ts
@@ -1,5 +1,6 @@
 import { Action, IAgentRuntime, Memory, State } from "@elizaos/core";
 import { ReservoirService } from "../services/reservoir";
+import { HandlerCallback } from "@elizaos/core";
 
 // Helper function to extract NFT details from the message
 function extractNFTDetails(text: string): {
@@ -34,7 +35,9 @@ export const sweepFloorAction = (nftService: ReservoirService): Action => {
         handler: async (
             runtime: IAgentRuntime,
             message: Memory,
-            state?: State
+            state: State,
+            options: any,
+            callback: HandlerCallback
         ) => {
             try {
                 const { collectionAddress, quantity } = extractNFTDetails(
@@ -80,7 +83,9 @@ export const sweepFloorAction = (nftService: ReservoirService): Action => {
                     `• Average Price: ${(totalPrice / quantity).toFixed(4)} ETH\n` +
                     `• Transaction Path: ${result.path}\n` +
                     `• Status: ${result.steps.map((step) => `${step.action} - ${step.status}`).join(", ")}`;
-
+                callback({
+                    text: response,
+                });
                 await runtime.messageManager.createMemory({
                     id: message.id,
                     content: { text: response },