@eko-ai/eko 1.0.9 → 1.0.11

package/dist/test/integration/workflow.generator.test.js

@@ -0,0 +1,207 @@
+// test/integration/workflow.generator.test.ts
+import { ClaudeProvider } from '../../src/services/llm/claude-provider';
+import { WorkflowGenerator } from '../../src/services/workflow/generator';
+import { ToolRegistry } from '../../src/core/tool-registry';
+import { WorkflowParser } from '../../src/services/parser/workflow-parser';
+import * as fs from 'fs/promises';
+import * as path from 'path';
+import dotenv from 'dotenv';
+dotenv.config();
+// Mock browser tool base class to avoid duplicate code
+class BrowserTool {
+    constructor(name, description, input_schema) {
+        this.name = name;
+        this.description = description;
+        this.input_schema = input_schema;
+    }
+    async execute(params) {
+        throw new Error('Not implemented');
+    }
+}
+// Create mock browser tools
+function createBrowserTools() {
+    return [
+        new BrowserTool('open_url', 'Opens a specified URL in the current browser tab', {
+            type: 'object',
+            properties: {
+                url: {
+                    type: 'string',
+                    description: 'The URL to open'
+                }
+            },
+            required: ['url']
+        }),
+        new BrowserTool('find_dom_object', 'Finds a DOM element using CSS selector, returns multiple elements if found', {
+            type: 'object',
+            properties: {
+                selector: {
+                    type: 'string',
+                    description: 'CSS selector to find elements'
+                },
+                waitForElement: {
+                    type: 'boolean',
+                    description: 'Whether to wait for elements to appear'
+                },
+                timeout: {
+                    type: 'integer',
+                    description: 'Maximum time to wait in milliseconds'
+                }
+            },
+            required: ['selector']
+        }),
+        new BrowserTool('click_dom_object', 'Clicks on a DOM element found by CSS selector', {
+            type: 'object',
+            properties: {
+                selector: {
+                    type: 'string',
+                    description: 'CSS selector of element to click'
+                }
+            },
+            required: ['selector']
+        }),
+        new BrowserTool('input_text', 'Types text into a form field', {
+            type: 'object',
+            properties: {
+                selector: {
+                    type: 'string',
+                    description: 'CSS selector of input element'
+                },
+                text: {
+                    type: 'string',
+                    description: 'Text to type'
+                },
+                clear: {
+                    type: 'boolean',
+                    description: 'Whether to clear existing text first'
+                }
+            },
+            required: ['selector', 'text']
+        }),
+        new BrowserTool('copy_dom_object_text', 'Extracts text content from DOM elements matching a selector', {
+            type: 'object',
+            properties: {
+                selector: {
+                    type: 'string',
+                    description: 'CSS selector of elements to copy text from'
+                }
+            },
+            required: ['selector']
+        }),
+        new BrowserTool('save_file', 'Saves content to a file', {
+            type: 'object',
+            properties: {
+                content: {
+                    type: 'string',
+                    description: 'Content to save'
+                },
+                filename: {
+                    type: 'string',
+                    description: 'Name of the file'
+                },
+                type: {
+                    type: 'string',
+                    description: 'File type',
+                    enum: ['text/plain', 'text/csv', 'text/html', 'application/json']
+                }
+            },
+            required: ['content', 'filename']
+        })
+    ];
+}
+const ENABLE_INTEGRATION_TESTS = process.env.ENABLE_INTEGRATION_TESTS === 'true';
+const describeIntegration = ENABLE_INTEGRATION_TESTS ? describe : describe.skip;
+describeIntegration('WorkflowGenerator Integration', () => {
+    let toolRegistry;
+    let generator;
+    // Helper function to save workflow DSL to file
+    async function saveWorkflowToFile(dsl, filename) {
+        const testOutputDir = path.join(__dirname, '../fixtures/generated');
+        await fs.mkdir(testOutputDir, { recursive: true });
+        await fs.writeFile(path.join(testOutputDir, filename), dsl, 'utf-8');
+    }
+    beforeAll(() => {
+        const ANTHROPIC_API_KEY = process.env.ANTHROPIC_API_KEY;
+        if (!ANTHROPIC_API_KEY) {
+            throw new Error('ANTHROPIC_API_KEY environment variable is required for integration tests');
+        }
+        // Set up registry with browser tools
+        toolRegistry = new ToolRegistry();
+        createBrowserTools().forEach(tool => toolRegistry.registerTool(tool));
+        // Create generator with Claude provider
+        const llmProvider = new ClaudeProvider(ANTHROPIC_API_KEY);
+        generator = new WorkflowGenerator(llmProvider, toolRegistry);
+    });
+    it('should generate workflow for finding Chromium developers', async () => {
+        const prompt = "Find Chromium developers from Github, collect the profiles, and summarize the results to CSV";
+        // Generate workflow
+        const workflow = await generator.generateWorkflow(prompt);
+        // Convert to DSL for validation and inspection
+        const dsl = WorkflowParser.serialize(workflow);
+        // Save DSL for human inspection
+        await saveWorkflowToFile(dsl, 'github_chromium_workflow.json');
+        // Validate the generated workflow
+        const validationResult = WorkflowParser.validate(JSON.parse(dsl));
+        // Log validation errors if any (helpful for debugging)
+        if (!validationResult.valid) {
+            console.error('Validation errors:', JSON.stringify(validationResult.errors, null, 2));
+        }
+        // Assert validation
+        expect(validationResult.valid).toBe(true);
+        expect(validationResult.errors).toHaveLength(0);
+        // Basic structure checks
+        expect(workflow.id).toBeDefined();
+        expect(workflow.name).toBeDefined();
+        expect(workflow.nodes).toBeDefined();
+        expect(workflow.nodes.length).toBeGreaterThan(0);
+        // Log workflow structure for inspection
+        console.log('\nGenerated Workflow Structure:');
+        console.log('ID:', workflow.id);
+        console.log('Name:', workflow.name);
+        console.log('Number of nodes:', workflow.nodes.length);
+        console.log('Nodes:', workflow.nodes.map(n => ({
+            id: n.id,
+            name: n.name,
+            dependencies: n.dependencies,
+            action: {
+                type: n.action.type,
+                tools: n.action.tools.map(t => t.name)
+            }
+        })));
+        // Verify tool usage
+        const usedTools = new Set();
+        workflow.nodes.forEach(node => {
+            node.action.tools.forEach(tool => {
+                usedTools.add(tool.name);
+            });
+        });
+        console.log('\nTools used:', Array.from(usedTools));
+        // Expected tools for this workflow
+        const expected_tools = new Set([
+            'open_url', // For navigating to Github
+            'input_text', // For entering search terms
+            'click_dom_object', // For interaction
+            'find_dom_object', // For finding profile elements
+            'copy_dom_object_text', // For extracting profile data
+            'save_file', // For saving the CSV
+            'write_context' // For storing data in context
+        ]);
+        // Verify reasonable tool usage
+        expect(usedTools.size).toBeGreaterThanOrEqual(3);
+        usedTools.forEach(tool => {
+            expect(expected_tools).toContain(tool);
+        });
+        // Verify workflow has proper node dependencies
+        expect(workflow.validateDAG()).toBe(true);
+        // The last node should use save_file tool to create CSV
+        const lastNode = workflow.nodes[workflow.nodes.length - 1];
+        expect(lastNode.action.tools.some(t => t.name === 'save_file')).toBe(true);
+        // First node should have no dependencies
+        const firstNode = workflow.nodes[0];
+        expect(firstNode.dependencies.length).toBe(0);
+        // Other nodes should have dependencies
+        workflow.nodes.slice(1).forEach(node => {
+            expect(node.dependencies.length).toBeGreaterThan(0);
+        });
+    }, 30000); // Increased timeout for LLM
+});
+//# sourceMappingURL=workflow.generator.test.js.map

package/dist/test/integration/workflow.generator.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow.generator.test.js","sourceRoot":"","sources":["../../../test/integration/workflow.generator.test.ts"],"names":[],"mappings":"AAAA,8CAA8C;AAE9C,OAAO,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAG5D,OAAO,EAAE,cAAc,EAAE,MAAM,2CAA2C,CAAC;AAC3E,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,MAAM,CAAC,MAAM,EAAE,CAAC;AAEhB,uDAAuD;AACvD,MAAM,WAAW;IACf,YACS,IAAY,EACZ,WAAmB,EACnB,YAAyB;QAFzB,SAAI,GAAJ,IAAI,CAAQ;QACZ,gBAAW,GAAX,WAAW,CAAQ;QACnB,iBAAY,GAAZ,YAAY,CAAa;IAC/B,CAAC;IAEJ,KAAK,CAAC,OAAO,CAAC,MAAe;QAC3B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,4BAA4B;AAC5B,SAAS,kBAAkB;IACzB,OAAO;QACL,IAAI,WAAW,CACb,UAAU,EACV,kDAAkD,EAClD;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iBAAiB;iBAC/B;aACF;YACD,QAAQ,EAAE,CAAC,KAAK,CAAC;SAClB,CACF;QACD,IAAI,WAAW,CACb,iBAAiB,EACjB,4EAA4E,EAC5E;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,+BAA+B;iBAC7C;gBACD,cAAc,EAAE;oBACd,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,wCAAwC;iBACtD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,sCAAsC;iBACpD;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,CAAC;SACvB,CACF;QACD,IAAI,WAAW,CACb,kBAAkB,EAClB,+CAA+C,EAC/C;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kCAAkC;iBAChD;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,CAAC;SACvB,CACF;QACD,IAAI,WAAW,CACb,YAAY,EACZ,8BAA8B,EAC9B;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,+BAA+B;iBAC7C;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,cAAc;iBAC5B;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,sCAAsC;iBACpD;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,EAAE,MAAM,CAAC;SAC/B,CACF;QACD,IAAI,WAAW,CACb,sBAAsB,EACtB,6DAA6D,EAC7D;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4CAA4C;iBAC1D;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,CAAC;SACvB,CACF;QACD,IAAI,WAAW,CACb,WAAW,EACX,yBAAyB,EACzB;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iBAAiB;iBAC/B;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kBAAkB;iBAChC;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,WAAW;oBACxB,IAAI,EAAE,CAAC,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,kBAAkB,CAAC;iBAClE;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC;SAClC,CACF;KACF,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,KAAK,MAAM,CAAC;AACjF,MAAM,mBAAmB,GAAG,wBAAwB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAEhF,mBAAmB,CAAC,+BAA+B,EAAE,GAAG,EAAE;IACxD,IAAI,YAA0B,CAAC;IAC/B,IAAI,SAA4B,CAAC;IAEjC,+CAA+C;IAC/C,KAAK,UAAU,kBAAkB,CAAC,GAAW,EAAE,QAAgB;QAC7D,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,uBAAuB,CAAC,CAAC;QACpE,MAAM,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IACvE,CAAC;IAED,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QACxD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,0EAA0E,CAAC,CAAC;QAC9F,CAAC;QAED,qCAAqC;QACrC,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,kBAAkB,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;QAEtE,wCAAwC;QACxC,MAAM,WAAW,GAAG,IAAI,cAAc,CAAC,iBAAiB,CAAC,CAAC;QAC1D,SAAS,GAAG,IAAI,iBAAiB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,MAAM,GAAG,8FAA8F,CAAC;QAE9G,oBAAoB;QACpB,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAE1D,+CAA+C;QAC/C,MAAM,GAAG,GAAG,cAAc,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAE/C,gCAAgC;QAChC,MAAM,kBAAkB,CAAC,GAAG,EAAE,+BAA+B,CAAC,CAAC;QAE/D,kCAAkC;QAClC,MAAM,gBAAgB,GAAqB,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;QAEpF,uDAAuD;QACvD,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;YAC5B,OAAO,CAAC,KAAK,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACxF,CAAC;QAED,oBAAoB;QACpB,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEhD,yBAAyB;QACzB,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAEjD,wCAAwC;QACxC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QAC/C,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC7C,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,YAAY,EAAE,CAAC,CAAC,YAAY;YAC5B,MAAM,EAAE;gBACN,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI;gBACnB,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aACvC;SACF,CAAC,CAAC,CAAC,CAAC;QAEL,oBAAoB;QACpB,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YAC5B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;gBAC/B,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QAEpD,mCAAmC;QACnC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC;YAC7B,UAAU,EAAU,2BAA2B;YAC/C,YAAY,EAAQ,4BAA4B;YAChD,kBAAkB,EAAE,kBAAkB;YACtC,iBAAiB,EAAG,+BAA+B;YACnD,sBAAsB,EAAE,8BAA8B;YACtD,WAAW,EAAU,qBAAqB;YAC1C,eAAe,CAAM,8BAA8B;SACpD,CAAC,CAAC;QAEH,+BAA+B;QAC/B,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QACjD,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YACvB,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,+CAA+C;QAC/C,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE1C,wDAAwD;QACxD,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC3D,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE3E,yCAAyC;QACzC,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE9C,uCAAuC;QACvC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YACrC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,4BAA4B;AACzC,CAAC,CAAC,CAAC"}

package/dist/test/unit/action.test.js

@@ -0,0 +1,186 @@
+import { ActionImpl } from '../../src/models/action';
+// Mock tool for testing
+class MockTool {
+    constructor(name, description = 'Mock tool for testing', shouldFail = false) {
+        this.name = name;
+        this.description = description;
+        this.shouldFail = shouldFail;
+        this.input_schema = {
+            type: 'object',
+            properties: {
+                testParam: { type: 'string' },
+            },
+            required: ['testParam'],
+        };
+    }
+    async execute(context, params) {
+        if (this.shouldFail) {
+            throw new Error('Tool execution failed');
+        }
+        return { success: true, params };
+    }
+}
+// Mock LLM provider
+class MockLLMProvider {
+    constructor(toolCallResponses = [], shouldFail = false, counter = 0) {
+        this.toolCallResponses = toolCallResponses;
+        this.shouldFail = shouldFail;
+        this.counter = counter;
+    }
+    async generateText() {
+        if (this.shouldFail) {
+            throw new Error('LLM generation failed');
+        }
+        return {
+            content: 'Test response',
+            toolCalls: this.toolCallResponses,
+        };
+    }
+    async generateStream(messages, params, handler) {
+        var _a, _b, _c, _d;
+        if (this.shouldFail) {
+            (_a = handler.onError) === null || _a === void 0 ? void 0 : _a.call(handler, new Error('Stream generation failed'));
+            return;
+        }
+        // Simulate thinking output
+        (_b = handler.onContent) === null || _b === void 0 ? void 0 : _b.call(handler, 'Thinking about the task...');
+        // Process each tool call
+        const toolCall = this.toolCallResponses[this.counter++];
+        (_c = handler.onToolUse) === null || _c === void 0 ? void 0 : _c.call(handler, {
+            id: `tool-${Math.random()}`,
+            name: toolCall.name,
+            input: toolCall.input,
+        });
+        // Final response
+        (_d = handler.onComplete) === null || _d === void 0 ? void 0 : _d.call(handler, {
+            content: [
+                { type: 'text', text: 'Thinking about the task...' },
+                {
+                    type: 'tool_use',
+                    id: `tool-${Math.random()}`,
+                    name: toolCall.name,
+                    input: toolCall.input
+                }
+            ],
+            toolCalls: [{
+                    id: `tool-${Math.random()}`,
+                    name: toolCall.name,
+                    input: toolCall.input
+                }],
+            stop_reason: 'tool_use',
+            textContent: null // No text content when using tools
+        });
+    }
+}
+describe('ActionImpl', () => {
+    let mockTool;
+    let mockLLMProvider;
+    let context;
+    beforeEach(() => {
+        mockTool = new MockTool('test_tool');
+        mockLLMProvider = new MockLLMProvider();
+        context = {
+            llmProvider: mockLLMProvider,
+            variables: new Map(),
+            tools: new Map(),
+        };
+    });
+    describe('constructor', () => {
+        it('should create an action with tools including write_context', () => {
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProvider);
+            expect(action.tools).toHaveLength(2); // Original tool + write_context
+            expect(action.tools.some((t) => t.name === 'write_context')).toBeTruthy();
+            expect(action.tools.some((t) => t.name === 'test_tool')).toBeTruthy();
+        });
+    });
+    describe('execute', () => {
+        it('should handle successful tool execution', async () => {
+            // Setup LLM to make a tool call
+            mockLLMProvider = new MockLLMProvider([
+                { name: 'test_tool', input: { testParam: 'test' } },
+                { name: 'return_output', input: { value: 'test return' } },
+            ]);
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProvider);
+            await action.execute('Test input', context);
+            // Tool was successful, no errors thrown
+        });
+        it('should handle tool execution failure', async () => {
+            // Setup failing tool
+            mockTool = new MockTool('test_tool', 'Mock tool', true);
+            mockLLMProvider = new MockLLMProvider([
+                { name: 'test_tool', input: { testParam: 'test' } },
+                { name: 'return_output', input: { value: 'test return' } },
+            ]);
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProvider);
+            await action.execute('Test input', context);
+            // Should handle tool failure gracefully, no error thrown
+        });
+        it('should handle LLM provider failure', async () => {
+            mockLLMProvider = new MockLLMProvider([], true);
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProvider);
+            await expect(action.execute('Test input', context)).resolves.toBeDefined();
+            // Should handle LLM failure gracefully
+        });
+        it('should properly use write_context tool', async () => {
+            // Setup LLM to make a write_context call
+            mockLLMProvider = new MockLLMProvider([
+                {
+                    name: 'write_context',
+                    input: { key: 'test_key', value: JSON.stringify({ data: 'test' }) },
+                },
+                { name: 'return_output', input: { value: 'test return' } },
+            ]);
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProvider);
+            await action.execute('Test input', context);
+            // Check if value was written to context
+            expect(context.variables.get('test_key')).toEqual({ data: 'test' });
+        });
+        it('should handle non-JSON values in write_context', async () => {
+            // Setup LLM to make a write_context call with string value
+            mockLLMProvider = new MockLLMProvider([
+                {
+                    name: 'write_context',
+                    input: { key: 'test_key', value: 'plain text value' },
+                },
+                { name: 'return_output', input: { value: 'test return' } },
+            ]);
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProvider);
+            await action.execute('Test input', context);
+            // Check if value was written to context as string
+            expect(context.variables.get('test_key')).toBe('plain text value');
+        });
+        it('should include context variables in user prompt', async () => {
+            // Setup context with some variables
+            context.variables.set('existingVar', 'test value');
+            // Create mock LLM provider that captures messages
+            const capturedMessages = [];
+            const mockLLMProviderWithCapture = new MockLLMProvider();
+            mockLLMProviderWithCapture.generateStream = async (messages, params, handler) => {
+                var _a;
+                capturedMessages.push(...messages);
+                // Continue with normal stream handling
+                await ((_a = handler.onContent) === null || _a === void 0 ? void 0 : _a.call(handler, 'Test content'));
+            };
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProviderWithCapture);
+            await action.execute('Test input', context);
+            // Verify system prompt includes context variables
+            const initialPrompt = capturedMessages[1].content;
+            expect(initialPrompt).toContain('existingVar');
+            expect(initialPrompt).toContain('test value');
+        });
+        it('should merge action tools with context tools', async () => {
+            var _a;
+            const contextTool = new MockTool('context_tool');
+            (_a = context.tools) === null || _a === void 0 ? void 0 : _a.set(contextTool.name, contextTool);
+            mockLLMProvider = new MockLLMProvider([
+                { name: 'context_tool', input: { testParam: 'test' } },
+                { name: 'test_tool', input: { testParam: 'test' } },
+                { name: 'return_output', input: { value: 'test return' } },
+            ]);
+            const action = ActionImpl.createPromptAction('test_action', 'This is an action for testing purposes', [mockTool], mockLLMProvider);
+            await action.execute('Test input', context);
+            // Both tools should have been accessible
+        });
+    });
+});
+//# sourceMappingURL=action.test.js.map

package/dist/test/unit/action.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"action.test.js","sourceRoot":"","sources":["../../../test/unit/action.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAIrD,wBAAwB;AACxB,MAAM,QAAQ;IACZ,YACS,IAAY,EACZ,cAAsB,uBAAuB,EAC7C,aAAsB,KAAK;QAF3B,SAAI,GAAJ,IAAI,CAAQ;QACZ,gBAAW,GAAX,WAAW,CAAkC;QAC7C,eAAU,GAAV,UAAU,CAAiB;QAGpC,iBAAY,GAAG;YACb,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC9B;YACD,QAAQ,EAAE,CAAC,WAAW,CAAC;SACT,CAAC;IARd,CAAC;IAUJ,KAAK,CAAC,OAAO,CAAC,OAAyB,EAAE,MAAe;QACtD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IACnC,CAAC;CACF;AAED,oBAAoB;AACpB,MAAM,eAAe;IACnB,YACU,oBAAyD,EAAE,EAC5D,aAAsB,KAAK,EAC3B,UAAkB,CAAC;QAFlB,sBAAiB,GAAjB,iBAAiB,CAA0C;QAC5D,eAAU,GAAV,UAAU,CAAiB;QAC3B,YAAO,GAAP,OAAO,CAAY;IACzB,CAAC;IAEJ,KAAK,CAAC,YAAY;QAChB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO;YACL,OAAO,EAAE,eAAe;YACxB,SAAS,EAAE,IAAI,CAAC,iBAAiB;SAClC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,QAAmB,EACnB,MAAqB,EACrB,OAAyB;;QAEzB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAA,OAAO,CAAC,OAAO,wDAAG,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC,CAAC;YACzD,OAAO;QACT,CAAC;QAED,2BAA2B;QAC3B,MAAA,OAAO,CAAC,SAAS,wDAAG,4BAA4B,CAAC,CAAC;QAElD,yBAAyB;QACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACxD,MAAA,OAAO,CAAC,SAAS,wDAAG;YAClB,EAAE,EAAE,QAAQ,IAAI,CAAC,MAAM,EAAE,EAAE;YAC3B,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC,CAAC;QAEH,iBAAiB;QACjB,MAAA,OAAO,CAAC,UAAU,wDAAG;YACjB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,4BAA4B,EAAE;gBACpD;oBACE,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,QAAQ,IAAI,CAAC,MAAM,EAAE,EAAE;oBAC3B,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;iBACtB;aACF;YACD,SAAS,EAAE,CAAC;oBACV,EAAE,EAAE,QAAQ,IAAI,CAAC,MAAM,EAAE,EAAE;oBAC3B,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;iBACtB,CAAC;YACF,WAAW,EAAE,UAAU;YACvB,WAAW,EAAE,IAAI,CAAE,mCAAmC;SACvD,CAAC,CAAC;IACP,CAAC;CACF;AAED,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,IAAI,QAAkB,CAAC;IACvB,IAAI,eAAgC,CAAC;IACrC,IAAI,OAAyB,CAAC;IAE9B,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,QAAQ,CAAC,WAAW,CAAC,CAAC;QACrC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QACxC,OAAO,GAAG;YACR,WAAW,EAAE,eAAe;YAC5B,SAAS,EAAE,IAAI,GAAG,EAAmB;YACrC,KAAK,EAAE,IAAI,GAAG,EAA0B;SACzC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,wCAAwC,EAAE,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,CAAC;YAEnI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,gCAAgC;YACtE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,gCAAgC;YAChC,eAAe,GAAG,IAAI,eAAe,CAAC;gBACpC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;gBACnD,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;aAC3D,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,wCAAwC,EAAE,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,CAAC;YAEnI,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC5C,wCAAwC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,qBAAqB;YACrB,QAAQ,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;YACxD,eAAe,GAAG,IAAI,eAAe,CAAC;gBACpC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;gBACnD,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;aAC7D,CAAC,CAAC;YAED,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,wCAAwC,EAAE,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,CAAC;YAEnI,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC5C,yDAAyD;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,eAAe,GAAG,IAAI,eAAe,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAEhD,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,wCAAwC,EAAE,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,CAAC;YAEnI,MAAM,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC3E,uCAAuC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,yCAAyC;YACzC,eAAe,GAAG,IAAI,eAAe,CAAC;gBACpC;oBACE,IAAI,EAAE,eAAe;oBACrB,KAAK,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE;iBACpE;gBACD,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;aAC3D,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,wCAAwC,EAAE,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,CAAC;YAEnI,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAE5C,wCAAwC;YACxC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,2DAA2D;YAC3D,eAAe,GAAG,IAAI,eAAe,CAAC;gBACpC;oBACE,IAAI,EAAE,eAAe;oBACrB,KAAK,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,kBAAkB,EAAE;iBACtD;gBACD,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;aAC3D,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,wCAAwC,EAAE,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,CAAC;YAEnI,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAE5C,kDAAkD;YAClD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,oCAAoC;YACpC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAEnD,kDAAkD;YAClD,MAAM,gBAAgB,GAAc,EAAE,CAAC;YACvC,MAAM,0BAA0B,GAAG,IAAI,eAAe,EAAE,CAAC;YACzD,0BAA0B,CAAC,cAAc,GAAG,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;;gBAC9E,gBAAgB,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;gBACnC,uCAAuC;gBACvC,MAAM,CAAA,MAAA,OAAO,CAAC,SAAS,wDAAG,cAAc,CAAC,CAAA,CAAC;YAC5C,CAAC,CAAC;YAEF,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAC1C,aAAa,EAAE,wCAAwC,EACvD,CAAC,QAAQ,CAAC,EACV,0BAA0B,CAC3B,CAAC;YAEF,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAE5C,kDAAkD;YAClD,MAAM,aAAa,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC,OAAiB,CAAC;YAC5D,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;;YAC5D,MAAM,WAAW,GAAG,IAAI,QAAQ,CAAC,cAAc,CAAC,CAAC;YACjD,MAAA,OAAO,CAAC,KAAK,0CAAE,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;YAElD,eAAe,GAAG,IAAI,eAAe,CAAC;gBACpC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;gBACtD,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;gBACnD,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;aAC3D,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC,aAAa,EAAE,wCAAwC,EAAE,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,CAAC;YAEnI,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC5C,yCAAyC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/test/unit/tool-registry.test.js

@@ -0,0 +1,99 @@
+import { ToolRegistry } from '../../src/core/tool-registry';
+class MockTool {
+    constructor(name, description = 'Mock tool description', input_schema = {
+        type: 'object',
+        properties: {
+            param: { type: 'string' }
+        }
+    }) {
+        this.name = name;
+        this.description = description;
+        this.input_schema = input_schema;
+    }
+    async execute(params) {
+        return { executed: true };
+    }
+}
+describe('ToolRegistry', () => {
+    let registry;
+    let mockTool;
+    beforeEach(() => {
+        registry = new ToolRegistry();
+        mockTool = new MockTool('mock_tool');
+    });
+    describe('tool management', () => {
+        test('should register a tool successfully', () => {
+            registry.registerTool(mockTool);
+            expect(registry.getTool('mock_tool')).toBe(mockTool);
+        });
+        test('should throw when registering duplicate tool', () => {
+            registry.registerTool(mockTool);
+            expect(() => registry.registerTool(mockTool)).toThrow();
+        });
+        test('should unregister a tool successfully', () => {
+            registry.registerTool(mockTool);
+            registry.unregisterTool('mock_tool');
+            expect(() => registry.getTool('mock_tool')).toThrow();
+        });
+        test('should throw when unregistering non-existent tool', () => {
+            expect(() => registry.unregisterTool('non_existent')).toThrow();
+        });
+        test('should check for tool existence correctly', () => {
+            registry.registerTool(mockTool);
+            expect(registry.hasTools(['mock_tool'])).toBe(true);
+            expect(registry.hasTools(['non_existent'])).toBe(false);
+            expect(registry.hasTools(['mock_tool', 'non_existent'])).toBe(false);
+        });
+    });
+    describe('tool enumeration', () => {
+        beforeEach(() => {
+            registry.registerTool(new MockTool('tool1'));
+            registry.registerTool(new MockTool('tool2'));
+        });
+        test('should get all tools', () => {
+            const tools = registry.getAllTools();
+            expect(tools).toHaveLength(2);
+            expect(tools.map(t => t.name)).toEqual(['tool1', 'tool2']);
+        });
+        test('should get tool definitions', () => {
+            const definitions = registry.getToolDefinitions();
+            expect(definitions).toHaveLength(2);
+            expect(definitions[0]).toHaveProperty('name', 'tool1');
+            expect(definitions[0]).toHaveProperty('description');
+            expect(definitions[0]).toHaveProperty('input_schema');
+        });
+        test('should get tool enum', () => {
+            const enumValues = registry.getToolEnum();
+            expect(enumValues).toEqual(['tool1', 'tool2']);
+        });
+    });
+    describe('workflow schema generation', () => {
+        beforeEach(() => {
+            registry.registerTool(new MockTool('tool1'));
+            registry.registerTool(new MockTool('tool2'));
+        });
+        test('should generate valid workflow schema', () => {
+            const schema = registry.getWorkflowSchema();
+            // Basic schema structure
+            expect(schema).toHaveProperty('type', 'object');
+            expect(schema.properties).toHaveProperty('nodes');
+            // Tool enum in action schema
+            const actionTools = schema.properties.nodes.items.properties.action.properties.tools;
+            expect(actionTools.items.enum).toEqual(['tool1', 'tool2']);
+        });
+        test('should update schema when tools change', () => {
+            let schema = registry.getWorkflowSchema();
+            expect(schema.properties.nodes.items.properties.action.properties.tools.items.enum)
+                .toEqual(['tool1', 'tool2']);
+            registry.registerTool(new MockTool('tool3'));
+            schema = registry.getWorkflowSchema();
+            expect(schema.properties.nodes.items.properties.action.properties.tools.items.enum)
+                .toEqual(['tool1', 'tool2', 'tool3']);
+            registry.unregisterTool('tool2');
+            schema = registry.getWorkflowSchema();
+            expect(schema.properties.nodes.items.properties.action.properties.tools.items.enum)
+                .toEqual(['tool1', 'tool3']);
+        });
+    });
+});
+//# sourceMappingURL=tool-registry.test.js.map

package/dist/test/unit/tool-registry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-registry.test.js","sourceRoot":"","sources":["../../../test/unit/tool-registry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAG5D,MAAM,QAAQ;IACZ,YACS,IAAY,EACZ,cAAsB,uBAAuB,EAC7C,eAA4B;QACjC,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1B;KACF;QAPM,SAAI,GAAJ,IAAI,CAAQ;QACZ,gBAAW,GAAX,WAAW,CAAkC;QAC7C,iBAAY,GAAZ,YAAY,CAKlB;IACA,CAAC;IAEJ,KAAK,CAAC,OAAO,CAAC,MAAe;QAC3B,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,QAAsB,CAAC;IAC3B,IAAI,QAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QAC9B,QAAQ,GAAG,IAAI,QAAQ,CAAC,WAAW,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,IAAI,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC/C,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAChC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACxD,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAChC,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE;YACjD,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAChC,QAAQ,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YACrC,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACrD,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAChC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,UAAU,CAAC,GAAG,EAAE;YACd,QAAQ,CAAC,YAAY,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YAC7C,QAAQ,CAAC,YAAY,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE;YAChC,MAAM,KAAK,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACvC,MAAM,WAAW,GAAG,QAAQ,CAAC,kBAAkB,EAAE,CAAC;YAClD,MAAM,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACvD,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YACrD,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE;YAChC,MAAM,UAAU,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC1C,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,UAAU,CAAC,GAAG,EAAE;YACd,QAAQ,CAAC,YAAY,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YAC7C,QAAQ,CAAC,YAAY,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE;YACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,iBAAiB,EAAS,CAAC;YAEnD,yBAAyB;YACzB,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAElD,6BAA6B;YAC7B,MAAM,WAAW,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC;YACrF,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAClD,IAAI,MAAM,GAAG,QAAQ,CAAC,iBAAiB,EAAS,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;iBAChF,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAE/B,QAAQ,CAAC,YAAY,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YAC7C,MAAM,GAAG,QAAQ,CAAC,iBAAiB,EAAS,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;iBAChF,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAExC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACjC,MAAM,GAAG,QAAQ,CAAC,iBAAiB,EAAS,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;iBAChF,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}