genaicode 0.14.2 → 1.0.0

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/send-message.js

@@ -0,0 +1,56 @@
+import { putAssistantMessage, putUserMessage } from '../../../../../../main/common/content-bus.js';
+import { askUserForInput } from '../../../../../../main/common/user-actions.js';
+export const sendMessageDef = {
+    name: 'sendMessage',
+    description: 'Send a message to the user, either to inform them about something, or ask them a question, or answer their question.',
+    parameters: {
+        type: 'object',
+        properties: {
+            message: {
+                type: 'string',
+                description: 'The message to send to the user.',
+            },
+            waitForUserResponse: {
+                type: 'boolean',
+                description: 'Whether the message is a question, and user input is expected.',
+            },
+        },
+        required: ['message', 'waitForUserResponse'],
+    },
+};
+export async function handleSendMessage(props) {
+    const { actionResult, taskExecutionPrompt, options } = props;
+    const args = actionResult.args;
+    taskExecutionPrompt.push({
+        type: 'assistant',
+        functionCalls: [actionResult],
+    });
+    putAssistantMessage(args.message);
+    if (args.waitForUserResponse) {
+        const response = await askUserForInput('Your answer', '', options, true);
+        putUserMessage(response.answer);
+        taskExecutionPrompt.push({
+            type: 'user',
+            text: response.answer,
+            functionResponses: [
+                {
+                    name: actionResult.name,
+                    call_id: actionResult.id || undefined,
+                },
+            ],
+        });
+    }
+    else {
+        taskExecutionPrompt.push({
+            type: 'user',
+            functionResponses: [
+                {
+                    name: actionResult.name,
+                    call_id: actionResult.id || undefined,
+                },
+            ],
+        });
+    }
+    return { shouldBreakOuter: false, commandsExecutedIncrement: 0 };
+}
+//# sourceMappingURL=send-message.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/send-message.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"send-message.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/commands/send-message.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,8CAA8C,CAAC;AACnG,OAAO,EAAE,eAAe,EAAE,MAAM,+CAA+C,CAAC;AAGhF,MAAM,CAAC,MAAM,cAAc,GAAgB;IACzC,IAAI,EAAE,aAAa;IACnB,WAAW,EACT,sHAAsH;IACxH,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kCAAkC;aAChD;YACD,mBAAmB,EAAE;gBACnB,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,gEAAgE;aAC9E;SACF;QACD,QAAQ,EAAE,CAAC,SAAS,EAAE,qBAAqB,CAAC;KAC7C;CACF,CAAC;AAIF,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,KAAwF;IAExF,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAC7D,MAAM,IAAI,GAAG,YAAY,CAAC,IAAuB,CAAC;IAElD,mBAAmB,CAAC,IAAI,CAAC;QACvB,IAAI,EAAE,WAAW;QACjB,aAAa,EAAE,CAAC,YAAY,CAAC;KAC9B,CAAC,CAAC;IAEH,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClC,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;QACzE,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAChC,mBAAmB,CAAC,IAAI,CAAC;YACvB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ,CAAC,MAAM;YACrB,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS;iBACtC;aACF;SACF,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,mBAAmB,CAAC,IAAI,CAAC;YACvB,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS;iBACtC;aACF;SACF,CAAC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,EAAE,CAAC;AACnE,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/set-execution-plan.d.ts

@@ -0,0 +1,4 @@
+import { FunctionDef } from '../../../../../../ai-service/common-types.js';
+import { CommandHandlerBaseProps, CommandHandlerResult } from '../container-commands-types.js';
+export declare const setExecutionPlanDef: FunctionDef;
+export declare function handleSetExecutionPlan(props: Pick<CommandHandlerBaseProps, 'actionResult' | 'taskExecutionPrompt'>): Promise<CommandHandlerResult>;

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/set-execution-plan.js

@@ -0,0 +1,56 @@
+import { putContainerLog } from '../../../../../../main/common/content-bus.js';
+import { updateExecutionPlanDef } from './update-execution-plan.js';
+export const setExecutionPlanDef = {
+    name: 'setExecutionPlan',
+    description: 'Record a concise execution plan to follow in subsequent steps.',
+    parameters: {
+        type: 'object',
+        properties: {
+            plan: {
+                type: 'array',
+                description: 'The steps of the execution plan. This is the latest most up to date version of the plan, which takes into account all prior context and findings.',
+                items: {
+                    type: 'object',
+                    description: 'A single step in the execution plan.',
+                    properties: {
+                        id: {
+                            type: 'string',
+                            description: 'A unique identifier for this step.',
+                        },
+                        description: {
+                            type: 'string',
+                            description: 'A brief description of this step.',
+                        },
+                        dependsOn: {
+                            type: 'array',
+                            description: 'The IDs of the steps that this step depends on.',
+                            items: {
+                                type: 'string',
+                            },
+                        },
+                        state: updateExecutionPlanDef.parameters.properties.state,
+                        statusUpdate: updateExecutionPlanDef.parameters.properties.statusUpdate,
+                    },
+                    required: ['id', 'description', 'dependsOn'],
+                },
+            },
+        },
+        required: ['plan'],
+    },
+};
+export async function handleSetExecutionPlan(props) {
+    const { actionResult, taskExecutionPrompt } = props;
+    const args = actionResult.args;
+    putContainerLog('info', '📝 Execution plan recorded.', args);
+    taskExecutionPrompt.push({
+        type: 'assistant',
+        text: 'Setting execution plan.',
+        functionCalls: [actionResult],
+    }, {
+        type: 'user',
+        text: 'Please follow the execution plan carefully.',
+        functionResponses: [{ name: actionResult.name, call_id: actionResult.id || undefined }],
+    });
+    return { shouldBreakOuter: false, commandsExecutedIncrement: 0 };
+}
+//# sourceMappingURL=set-execution-plan.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/set-execution-plan.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"set-execution-plan.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/commands/set-execution-plan.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,8CAA8C,CAAC;AAE/E,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAEpE,MAAM,CAAC,MAAM,mBAAmB,GAAgB;IAC9C,IAAI,EAAE,kBAAkB;IACxB,WAAW,EAAE,gEAAgE;IAC7E,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,OAAO;gBACb,WAAW,EACT,mJAAmJ;gBACrJ,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,sCAAsC;oBACnD,UAAU,EAAE;wBACV,EAAE,EAAE;4BACF,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,oCAAoC;yBAClD;wBACD,WAAW,EAAE;4BACX,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,mCAAmC;yBACjD;wBACD,SAAS,EAAE;4BACT,IAAI,EAAE,OAAO;4BACb,WAAW,EAAE,iDAAiD;4BAC9D,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;6BACf;yBACF;wBACD,KAAK,EAAE,sBAAsB,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK;wBACzD,YAAY,EAAE,sBAAsB,CAAC,UAAU,CAAC,UAAU,CAAC,YAAY;qBACxE;oBACD,QAAQ,EAAE,CAAC,IAAI,EAAE,aAAa,EAAE,WAAW,CAAC;iBAC7C;aACF;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB;CACF,CAAC;AAIF,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,KAA4E;IAE5E,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,GAAG,KAAK,CAAC;IACpD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAwC,CAAC;IACnE,eAAe,CAAC,MAAM,EAAE,6BAA6B,EAAE,IAAI,CAAC,CAAC;IAE7D,mBAAmB,CAAC,IAAI,CACtB;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,yBAAyB;QAC/B,aAAa,EAAE,CAAC,YAAY,CAAC;KAC9B,EACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,6CAA6C;QACnD,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS,EAAE,CAAC;KACxF,CACF,CAAC;IACF,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,EAAE,CAAC;AACnE,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/update-execution-plan.d.ts

@@ -0,0 +1,4 @@
+import { FunctionDef } from '../../../../../../ai-service/common-types.js';
+import { CommandHandlerBaseProps, CommandHandlerResult } from '../container-commands-types.js';
+export declare const updateExecutionPlanDef: FunctionDef;
+export declare function handleUpdateExecutionPlan(props: Pick<CommandHandlerBaseProps, 'actionResult' | 'taskExecutionPrompt'>): Promise<CommandHandlerResult>;

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/update-execution-plan.js

@@ -0,0 +1,39 @@
+import { putContainerLog } from '../../../../../../main/common/content-bus.js';
+export const updateExecutionPlanDef = {
+    name: 'updateExecutionPlan',
+    description: 'Update progress/status against the previously set execution plan.',
+    parameters: {
+        type: 'object',
+        properties: {
+            id: {
+                type: 'string',
+                description: 'The ID of the step to update.',
+            },
+            statusUpdate: {
+                type: 'string',
+                description: 'The updated status of the step.',
+            },
+            state: {
+                type: 'string',
+                description: 'The state of the step, e.g., "pending", "in-progress", "completed", "failed", "skipped".',
+                enum: ['pending', 'in-progress', 'completed', 'failed', 'skipped'],
+            },
+        },
+        required: ['id', 'statusUpdate', 'state'],
+    },
+};
+export async function handleUpdateExecutionPlan(props) {
+    const { actionResult, taskExecutionPrompt } = props;
+    const args = actionResult.args;
+    putContainerLog('info', '📈 Execution plan updated.', args);
+    taskExecutionPrompt.push({
+        type: 'assistant',
+        text: `Updating execution plan for step ${args.id}.`,
+        functionCalls: [actionResult],
+    }, {
+        type: 'user',
+        functionResponses: [{ name: actionResult.name, call_id: actionResult.id || undefined }],
+    });
+    return { shouldBreakOuter: false, commandsExecutedIncrement: 0 };
+}
+//# sourceMappingURL=update-execution-plan.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/update-execution-plan.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-execution-plan.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/commands/update-execution-plan.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,8CAA8C,CAAC;AAG/E,MAAM,CAAC,MAAM,sBAAsB,GAAgB;IACjD,IAAI,EAAE,qBAAqB;IAC3B,WAAW,EAAE,mEAAmE;IAChF,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,EAAE,EAAE;gBACF,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,+BAA+B;aAC7C;YACD,YAAY,EAAE;gBACZ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;aAC/C;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0FAA0F;gBACvG,IAAI,EAAE,CAAC,SAAS,EAAE,aAAa,EAAE,WAAW,EAAE,QAAQ,EAAE,SAAS,CAAC;aACnE;SACF;QACD,QAAQ,EAAE,CAAC,IAAI,EAAE,cAAc,EAAE,OAAO,CAAC;KAC1C;CACF,CAAC;AAQF,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,KAA4E;IAE5E,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,GAAG,KAAK,CAAC;IACpD,MAAM,IAAI,GAAG,YAAY,CAAC,IAA+B,CAAC;IAC1D,eAAe,CAAC,MAAM,EAAE,4BAA4B,EAAE,IAAI,CAAC,CAAC;IAC5D,mBAAmB,CAAC,IAAI,CACtB;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,oCAAoC,IAAI,CAAC,EAAE,GAAG;QACpD,aAAa,EAAE,CAAC,YAAY,CAAC;KAC9B,EACD;QACE,IAAI,EAAE,MAAM;QACZ,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS,EAAE,CAAC;KACxF,CACF,CAAC;IACF,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,EAAE,CAAC;AACnE,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/view-file.d.ts

@@ -0,0 +1,10 @@
+import { FunctionDef } from '../../../../../../ai-service/common-types.js';
+import { CommandHandlerBaseProps, CommandHandlerResult } from '../container-commands-types.js';
+export declare const viewFileDef: FunctionDef;
+export type ViewFileArgs = {
+    reasoning: string;
+    filePath: string;
+    startLine?: number;
+    endLine?: number;
+};
+export declare function handleViewFile(props: CommandHandlerBaseProps): Promise<CommandHandlerResult>;

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/view-file.js

@@ -0,0 +1,95 @@
+import { putContainerLog } from '../../../../../../main/common/content-bus.js';
+import { getFileContentFromContainer, executeCommand } from '../utils/docker-utils.js';
+export const viewFileDef = {
+    name: 'viewFile',
+    description: 'View a text file or list directory contents inside the container. You can view the full file or a specific range of lines.',
+    parameters: {
+        type: 'object',
+        properties: {
+            reasoning: {
+                type: 'string',
+                description: 'Explanation of why this path needs to be viewed.',
+            },
+            filePath: {
+                type: 'string',
+                description: 'The absolute path to the file or directory inside the container.',
+            },
+            startLine: {
+                type: 'number',
+                description: 'The starting line number to view (1-indexed). Only applicable for files.',
+            },
+            endLine: {
+                type: 'number',
+                description: 'The ending line number to view (1-indexed). Only applicable for files.',
+            },
+        },
+        required: ['reasoning', 'filePath'],
+    },
+};
+export async function handleViewFile(props) {
+    const { actionResult, taskExecutionPrompt, container } = props;
+    const args = actionResult.args;
+    const { reasoning, filePath, startLine, endLine } = args;
+    putContainerLog('info', `Viewing path: ${reasoning}`, args, 'command');
+    try {
+        const dirCheck = await executeCommand(container, '/bin/sh', `test -d "${filePath}"`, '', '/');
+        let finalContent;
+        if (dirCheck.exitCode === 0) {
+            if (startLine || endLine) {
+                throw new Error('startLine and endLine parameters are not applicable for directories.');
+            }
+            const { output, exitCode } = await executeCommand(container, '/bin/sh', `ls -la "${filePath}"`, '', '/');
+            if (exitCode !== 0) {
+                throw new Error(`Failed to list directory contents for ${filePath}: ${output}`);
+            }
+            finalContent = `Directory listing for ${filePath}:\n\n${output}`;
+            putContainerLog('info', `Successfully listed directory ${filePath}`, { filePath }, 'command');
+        }
+        else {
+            let content = await getFileContentFromContainer(container, filePath);
+            const lines = content.split('\n');
+            let resultMessage = `File content of ${filePath}:\n\n`;
+            if (startLine || endLine) {
+                const start = startLine ? startLine - 1 : 0;
+                const end = endLine ? endLine : lines.length;
+                const slicedLines = lines.slice(start, end);
+                content = slicedLines.join('\n');
+                resultMessage = `File content of ${filePath} from line ${startLine || 1} to ${endLine || lines.length}:\n\n`;
+            }
+            finalContent = resultMessage + content;
+            putContainerLog('info', `Successfully viewed file ${filePath}`, { filePath }, 'command');
+        }
+        taskExecutionPrompt.push({
+            type: 'assistant',
+            functionCalls: [actionResult],
+        }, {
+            type: 'user',
+            functionResponses: [
+                {
+                    name: actionResult.name,
+                    call_id: actionResult.id,
+                    content: finalContent,
+                },
+            ],
+        });
+    }
+    catch (e) {
+        const errMessage = e instanceof Error ? e.message : String(e);
+        putContainerLog('error', 'An error occurred while viewing the path', { error: errMessage });
+        taskExecutionPrompt.push({
+            type: 'assistant',
+            functionCalls: [actionResult],
+        }, {
+            type: 'user',
+            functionResponses: [
+                {
+                    name: actionResult.name,
+                    call_id: actionResult.id,
+                    content: `View path failed: ${errMessage}`,
+                },
+            ],
+        });
+    }
+    return { commandsExecutedIncrement: 1 };
+}
+//# sourceMappingURL=view-file.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/view-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"view-file.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/commands/view-file.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,8CAA8C,CAAC;AAC/E,OAAO,EAAE,2BAA2B,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAGvF,MAAM,CAAC,MAAM,WAAW,GAAgB;IACtC,IAAI,EAAE,UAAU;IAChB,WAAW,EACT,4HAA4H;IAC9H,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kDAAkD;aAChE;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kEAAkE;aAChF;YACD,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0EAA0E;aACxF;YACD,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,wEAAwE;aACtF;SACF;QACD,QAAQ,EAAE,CAAC,WAAW,EAAE,UAAU,CAAC;KACpC;CACF,CAAC;AASF,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,KAA8B;IACjE,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IAC/D,MAAM,IAAI,GAAG,YAAY,CAAC,IAAoB,CAAC;IAC/C,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAEzD,eAAe,CAAC,MAAM,EAAE,iBAAiB,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IAEvE,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,SAAS,EAAE,SAAS,EAAE,YAAY,QAAQ,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;QAC9F,IAAI,YAAoB,CAAC;QAEzB,IAAI,QAAQ,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC5B,IAAI,SAAS,IAAI,OAAO,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,sEAAsE,CAAC,CAAC;YAC1F,CAAC;YACD,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,cAAc,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,QAAQ,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YACzG,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,yCAAyC,QAAQ,KAAK,MAAM,EAAE,CAAC,CAAC;YAClF,CAAC;YACD,YAAY,GAAG,yBAAyB,QAAQ,QAAQ,MAAM,EAAE,CAAC;YACjE,eAAe,CAAC,MAAM,EAAE,iCAAiC,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC;QAChG,CAAC;aAAM,CAAC;YACN,IAAI,OAAO,GAAG,MAAM,2BAA2B,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACrE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,aAAa,GAAG,mBAAmB,QAAQ,OAAO,CAAC;YAEvD,IAAI,SAAS,IAAI,OAAO,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5C,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;gBAC7C,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;gBAC5C,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjC,aAAa,GAAG,mBAAmB,QAAQ,cAAc,SAAS,IAAI,CAAC,OAAO,OAAO,IAAI,KAAK,CAAC,MAAM,OAAO,CAAC;YAC/G,CAAC;YACD,YAAY,GAAG,aAAa,GAAG,OAAO,CAAC;YACvC,eAAe,CAAC,MAAM,EAAE,4BAA4B,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC;QAC3F,CAAC;QAED,mBAAmB,CAAC,IAAI,CACtB;YACE,IAAI,EAAE,WAAW;YACjB,aAAa,EAAE,CAAC,YAAY,CAAC;SAC9B,EACD;YACE,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,OAAO,EAAE,YAAY,CAAC,EAAE;oBACxB,OAAO,EAAE,YAAY;iBACtB;aACF;SACF,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,UAAU,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAC9D,eAAe,CAAC,OAAO,EAAE,0CAA0C,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5F,mBAAmB,CAAC,IAAI,CACtB;YACE,IAAI,EAAE,WAAW;YACjB,aAAa,EAAE,CAAC,YAAY,CAAC;SAC9B,EACD;YACE,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,OAAO,EAAE,YAAY,CAAC,EAAE;oBACxB,OAAO,EAAE,qBAAqB,UAAU,EAAE;iBAC3C;aACF;SACF,CACF,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,yBAAyB,EAAE,CAAC,EAAE,CAAC;AAC1C,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/web-search.d.ts

@@ -0,0 +1,2 @@
+import { CommandHandlerBaseProps, CommandHandlerResult } from '../container-commands-types.js';
+export declare function handleWebSearch(props: CommandHandlerBaseProps): Promise<CommandHandlerResult>;

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/web-search.js

@@ -0,0 +1,96 @@
+import { ModelType } from '../../../../../../ai-service/common-types.js';
+import { putContainerLog } from '../../../../../../main/common/content-bus.js';
+export async function handleWebSearch(props) {
+    const { actionResult, generateContentFn, options, taskExecutionPrompt } = props;
+    const { query } = actionResult.args;
+    taskExecutionPrompt.push({
+        type: 'assistant',
+        functionCalls: [actionResult],
+    });
+    if (!query) {
+        putContainerLog('warn', 'Web search command called without a query.');
+        taskExecutionPrompt.push({
+            type: 'user',
+            functionResponses: [
+                {
+                    name: actionResult.name,
+                    call_id: actionResult.id || undefined,
+                    content: 'Web search failed: Missing query',
+                },
+            ],
+        });
+        return {
+            shouldBreakOuter: false,
+            commandsExecutedIncrement: 1,
+        };
+    }
+    putContainerLog('info', `Performing web search for: "${query}"`);
+    try {
+        const searchResponse = await generateContentFn([
+            {
+                type: 'user',
+                text: query,
+            },
+        ], {
+            temperature: 0.7,
+            modelType: ModelType.LITE,
+            expectedResponseType: {
+                webSearch: true,
+            },
+        }, options);
+        const webSearchResult = searchResponse.find((item) => item.type === 'webSearch');
+        if (webSearchResult) {
+            putContainerLog('info', 'Web search completed successfully.', webSearchResult);
+            taskExecutionPrompt.push({
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: actionResult.name,
+                        call_id: actionResult.id || undefined,
+                        content: JSON.stringify(webSearchResult),
+                    },
+                ],
+            });
+            return {
+                shouldBreakOuter: false,
+                commandsExecutedIncrement: 1,
+            };
+        }
+        else {
+            putContainerLog('warn', 'Web search did not return any results.');
+            taskExecutionPrompt.push({
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: actionResult.name,
+                        call_id: actionResult.id || undefined,
+                        content: 'Web search failed: No results found',
+                    },
+                ],
+            });
+            return {
+                shouldBreakOuter: false,
+                commandsExecutedIncrement: 1,
+            };
+        }
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : 'An unknown error occurred';
+        putContainerLog('error', `Web search failed: ${errorMessage}`, { error });
+        taskExecutionPrompt.push({
+            type: 'user',
+            functionResponses: [
+                {
+                    name: actionResult.name,
+                    call_id: actionResult.id || undefined,
+                    content: `Web search failed: ${errorMessage}`,
+                },
+            ],
+        });
+        return {
+            shouldBreakOuter: false,
+            commandsExecutedIncrement: 1,
+        };
+    }
+}
+//# sourceMappingURL=web-search.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/web-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"web-search.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/commands/web-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,8CAA8C,CAAC;AACzE,OAAO,EAAE,eAAe,EAAE,MAAM,8CAA8C,CAAC;AAI/E,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,KAA8B;IAClE,MAAM,EAAE,YAAY,EAAE,iBAAiB,EAAE,OAAO,EAAE,mBAAmB,EAAE,GAAG,KAAK,CAAC;IAChF,MAAM,EAAE,KAAK,EAAE,GAAG,YAAY,CAAC,IAAqB,CAAC;IAErD,mBAAmB,CAAC,IAAI,CAAC;QACvB,IAAI,EAAE,WAAW;QACjB,aAAa,EAAE,CAAC,YAAY,CAAC;KAC9B,CAAC,CAAC;IAEH,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,eAAe,CAAC,MAAM,EAAE,4CAA4C,CAAC,CAAC;QACtE,mBAAmB,CAAC,IAAI,CAAC;YACvB,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS;oBACrC,OAAO,EAAE,kCAAkC;iBAC5C;aACF;SACF,CAAC,CAAC;QACH,OAAO;YACL,gBAAgB,EAAE,KAAK;YACvB,yBAAyB,EAAE,CAAC;SAC7B,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,MAAM,EAAE,+BAA+B,KAAK,GAAG,CAAC,CAAC;IAEjE,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAC5C;YACE;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,KAAK;aACZ;SACF,EACD;YACE,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,SAAS,CAAC,IAAI;YACzB,oBAAoB,EAAE;gBACpB,SAAS,EAAE,IAAI;aAChB;SACF,EACD,OAAO,CACR,CAAC;QAEF,MAAM,eAAe,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAEjF,IAAI,eAAe,EAAE,CAAC;YACpB,eAAe,CAAC,MAAM,EAAE,oCAAoC,EAAE,eAAe,CAAC,CAAC;YAC/E,mBAAmB,CAAC,IAAI,CAAC;gBACvB,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,YAAY,CAAC,IAAI;wBACvB,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS;wBACrC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC;qBACzC;iBACF;aACF,CAAC,CAAC;YACH,OAAO;gBACL,gBAAgB,EAAE,KAAK;gBACvB,yBAAyB,EAAE,CAAC;aAC7B,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,MAAM,EAAE,wCAAwC,CAAC,CAAC;YAClE,mBAAmB,CAAC,IAAI,CAAC;gBACvB,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,YAAY,CAAC,IAAI;wBACvB,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS;wBACrC,OAAO,EAAE,qCAAqC;qBAC/C;iBACF;aACF,CAAC,CAAC;YACH,OAAO;gBACL,gBAAgB,EAAE,KAAK;gBACvB,yBAAyB,EAAE,CAAC;aAC7B,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,2BAA2B,CAAC;QAC1F,eAAe,CAAC,OAAO,EAAE,sBAAsB,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1E,mBAAmB,CAAC,IAAI,CAAC;YACvB,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,OAAO,EAAE,YAAY,CAAC,EAAE,IAAI,SAAS;oBACrC,OAAO,EAAE,sBAAsB,YAAY,EAAE;iBAC9C;aACF;SACF,CAAC,CAAC;QACH,OAAO;YACL,gBAAgB,EAAE,KAAK;YACvB,yBAAyB,EAAE,CAAC;SAC7B,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/wrap-context.d.ts

@@ -0,0 +1,4 @@
+import { GenerateContentFunction, PromptItem } from '../../../../../../ai-service/common-types.js';
+import { ActionHandlerProps } from '../../../step-iterate-types.js';
+import Docker from 'dockerode';
+export declare function maybeWrapContext(systemPrompt: PromptItem, taskPrompt: PromptItem, taskExecutionPrompt: PromptItem[], generateContentFn: GenerateContentFunction, container: Docker.Container, options: ActionHandlerProps['options']): Promise<void>;

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/wrap-context.js

@@ -0,0 +1,123 @@
+import { ModelType, } from '../../../../../../ai-service/common-types.js';
+import { putContainerLog } from '../../../../../../main/common/content-bus.js';
+import { setExecutionPlanDef } from './set-execution-plan.js';
+const MAX_CONTEXT_ITEMS = 24;
+const WRAP_RATIO = 2 / 3;
+export async function maybeWrapContext(systemPrompt, taskPrompt, taskExecutionPrompt, generateContentFn, container, options) {
+    if (taskExecutionPrompt.length < MAX_CONTEXT_ITEMS) {
+        return;
+    }
+    putContainerLog('info', 'Context is getting long, wrapping it up.');
+    await handleWrapContext({
+        actionResult: {
+            name: 'wrapContext',
+            args: {}, // This will be populated by the LLM inside handleWrapContext
+        },
+        systemPrompt,
+        taskPrompt,
+        taskExecutionPrompt,
+        container,
+        options,
+        generateContentFn,
+        maxContextItems: MAX_CONTEXT_ITEMS,
+        maxContextSize: 0, // not used
+        maxOutputLength: 0, // not used
+    });
+}
+const wrapContextInternalDef = {
+    name: 'wrapContextInternal',
+    description: 'Internal tool to summarize the conversation history.',
+    parameters: {
+        type: 'object',
+        properties: {
+            summary: {
+                type: 'string',
+                description: 'A summary of prior steps and important findings to keep for next actions.',
+            },
+            plan: setExecutionPlanDef.parameters.properties.plan,
+            progress: {
+                type: 'string',
+                description: 'What was achieved so far? What are the outcomes?',
+            },
+            importantFiles: {
+                type: 'array',
+                items: {
+                    type: 'string',
+                    description: `Paths to important files or directories to keep in mind for next actions.
+It is critically IMPORTANT to keep track of files which are essential for the task. For example the files that were modified or created during the task so far.`,
+                },
+            },
+            nextStep: {
+                type: 'string',
+                description: 'The next step to take in the process.',
+            },
+        },
+        required: ['summary', 'plan', 'progress', 'importantFiles', 'nextStep'],
+    },
+};
+async function handleWrapContext(props) {
+    const { taskExecutionPrompt, generateContentFn, options, systemPrompt, taskPrompt } = props;
+    let splitPoint = Math.floor(taskExecutionPrompt.length * WRAP_RATIO);
+    // Adjust split point to not separate a function call from its response.
+    // Iterate backwards from the middle to find a safe point that is not right after an assistant's function call.
+    while (splitPoint > 0) {
+        const precedingItem = taskExecutionPrompt[splitPoint - 1];
+        if (precedingItem.type === 'assistant' && precedingItem.functionCalls?.length) {
+            // This is an assistant message with a function call. It's unsafe to split right after it.
+            // Move the split point before this message to keep the call and its subsequent response together.
+            splitPoint--;
+        }
+        else {
+            // This is a safe split point (e.g., after a user message or an assistant message without a call).
+            break;
+        }
+    }
+    const partToWrap = taskExecutionPrompt.slice(0, splitPoint);
+    const partToKeep = taskExecutionPrompt.slice(splitPoint);
+    if (partToWrap.length === 0) {
+        // Nothing to wrap, probably because the first half was all function calls.
+        return { shouldBreakOuter: false, commandsExecutedIncrement: 0 };
+    }
+    const summarizationPrompt = [
+        systemPrompt,
+        taskPrompt,
+        ...partToWrap,
+        {
+            type: 'user',
+            text: 'Please summarize the conversation so far. Include the current plan, progress, important files, and what the next immediate step should be, based on the last message.',
+        },
+    ];
+    const response = await generateContentFn(summarizationPrompt, {
+        functionDefs: [wrapContextInternalDef],
+        requiredFunctionName: wrapContextInternalDef.name,
+        modelType: ModelType.LITE,
+    }, options);
+    const wrapContextCall = response
+        .filter((item) => item.type === 'functionCall')
+        .filter((item) => item.functionCall.name === wrapContextInternalDef.name)
+        .map((item) => item.functionCall)[0];
+    if (!wrapContextCall || !wrapContextCall.args) {
+        putContainerLog('warn', 'Failed to generate a valid context summary.');
+        // If summarization fails, we just keep the context as is to avoid breaking the flow.
+        return { shouldBreakOuter: false, commandsExecutedIncrement: 0 };
+    }
+    putContainerLog('info', '🗂️ Context wrapped by internal operator.', wrapContextCall.args);
+    const assistantMsg = {
+        type: 'assistant',
+        text: 'Wrapping context for continued processing. This summary will help maintain continuity, it contains the current plan, list of important files, and progress made so far, as well as the next step to take.',
+        functionCalls: [wrapContextCall],
+    };
+    const userResp = {
+        type: 'user',
+        text: `Please continue with the next step: ${wrapContextCall.args.nextStep}`,
+        functionResponses: [
+            {
+                name: wrapContextCall.name,
+                call_id: wrapContextCall.id || undefined,
+            },
+        ],
+    };
+    taskExecutionPrompt.splice(0, taskExecutionPrompt.length, assistantMsg, userResp, ...partToKeep);
+    return { shouldBreakOuter: false, commandsExecutedIncrement: 0 };
+}
+//# sourceMappingURL=wrap-context.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/commands/wrap-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrap-context.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/commands/wrap-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,SAAS,GAEV,MAAM,8CAA8C,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,8CAA8C,CAAC;AAE/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAI9D,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,UAAU,GAAG,CAAC,GAAG,CAAC,CAAC;AAEzB,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,YAAwB,EACxB,UAAsB,EACtB,mBAAiC,EACjC,iBAA0C,EAC1C,SAA2B,EAC3B,OAAsC;IAEtC,IAAI,mBAAmB,CAAC,MAAM,GAAG,iBAAiB,EAAE,CAAC;QACnD,OAAO;IACT,CAAC;IAED,eAAe,CAAC,MAAM,EAAE,0CAA0C,CAAC,CAAC;IAEpE,MAAM,iBAAiB,CAAC;QACtB,YAAY,EAAE;YACZ,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,EAAE,EAAE,6DAA6D;SACxE;QACD,YAAY;QACZ,UAAU;QACV,mBAAmB;QACnB,SAAS;QACT,OAAO;QACP,iBAAiB;QACjB,eAAe,EAAE,iBAAiB;QAClC,cAAc,EAAE,CAAC,EAAE,WAAW;QAC9B,eAAe,EAAE,CAAC,EAAE,WAAW;KAChC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,sBAAsB,GAAgB;IAC1C,IAAI,EAAE,qBAAqB;IAC3B,WAAW,EAAE,sDAAsD;IACnE,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,2EAA2E;aACzF;YACD,IAAI,EAAE,mBAAmB,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI;YACpD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kDAAkD;aAChE;YACD,cAAc,EAAE;gBACd,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE;gKACyI;iBACvJ;aACF;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,uCAAuC;aACrD;SACF;QACD,QAAQ,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,UAAU,CAAC;KACxE;CACF,CAAC;AAIF,KAAK,UAAU,iBAAiB,CAAC,KAA6B;IAC5D,MAAM,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,KAAK,CAAC;IAE5F,IAAI,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC;IAErE,wEAAwE;IACxE,+GAA+G;IAC/G,OAAO,UAAU,GAAG,CAAC,EAAE,CAAC;QACtB,MAAM,aAAa,GAAG,mBAAmB,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QAC1D,IAAI,aAAa,CAAC,IAAI,KAAK,WAAW,IAAI,aAAa,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;YAC9E,0FAA0F;YAC1F,kGAAkG;YAClG,UAAU,EAAE,CAAC;QACf,CAAC;aAAM,CAAC;YACN,kGAAkG;YAClG,MAAM;QACR,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,mBAAmB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAEzD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,2EAA2E;QAC3E,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,EAAE,CAAC;IACnE,CAAC;IAED,MAAM,mBAAmB,GAAiB;QACxC,YAAY;QACZ,UAAU;QACV,GAAG,UAAU;QACb;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,uKAAuK;SAC9K;KACF,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACtC,mBAAmB,EACnB;QACE,YAAY,EAAE,CAAC,sBAAsB,CAAC;QACtC,oBAAoB,EAAE,sBAAsB,CAAC,IAAI;QACjD,SAAS,EAAE,SAAS,CAAC,IAAI;KAC1B,EACD,OAAO,CACR,CAAC;IAEF,MAAM,eAAe,GACnB,QAAQ;SACL,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,KAAK,sBAAsB,CAAC,IAAI,CAAC;SACxE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CACnC,CAAC,CAAC,CAAC,CAAC;IAEL,IAAI,CAAC,eAAe,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;QAC9C,eAAe,CAAC,MAAM,EAAE,6CAA6C,CAAC,CAAC;QACvE,qFAAqF;QACrF,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,EAAE,CAAC;IACnE,CAAC;IAED,eAAe,CAAC,MAAM,EAAE,2CAA2C,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;IAE3F,MAAM,YAAY,GAAe;QAC/B,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,2MAA2M;QACjN,aAAa,EAAE,CAAC,eAAe,CAAC;KACjC,CAAC;IACF,MAAM,QAAQ,GAAe;QAC3B,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,uCAAuC,eAAe,CAAC,IAAI,CAAC,QAAQ,EAAE;QAC5E,iBAAiB,EAAE;YACjB;gBACE,IAAI,EAAE,eAAe,CAAC,IAAI;gBAC1B,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,SAAS;aACzC;SACF;KACF,CAAC;IAEF,mBAAmB,CAAC,MAAM,CAAC,CAAC,EAAE,mBAAmB,CAAC,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,CAAC;IAEjG,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,EAAE,CAAC;AACnE,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/container-commands-registry.d.ts

@@ -0,0 +1,11 @@
+import { FunctionDef } from '../../../../../ai-service/common-types.js';
+import { CommandHandler } from './container-commands-types.js';
+/**
+ * Gets all function definitions for the container commands.
+ * It resolves any definition functions into concrete FunctionDef objects.
+ */
+export declare function getContainerCommandDefs(): FunctionDef[];
+/**
+ * Gets the handler for a specific command by name.
+ */
+export declare function getContainerCommandHandler(name: string): CommandHandler | undefined;

package/dist/prompt/steps/step-iterate/handlers/container-task/container-commands-registry.js

@@ -0,0 +1,50 @@
+import { completeTaskDef, handleCompleteTask } from './commands/complete-task.js';
+import { failTaskDef, handleFailTask } from './commands/fail-task.js';
+import { setExecutionPlanDef, handleSetExecutionPlan } from './commands/set-execution-plan.js';
+import { updateExecutionPlanDef, handleUpdateExecutionPlan } from './commands/update-execution-plan.js';
+import { sendMessageDef, handleSendMessage } from './commands/send-message.js';
+import { runCommandDef, handleRunCommand } from './commands/run-command.js';
+import { getCopyToContainerDef, handleCopyToContainer } from './commands/copy-to-container.js';
+import { getCopyFromContainerDef, handleCopyFromContainer } from './commands/copy-from-container.js';
+import { handleWebSearch } from './commands/web-search.js';
+import { webSearchDef } from '../../../../function-defs/web-search.js';
+import { requestSecretDef, handleRequestSecret } from './commands/request-secret.js';
+import { gainKnowledgeDef, handleGainKnowledge } from './commands/gain-knowledge.js';
+import { queryKnowledgeDef, handleQueryKnowledge } from './commands/query-knowledge.js';
+import { viewFileDef, handleViewFile } from './commands/view-file.js';
+import { editFileDef, handleEditFile } from './commands/edit-file.js';
+const commandRegistry = new Map();
+function registerCommand(name, def, handler) {
+    commandRegistry.set(name, { def, handler: handler });
+}
+// Register all available commands
+registerCommand('completeTask', completeTaskDef, handleCompleteTask);
+registerCommand('failTask', failTaskDef, handleFailTask);
+registerCommand('setExecutionPlan', setExecutionPlanDef, handleSetExecutionPlan);
+registerCommand('updateExecutionPlan', updateExecutionPlanDef, handleUpdateExecutionPlan);
+registerCommand('sendMessage', sendMessageDef, handleSendMessage);
+registerCommand('runCommand', runCommandDef, handleRunCommand);
+registerCommand('copyToContainer', getCopyToContainerDef, handleCopyToContainer);
+registerCommand('copyFromContainer', getCopyFromContainerDef, handleCopyFromContainer);
+registerCommand('webSearch', webSearchDef, handleWebSearch);
+registerCommand('requestSecret', requestSecretDef, handleRequestSecret);
+registerCommand('gainKnowledge', gainKnowledgeDef, handleGainKnowledge);
+registerCommand('queryKnowledge', queryKnowledgeDef, handleQueryKnowledge);
+registerCommand('viewFile', viewFileDef, handleViewFile);
+registerCommand('editFile', editFileDef, handleEditFile);
+/**
+ * Gets all function definitions for the container commands.
+ * It resolves any definition functions into concrete FunctionDef objects.
+ */
+export function getContainerCommandDefs() {
+    return Array.from(commandRegistry.values()).map((command) => {
+        return typeof command.def === 'function' ? command.def() : command.def;
+    });
+}
+/**
+ * Gets the handler for a specific command by name.
+ */
+export function getContainerCommandHandler(name) {
+    return commandRegistry.get(name)?.handler;
+}
+//# sourceMappingURL=container-commands-registry.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/container-commands-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"container-commands-registry.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-iterate/handlers/container-task/container-commands-registry.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAClF,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,kCAAkC,CAAC;AAC/F,OAAO,EAAE,sBAAsB,EAAE,yBAAyB,EAAE,MAAM,qCAAqC,CAAC;AACxG,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/E,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC5E,OAAO,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAC/F,OAAO,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,mCAAmC,CAAC;AACrG,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AACvE,OAAO,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AACrF,OAAO,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AACrF,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AACxF,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAEtE,MAAM,eAAe,GAAG,IAAI,GAAG,EAAmB,CAAC;AAEnD,SAAS,eAAe,CACtB,IAAY,EACZ,GAAsC,EACtC,OAA0B;IAE1B,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAkD,EAAE,CAAC,CAAC;AAClG,CAAC;AAED,kCAAkC;AAClC,eAAe,CAAC,cAAc,EAAE,eAAe,EAAE,kBAAkB,CAAC,CAAC;AACrE,eAAe,CAAC,UAAU,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;AACzD,eAAe,CAAC,kBAAkB,EAAE,mBAAmB,EAAE,sBAAsB,CAAC,CAAC;AACjF,eAAe,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,yBAAyB,CAAC,CAAC;AAC1F,eAAe,CAAC,aAAa,EAAE,cAAc,EAAE,iBAAiB,CAAC,CAAC;AAClE,eAAe,CAAC,YAAY,EAAE,aAAa,EAAE,gBAAgB,CAAC,CAAC;AAC/D,eAAe,CAAC,iBAAiB,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;AACjF,eAAe,CAAC,mBAAmB,EAAE,uBAAuB,EAAE,uBAAuB,CAAC,CAAC;AACvF,eAAe,CAAC,WAAW,EAAE,YAAY,EAAE,eAAe,CAAC,CAAC;AAC5D,eAAe,CAAC,eAAe,EAAE,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;AACxE,eAAe,CAAC,eAAe,EAAE,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;AACxE,eAAe,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,oBAAoB,CAAC,CAAC;AAC3E,eAAe,CAAC,UAAU,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;AACzD,eAAe,CAAC,UAAU,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;AAEzD;;;GAGG;AACH,MAAM,UAAU,uBAAuB;IACrC,OAAO,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QAC1D,OAAO,OAAO,OAAO,CAAC,GAAG,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;IACzE,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CAAC,IAAY;IACrD,OAAO,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;AAC5C,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/container-commands-types.d.ts

@@ -0,0 +1,28 @@
+import Docker from 'dockerode';
+import { FunctionCall, GenerateContentFunction, PromptItem, FunctionDef } from '../../../../../ai-service/common-types.js';
+import { CodegenOptions } from '../../../../../main/codegen-types.js';
+export interface CommandHandlerBaseProps {
+    actionResult: FunctionCall;
+    systemPrompt: PromptItem;
+    taskPrompt: PromptItem;
+    taskExecutionPrompt: PromptItem[];
+    generateContentFn: GenerateContentFunction;
+    container: Docker.Container;
+    options: CodegenOptions;
+    maxContextItems: number;
+    maxContextSize: number;
+    maxOutputLength: number;
+}
+export type CommandHandlerResult = void | {
+    success?: boolean;
+    summary?: string;
+    commandsExecutedIncrement?: number;
+    shouldBreakOuter?: boolean;
+};
+export interface CommandHandler<T extends CommandHandlerBaseProps = CommandHandlerBaseProps> {
+    (props: T): Promise<CommandHandlerResult>;
+}
+export interface Command<T extends CommandHandlerBaseProps = CommandHandlerBaseProps> {
+    def: FunctionDef | (() => FunctionDef);
+    handler: CommandHandler<T>;
+}