@polka-codes/core 0.10.13 → 0.10.17

package/dist/workflow/agent.workflow.js

@@ -0,0 +1,166 @@
+// generated by polka.codes
+import { jsonSchema } from 'ai';
+import { toJSONSchema, z } from 'zod';
+import { parseJsonFromMarkdown } from '../Agent/parseJsonFromMarkdown';
+import { TaskEventKind } from './types';
+function isValidToolResultOutput(value) {
+    return (typeof value === 'object' &&
+        value !== null &&
+        'type' in value &&
+        (value.type === 'text' || value.type === 'content' || value.type === 'execution-denied'));
+}
+export const agentWorkflow = async (input, { step, tools, logger }) => {
+    const event = (name, event) => step(name, () => tools.taskEvent(event));
+    const { tools: toolInfo, maxToolRoundTrips = 200 } = input;
+    const messages = 'systemPrompt' in input ? [{ role: 'system', content: input.systemPrompt }] : input.messages;
+    await event('start-task', { kind: TaskEventKind.StartTask, systemPrompt: 'systemPrompt' in input ? input.systemPrompt : '' });
+    const toolSet = {};
+    for (const tool of toolInfo) {
+        toolSet[tool.name] = {
+            description: tool.description,
+            inputSchema: jsonSchema(toJSONSchema(tool.parameters)),
+        };
+    }
+    let nextMessage = input.userMessage;
+    for (let i = 0; i < maxToolRoundTrips; i++) {
+        messages.push(...nextMessage);
+        await event(`start-round-${i}`, { kind: TaskEventKind.StartRequest, userMessage: nextMessage });
+        const assistantMessage = await step(`agent-round-${i}`, { retry: 2 }, async () => {
+            return await tools.generateText({
+                messages,
+                tools: toolSet,
+                model: input.model,
+            });
+        });
+        messages.push(...assistantMessage);
+        const toolCalls = [];
+        for (const msg of assistantMessage) {
+            if (typeof msg.content === 'string') {
+                continue;
+            }
+            if (msg.content) {
+                for (const part of msg.content) {
+                    if (part.type === 'tool-call') {
+                        toolCalls.push(part);
+                    }
+                }
+            }
+        }
+        const textContent = assistantMessage
+            .flatMap((m) => {
+            if (typeof m.content === 'string') {
+                return [m.content];
+            }
+            if (m.role === 'assistant' && Array.isArray(m.content)) {
+                return m.content.map((part) => {
+                    if (part.type === 'text') {
+                        return part.text;
+                    }
+                    return '';
+                });
+            }
+            return [];
+        })
+            .join('\n');
+        await event(`end-round-${i}`, { kind: TaskEventKind.EndRequest, message: textContent });
+        if (toolCalls.length === 0) {
+            if (!input.outputSchema) {
+                const exitReason = { type: 'Exit', message: textContent, messages };
+                await event('end-task', { kind: TaskEventKind.EndTask, exitReason });
+                return exitReason;
+            }
+            const parsed = parseJsonFromMarkdown(textContent);
+            if (!parsed.success) {
+                const errorMessage = `Failed to parse JSON from markdown. Error: ${parsed.error}. Please correct the output. It MUST be in valid JSON format.`;
+                nextMessage = [{ role: 'user', content: errorMessage }];
+                continue;
+            }
+            const validated = input.outputSchema.safeParse(parsed.data);
+            if (!validated.success) {
+                const errorMessage = `Output validation failed. Error: ${z.prettifyError(validated.error)}. Please correct the output.`;
+                nextMessage = [{ role: 'user', content: errorMessage }];
+                continue;
+            }
+            const exitReason = { type: 'Exit', message: textContent, object: validated.data, messages };
+            await event('end-task', { kind: TaskEventKind.EndTask, exitReason });
+            return exitReason;
+        }
+        const toolResults = [];
+        for (const toolCall of toolCalls) {
+            if (!toolSet[toolCall.toolName]) {
+                logger.warn('Tool not found.', toolCall);
+                await event(`event-tool-error-${toolCall.toolName}-${toolCall.toolCallId}`, {
+                    kind: TaskEventKind.ToolError,
+                    tool: toolCall.toolName,
+                    error: {
+                        type: 'error-text',
+                        value: `Tool '${toolCall.toolName}' not found.`,
+                    },
+                });
+                toolResults.push({
+                    toolCallId: toolCall.toolCallId,
+                    toolName: toolCall.toolName,
+                    output: {
+                        type: 'error-text',
+                        value: `Error: Tool '${toolCall.toolName}' not found.`,
+                    },
+                });
+                continue;
+            }
+            await event(`event-tool-use-${toolCall.toolName}-${toolCall.toolCallId}`, {
+                kind: TaskEventKind.ToolUse,
+                tool: toolCall.toolName,
+                params: toolCall.input,
+            });
+            const toolResponse = await step(`invoke-tool-${toolCall.toolName}-${toolCall.toolCallId}`, async () => {
+                return await tools.invokeTool({
+                    toolName: toolCall.toolName,
+                    input: toolCall.input,
+                });
+            });
+            if (toolResponse.success) {
+                await event(`event-tool-reply-${toolCall.toolName}-${toolCall.toolCallId}`, {
+                    kind: TaskEventKind.ToolReply,
+                    tool: toolCall.toolName,
+                    content: toolResponse.message,
+                });
+                if (!isValidToolResultOutput(toolResponse.message)) {
+                    throw new Error(`Invalid tool result format from ${toolCall.toolName}`);
+                }
+                toolResults.push({
+                    toolCallId: toolCall.toolCallId,
+                    toolName: toolCall.toolName,
+                    output: toolResponse.message,
+                });
+            }
+            else {
+                await event(`event-tool-error-${toolCall.toolName}-${toolCall.toolCallId}`, {
+                    kind: TaskEventKind.ToolError,
+                    tool: toolCall.toolName,
+                    error: toolResponse.message ?? 'Unknown error',
+                });
+                if (!isValidToolResultOutput(toolResponse.message)) {
+                    throw new Error(`Invalid tool result format from ${toolCall.toolName}`);
+                }
+                toolResults.push({
+                    toolCallId: toolCall.toolCallId,
+                    toolName: toolCall.toolName,
+                    output: toolResponse.message,
+                });
+            }
+        }
+        nextMessage = [
+            {
+                role: 'tool',
+                content: toolResults.map((r) => ({
+                    type: 'tool-result',
+                    ...r,
+                })),
+            },
+        ];
+    }
+    const exitReason = { type: 'UsageExceeded', messages };
+    await event('end-task', { kind: TaskEventKind.EndTask, exitReason });
+    return exitReason;
+};
+//# sourceMappingURL=agent.workflow.js.map

package/dist/workflow/agent.workflow.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.workflow.js","sourceRoot":"","sources":["../../src/workflow/agent.workflow.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAG3B,OAAO,EAAE,UAAU,EAAmC,MAAM,IAAI,CAAA;AAChE,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACrC,OAAO,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAA;AAGtE,OAAO,EAAmC,aAAa,EAAE,MAAM,SAAS,CAAA;AAGxE,SAAS,uBAAuB,CAAC,KAAc;IAC7C,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,MAAM,IAAI,KAAK;QACf,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,CAAC,CACzF,CAAA;AACH,CAAC;AAgCD,MAAM,CAAC,MAAM,aAAa,GAAkE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE;IACnI,MAAM,KAAK,GAAG,CAAC,IAAY,EAAE,KAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAA;IAE1F,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,iBAAiB,GAAG,GAAG,EAAE,GAAG,KAAK,CAAA;IAE1D,MAAM,QAAQ,GAAuB,cAAc,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAA;IAEjI,MAAM,KAAK,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,aAAa,CAAC,SAAS,EAAE,YAAY,EAAE,cAAc,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;IAE7H,MAAM,OAAO,GAAY,EAAE,CAAA;IAC3B,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC5B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;YACnB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,WAAW,EAAE,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;SACvD,CAAA;IACH,CAAC;IAED,IAAI,WAAW,GAAgC,KAAK,CAAC,WAAW,CAAA;IAEhE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,iBAAiB,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3C,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAA;QAE7B,MAAM,KAAK,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,aAAa,CAAC,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC,CAAA;QAC/F,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC/E,OAAO,MAAM,KAAK,CAAC,YAAY,CAAC;gBAC9B,QAAQ;gBACR,KAAK,EAAE,OAAO;gBACd,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,QAAQ,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAA;QAElC,MAAM,SAAS,GAAmB,EAAE,CAAA;QACpC,KAAK,MAAM,GAAG,IAAI,gBAAgB,EAAE,CAAC;YACnC,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACpC,SAAQ;YACV,CAAC;YACD,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChB,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBAC9B,SAAS,CAAC,IAAI,CAAC,IAAoB,CAAC,CAAA;oBACtC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,gBAAgB;aACjC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACb,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAClC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAA;YACpB,CAAC;YAED,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;gBACvD,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;oBAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACzB,OAAO,IAAI,CAAC,IAAI,CAAA;oBAClB,CAAC;oBACD,OAAO,EAAE,CAAA;gBACX,CAAC,CAAC,CAAA;YACJ,CAAC;YACD,OAAO,EAAE,CAAA;QACX,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAA;QAEb,MAAM,KAAK,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAA;QAEvF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;gBACxB,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAA;gBAC/E,MAAM,KAAK,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAA;gBACpE,OAAO,UAAU,CAAA;YACnB,CAAC;YAED,MAAM,MAAM,GAAG,qBAAqB,CAAC,WAAW,CAAC,CAAA;YACjD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,YAAY,GAAG,8CAA8C,MAAM,CAAC,KAAK,+DAA+D,CAAA;gBAC9I,WAAW,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAA;gBACvD,SAAQ;YACV,CAAC;YAED,MAAM,SAAS,GAAG,KAAK,CAAC,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAC3D,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;gBACvB,MAAM,YAAY,GAAG,oCAAoC,CAAC,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,8BAA8B,CAAA;gBACvH,WAAW,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAA;gBACvD,SAAQ;YACV,CAAC;YACD,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAA;YACvG,MAAM,KAAK,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAA;YACpE,OAAO,UAAU,CAAA;QACnB,CAAC;QAED,MAAM,WAAW,GAAwF,EAAE,CAAA;QAC3G,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAChC,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAA;gBACxC,MAAM,KAAK,CAAC,oBAAoB,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE;oBAC1E,IAAI,EAAE,aAAa,CAAC,SAAS;oBAC7B,IAAI,EAAE,QAAQ,CAAC,QAAQ;oBACvB,KAAK,EAAE;wBACL,IAAI,EAAE,YAAY;wBAClB,KAAK,EAAE,SAAS,QAAQ,CAAC,QAAQ,cAAc;qBAChD;iBACF,CAAC,CAAA;gBACF,WAAW,CAAC,IAAI,CAAC;oBACf,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;oBAC3B,MAAM,EAAE;wBACN,IAAI,EAAE,YAAY;wBAClB,KAAK,EAAE,gBAAgB,QAAQ,CAAC,QAAQ,cAAc;qBACvD;iBACF,CAAC,CAAA;gBACF,SAAQ;YACV,CAAC;YAED,MAAM,KAAK,CAAC,kBAAkB,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE;gBACxE,IAAI,EAAE,aAAa,CAAC,OAAO;gBAC3B,IAAI,EAAE,QAAQ,CAAC,QAAQ;gBACvB,MAAM,EAAE,QAAQ,CAAC,KAA4B;aAC9C,CAAC,CAAA;YACF,MAAM,YAAY,GAAiB,MAAM,IAAI,CAAC,eAAe,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBAClH,OAAO,MAAM,KAAK,CAAC,UAAU,CAAC;oBAC5B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;oBAC3B,KAAK,EAAE,QAAQ,CAAC,KAAK;iBACtB,CAAC,CAAA;YACJ,CAAC,CAAC,CAAA;YAEF,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBACzB,MAAM,KAAK,CAAC,oBAAoB,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE;oBAC1E,IAAI,EAAE,aAAa,CAAC,SAAS;oBAC7B,IAAI,EAAE,QAAQ,CAAC,QAAQ;oBACvB,OAAO,EAAE,YAAY,CAAC,OAAO;iBAC9B,CAAC,CAAA;gBACF,IAAI,CAAC,uBAAuB,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;oBACnD,MAAM,IAAI,KAAK,CAAC,mCAAmC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAA;gBACzE,CAAC;gBACD,WAAW,CAAC,IAAI,CAAC;oBACf,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;oBAC3B,MAAM,EAAE,YAAY,CAAC,OAAO;iBAC7B,CAAC,CAAA;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC,oBAAoB,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE;oBAC1E,IAAI,EAAE,aAAa,CAAC,SAAS;oBAC7B,IAAI,EAAE,QAAQ,CAAC,QAAQ;oBACvB,KAAK,EAAE,YAAY,CAAC,OAAO,IAAI,eAAe;iBAC/C,CAAC,CAAA;gBACF,IAAI,CAAC,uBAAuB,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;oBACnD,MAAM,IAAI,KAAK,CAAC,mCAAmC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAA;gBACzE,CAAC;gBACD,WAAW,CAAC,IAAI,CAAC;oBACf,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;oBAC3B,MAAM,EAAE,YAAY,CAAC,OAAO;iBAC7B,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;QAED,WAAW,GAAG;YACZ;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC/B,IAAI,EAAE,aAAa;oBACnB,GAAG,CAAC;iBACL,CAAC,CAAC;aACJ;SACF,CAAA;IACH,CAAC;IAED,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAA;IAClE,MAAM,KAAK,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAA;IACpE,OAAO,UAAU,CAAA;AACnB,CAAC,CAAA"}

package/dist/workflow/agent.workflow.test.d.ts

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

package/dist/workflow/agent.workflow.test.js

@@ -0,0 +1,175 @@
+// generated by polka.codes
+import { expect, spyOn, test } from 'bun:test';
+import { z } from 'zod';
+import { agentWorkflow } from './agent.workflow';
+import { toJsonModelMessage } from './json-ai-types';
+import { createContext } from './workflow';
+const createMockTool = (name, description, handler) => ({
+    name,
+    description,
+    parameters: z.object({
+        path: z.string().describe('The path to list files from'),
+    }),
+    handler: (_, args) => handler(args),
+});
+const listFilesTool = createMockTool('listFiles', 'List files in a directory', async ({ path }) => {
+    if (path === './src') {
+        return {
+            success: true,
+            message: {
+                type: 'text',
+                value: JSON.stringify(['index.ts', 'agent.workflow.ts']),
+            },
+        };
+    }
+    return {
+        success: true,
+        message: { type: 'text', value: JSON.stringify([]) },
+    };
+});
+test('should run agent workflow with a tool call and reply', async () => {
+    const mockResponses = [
+        {
+            role: 'assistant',
+            content: [
+                {
+                    type: 'tool-call',
+                    toolCallId: 'tool-call-1',
+                    toolName: 'listFiles',
+                    input: { path: './src' },
+                },
+            ],
+        },
+        {
+            role: 'assistant',
+            content: 'The files are index.ts and agent.workflow.ts',
+        },
+    ];
+    const initialMessages = [{ role: 'user', content: 'What are the files in the src directory?' }];
+    const allTools = [listFilesTool];
+    const spied = { fn: (_) => { } };
+    const taskEventSpy = spyOn(spied, 'fn');
+    const tools = {
+        generateText: async () => {
+            const response = mockResponses.shift();
+            return [response];
+        },
+        invokeTool: async (input) => {
+            const { toolName, input: toolInput } = input;
+            const toolInfo = allTools.find((t) => t.name === toolName);
+            if (!toolInfo)
+                throw new Error(`Tool not found: ${toolName}`);
+            // Type assertion needed for test mock - real tool handlers validate their own inputs
+            return toolInfo.handler({}, toolInput);
+        },
+        taskEvent: async (input) => {
+            taskEventSpy(input);
+        },
+    };
+    const result = await agentWorkflow({
+        userMessage: initialMessages.map(toJsonModelMessage),
+        tools: allTools,
+        systemPrompt: 'You are a helpful assistant.',
+    }, createContext(tools));
+    expect(result).toMatchObject({
+        type: 'Exit',
+        message: 'The files are index.ts and agent.workflow.ts',
+    });
+});
+test('should fail if maxToolRoundTrips is exceeded', async () => {
+    const mockResponses = Array(12).fill({
+        role: 'assistant',
+        content: [
+            {
+                type: 'tool-call',
+                toolCallId: 'tool-call-1',
+                toolName: 'listFiles',
+                input: { path: './src' },
+            },
+        ],
+    });
+    const allTools = [listFilesTool];
+    const tools = {
+        generateText: async () => {
+            const response = mockResponses.shift();
+            return [response];
+        },
+        invokeTool: async (input) => {
+            const { toolName, input: toolInput } = input;
+            const toolInfo = allTools.find((t) => t.name === toolName);
+            if (!toolInfo)
+                throw new Error(`Tool not found: ${toolName}`);
+            // Type assertion needed for test mock - real tool handlers validate their own inputs
+            return toolInfo.handler({}, toolInput);
+        },
+        taskEvent: async () => { },
+    };
+    const result = await agentWorkflow({
+        userMessage: [toJsonModelMessage({ role: 'user', content: 'List files.' })],
+        tools: allTools,
+        systemPrompt: 'You are a helpful assistant.',
+        maxToolRoundTrips: 10,
+    }, createContext(tools));
+    expect(result.type).toBe('UsageExceeded');
+    expect(result.messages).toBeDefined();
+    expect(result.messages.length).toBeGreaterThan(0);
+});
+test('should handle mixed valid and invalid tool calls by returning results for both', async () => {
+    const mockResponses = [
+        {
+            role: 'assistant',
+            content: [
+                {
+                    type: 'tool-call',
+                    toolCallId: 'call-1',
+                    toolName: 'listFiles',
+                    input: { path: './src' },
+                },
+                {
+                    type: 'tool-call',
+                    toolCallId: 'call-2',
+                    toolName: 'invalidTool',
+                    input: { some: 'param' },
+                },
+            ],
+        },
+        {
+            role: 'assistant',
+            content: 'Result received.',
+        },
+    ];
+    const allTools = [listFilesTool];
+    // We need to capture the tool results sent back to the model
+    let capturedToolResults = [];
+    const tools = {
+        generateText: async ({ messages }) => {
+            // Check if the last message is a tool result message and capture it
+            const lastMsg = messages[messages.length - 1];
+            if (lastMsg.role === 'tool') {
+                capturedToolResults = lastMsg.content;
+            }
+            const response = mockResponses.shift();
+            if (!response) {
+                // Return a dummy response to stop the loop if we run out of mocks
+                return [{ role: 'assistant', content: 'Done' }];
+            }
+            return [response];
+        },
+        invokeTool: async (input) => {
+            const { toolName, input: toolInput } = input;
+            const toolInfo = allTools.find((t) => t.name === toolName);
+            if (!toolInfo)
+                throw new Error(`Tool not found: ${toolName}`);
+            // Type assertion needed for test mock - real tool handlers validate their own inputs
+            return toolInfo.handler({}, toolInput);
+        },
+        taskEvent: async () => { },
+    };
+    await agentWorkflow({
+        userMessage: [toJsonModelMessage({ role: 'user', content: 'Do things.' })],
+        tools: allTools,
+        systemPrompt: 'You are a helpful assistant.',
+    }, createContext(tools));
+    expect(capturedToolResults).toMatchSnapshot();
+});
+//# sourceMappingURL=agent.workflow.test.js.map

package/dist/workflow/agent.workflow.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.workflow.test.js","sourceRoot":"","sources":["../../src/workflow/agent.workflow.test.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAE3B,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAE9C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAA0B,aAAa,EAAE,MAAM,kBAAkB,CAAA;AACxE,OAAO,EAA4B,kBAAkB,EAAE,MAAM,iBAAiB,CAAA;AAC9E,OAAO,EAAE,aAAa,EAAsB,MAAM,YAAY,CAAA;AAE9D,MAAM,cAAc,GAAG,CAAC,IAAY,EAAE,WAAmB,EAAE,OAA6C,EAAgB,EAAE,CAAC,CAAC;IAC1H,IAAI;IACJ,WAAW;IACX,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;QACnB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;KACzD,CAAC;IACF,OAAO,EAAE,CAAC,CAAM,EAAE,IAAS,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;CAC9C,CAAC,CAAA;AAEF,MAAM,aAAa,GAAG,cAAc,CAAC,WAAW,EAAE,2BAA2B,EAAE,KAAK,EAAE,EAAE,IAAI,EAAoB,EAAE,EAAE;IAClH,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QACrB,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE;gBACP,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;aACzD;SACF,CAAA;IACH,CAAC;IACD,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE;KACrD,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,IAAI,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;IACtE,MAAM,aAAa,GAAmB;QACpC;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,WAAW;oBACjB,UAAU,EAAE,aAAa;oBACzB,QAAQ,EAAE,WAAW;oBACrB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;iBACzB;aACF;SACF;QACD;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,8CAA8C;SACxD;KACF,CAAA;IAED,MAAM,eAAe,GAAuB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,0CAA0C,EAAE,CAAC,CAAA;IAEnH,MAAM,QAAQ,GAAG,CAAC,aAAa,CAAC,CAAA;IAChC,MAAM,KAAK,GAAG,EAAE,EAAE,EAAE,CAAC,CAAM,EAAE,EAAE,GAAE,CAAC,EAAE,CAAA;IACpC,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAA;IAEvC,MAAM,KAAK,GAAqC;QAC9C,YAAY,EAAE,KAAK,IAAI,EAAE;YACvB,MAAM,QAAQ,GAAG,aAAa,CAAC,KAAK,EAAE,CAAA;YACtC,OAAO,CAAC,QAAS,CAA0B,CAAA;QAC7C,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC1B,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,KAAK,CAAA;YAC5C,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAA;YAC1D,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAA;YAC7D,qFAAqF;YACrF,OAAO,QAAQ,CAAC,OAAO,CAAC,EAAS,EAAE,SAAgB,CAAC,CAAA;QACtD,CAAC;QACD,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACzB,YAAY,CAAC,KAAK,CAAC,CAAA;QACrB,CAAC;KACF,CAAA;IAED,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC;QACE,WAAW,EAAE,eAAe,CAAC,GAAG,CAAC,kBAAkB,CAAQ;QAC3D,KAAK,EAAE,QAAQ;QACf,YAAY,EAAE,8BAA8B;KAC7C,EACD,aAAa,CAAC,KAAK,CAAC,CACrB,CAAA;IAED,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC;QAC3B,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,8CAA8C;KACxD,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,IAAI,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;IAC9D,MAAM,aAAa,GAAmB,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC;QACnD,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,aAAa;gBACzB,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;aACzB;SACF;KACF,CAAC,CAAA;IAEF,MAAM,QAAQ,GAAG,CAAC,aAAa,CAAC,CAAA;IAEhC,MAAM,KAAK,GAAqC;QAC9C,YAAY,EAAE,KAAK,IAAI,EAAE;YACvB,MAAM,QAAQ,GAAG,aAAa,CAAC,KAAK,EAAE,CAAA;YACtC,OAAO,CAAC,QAAS,CAA0B,CAAA;QAC7C,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC1B,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,KAAK,CAAA;YAC5C,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAA;YAC1D,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAA;YAC7D,qFAAqF;YACrF,OAAO,QAAQ,CAAC,OAAO,CAAC,EAAS,EAAE,SAAgB,CAAC,CAAA;QACtD,CAAC;QACD,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;KAC1B,CAAA;IAED,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC;QACE,WAAW,EAAE,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAQ;QAClF,KAAK,EAAE,QAAQ;QACf,YAAY,EAAE,8BAA8B;QAC5C,iBAAiB,EAAE,EAAE;KACtB,EACD,aAAa,CAAC,KAAK,CAAC,CACrB,CAAA;IAED,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;IACzC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;IACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;AACnD,CAAC,CAAC,CAAA;AAEF,IAAI,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;IAChG,MAAM,aAAa,GAAmB;QACpC;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,WAAW;oBACjB,UAAU,EAAE,QAAQ;oBACpB,QAAQ,EAAE,WAAW;oBACrB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;iBACzB;gBACD;oBACE,IAAI,EAAE,WAAW;oBACjB,UAAU,EAAE,QAAQ;oBACpB,QAAQ,EAAE,aAAa;oBACvB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;iBACzB;aACF;SACF;QACD;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,kBAAkB;SAC5B;KACF,CAAA;IAED,MAAM,QAAQ,GAAG,CAAC,aAAa,CAAC,CAAA;IAEhC,6DAA6D;IAC7D,IAAI,mBAAmB,GAAU,EAAE,CAAA;IAEnC,MAAM,KAAK,GAAqC;QAC9C,YAAY,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;YACnC,oEAAoE;YACpE,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;YAC7C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,mBAAmB,GAAG,OAAO,CAAC,OAAgB,CAAA;YAChD,CAAC;YAED,MAAM,QAAQ,GAAG,aAAa,CAAC,KAAK,EAAE,CAAA;YACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,kEAAkE;gBAClE,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,CAA0B,CAAA;YAC1E,CAAC;YACD,OAAO,CAAC,QAAQ,CAA0B,CAAA;QAC5C,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC1B,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,KAAK,CAAA;YAC5C,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAA;YAC1D,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAA;YAC7D,qFAAqF;YACrF,OAAO,QAAQ,CAAC,OAAO,CAAC,EAAS,EAAE,SAAgB,CAAC,CAAA;QACtD,CAAC;QACD,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;KAC1B,CAAA;IAED,MAAM,aAAa,CACjB;QACE,WAAW,EAAE,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAQ;QACjF,KAAK,EAAE,QAAQ;QACf,YAAY,EAAE,8BAA8B;KAC7C,EACD,aAAa,CAAC,KAAK,CAAC,CACrB,CAAA;IAED,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,EAAE,CAAA;AAC/C,CAAC,CAAC,CAAA"}

package/dist/workflow/control-flow.test.d.ts

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