fix: plugin-lightning

packages/plugin-lightning/src/actions/createInvoice.ts

@@ -40,7 +40,7 @@ export const createInvoiceAction = {
         runtime: IAgentRuntime,
         _message: Memory,
         state: State,
-        _options: any,
+        _options: Record<string, unknown>,
         callback?: (response: {
             text: string;
             content?: { success: boolean; invoice?: string };

packages/plugin-lightning/src/actions/payInvoice.ts

@@ -13,6 +13,7 @@ import {
 import type { PayResult } from "astra-lightning";
 import type { PayArgs } from "../types";
 import { payInvoiceTemplate } from "../templates";
+import { z } from "zod";
 
 export { payInvoiceTemplate };
 
@@ -54,15 +55,26 @@ export class PayInvoiceAction {
     }
 }
 
+// Define the schema type
+const payInvoiceSchema = z.object({
+    request: z.string(),
+    outgoing_channel: z.string()
+});
+
+type PayInvoiceContent = z.infer<typeof payInvoiceSchema>;
+
 export const payInvoiceAction = {
     name: "PAY_INVOICE",
     description: "Make a payment.",
     handler: async (
         runtime: IAgentRuntime,
         _message: Memory,
         state: State,
-        _options: any,
-        callback?: any
+        _options: Record<string, unknown>,
+        callback?: (response: {
+            text: string;
+            content?: { success: boolean };
+        }) => void
     ) => {
         elizaLogger.log("payInvoice action handler called");
         const lightningProvider = await initLightningProvider(runtime);
@@ -76,28 +88,30 @@ export const payInvoiceAction = {
         const content = await generateObject({
             runtime,
             context: payInvoiceContext,
+            schema: payInvoiceSchema as z.ZodType,
             modelClass: ModelClass.LARGE,
         });
 
+        const payInvoiceContent = content.object as PayInvoiceContent;
+
         const payInvoiceOptions: PayArgs = {
-            request: content.request,
-            outgoing_channel: content.outgoing_channel,
+            request: payInvoiceContent.request,
+            outgoing_channel: payInvoiceContent.outgoing_channel,
         };
 
         try {
             const payInvoiceResp = await action.payInvoice(payInvoiceOptions);
             elizaLogger.log("🚀 ~ payInvoiceResp:", payInvoiceResp);
 
             if (callback) {
-                const text = "";
                 if (payInvoiceResp.is_confirmed) {
                     callback({
-                        text: `Successfully paid invoice ${content.request} from ${payInvoiceResp.outgoing_channel};\nAmount: ${payInvoiceResp.tokens};\nFee: ${payInvoiceResp.fee};\nPayment Hash: ${payInvoiceResp.id};`,
+                        text: `Successfully paid invoice ${payInvoiceContent.request} from ${payInvoiceResp.outgoing_channel};\nAmount: ${payInvoiceResp.tokens};\nFee: ${payInvoiceResp.fee};\nPayment Hash: ${payInvoiceResp.id};`,
                         content: { success: true },
                     });
                 } else {
                     callback({
-                        text: `Failed to payInvoice ${content.request} from ${content.outgoing_channel};\r\n Amount: ${payInvoiceResp.tokens};`,
+                        text: `Failed to payInvoice ${payInvoiceContent.request} from ${payInvoiceContent.outgoing_channel};\r\n Amount: ${payInvoiceResp.tokens};`,
                         content: {
                             success: false,
                         },