@polka-codes/core 0.10.11 → 0.10.16

package/dist/workflow/try-catch.test.js

@@ -0,0 +1,443 @@
+import { describe, expect, it } from 'bun:test';
+import { createDynamicWorkflow, validateWorkflowFile } from './dynamic';
+describe('Dynamic Workflow Try/Catch Blocks', () => {
+    const mockToolInfo = [
+        {
+            name: 'errorTool',
+            description: 'A tool that throws errors',
+            parameters: { type: 'object' },
+            handler: async () => {
+                throw new Error('Tool execution failed');
+            },
+        },
+        {
+            name: 'successTool',
+            description: 'A tool that succeeds',
+            parameters: { type: 'object' },
+            handler: async () => ({
+                success: true,
+                message: { type: 'text', value: 'Success' },
+            }),
+        },
+    ];
+    describe('validation', () => {
+        it('should accept try/catch with empty catch steps', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Try with empty catch',
+                        steps: [
+                            {
+                                id: 'tryBlock',
+                                try: {
+                                    trySteps: [{ id: 'attempt', task: 'Try something' }],
+                                    catchSteps: [],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const result = validateWorkflowFile(workflowDef);
+            expect(result.success).toBe(true);
+        });
+        it('should accept try/catch with empty try steps', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Try with empty try',
+                        steps: [
+                            {
+                                id: 'tryBlock',
+                                try: {
+                                    trySteps: [],
+                                    catchSteps: [{ id: 'recover', task: 'Recover' }],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const result = validateWorkflowFile(workflowDef);
+            expect(result.success).toBe(true);
+        });
+        it('should accept nested try/catch in try block', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Nested try in try',
+                        steps: [
+                            {
+                                id: 'outerTry',
+                                try: {
+                                    trySteps: [
+                                        {
+                                            id: 'innerTry',
+                                            try: {
+                                                trySteps: [{ id: 'danger', task: 'Danger' }],
+                                                catchSteps: [{ id: 'innerRecover', task: 'Inner recover' }],
+                                            },
+                                        },
+                                    ],
+                                    catchSteps: [{ id: 'outerRecover', task: 'Outer recover' }],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const result = validateWorkflowFile(workflowDef);
+            expect(result.success).toBe(true);
+        });
+        it('should accept nested try/catch in catch block', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Nested try in catch',
+                        steps: [
+                            {
+                                id: 'outerTry',
+                                try: {
+                                    trySteps: [{ id: 'danger', task: 'Danger' }],
+                                    catchSteps: [
+                                        {
+                                            id: 'innerTry',
+                                            try: {
+                                                trySteps: [{ id: 'moreDanger', task: 'More danger' }],
+                                                catchSteps: [{ id: 'innerRecover', task: 'Inner recover' }],
+                                            },
+                                        },
+                                    ],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const result = validateWorkflowFile(workflowDef);
+            expect(result.success).toBe(true);
+        });
+        it('should accept break inside try block within loop', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Break in try within loop',
+                        steps: [
+                            {
+                                id: 'loop',
+                                while: {
+                                    condition: 'true',
+                                    steps: [
+                                        {
+                                            id: 'tryBlock',
+                                            try: {
+                                                trySteps: [{ break: true }],
+                                                catchSteps: [{ id: 'recover', task: 'Recover' }],
+                                            },
+                                        },
+                                    ],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const result = validateWorkflowFile(workflowDef);
+            expect(result.success).toBe(true);
+        });
+        it('should accept break inside catch block within loop', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Break in catch within loop',
+                        steps: [
+                            {
+                                id: 'loop',
+                                while: {
+                                    condition: 'true',
+                                    steps: [
+                                        {
+                                            id: 'tryBlock',
+                                            try: {
+                                                trySteps: [{ id: 'attempt', task: 'Attempt' }],
+                                                catchSteps: [{ break: true }],
+                                            },
+                                        },
+                                    ],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const result = validateWorkflowFile(workflowDef);
+            expect(result.success).toBe(true);
+        });
+    });
+    describe('structure', () => {
+        it('should create workflow with try/catch', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Test try/catch',
+                        steps: [
+                            {
+                                id: 'riskyOperation',
+                                try: {
+                                    trySteps: [
+                                        {
+                                            id: 'attemptTask',
+                                            task: 'Attempt a risky operation',
+                                            tools: ['internet'],
+                                        },
+                                    ],
+                                    catchSteps: [
+                                        {
+                                            id: 'handleError',
+                                            task: 'Handle the error gracefully',
+                                            output: 'fallbackResult',
+                                        },
+                                    ],
+                                },
+                                output: 'result',
+                            },
+                        ],
+                    },
+                },
+            };
+            const workflow = createDynamicWorkflow(workflowDef, {
+                toolInfo: mockToolInfo,
+            });
+            expect(workflow).toBeDefined();
+        });
+        it('should handle multiple sequential try/catch blocks', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Multiple try/catch',
+                        steps: [
+                            {
+                                id: 'firstTry',
+                                try: {
+                                    trySteps: [{ id: 'firstAttempt', task: 'First attempt' }],
+                                    catchSteps: [{ id: 'firstRecover', task: 'First recovery' }],
+                                },
+                            },
+                            {
+                                id: 'secondTry',
+                                try: {
+                                    trySteps: [{ id: 'secondAttempt', task: 'Second attempt' }],
+                                    catchSteps: [{ id: 'secondRecover', task: 'Second recovery' }],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const workflow = createDynamicWorkflow(workflowDef, {
+                toolInfo: mockToolInfo,
+            });
+            expect(workflow).toBeDefined();
+        });
+        it('should handle try/catch inside if/else', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Try/catch in if',
+                        steps: [
+                            {
+                                id: 'check',
+                                if: {
+                                    condition: 'true',
+                                    thenBranch: [
+                                        {
+                                            id: 'tryBlock',
+                                            try: {
+                                                trySteps: [{ id: 'attempt', task: 'Attempt' }],
+                                                catchSteps: [{ id: 'recover', task: 'Recover' }],
+                                            },
+                                        },
+                                    ],
+                                    elseBranch: [
+                                        {
+                                            id: 'elseTry',
+                                            try: {
+                                                trySteps: [{ id: 'elseAttempt', task: 'Else attempt' }],
+                                                catchSteps: [{ id: 'elseRecover', task: 'Else recover' }],
+                                            },
+                                        },
+                                    ],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const workflow = createDynamicWorkflow(workflowDef, {
+                toolInfo: mockToolInfo,
+            });
+            expect(workflow).toBeDefined();
+        });
+        it('should handle try/catch with output from catch block', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Try/catch with output',
+                        steps: [
+                            {
+                                id: 'tryBlock',
+                                try: {
+                                    trySteps: [
+                                        {
+                                            id: 'attempt',
+                                            task: 'Attempt',
+                                            output: 'tryResult',
+                                        },
+                                    ],
+                                    catchSteps: [
+                                        {
+                                            id: 'recover',
+                                            task: 'Recover',
+                                            output: 'catchResult',
+                                        },
+                                    ],
+                                },
+                                output: 'finalResult',
+                            },
+                        ],
+                        output: 'finalResult',
+                    },
+                },
+            };
+            const workflow = createDynamicWorkflow(workflowDef, {
+                toolInfo: mockToolInfo,
+            });
+            expect(workflow).toBeDefined();
+        });
+        it('should handle deeply nested try/catch structures', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Deeply nested try/catch',
+                        steps: [
+                            {
+                                id: 'level1',
+                                try: {
+                                    trySteps: [
+                                        {
+                                            id: 'level2',
+                                            while: {
+                                                condition: 'state.i < 2',
+                                                steps: [
+                                                    {
+                                                        id: 'level3',
+                                                        if: {
+                                                            condition: 'true',
+                                                            thenBranch: [
+                                                                {
+                                                                    id: 'level4',
+                                                                    try: {
+                                                                        trySteps: [{ id: 'core', task: 'Core operation' }],
+                                                                        catchSteps: [{ id: 'coreRecover', task: 'Core recovery' }],
+                                                                    },
+                                                                },
+                                                            ],
+                                                        },
+                                                    },
+                                                ],
+                                            },
+                                        },
+                                    ],
+                                    catchSteps: [{ id: 'level1Recover', task: 'Level 1 recovery' }],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const workflow = createDynamicWorkflow(workflowDef, {
+                toolInfo: mockToolInfo,
+            });
+            expect(workflow).toBeDefined();
+        });
+    });
+    describe('output handling', () => {
+        it('should allow try/catch without output field', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Try/catch without output',
+                        steps: [
+                            {
+                                id: 'tryBlock',
+                                try: {
+                                    trySteps: [{ id: 'attempt', task: 'Attempt' }],
+                                    catchSteps: [{ id: 'recover', task: 'Recover' }],
+                                },
+                            },
+                            {
+                                id: 'nextStep',
+                                task: 'Next step',
+                            },
+                        ],
+                    },
+                },
+            };
+            const workflow = createDynamicWorkflow(workflowDef, {
+                toolInfo: mockToolInfo,
+            });
+            expect(workflow).toBeDefined();
+        });
+        it('should use step id as default output key', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Default output key',
+                        steps: [
+                            {
+                                id: 'myTryBlock',
+                                try: {
+                                    trySteps: [
+                                        {
+                                            id: 'attempt',
+                                            task: 'Attempt',
+                                            output: 'value',
+                                        },
+                                    ],
+                                    catchSteps: [{ id: 'recover', task: 'Recover' }],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const workflow = createDynamicWorkflow(workflowDef, {
+                toolInfo: mockToolInfo,
+            });
+            expect(workflow).toBeDefined();
+        });
+    });
+    describe('error scenarios', () => {
+        it('should handle empty try and catch steps', () => {
+            const workflowDef = {
+                workflows: {
+                    test: {
+                        task: 'Empty try/catch',
+                        steps: [
+                            {
+                                id: 'emptyBlock',
+                                try: {
+                                    trySteps: [],
+                                    catchSteps: [],
+                                },
+                            },
+                        ],
+                    },
+                },
+            };
+            const result = validateWorkflowFile(workflowDef);
+            expect(result.success).toBe(true);
+        });
+    });
+});
+//# sourceMappingURL=try-catch.test.js.map

package/dist/workflow/try-catch.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"try-catch.test.js","sourceRoot":"","sources":["../../src/workflow/try-catch.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,UAAU,CAAA;AAE/C,OAAO,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAA;AAGvE,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,MAAM,YAAY,GAAmB;QACnC;YACE,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,2BAA2B;YACxC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAS;YACrC,OAAO,EAAE,KAAK,IAAI,EAAE;gBAClB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAA;YAC1C,CAAC;SACF;QACD;YACE,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,sBAAsB;YACnC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAS;YACrC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBACpB,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE;aAC5C,CAAC;SACH;KACF,CAAA;IAED,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,sBAAsB;wBAC5B,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,UAAU;gCACd,GAAG,EAAE;oCACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;oCACpD,UAAU,EAAE,EAAE;iCACf;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,oBAAoB;wBAC1B,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,UAAU;gCACd,GAAG,EAAE;oCACH,QAAQ,EAAE,EAAE;oCACZ,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;iCACjD;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,mBAAmB;wBACzB,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,UAAU;gCACd,GAAG,EAAE;oCACH,QAAQ,EAAE;wCACR;4CACE,EAAE,EAAE,UAAU;4CACd,GAAG,EAAE;gDACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gDAC5C,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;6CAC5D;yCACF;qCACF;oCACD,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;iCAC5D;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,qBAAqB;wBAC3B,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,UAAU;gCACd,GAAG,EAAE;oCACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;oCAC5C,UAAU,EAAE;wCACV;4CACE,EAAE,EAAE,UAAU;4CACd,GAAG,EAAE;gDACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;gDACrD,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;6CAC5D;yCACF;qCACF;iCACF;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,0BAA0B;wBAChC,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,MAAM;gCACV,KAAK,EAAE;oCACL,SAAS,EAAE,MAAM;oCACjB,KAAK,EAAE;wCACL;4CACE,EAAE,EAAE,UAAU;4CACd,GAAG,EAAE;gDACH,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gDAC3B,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;6CACjD;yCACF;qCACF;iCACF;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,4BAA4B;wBAClC,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,MAAM;gCACV,KAAK,EAAE;oCACL,SAAS,EAAE,MAAM;oCACjB,KAAK,EAAE;wCACL;4CACE,EAAE,EAAE,UAAU;4CACd,GAAG,EAAE;gDACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;gDAC9C,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;6CAC9B;yCACF;qCACF;iCACF;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,gBAAgB;wBACtB,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,gBAAgB;gCACpB,GAAG,EAAE;oCACH,QAAQ,EAAE;wCACR;4CACE,EAAE,EAAE,aAAa;4CACjB,IAAI,EAAE,2BAA2B;4CACjC,KAAK,EAAE,CAAC,UAAU,CAAC;yCACpB;qCACF;oCACD,UAAU,EAAE;wCACV;4CACE,EAAE,EAAE,aAAa;4CACjB,IAAI,EAAE,6BAA6B;4CACnC,MAAM,EAAE,gBAAgB;yCACzB;qCACF;iCACF;gCACD,MAAM,EAAE,QAAQ;6BACjB;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,EAAE;gBAClD,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAA;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,oBAAoB;wBAC1B,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,UAAU;gCACd,GAAG,EAAE;oCACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;oCACzD,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;iCAC7D;6BACF;4BACD;gCACE,EAAE,EAAE,WAAW;gCACf,GAAG,EAAE;oCACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;oCAC3D,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;iCAC/D;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,EAAE;gBAClD,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAA;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,iBAAiB;wBACvB,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,OAAO;gCACX,EAAE,EAAE;oCACF,SAAS,EAAE,MAAM;oCACjB,UAAU,EAAE;wCACV;4CACE,EAAE,EAAE,UAAU;4CACd,GAAG,EAAE;gDACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;gDAC9C,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;6CACjD;yCACF;qCACF;oCACD,UAAU,EAAE;wCACV;4CACE,EAAE,EAAE,SAAS;4CACb,GAAG,EAAE;gDACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;gDACvD,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;6CAC1D;yCACF;qCACF;iCACF;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,EAAE;gBAClD,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAA;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,uBAAuB;wBAC7B,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,UAAU;gCACd,GAAG,EAAE;oCACH,QAAQ,EAAE;wCACR;4CACE,EAAE,EAAE,SAAS;4CACb,IAAI,EAAE,SAAS;4CACf,MAAM,EAAE,WAAW;yCACpB;qCACF;oCACD,UAAU,EAAE;wCACV;4CACE,EAAE,EAAE,SAAS;4CACb,IAAI,EAAE,SAAS;4CACf,MAAM,EAAE,aAAa;yCACtB;qCACF;iCACF;gCACD,MAAM,EAAE,aAAa;6BACtB;yBACF;wBACD,MAAM,EAAE,aAAa;qBACtB;iBACF;aACF,CAAA;YAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,EAAE;gBAClD,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAA;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,yBAAyB;wBAC/B,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,QAAQ;gCACZ,GAAG,EAAE;oCACH,QAAQ,EAAE;wCACR;4CACE,EAAE,EAAE,QAAQ;4CACZ,KAAK,EAAE;gDACL,SAAS,EAAE,aAAa;gDACxB,KAAK,EAAE;oDACL;wDACE,EAAE,EAAE,QAAQ;wDACZ,EAAE,EAAE;4DACF,SAAS,EAAE,MAAM;4DACjB,UAAU,EAAE;gEACV;oEACE,EAAE,EAAE,QAAQ;oEACZ,GAAG,EAAE;wEACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;wEAClD,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;qEAC3D;iEACF;6DACF;yDACF;qDACF;iDACF;6CACF;yCACF;qCACF;oCACD,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;iCAChE;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,EAAE;gBAClD,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAA;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,0BAA0B;wBAChC,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,UAAU;gCACd,GAAG,EAAE;oCACH,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;oCAC9C,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;iCACjD;6BACF;4BACD;gCACE,EAAE,EAAE,UAAU;gCACd,IAAI,EAAE,WAAW;6BAClB;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,EAAE;gBAClD,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAA;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,oBAAoB;wBAC1B,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,YAAY;gCAChB,GAAG,EAAE;oCACH,QAAQ,EAAE;wCACR;4CACE,EAAE,EAAE,SAAS;4CACb,IAAI,EAAE,SAAS;4CACf,MAAM,EAAE,OAAO;yCAChB;qCACF;oCACD,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;iCACjD;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,EAAE;gBAClD,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAA;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,WAAW,GAAiB;gBAChC,SAAS,EAAE;oBACT,IAAI,EAAE;wBACJ,IAAI,EAAE,iBAAiB;wBACvB,KAAK,EAAE;4BACL;gCACE,EAAE,EAAE,YAAY;gCAChB,GAAG,EAAE;oCACH,QAAQ,EAAE,EAAE;oCACZ,UAAU,EAAE,EAAE;iCACf;6BACF;yBACF;qBACF;iBACF;aACF,CAAA;YAED,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/workflow/types.d.ts

@@ -0,0 +1,103 @@
+import type { ToolResponseResult } from '../tool';
+import type { JsonModelMessage } from './json-ai-types';
+/**
+ * Enum representing different kinds of task events
+ */
+export declare enum TaskEventKind {
+    StartTask = "StartTask",
+    StartRequest = "StartRequest",
+    EndRequest = "EndRequest",
+    Text = "Text",
+    Reasoning = "Reasoning",
+    ToolUse = "ToolUse",
+    ToolReply = "ToolReply",
+    ToolError = "ToolError",
+    UsageExceeded = "UsageExceeded",
+    EndTask = "EndTask"
+}
+/**
+ * Base interface for all task events
+ */
+export interface TaskEventBase {
+    kind: TaskEventKind;
+}
+/**
+ * Event for task start
+ */
+export interface TaskEventStartTask extends TaskEventBase {
+    kind: TaskEventKind.StartTask;
+    systemPrompt: string;
+}
+/**
+ * Event for request start
+ */
+export interface TaskEventStartRequest extends TaskEventBase {
+    kind: TaskEventKind.StartRequest;
+    userMessage: readonly JsonModelMessage[];
+}
+/**
+ * Event for request end
+ */
+export interface TaskEventEndRequest extends TaskEventBase {
+    kind: TaskEventKind.EndRequest;
+    message: string;
+}
+/**
+ * Event for text/reasoning updates
+ */
+export interface TaskEventText extends TaskEventBase {
+    kind: TaskEventKind.Text | TaskEventKind.Reasoning;
+    newText: string;
+}
+/**
+ * Event for tool-related updates
+ */
+export interface TaskEventToolUse extends TaskEventBase {
+    kind: TaskEventKind.ToolUse;
+    tool: string;
+    params: Record<string, any>;
+}
+export interface TaskEventToolResult extends TaskEventBase {
+    kind: TaskEventKind.ToolReply;
+    tool: string;
+    content: ToolResponseResult;
+}
+export interface TaskEventToolError extends TaskEventBase {
+    kind: TaskEventKind.ToolError;
+    tool: string;
+    error: ToolResponseResult;
+}
+/**
+ * Event for task usage exceeded
+ */
+export interface TaskEventUsageExceeded extends TaskEventBase {
+    kind: TaskEventKind.UsageExceeded;
+}
+/**
+ * Event for task end
+ */
+export interface TaskEventEndTask extends TaskEventBase {
+    kind: TaskEventKind.EndTask;
+    exitReason: ExitReason;
+}
+/**
+ * Union type of all possible task events
+ */
+export type TaskEvent = TaskEventStartTask | TaskEventStartRequest | TaskEventEndRequest | TaskEventText | TaskEventToolUse | TaskEventToolResult | TaskEventToolError | TaskEventUsageExceeded | TaskEventEndTask;
+export type TaskEventCallback = (event: TaskEvent) => void | Promise<void>;
+export type ExitReason = {
+    type: 'UsageExceeded';
+    messages: JsonModelMessage[];
+} | {
+    type: 'Exit';
+    message: string;
+    object?: any;
+    messages: JsonModelMessage[];
+} | {
+    type: 'Error';
+    error: {
+        message: string;
+        stack?: string;
+    };
+    messages: JsonModelMessage[];
+};

package/dist/workflow/types.js

@@ -0,0 +1,17 @@
+/**
+ * Enum representing different kinds of task events
+ */
+export var TaskEventKind;
+(function (TaskEventKind) {
+    TaskEventKind["StartTask"] = "StartTask";
+    TaskEventKind["StartRequest"] = "StartRequest";
+    TaskEventKind["EndRequest"] = "EndRequest";
+    TaskEventKind["Text"] = "Text";
+    TaskEventKind["Reasoning"] = "Reasoning";
+    TaskEventKind["ToolUse"] = "ToolUse";
+    TaskEventKind["ToolReply"] = "ToolReply";
+    TaskEventKind["ToolError"] = "ToolError";
+    TaskEventKind["UsageExceeded"] = "UsageExceeded";
+    TaskEventKind["EndTask"] = "EndTask";
+})(TaskEventKind || (TaskEventKind = {}));
+//# sourceMappingURL=types.js.map

package/dist/workflow/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/workflow/types.ts"],"names":[],"mappings":"AAGA;;GAEG;AACH,MAAM,CAAN,IAAY,aAWX;AAXD,WAAY,aAAa;IACvB,wCAAuB,CAAA;IACvB,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,8BAAa,CAAA;IACb,wCAAuB,CAAA;IACvB,oCAAmB,CAAA;IACnB,wCAAuB,CAAA;IACvB,wCAAuB,CAAA;IACvB,gDAA+B,CAAA;IAC/B,oCAAmB,CAAA;AACrB,CAAC,EAXW,aAAa,KAAb,aAAa,QAWxB"}

package/dist/workflow/workflow.d.ts

@@ -0,0 +1,29 @@
+export interface Logger {
+    debug: (...args: any[]) => void;
+    info: (...args: any[]) => void;
+    warn: (...args: any[]) => void;
+    error: (...args: any[]) => void;
+}
+export type ToolSignature<I, O> = {
+    input: I;
+    output: O;
+};
+export type ToolRegistry = Record<string, ToolSignature<any, any>>;
+export type WorkflowTools<TTools extends ToolRegistry> = {
+    [K in keyof TTools]: (input: TTools[K]['input']) => Promise<TTools[K]['output']>;
+};
+export type StepOptions = {
+    retry?: number;
+};
+export interface StepFn {
+    <T>(name: string, fn: () => Promise<T>): Promise<T>;
+    <T>(name: string, options: StepOptions, fn: () => Promise<T>): Promise<T>;
+}
+export interface BaseWorkflowContext<TTools extends ToolRegistry> {
+    step: StepFn;
+    logger: Logger;
+    tools: WorkflowTools<TTools>;
+}
+export type WorkflowFn<TInput, TOutput, TTools extends ToolRegistry, TContext extends BaseWorkflowContext<TTools> = BaseWorkflowContext<TTools>> = (input: TInput, context: TContext) => Promise<TOutput>;
+export declare function createContext<TTools extends ToolRegistry>(tools: WorkflowTools<TTools>, stepFn?: StepFn, logger?: Logger): BaseWorkflowContext<TTools>;
+export declare const makeStepFn: () => StepFn;

package/dist/workflow/workflow.js

@@ -0,0 +1,57 @@
+// Create a default silent logger
+const silentLogger = {
+    debug: () => { },
+    info: () => { },
+    warn: () => { },
+    error: () => { },
+};
+export function createContext(tools, stepFn, logger = silentLogger) {
+    if (!stepFn) {
+        // simple default step function
+        stepFn = async (_name, arg2, arg3) => {
+            const fn = typeof arg2 === 'function' ? arg2 : arg3;
+            return fn();
+        };
+    }
+    return { step: stepFn, logger, tools };
+}
+export const makeStepFn = () => {
+    const results = new Map();
+    const callStack = [];
+    return async (name, arg2, arg3) => {
+        let fn;
+        let options;
+        if (typeof arg2 === 'function') {
+            fn = arg2;
+            options = arg3;
+        }
+        else {
+            options = arg2;
+            fn = arg3;
+        }
+        callStack.push(name);
+        const key = callStack.join('>');
+        try {
+            if (results.has(key)) {
+                return results.get(key);
+            }
+            const maxRetryCount = options?.retry ?? 0;
+            let lastError;
+            for (let retryCount = 0; retryCount <= maxRetryCount; retryCount++) {
+                try {
+                    const result = await fn();
+                    results.set(key, result);
+                    return result;
+                }
+                catch (error) {
+                    lastError = error;
+                }
+            }
+            throw lastError;
+        }
+        finally {
+            callStack.pop();
+        }
+    };
+};
+//# sourceMappingURL=workflow.js.map

package/dist/workflow/workflow.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow.js","sourceRoot":"","sources":["../../src/workflow/workflow.ts"],"names":[],"mappings":"AAwCA,iCAAiC;AACjC,MAAM,YAAY,GAAW;IAC3B,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;IACf,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;CAChB,CAAA;AAED,MAAM,UAAU,aAAa,CAC3B,KAA4B,EAC5B,MAAe,EACf,SAAiB,YAAY;IAE7B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,+BAA+B;QAC/B,MAAM,GAAG,KAAK,EAAK,KAAa,EAAE,IAAsC,EAAE,IAAuC,EAAE,EAAE;YACnH,MAAM,EAAE,GAAG,OAAO,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAE,IAAyB,CAAA;YACzE,OAAO,EAAE,EAAE,CAAA;QACb,CAAC,CAAA;IACH,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,CAAA;AACxC,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,GAAW,EAAE;IACrC,MAAM,OAAO,GAAqB,IAAI,GAAG,EAAE,CAAA;IAC3C,MAAM,SAAS,GAAa,EAAE,CAAA;IAE9B,OAAO,KAAK,EAAK,IAAY,EAAE,IAAsC,EAAE,IAAuC,EAAc,EAAE;QAC5H,IAAI,EAAoB,CAAA;QACxB,IAAI,OAAgC,CAAA;QAEpC,IAAI,OAAO,IAAI,KAAK,UAAU,EAAE,CAAC;YAC/B,EAAE,GAAG,IAAwB,CAAA;YAC7B,OAAO,GAAG,IAA+B,CAAA;QAC3C,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,IAAmB,CAAA;YAC7B,EAAE,GAAG,IAAwB,CAAA;QAC/B,CAAC;QAED,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACpB,MAAM,GAAG,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAE/B,IAAI,CAAC;YACH,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrB,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAM,CAAA;YAC9B,CAAC;YAED,MAAM,aAAa,GAAG,OAAO,EAAE,KAAK,IAAI,CAAC,CAAA;YACzC,IAAI,SAAkB,CAAA;YAEtB,KAAK,IAAI,UAAU,GAAG,CAAC,EAAE,UAAU,IAAI,aAAa,EAAE,UAAU,EAAE,EAAE,CAAC;gBACnE,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,EAAE,EAAE,CAAA;oBACzB,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;oBACxB,OAAO,MAAM,CAAA;gBACf,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,SAAS,GAAG,KAAK,CAAA;gBACnB,CAAC;YACH,CAAC;YACD,MAAM,SAAS,CAAA;QACjB,CAAC;gBAAS,CAAC;YACT,SAAS,CAAC,GAAG,EAAE,CAAA;QACjB,CAAC;IACH,CAAC,CAAA;AACH,CAAC,CAAA"}