@dainprotocol/service-sdk 1.1.11 → 1.1.14

package/dist/__tests__/crypto-plugin.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/__tests__/crypto-plugin.test.js

@@ -0,0 +1,358 @@
+"use strict";
+// File: src/__tests__/crypto-plugin.test.ts
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+const client_auth_1 = require("@/client/client-auth");
+const client_1 = require("@/client/client");
+const ed25519_1 = require("@noble/curves/ed25519");
+const bs58_1 = tslib_1.__importDefault(require("bs58"));
+const nodeService_1 = require("@/service/nodeService");
+const core_1 = require("@/service/core");
+const zod_1 = require("zod");
+const crypto_plugin_1 = require("@/plugins/crypto-plugin");
+const cuid2_1 = require("@paralleldrive/cuid2");
+describe("DAIN Framework CryptoPlugin System", () => {
+    // Generate test keys
+    const servicePrivateKey = ed25519_1.ed25519.utils.randomPrivateKey();
+    const servicePublicKey = ed25519_1.ed25519.getPublicKey(servicePrivateKey);
+    const serviceAddress = bs58_1.default.encode(servicePublicKey);
+    const clientPrivateKey = ed25519_1.ed25519.utils.randomPrivateKey();
+    // Set up client authentication
+    const agentAuth = new client_auth_1.DainClientAuth({
+        privateKeyBase58: bs58_1.default.encode(clientPrivateKey),
+        agentId: "agent-crypto-test",
+        orgId: "org-crypto-test",
+    });
+    // Create a test wallet configuration
+    const testWallets = [
+        { chain: "sol", address: "Sol11111111111111111111111111111111111111111" },
+        { chain: "eth", address: "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266" },
+        { chain: "arb", address: "0x70997970C51812dc3A010C7d01b50e0d17dc79C8" }
+    ];
+    // Create separate CryptoPlugin instances for client and service
+    const clientCryptoPlugin = new crypto_plugin_1.CryptoPlugin(testWallets);
+    const serviceCryptoPlugin = new crypto_plugin_1.CryptoPlugin(); // Service doesn't need wallet information
+    // Initialize the connection to the DAIN service with the client crypto plugin
+    const dainConnection = new client_1.DainServiceConnection("http://localhost:4485", agentAuth, {
+        plugins: [clientCryptoPlugin]
+    });
+    // Create a mock Solana transaction (added id field)
+    const mockSolanaTransaction = {
+        encodedTx: "AQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==",
+        chain: "sol",
+        signer: "Sol11111111111111111111111111111111111111111",
+        metadata: {
+            description: "Test SOL Transfer",
+            amount: 0.01,
+            recipient: "SolRecipient111111111111111111111111111111111"
+        }
+    };
+    // Create a mock Ethereum transaction (added id field)
+    const mockEthereumTransaction = {
+        encodedTx: "0xf86c0a85046c7cfe0083016dea94d1310c1e0aa1b9d5dfe80aa9ee4a8b0a48d0f2a780a46057361d00000000000000000000000000000000000000000000000000000000000003e880820a95a0f5d4c5201602ddf5110f112a85fb0cf39a85e3a3174cd237b8383670cab7a3b5a048f2be5cc9f0a0b3f22e60f709c649d5a6a4ad3d70ce44ec80b96ffe7843508c",
+        chain: "eth",
+        signer: "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
+        metadata: {
+            description: "Test ETH Transfer",
+            amount: 0.1,
+            recipient: "0x70997970C51812dc3A010C7d01b50e0d17dc79C8",
+            gasLimit: 21000
+        }
+    };
+    // Create a test tool that uses wallet information and creates unsigned transactions
+    const cryptoTransferTool = (0, core_1.createTool)({
+        id: "crypto-transfer",
+        name: "Cryptocurrency Transfer Tool",
+        description: "Creates unsigned transactions for cryptocurrency transfers",
+        input: zod_1.z.object({
+            amount: zod_1.z.number().positive(),
+            chain: zod_1.z.enum(["sol", "eth", "arb"]),
+            recipientAddress: zod_1.z.string()
+        }),
+        output: zod_1.z.object({
+            success: zod_1.z.boolean(),
+            message: zod_1.z.string(),
+            signatureRequests: zod_1.z.array(zod_1.z.object({
+                txs: zod_1.z.array(zod_1.z.object({
+                    id: zod_1.z.string(),
+                    encodedTx: zod_1.z.string(),
+                    chain: zod_1.z.enum(["sol", "eth", "arb"]),
+                    signer: zod_1.z.string(),
+                    metadata: zod_1.z.any().optional()
+                }))
+            })).optional()
+        }),
+        handler: async ({ amount, chain, recipientAddress }, agentInfo, context) => {
+            // Use the SERVICE-SIDE plugin (serviceCryptoPlugin) helper to get the user's wallets
+            const wallets = serviceCryptoPlugin.getUserWallets(context);
+            console.log(`[SERVICE] Available user wallets:`, wallets);
+            // Get the wallet for the specified chain
+            const userWallet = serviceCryptoPlugin.getUserWalletForChain(context, chain);
+            if (!userWallet) {
+                return {
+                    text: `No wallet found for ${chain} chain`,
+                    data: {
+                        success: false,
+                        message: `No ${chain} wallet found for this user. Please connect a ${chain} wallet.`
+                    },
+                    ui: null
+                };
+            }
+            console.log(`[SERVICE] Using ${chain} wallet: ${userWallet} for transfer of ${amount} to ${recipientAddress}`);
+            // Create mock transactions based on the chain
+            let mockTransaction;
+            if (chain === "sol") {
+                mockTransaction = {
+                    ...mockSolanaTransaction,
+                    signer: userWallet,
+                    metadata: {
+                        ...mockSolanaTransaction.metadata,
+                        amount: amount,
+                        recipient: recipientAddress
+                    }
+                };
+            }
+            else if (chain === "eth" || chain === "arb") {
+                mockTransaction = {
+                    ...mockEthereumTransaction,
+                    chain: chain,
+                    signer: userWallet,
+                    metadata: {
+                        ...mockEthereumTransaction.metadata,
+                        amount: amount,
+                        recipient: recipientAddress
+                    }
+                };
+            }
+            else {
+                return {
+                    text: `Unsupported chain: ${chain}`,
+                    data: {
+                        success: false,
+                        message: `Chain ${chain} is not supported yet.`
+                    },
+                    ui: null
+                };
+            }
+            // Create a transaction with a unique ID using the service plugin helper
+            const transactionWithId = serviceCryptoPlugin.createTransaction(mockTransaction);
+            // Log the transaction ID for debugging
+            console.log(`[SERVICE] Created transaction with ID: ${transactionWithId.id}`);
+            // Create a signature request using the SERVICE-SIDE plugin helper
+            const signatureRequest = serviceCryptoPlugin.createTransactionRequest([transactionWithId]);
+            return {
+                text: `Created unsigned ${chain} transaction for ${amount} to ${recipientAddress}`,
+                data: {
+                    success: true,
+                    message: `Created unsigned ${chain} transaction. Please sign to complete the transfer.`,
+                    signatureRequests: [signatureRequest]
+                },
+                ui: null
+            };
+        },
+    });
+    // Initialize the DAIN service with the SERVICE crypto plugin instance
+    const dainService = (0, nodeService_1.defineDAINService)({
+        metadata: {
+            title: "Crypto Plugin Test Service",
+            description: "A DAIN service for testing the crypto plugin",
+            version: "1.0.0",
+            author: "Test Author",
+            tags: ["crypto", "plugin", "test"],
+        },
+        identity: {
+            publicKey: bs58_1.default.encode(servicePublicKey),
+            agentId: "crypto-agent",
+            orgId: "crypto-org",
+            privateKey: bs58_1.default.encode(servicePrivateKey),
+        },
+        tools: [cryptoTransferTool],
+        plugins: [serviceCryptoPlugin] // Using service-side plugin instance
+    });
+    let server;
+    beforeAll(async () => {
+        server = await dainService.startNode({ port: 4485 });
+        await new Promise(resolve => setTimeout(resolve, 500)); // Give server time to start
+    });
+    afterAll(async () => {
+        await server.shutdown();
+    });
+    it("should assign unique IDs to transactions", async () => {
+        console.log("[CLIENT] Using client plugin with wallets:", clientCryptoPlugin.getWallets());
+        console.log("[SERVICE] Using service plugin with wallets:", serviceCryptoPlugin.getWallets());
+        // Call the tool WITHOUT providing wallet information in the params
+        // The client plugin should automatically add wallet info
+        const params = {
+            amount: 0.05,
+            chain: "sol",
+            recipientAddress: "SolRecipient222222222222222222222222222222222"
+            // No wallets provided - they should come from the client plugin
+        };
+        // Call the tool
+        const result = await dainConnection.callTool("crypto-transfer", params);
+        console.log("[CLIENT] Received response:", JSON.stringify(result, null, 2));
+        // Verify the response
+        expect(result.data.success).toBe(true);
+        expect(result.data.message).toContain("Created unsigned sol transaction");
+        // Use the CLIENT plugin helper to extract transaction requests
+        const signatureRequests = clientCryptoPlugin.getRequestedCryptoSignatures(result);
+        console.log("[CLIENT] Extracted signature requests:", JSON.stringify(signatureRequests, null, 2));
+        expect(signatureRequests).toBeDefined();
+        expect(signatureRequests.length).toBe(1);
+        expect(signatureRequests[0].txs.length).toBe(1);
+        const transaction = signatureRequests[0].txs[0];
+        expect(transaction.chain).toBe("sol");
+        expect(transaction.signer).toBe(testWallets[0].address);
+        expect(transaction.encodedTx).toBeDefined();
+        // Verify the transaction has an ID
+        expect(transaction.id).toBeDefined();
+        expect(typeof transaction.id).toBe("string");
+        expect(transaction.id.length).toBeGreaterThan(10);
+        // Save the transaction ID for later use
+        const txId = transaction.id;
+        console.log(`[CLIENT] Transaction ID: ${txId}`);
+        // Test getTransactionById method
+        const foundTx = clientCryptoPlugin.getTransactionById(result, txId);
+        expect(foundTx).toBeDefined();
+        expect(foundTx?.id).toBe(txId);
+    });
+    it("should handle Ethereum transactions with unique IDs", async () => {
+        // Prepare the tool parameters without wallet information
+        const params = {
+            amount: 0.25,
+            chain: "eth",
+            recipientAddress: "0xRecipientEthereumAddress0123456789abcdef"
+            // No wallets provided - they should come from the client plugin
+        };
+        // Call the tool
+        const result = await dainConnection.callTool("crypto-transfer", params);
+        // Verify the response
+        expect(result.data.success).toBe(true);
+        expect(result.data.message).toContain("Created unsigned eth transaction");
+        // Get transaction requests using the CLIENT plugin
+        const signatureRequests = clientCryptoPlugin.getRequestedCryptoSignatures(result);
+        const transaction = signatureRequests[0].txs[0];
+        expect(transaction.chain).toBe("eth");
+        expect(transaction.signer).toBe(testWallets[1].address);
+        expect(transaction.metadata?.amount).toBe(0.25);
+        expect(transaction.metadata?.recipient).toBe("0xRecipientEthereumAddress0123456789abcdef");
+        // Verify the transaction has an ID
+        expect(transaction.id).toBeDefined();
+        expect(typeof transaction.id).toBe("string");
+        console.log(`[CLIENT] Ethereum transaction ID: ${transaction.id}`);
+    });
+    it("should show wallet information flow and transaction IDs", async () => {
+        // Add a new custom wallet to the client plugin
+        clientCryptoPlugin.addWallet({
+            chain: "sol",
+            address: "CustomSolAddress123456789"
+        });
+        // Call the tool with the updated client plugin
+        const result = await dainConnection.callTool("crypto-transfer", {
+            amount: 0.75,
+            chain: "sol",
+            recipientAddress: "NewRecipientAddress"
+        });
+        // Verify the result uses the updated wallet
+        const signatureRequests = clientCryptoPlugin.getRequestedCryptoSignatures(result);
+        const transaction = signatureRequests[0].txs[0];
+        // The address should be the new custom one we just added
+        expect(transaction.signer).toBe("CustomSolAddress123456789");
+        expect(transaction.metadata?.amount).toBe(0.75);
+        // Verify the transaction has an ID
+        expect(transaction.id).toBeDefined();
+        console.log(`[CLIENT] Updated wallet transaction ID: ${transaction.id}`);
+    });
+    it("should demonstrate creating and accessing transactions with IDs", async () => {
+        // Create a transaction directly with the createTransaction method
+        const txData = {
+            encodedTx: "0xManuallyCreatedTx",
+            chain: "eth",
+            signer: "0xManualAddress",
+            metadata: { test: true, purpose: "Demo" }
+        };
+        // Create a transaction with the client plugin
+        const tx1 = clientCryptoPlugin.createTransaction(txData);
+        // Create another transaction with the same data
+        const tx2 = clientCryptoPlugin.createTransaction(txData);
+        // Create a transaction with a predefined ID
+        const customId = (0, cuid2_1.createId)();
+        const tx3 = clientCryptoPlugin.createTransaction({
+            ...txData,
+            id: customId
+        });
+        // Verify each transaction has a unique ID
+        expect(tx1.id).toBeDefined();
+        expect(tx2.id).toBeDefined();
+        expect(tx3.id).toBeDefined();
+        // Transaction IDs should be unique
+        expect(tx1.id).not.toBe(tx2.id);
+        // Custom ID should be preserved
+        expect(tx3.id).toBe(customId);
+        console.log(`[TEST] Manual transaction IDs:`, {
+            tx1Id: tx1.id,
+            tx2Id: tx2.id,
+            tx3Id: tx3.id
+        });
+        // Create a signature request with multiple transactions
+        const signatureRequest = clientCryptoPlugin.createTransactionRequest([tx1, tx2, tx3]);
+        // Verify the signature request contains all transactions
+        expect(signatureRequest.txs.length).toBe(3);
+        expect(signatureRequest.txs.map(tx => tx.id)).toContain(tx1.id);
+        expect(signatureRequest.txs.map(tx => tx.id)).toContain(tx2.id);
+        expect(signatureRequest.txs.map(tx => tx.id)).toContain(tx3.id);
+    });
+    it("should allow service plugin to process transactions with additional data", async () => {
+        // Create a new test with a local function that adds metadata
+        // We'll use a real function instead of adding to the plugin instance
+        const addServiceMetadata = (tx) => {
+            return {
+                ...tx,
+                metadata: {
+                    ...tx.metadata,
+                    timestamp: Date.now(),
+                    serviceId: "test-service-id",
+                    enhanced: true
+                }
+            };
+        };
+        // Create a simple tool handler that would use this function
+        // This simulates what a service would do
+        const processTransaction = async (chain, userWallet) => {
+            // Create a basic transaction with ID
+            const baseTx = {
+                encodedTx: "0xBasicTransaction",
+                chain: chain,
+                signer: userWallet,
+                metadata: { basic: true }
+            };
+            // Create transaction with ID using the helper
+            const tx = serviceCryptoPlugin.createTransaction(baseTx);
+            // Enhance it with service-side metadata
+            const enhancedTx = addServiceMetadata(tx);
+            // Create the signature request
+            const request = {
+                txs: [enhancedTx]
+            };
+            return {
+                success: true,
+                signatureRequests: [request]
+            };
+        };
+        // Simulate processing with a simple test
+        const testWallet = "0xTestWalletAddress";
+        const result = await processTransaction("eth", testWallet);
+        // Verify the metadata was added
+        expect(result.signatureRequests[0].txs[0].metadata?.enhanced).toBe(true);
+        expect(result.signatureRequests[0].txs[0].metadata?.serviceId).toBe("test-service-id");
+        expect(result.signatureRequests[0].txs[0].metadata?.timestamp).toBeDefined();
+        expect(result.signatureRequests[0].txs[0].metadata?.basic).toBe(true); // Original data preserved
+        // Make sure the transaction info is correct
+        expect(result.signatureRequests[0].txs[0].chain).toBe("eth");
+        expect(result.signatureRequests[0].txs[0].signer).toBe(testWallet);
+        // Verify the transaction has an ID
+        expect(result.signatureRequests[0].txs[0].id).toBeDefined();
+        console.log(`[TEST] Service processed transaction ID: ${result.signatureRequests[0].txs[0].id}`);
+    });
+});
+//# sourceMappingURL=crypto-plugin.test.js.map

package/dist/__tests__/crypto-plugin.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"crypto-plugin.test.js","sourceRoot":"","sources":["../../src/__tests__/crypto-plugin.test.ts"],"names":[],"mappings":";AAAA,4CAA4C;;;AAE5C,sDAAsD;AACtD,4CAAwD;AACxD,mDAAgD;AAChD,wDAAwB;AACxB,uDAA0D;AAC1D,yCAA4C;AAC5C,6BAAwB;AACxB,2DAAqH;AACrH,gDAAgD;AAEhD,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,qBAAqB;IACrB,MAAM,iBAAiB,GAAG,iBAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC;IAC3D,MAAM,gBAAgB,GAAG,iBAAO,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC;IACjE,MAAM,cAAc,GAAG,cAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;IACrD,MAAM,gBAAgB,GAAG,iBAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC;IAE1D,+BAA+B;IAC/B,MAAM,SAAS,GAAG,IAAI,4BAAc,CAAC;QACnC,gBAAgB,EAAE,cAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;QAC/C,OAAO,EAAE,mBAAmB;QAC5B,KAAK,EAAE,iBAAiB;KACzB,CAAC,CAAC;IAEH,qCAAqC;IACrC,MAAM,WAAW,GAAiB;QAChC,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,8CAA8C,EAAE;QACzE,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,4CAA4C,EAAE;QACvE,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,4CAA4C,EAAE;KACxE,CAAC;IAEF,gEAAgE;IAChE,MAAM,kBAAkB,GAAG,IAAI,4BAAY,CAAC,WAAW,CAAC,CAAC;IACzD,MAAM,mBAAmB,GAAG,IAAI,4BAAY,EAAE,CAAC,CAAC,0CAA0C;IAE1F,8EAA8E;IAC9E,MAAM,cAAc,GAAG,IAAI,8BAAqB,CAC9C,uBAAuB,EACvB,SAAS,EACT;QACE,OAAO,EAAE,CAAC,kBAAkB,CAAC;KAC9B,CACF,CAAC;IAEF,oDAAoD;IACpD,MAAM,qBAAqB,GAAoC;QAC7D,SAAS,EAAE,0FAA0F;QACrG,KAAK,EAAE,KAAK;QACZ,MAAM,EAAE,8CAA8C;QACtD,QAAQ,EAAE;YACR,WAAW,EAAE,mBAAmB;YAChC,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,+CAA+C;SAC3D;KACF,CAAC;IAEF,sDAAsD;IACtD,MAAM,uBAAuB,GAAoC;QAC/D,SAAS,EAAE,gSAAgS;QAC3S,KAAK,EAAE,KAAK;QACZ,MAAM,EAAE,4CAA4C;QACpD,QAAQ,EAAE;YACR,WAAW,EAAE,mBAAmB;YAChC,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,4CAA4C;YACvD,QAAQ,EAAE,KAAK;SAChB;KACF,CAAC;IAEF,oFAAoF;IACpF,MAAM,kBAAkB,GAAG,IAAA,iBAAU,EAAC;QACpC,EAAE,EAAE,iBAAiB;QACrB,IAAI,EAAE,8BAA8B;QACpC,WAAW,EAAE,4DAA4D;QACzE,KAAK,EAAE,OAAC,CAAC,MAAM,CAAC;YACd,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;YAC7B,KAAK,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;YACpC,gBAAgB,EAAE,OAAC,CAAC,MAAM,EAAE;SAC7B,CAAC;QACF,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC;YACf,OAAO,EAAE,OAAC,CAAC,OAAO,EAAE;YACpB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;YACnB,iBAAiB,EAAE,OAAC,CAAC,KAAK,CACxB,OAAC,CAAC,MAAM,CAAC;gBACP,GAAG,EAAE,OAAC,CAAC,KAAK,CACV,OAAC,CAAC,MAAM,CAAC;oBACP,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE;oBACd,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE;oBACrB,KAAK,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;oBACpC,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE;oBAClB,QAAQ,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;iBAC7B,CAAC,CACH;aACF,CAAC,CACH,CAAC,QAAQ,EAAE;SACb,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;YACzE,qFAAqF;YACrF,MAAM,OAAO,GAAG,mBAAmB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAC5D,OAAO,CAAC,GAAG,CAAC,mCAAmC,EAAE,OAAO,CAAC,CAAC;YAE1D,yCAAyC;YACzC,MAAM,UAAU,GAAG,mBAAmB,CAAC,qBAAqB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAE7E,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO;oBACL,IAAI,EAAE,uBAAuB,KAAK,QAAQ;oBAC1C,IAAI,EAAE;wBACJ,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,MAAM,KAAK,iDAAiD,KAAK,UAAU;qBACrF;oBACD,EAAE,EAAE,IAAI;iBACT,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,mBAAmB,KAAK,YAAY,UAAU,oBAAoB,MAAM,OAAO,gBAAgB,EAAE,CAAC,CAAC;YAE/G,8CAA8C;YAC9C,IAAI,eAAgD,CAAC;YAErD,IAAI,KAAK,KAAK,KAAK,EAAE,CAAC;gBACpB,eAAe,GAAG;oBAChB,GAAG,qBAAqB;oBACxB,MAAM,EAAE,UAAU;oBAClB,QAAQ,EAAE;wBACR,GAAG,qBAAqB,CAAC,QAAQ;wBACjC,MAAM,EAAE,MAAM;wBACd,SAAS,EAAE,gBAAgB;qBAC5B;iBACF,CAAC;YACJ,CAAC;iBAAM,IAAI,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,KAAK,EAAE,CAAC;gBAC9C,eAAe,GAAG;oBAChB,GAAG,uBAAuB;oBAC1B,KAAK,EAAE,KAAK;oBACZ,MAAM,EAAE,UAAU;oBAClB,QAAQ,EAAE;wBACR,GAAG,uBAAuB,CAAC,QAAQ;wBACnC,MAAM,EAAE,MAAM;wBACd,SAAS,EAAE,gBAAgB;qBAC5B;iBACF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO;oBACL,IAAI,EAAE,sBAAsB,KAAK,EAAE;oBACnC,IAAI,EAAE;wBACJ,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,SAAS,KAAK,wBAAwB;qBAChD;oBACD,EAAE,EAAE,IAAI;iBACT,CAAC;YACJ,CAAC;YAED,wEAAwE;YACxE,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;YAEjF,uCAAuC;YACvC,OAAO,CAAC,GAAG,CAAC,0CAA0C,iBAAiB,CAAC,EAAE,EAAE,CAAC,CAAC;YAE9E,kEAAkE;YAClE,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,wBAAwB,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAE3F,OAAO;gBACL,IAAI,EAAE,oBAAoB,KAAK,oBAAoB,MAAM,OAAO,gBAAgB,EAAE;gBAClF,IAAI,EAAE;oBACJ,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,oBAAoB,KAAK,qDAAqD;oBACvF,iBAAiB,EAAE,CAAC,gBAAgB,CAAC;iBACtC;gBACD,EAAE,EAAE,IAAI;aACT,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,sEAAsE;IACtE,MAAM,WAAW,GAAG,IAAA,+BAAiB,EAAC;QACpC,QAAQ,EAAE;YACR,KAAK,EAAE,4BAA4B;YACnC,WAAW,EAAE,8CAA8C;YAC3D,OAAO,EAAE,OAAO;YAChB,MAAM,EAAE,aAAa;YACrB,IAAI,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC;SACnC;QACD,QAAQ,EAAE;YACR,SAAS,EAAE,cAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;YACxC,OAAO,EAAE,cAAc;YACvB,KAAK,EAAE,YAAY;YACnB,UAAU,EAAE,cAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC;SAC3C;QACD,KAAK,EAAE,CAAC,kBAAkB,CAAC;QAC3B,OAAO,EAAE,CAAC,mBAAmB,CAAC,CAAC,qCAAqC;KACrE,CAAC,CAAC;IAEH,IAAI,MAAW,CAAC;IAEhB,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,GAAG,MAAM,WAAW,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,4BAA4B;IACtF,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,MAAM,CAAC,QAAQ,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,OAAO,CAAC,GAAG,CAAC,4CAA4C,EAAE,kBAAkB,CAAC,UAAU,EAAE,CAAC,CAAC;QAC3F,OAAO,CAAC,GAAG,CAAC,8CAA8C,EAAE,mBAAmB,CAAC,UAAU,EAAE,CAAC,CAAC;QAE9F,mEAAmE;QACnE,yDAAyD;QACzD,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,KAAK;YACZ,gBAAgB,EAAE,+CAA+C;YACjE,gEAAgE;SACjE,CAAC;QAEF,gBAAgB;QAChB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAExE,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAE5E,sBAAsB;QACtB,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;QAE1E,+DAA+D;QAC/D,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;QAElF,OAAO,CAAC,GAAG,CAAC,wCAAwC,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAElG,MAAM,CAAC,iBAAiB,CAAC,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEhD,MAAM,WAAW,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QAE5C,mCAAmC;QACnC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,WAAW,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAElD,wCAAwC;QACxC,MAAM,IAAI,GAAG,WAAW,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;QAEhD,iCAAiC;QACjC,MAAM,OAAO,GAAG,kBAAkB,CAAC,kBAAkB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACpE,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,yDAAyD;QACzD,MAAM,MAAM,GAAG;YACb,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,KAAK;YACZ,gBAAgB,EAAE,4CAA4C;YAC9D,gEAAgE;SACjE,CAAC;QAEF,gBAAgB;QAChB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAExE,sBAAsB;QACtB,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;QAE1E,mDAAmD;QACnD,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;QAElF,MAAM,WAAW,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;QAE3F,mCAAmC;QACnC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,WAAW,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,qCAAqC,WAAW,CAAC,EAAE,EAAE,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,+CAA+C;QAC/C,kBAAkB,CAAC,SAAS,CAAC;YAC3B,KAAK,EAAE,KAAK;YACZ,OAAO,EAAE,2BAA2B;SACrC,CAAC,CAAC;QAEH,+CAA+C;QAC/C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,iBAAiB,EAAE;YAC9D,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,KAAK;YACZ,gBAAgB,EAAE,qBAAqB;SACxC,CAAC,CAAC;QAEH,4CAA4C;QAC5C,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;QAClF,MAAM,WAAW,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAEhD,yDAAyD;QACzD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAC7D,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhD,mCAAmC;QACnC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACrC,OAAO,CAAC,GAAG,CAAC,2CAA2C,WAAW,CAAC,EAAE,EAAE,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,kEAAkE;QAClE,MAAM,MAAM,GAAoC;YAC9C,SAAS,EAAE,qBAAqB;YAChC,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,iBAAiB;YACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;SAC1C,CAAC;QAEF,8CAA8C;QAC9C,MAAM,GAAG,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAEzD,gDAAgD;QAChD,MAAM,GAAG,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAEzD,4CAA4C;QAC5C,MAAM,QAAQ,GAAG,IAAA,gBAAQ,GAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,kBAAkB,CAAC,iBAAiB,CAAC;YAC/C,GAAG,MAAM;YACT,EAAE,EAAE,QAAQ;SACb,CAAC,CAAC;QAEH,0CAA0C;QAC1C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAE7B,mCAAmC;QACnC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEhC,gCAAgC;QAChC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE9B,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE;YAC5C,KAAK,EAAE,GAAG,CAAC,EAAE;YACb,KAAK,EAAE,GAAG,CAAC,EAAE;YACb,KAAK,EAAE,GAAG,CAAC,EAAE;SACd,CAAC,CAAC;QAEH,wDAAwD;QACxD,MAAM,gBAAgB,GAAG,kBAAkB,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAEtF,yDAAyD;QACzD,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAChE,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAChE,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,6DAA6D;QAC7D,qEAAqE;QACrE,MAAM,kBAAkB,GAAG,CAAC,EAAuB,EAAuB,EAAE;YAC1E,OAAO;gBACL,GAAG,EAAE;gBACL,QAAQ,EAAE;oBACR,GAAG,EAAE,CAAC,QAAQ;oBACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,SAAS,EAAE,iBAAiB;oBAC5B,QAAQ,EAAE,IAAI;iBACf;aACF,CAAC;QACJ,CAAC,CAAC;QAEF,4DAA4D;QAC5D,yCAAyC;QACzC,MAAM,kBAAkB,GAAG,KAAK,EAC9B,KAAa,EACb,UAAkB,EAC6D,EAAE;YACjF,qCAAqC;YACrC,MAAM,MAAM,GAAoC;gBAC9C,SAAS,EAAE,oBAAoB;gBAC/B,KAAK,EAAE,KAAY;gBACnB,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;aAC1B,CAAC;YAEF,8CAA8C;YAC9C,MAAM,EAAE,GAAG,mBAAmB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAEzD,wCAAwC;YACxC,MAAM,UAAU,GAAG,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAE1C,+BAA+B;YAC/B,MAAM,OAAO,GAAG;gBACd,GAAG,EAAE,CAAC,UAAU,CAAC;aAClB,CAAC;YAEF,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,iBAAiB,EAAE,CAAC,OAAO,CAAC;aAC7B,CAAC;QACJ,CAAC,CAAC;QAEF,yCAAyC;QACzC,MAAM,UAAU,GAAG,qBAAqB,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAE3D,gCAAgC;QAChC,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzE,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACvF,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7E,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,0BAA0B;QAEjG,4CAA4C;QAC5C,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAEnE,mCAAmC;QACnC,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5D,OAAO,CAAC,GAAG,CAAC,4CAA4C,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACnG,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/integration.test.js

@@ -215,6 +215,42 @@ describe("DAIN Framework Integration", () => {
             ui: { type: "text" },
         }),
     });
+    // Create a dedicated error tool for testing
+    const errorTool = (0, nodeService_1.createTool)({
+        id: "integration-error-tool",
+        name: "Integration Error Tool",
+        description: "Tool that deliberately throws errors for testing",
+        input: zod_1.z.object({
+            errorType: zod_1.z.enum(["none", "synchronous", "async", "input"]),
+            errorMessage: zod_1.z.string().optional()
+        }),
+        output: zod_1.z.object({ result: zod_1.z.string() }),
+        handler: async ({ errorType, errorMessage = "Deliberate test error" }, agentInfo, context) => {
+            // Send UI update first to test partial success
+            if (context.updateUI) {
+                await context.updateUI({ ui: { type: "loading", message: "Starting error test..." } });
+                // Small delay to ensure UI update goes through
+                await new Promise(resolve => setTimeout(resolve, 200));
+            }
+            // Handle different error types
+            switch (errorType) {
+                case "none":
+                    return {
+                        text: "No error occurred",
+                        data: { result: "success" },
+                        ui: null
+                    };
+                case "synchronous":
+                    throw new Error(errorMessage);
+                case "async":
+                    await new Promise(resolve => setTimeout(resolve, 300));
+                    throw new Error(errorMessage);
+                case "input":
+                    // This simulates validation errors
+                    throw new Error(`Invalid input: ${errorMessage}`);
+            }
+        },
+    });
     const dainService = (0, nodeService_1.defineDAINService)({
         metadata: {
             title: "Weather DAIN Service",
@@ -230,7 +266,7 @@ describe("DAIN Framework Integration", () => {
             privateKey: bs58_1.default.encode(privateKey),
         },
         services: [weatherService],
-        tools: [weatherTool, extraDataTool, confirmationTool, optionalParamsTool, noParamsTool],
+        tools: [weatherTool, extraDataTool, confirmationTool, optionalParamsTool, noParamsTool, errorTool],
         toolboxes: [weatherToolbox],
         contexts: [weatherContext],
         pinnables: [weatherPinnable],
@@ -241,7 +277,25 @@ describe("DAIN Framework Integration", () => {
         server = await dainService.startNode({ port: 3003 });
     });
     afterAll(async () => {
-        await server.shutdown();
+        // Use a more robust approach to shutting down with proper cleanup
+        if (server) {
+            // Set a timeout for shutdown to complete
+            const timeout = setTimeout(() => {
+                console.warn('Server shutdown timed out, process may still have open handles');
+            }, 5000);
+            try {
+                await server.shutdown();
+            }
+            catch (err) {
+                console.error('Error shutting down server:', err);
+            }
+            finally {
+                clearTimeout(timeout);
+                server = null;
+                // Wait a moment to allow any lingering connections to close
+                await new Promise(resolve => setTimeout(resolve, 500));
+            }
+        }
     });
     test("Service metadata is correct", () => {
         const metadata = dainService.getMetadata();
@@ -249,7 +303,7 @@ describe("DAIN Framework Integration", () => {
         expect(metadata.tags).toContain("weather");
     });
     test("Service has correct tools and services", () => {
-        expect(dainService.getTools()).toHaveLength(5);
+        expect(dainService.getTools()).toHaveLength(6);
         expect(dainService.getServices()).toHaveLength(1);
         expect(dainService.getToolboxes()).toHaveLength(1);
     });
@@ -362,6 +416,7 @@ describe("DAIN Framework Integration", () => {
         const port = app.port;
         const response = await dainConnection.makeRequest("GET", "/getAllToolsAsJsonSchema", {});
         console.log("JSON SCHEMA", JSON.stringify(response, null, 2));
+        // fs.writeFileSync("jsonSchema.json", JSON.stringify(response, null, 2));
         expect(response).toEqual({
             "tools": [
                 {
@@ -460,16 +515,13 @@ describe("DAIN Framework Integration", () => {
                         "type": "object",
                         "properties": {
                             "param1": {
-                                "type": "string",
-                                "optional": true
+                                "type": "string"
                             },
                             "param2": {
-                                "type": "number",
-                                "optional": true
+                                "type": "number"
                             },
                             "param3": {
-                                "type": "boolean",
-                                "optional": true
+                                "type": "boolean"
                             }
                         },
                         "additionalProperties": false,
@@ -501,6 +553,9 @@ describe("DAIN Framework Integration", () => {
                     "name": "No Parameters Tool",
                     "description": "Tool that requires no parameters",
                     "inputSchema": {
+                        "type": "object",
+                        "properties": {},
+                        "additionalProperties": false,
                         "$schema": "http://json-schema.org/draft-07/schema#"
                     },
                     "outputSchema": {
@@ -516,6 +571,46 @@ describe("DAIN Framework Integration", () => {
                         "additionalProperties": false,
                         "$schema": "http://json-schema.org/draft-07/schema#"
                     }
+                },
+                {
+                    "id": "integration-error-tool",
+                    "name": "Integration Error Tool",
+                    "description": "Tool that deliberately throws errors for testing",
+                    "inputSchema": {
+                        "type": "object",
+                        "properties": {
+                            "errorType": {
+                                "type": "string",
+                                "enum": [
+                                    "none",
+                                    "synchronous",
+                                    "async",
+                                    "input"
+                                ]
+                            },
+                            "errorMessage": {
+                                "type": "string"
+                            }
+                        },
+                        "required": [
+                            "errorType"
+                        ],
+                        "additionalProperties": false,
+                        "$schema": "http://json-schema.org/draft-07/schema#"
+                    },
+                    "outputSchema": {
+                        "type": "object",
+                        "properties": {
+                            "result": {
+                                "type": "string"
+                            }
+                        },
+                        "required": [
+                            "result"
+                        ],
+                        "additionalProperties": false,
+                        "$schema": "http://json-schema.org/draft-07/schema#"
+                    }
                 }
             ],
             "reccomendedPrompts": [
@@ -828,5 +923,96 @@ describe("DAIN Framework Integration", () => {
             expect(result.data.message).toBe("Success - no parameters needed");
         });
     });
+    // Add a new test suite for error handling
+    describe("Error Handling", () => {
+        it("Handles synchronous errors in non-streaming mode", async () => {
+            const errorResult = await dainConnection.callTool("integration-error-tool", {
+                errorType: "synchronous",
+                errorMessage: "Test sync error"
+            });
+            // Verify the error format matches our expected structure
+            expect(errorResult).toHaveProperty('error');
+            expect(errorResult).toHaveProperty('text');
+            expect(errorResult.error).toBe("Test sync error");
+            expect(errorResult.text).toBe("Error: Test sync error");
+            expect(errorResult.data).toBeNull();
+            expect(errorResult.ui).toBeNull();
+        });
+        it("Handles asynchronous errors in non-streaming mode", async () => {
+            const errorResult = await dainConnection.callTool("integration-error-tool", {
+                errorType: "async",
+                errorMessage: "Test async error"
+            });
+            expect(errorResult).toHaveProperty('error');
+            expect(errorResult.error).toBe("Test async error");
+            expect(errorResult.data).toBeNull();
+        });
+        it("Formats errors correctly in context endpoint", async () => {
+            const errorResult = await dainConnection.callToolAndGetNewContext("integration-error-tool", {
+                errorType: "synchronous",
+                errorMessage: "Test context error"
+            });
+            // Verify the context endpoint error format
+            expect(errorResult).toHaveProperty('toolResult');
+            expect(errorResult).toHaveProperty('context');
+            expect(errorResult.toolResult).toHaveProperty('error');
+            expect(errorResult.toolResult.error).toBe("Test context error");
+            expect(errorResult.toolResult.text).toBe("Error: Test context error");
+            expect(errorResult.toolResult.data).toBeNull();
+            expect(errorResult.toolResult.ui).toBeNull();
+            expect(Array.isArray(errorResult.context)).toBe(true);
+            expect(errorResult.context.length).toBe(0);
+        });
+        it("Handles errors in streaming mode and sends error events", async () => {
+            const events = [];
+            const errorResult = await dainConnection.callTool("integration-error-tool", {
+                errorType: "async",
+                errorMessage: "Test streaming error"
+            }, {
+                onUIUpdate: (update) => {
+                    events.push({ type: 'ui', data: update.ui });
+                }
+            });
+            // Verify we received UI updates before the error
+            expect(events.length).toBeGreaterThan(0);
+            expect(events[0].type).toBe('ui');
+            // Verify the error format in streaming mode
+            expect(errorResult).toHaveProperty('error');
+            expect(errorResult.error).toBe("Test streaming error");
+            expect(errorResult.text).toBe("Error: Test streaming error");
+            expect(errorResult.data).toBeNull();
+            expect(errorResult.ui).toBeNull();
+        });
+        it("Handles streaming errors in context endpoint", async () => {
+            const events = [];
+            const errorResult = await dainConnection.callToolAndGetNewContext("integration-error-tool", {
+                errorType: "async",
+                errorMessage: "Test streaming context error"
+            }, {
+                onUIUpdate: (update) => {
+                    events.push({ type: 'ui', data: update.ui });
+                }
+            });
+            // Verify we received UI updates before the error
+            expect(events.length).toBeGreaterThan(0);
+            // Verify the context endpoint error format in streaming mode
+            expect(errorResult).toHaveProperty('toolResult');
+            expect(errorResult).toHaveProperty('context');
+            expect(errorResult.toolResult).toHaveProperty('error');
+            expect(errorResult.toolResult.error).toBe("Test streaming context error");
+            expect(errorResult.toolResult.data).toBeNull();
+            expect(Array.isArray(errorResult.context)).toBe(true);
+        });
+        it("Verifies successful execution still works after error handling improvements", async () => {
+            const result = await dainConnection.callTool("integration-error-tool", {
+                errorType: "none"
+            });
+            expect(result).toEqual({
+                text: "No error occurred",
+                data: { result: "success" },
+                ui: null
+            });
+        });
+    });
 });
 //# sourceMappingURL=integration.test.js.map