genaicode 0.14.1 → 0.15.0

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>;
+}

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

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=container-commands-types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"container-commands-types.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-iterate/handlers/container-task/container-commands-types.ts"],"names":[],"mappings":""}

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

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../../step-iterate-types.js';
+export declare function runContainerTaskOrchestrator({ iterateCall, prompt, generateContentFn, options, waitIfPaused, }: ActionHandlerProps): Promise<ActionResult>;

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

@@ -0,0 +1,202 @@
+import Docker from 'dockerode';
+import { ModelType } from '../../../../../ai-service/common-types.js';
+import { putAssistantMessage, putContainerLog, putSystemMessage, putUserMessage, } from '../../../../../main/common/content-bus.js';
+import { askUserForConfirmationWithAnswer } from '../../../../../main/common/user-actions.js';
+import { abortController } from '../../../../../main/common/abort-controller.js';
+import { runContainerTaskDef } from '../../../../function-defs/run-container-task.js';
+import { cleanupOrphanedContainers, createAndStartContainer, executeCommand, pullImage, stopContainer, } from './utils/docker-utils.js';
+import { commandExecutionLoop } from './command-execution-loop.js';
+import { getFunctionDefs } from '../../../../function-calling.js';
+export async function runContainerTaskOrchestrator({ iterateCall, prompt, generateContentFn, options, waitIfPaused, }) {
+    const docker = new Docker({ socketPath: '/var/run/docker.sock' });
+    await cleanupOrphanedContainers(docker);
+    if (abortController?.signal.aborted) {
+        putContainerLog('warn', 'Container task aborted before start.');
+        putSystemMessage('Container task aborted before start.');
+        return { breakLoop: false, items: [] };
+    }
+    try {
+        putContainerLog('info', 'Container task: generating proper task request');
+        const request = [
+            [
+                ...prompt,
+                {
+                    type: 'assistant',
+                    text: iterateCall.args.message,
+                },
+                {
+                    type: 'user',
+                    text: 'I understand that you want to run a container task. Please provide the Docker image and task description.',
+                },
+            ],
+            {
+                functionDefs: getFunctionDefs(),
+                requiredFunctionName: runContainerTaskDef.name,
+                temperature: 0.7,
+                modelType: ModelType.CHEAP,
+                expectedResponseType: {
+                    text: false,
+                    functionCall: true,
+                    media: false,
+                },
+            },
+            options,
+        ];
+        const response = await generateContentFn(...request);
+        const runContainerTaskCall = response
+            .filter((item) => item.type === 'functionCall')
+            .map((item) => item.functionCall).find((call) => call.name === 'runContainerTask');
+        if (!runContainerTaskCall?.args?.image || !runContainerTaskCall.args.taskDescription) {
+            putContainerLog('error', 'Failed to get valid runContainerTask request');
+            putSystemMessage('❌ Failed to get valid runContainerTask request');
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+            }, {
+                type: 'user',
+                text: 'Failed to get valid runContainerTask request',
+            });
+            return { breakLoop: false, items: [] };
+        }
+        putAssistantMessage(`I would like to run a container task with image: ${runContainerTaskCall.args.image}`, runContainerTaskCall.args);
+        const confirmation = await askUserForConfirmationWithAnswer(`Do you want to run the task?`, 'Run task', 'Reject', true, options);
+        if (!confirmation.confirmed) {
+            putContainerLog('warn', 'Container task cancelled by user.');
+            putSystemMessage('Container task cancelled by user.');
+            if (confirmation.answer) {
+                putUserMessage(confirmation.answer);
+            }
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+            }, {
+                type: 'user',
+                text: 'I reject running the container task.' + (confirmation.answer ? ` ${confirmation.answer}` : ''),
+            });
+            return { breakLoop: false, items: [] };
+        }
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args.message,
+        }, {
+            type: 'user',
+            text: 'I accept running the container task.' + (confirmation.answer ? ` ${confirmation.answer}` : ''),
+        });
+        const { image, taskDescription, workingDir } = runContainerTaskCall.args;
+        putContainerLog('info', `Starting container task with image: ${image}`);
+        putSystemMessage(`🐳 Starting container task with image: ${image}`);
+        if (confirmation.answer) {
+            putUserMessage(confirmation.answer);
+        }
+        if (abortController?.signal.aborted) {
+            putContainerLog('warn', 'Container task aborted before pulling image.');
+            putSystemMessage('Container task aborted by user.');
+            return { breakLoop: false, items: [] };
+        }
+        try {
+            await pullImage(docker, image);
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            putSystemMessage('❌ Failed to pull Docker image', { error: errorMessage });
+            // The pullImage function already logs the detailed error to the terminal.
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+                functionCalls: [runContainerTaskCall],
+            }, {
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: 'runContainerTask',
+                        call_id: runContainerTaskCall.id || undefined,
+                        content: `Failed to pull Docker image: ${errorMessage}`,
+                    },
+                ],
+            });
+            return { breakLoop: false, items: [] };
+        }
+        let container;
+        let onAbort;
+        try {
+            container = await createAndStartContainer(docker, image);
+            onAbort = () => {
+                putContainerLog('warn', 'Execution interrupted by user. Stopping container...', undefined, 'docker');
+                // Best-effort stop; stopContainer is idempotent/safe via try/catch
+                if (container) {
+                    stopContainer(container).catch(() => { });
+                }
+            };
+            abortController?.signal.addEventListener('abort', onAbort);
+            putContainerLog('info', 'Ensuring working directory exists.');
+            await executeCommand(container, '/bin/sh', `mkdir -p ${workingDir}`, '', '/');
+            putContainerLog('info', 'Entering command execution loop.');
+            const { success, summary } = await commandExecutionLoop(runContainerTaskCall.args, container, taskDescription, generateContentFn, options, waitIfPaused);
+            const finalMessage = `✅ Task finished with status: ${success ? 'Success' : 'Failed'}.
+
+**Summary:**
+${summary}`;
+            putContainerLog(success ? 'success' : 'error', `Task finished. Summary: ${summary}`);
+            putSystemMessage(finalMessage);
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+                functionCalls: [runContainerTaskCall],
+            }, {
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: 'runContainerTask',
+                        call_id: runContainerTaskCall.id || undefined,
+                        content: finalMessage,
+                    },
+                ],
+            });
+            return { breakLoop: false, items: [] };
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            putContainerLog('error', 'An error occurred during the container task', { error: errorMessage });
+            putSystemMessage('❌ An error occurred during the container task', {
+                error: errorMessage,
+            });
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+                functionCalls: [runContainerTaskCall],
+            }, {
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: 'runContainerTask',
+                        call_id: runContainerTaskCall.id || undefined,
+                        content: `An error occurred during the container task: ${errorMessage}`,
+                    },
+                ],
+            });
+            return { breakLoop: false, items: [] };
+        }
+        finally {
+            if (container) {
+                await stopContainer(container);
+                if (onAbort) {
+                    abortController?.signal.removeEventListener('abort', onAbort);
+                }
+            }
+        }
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        putContainerLog('error', 'Error during container task initialization', { error: errorMessage });
+        putSystemMessage('Error during container task initialization', { error: errorMessage });
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args.message,
+        }, {
+            type: 'user',
+            text: `Error during container task initialization: ${errorMessage}`,
+        });
+        return { breakLoop: false, items: [] };
+    }
+}
+//# sourceMappingURL=container-task-orchestrator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"container-task-orchestrator.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-iterate/handlers/container-task/container-task-orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,WAAW,CAAC;AAC/B,OAAO,EAAqC,SAAS,EAAE,MAAM,2CAA2C,CAAC;AACzG,OAAO,EACL,mBAAmB,EACnB,eAAe,EACf,gBAAgB,EAChB,cAAc,GACf,MAAM,2CAA2C,CAAC;AACnD,OAAO,EAAE,gCAAgC,EAAE,MAAM,4CAA4C,CAAC;AAC9F,OAAO,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAC;AACjF,OAAO,EAAwB,mBAAmB,EAAE,MAAM,iDAAiD,CAAC;AAC5G,OAAO,EACL,yBAAyB,EACzB,uBAAuB,EACvB,cAAc,EACd,SAAS,EACT,aAAa,GACd,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAElE,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACjD,WAAW,EACX,MAAM,EACN,iBAAiB,EACjB,OAAO,EACP,YAAY,GACO;IACnB,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,UAAU,EAAE,sBAAsB,EAAE,CAAC,CAAC;IAClE,MAAM,yBAAyB,CAAC,MAAM,CAAC,CAAC;IAExC,IAAI,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QACpC,eAAe,CAAC,MAAM,EAAE,sCAAsC,CAAC,CAAC;QAChE,gBAAgB,CAAC,sCAAsC,CAAC,CAAC;QACzD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,IAAI,CAAC;QACH,eAAe,CAAC,MAAM,EAAE,gDAAgD,CAAC,CAAC;QAE1E,MAAM,OAAO,GAAwB;YACnC;gBACE,GAAG,MAAM;gBACT;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;iBAChC;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,2GAA2G;iBAClH;aACF;YACD;gBACE,YAAY,EAAE,eAAe,EAAE;gBAC/B,oBAAoB,EAAE,mBAAmB,CAAC,IAAI;gBAC9C,WAAW,EAAE,GAAG;gBAChB,SAAS,EAAE,SAAS,CAAC,KAAK;gBAC1B,oBAAoB,EAAE;oBACpB,IAAI,EAAE,KAAK;oBACX,YAAY,EAAE,IAAI;oBAClB,KAAK,EAAE,KAAK;iBACb;aACF;YACD,OAAO;SACR,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CAAC,GAAG,OAAO,CAAC,CAAC;QACrD,MAAM,oBAAoB,GACxB,QAAQ;aACL,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;aAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CACnC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,kBAAkB,CAAC,CAAC;QAEnD,IAAI,CAAC,oBAAoB,EAAE,IAAI,EAAE,KAAK,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YACrF,eAAe,CAAC,OAAO,EAAE,8CAA8C,CAAC,CAAC;YACzE,gBAAgB,CAAC,gDAAgD,CAAC,CAAC;YAEnE,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;aAChC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,8CAA8C;aACrD,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,mBAAmB,CACjB,oDAAoD,oBAAoB,CAAC,IAAI,CAAC,KAAK,EAAE,EACrF,oBAAoB,CAAC,IAAI,CAC1B,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,gCAAgC,CACzD,8BAA8B,EAC9B,UAAU,EACV,QAAQ,EACR,IAAI,EACJ,OAAO,CACR,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;YAC5B,eAAe,CAAC,MAAM,EAAE,mCAAmC,CAAC,CAAC;YAC7D,gBAAgB,CAAC,mCAAmC,CAAC,CAAC;YACtD,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;gBACxB,cAAc,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YACtC,CAAC;YAED,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;aAChC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,sCAAsC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACtG,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;SAChC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,sCAAsC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACtG,CACF,CAAC;QAEF,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,IAAI,CAAC;QAEzE,eAAe,CAAC,MAAM,EAAE,uCAAuC,KAAK,EAAE,CAAC,CAAC;QACxE,gBAAgB,CAAC,0CAA0C,KAAK,EAAE,CAAC,CAAC;QAEpE,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;YACxB,cAAc,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,eAAe,CAAC,MAAM,EAAE,8CAA8C,CAAC,CAAC;YACxE,gBAAgB,CAAC,iCAAiC,CAAC,CAAC;YACpD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,gBAAgB,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YAC3E,0EAA0E;YAE1E,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;gBAC/B,aAAa,EAAE,CAAC,oBAAoB,CAAC;aACtC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,oBAAoB,CAAC,EAAE,IAAI,SAAS;wBAC7C,OAAO,EAAE,gCAAgC,YAAY,EAAE;qBACxD;iBACF;aACF,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,IAAI,SAAuC,CAAC;QAC5C,IAAI,OAAiC,CAAC;QACtC,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,uBAAuB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAEzD,OAAO,GAAG,GAAG,EAAE;gBACb,eAAe,CAAC,MAAM,EAAE,sDAAsD,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACrG,mEAAmE;gBACnE,IAAI,SAAS,EAAE,CAAC;oBACd,aAAa,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC,CAAC;YACF,eAAe,EAAE,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE3D,eAAe,CAAC,MAAM,EAAE,oCAAoC,CAAC,CAAC;YAC9D,MAAM,cAAc,CAAC,SAAS,EAAE,SAAS,EAAE,YAAY,UAAU,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YAE9E,eAAe,CAAC,MAAM,EAAE,kCAAkC,CAAC,CAAC;YAC5D,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,MAAM,oBAAoB,CACrD,oBAAoB,CAAC,IAAI,EACzB,SAAS,EACT,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,YAAY,CACb,CAAC;YAEF,MAAM,YAAY,GAAG,gCAAgC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;;;EAGvF,OAAO,EAAE,CAAC;YAEN,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,2BAA2B,OAAO,EAAE,CAAC,CAAC;YACrF,gBAAgB,CAAC,YAAY,CAAC,CAAC;YAE/B,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;gBAC/B,aAAa,EAAE,CAAC,oBAAoB,CAAC;aACtC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,oBAAoB,CAAC,EAAE,IAAI,SAAS;wBAC7C,OAAO,EAAE,YAAY;qBACtB;iBACF;aACF,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,eAAe,CAAC,OAAO,EAAE,6CAA6C,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YACjG,gBAAgB,CAAC,+CAA+C,EAAE;gBAChE,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;gBAC/B,aAAa,EAAE,CAAC,oBAAoB,CAAC;aACtC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,oBAAoB,CAAC,EAAE,IAAI,SAAS;wBAC7C,OAAO,EAAE,gDAAgD,YAAY,EAAE;qBACxE;iBACF;aACF,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;gBAAS,CAAC;YACT,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,aAAa,CAAC,SAAS,CAAC,CAAC;gBAC/B,IAAI,OAAO,EAAE,CAAC;oBACZ,eAAe,EAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,eAAe,CAAC,OAAO,EAAE,4CAA4C,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAChG,gBAAgB,CAAC,4CAA4C,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAExF,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;SAChC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,+CAA+C,YAAY,EAAE;SACpE,CACF,CAAC;QAEF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.d.ts

@@ -0,0 +1,4 @@
+/**
+ * Check if Docker is available
+ */
+export declare function checkDockerAvailability(): Promise<boolean>;

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.js

@@ -0,0 +1,15 @@
+import Docker from 'dockerode';
+/**
+ * Check if Docker is available
+ */
+export async function checkDockerAvailability() {
+    try {
+        const docker = new Docker({ socketPath: '/var/run/docker.sock' });
+        await docker.ping();
+        return true;
+    }
+    catch (error) {
+        return false;
+    }
+}
+//# sourceMappingURL=docker-check.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"docker-check.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,WAAW,CAAC;AAE/B;;GAEG;AAEH,MAAM,CAAC,KAAK,UAAU,uBAAuB;IAC3C,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,UAAU,EAAE,sBAAsB,EAAE,CAAC,CAAC;QAClE,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-utils.d.ts

@@ -0,0 +1,50 @@
+import Docker from 'dockerode';
+import { AllowedDockerImage } from '../../../../../function-defs/run-container-task.js';
+/**
+ * Pull a Docker image
+ */
+export declare function pullImage(docker: Docker, image: AllowedDockerImage): Promise<void>;
+/**
+ * Create and start a Docker container
+ */
+export declare function createAndStartContainer(docker: Docker, image: AllowedDockerImage): Promise<Docker.Container>;
+/**
+ * Execute a command in a Docker container
+ */
+export declare function executeCommand(container: Docker.Container, shell: '/bin/sh' | '/bin/bash', command: string, stdin: string | undefined, workingDir: string, abortSignal?: AbortSignal): Promise<{
+    output: string;
+    exitCode: number;
+}>;
+/**
+ * Stop and clean up a Docker container
+ */
+export declare function stopContainer(container: Docker.Container): Promise<void>;
+/**
+ * Clean up orphaned containers left over from previous runs.
+ */
+export declare function cleanupOrphanedContainers(docker: Docker): Promise<void>;
+/**
+ * Copy a file or directory from the host to the container.
+ */
+export declare function copyToContainer(container: Docker.Container, hostPath: string, containerPath: string): Promise<void>;
+export interface ArchiveEntry {
+    name: string;
+    type: 'file' | 'directory';
+}
+/**
+ * Lists the files within a container archive without extracting them.
+ */
+export declare function listFilesInContainerArchive(container: Docker.Container, containerPath: string): Promise<ArchiveEntry[]>;
+/**
+ * Extracts a tar stream to a specified directory, with path validation.
+ * @param tarStream The readable tar stream.
+ * @param destinationHostPath The absolute path on the host to extract files to.
+ * @param pathValidator A function to validate each file path before writing.
+ */
+export declare function extractTarStreamToDirectory(tarStream: NodeJS.ReadableStream, destinationHostPath: string, pathValidator: (filePath: string) => boolean): Promise<void>;
+/**
+ * Copy a file or directory from the container to the host.
+ */
+export declare function copyFromContainer(container: Docker.Container, containerPath: string, hostPath: string): Promise<void>;
+export declare function checkPathExistsInContainer(container: Docker.Container, containerPath: string): Promise<boolean>;
+export declare function getFileContentFromContainer(container: Docker.Container, filePath: string): Promise<string>;