genaicode 0.14.2 → 0.15.0

package/dist/prompt/steps/step-iterate/handlers/handle-search-code.js

@@ -0,0 +1,96 @@
+import { searchSourceCode } from '../../../../files/search-files.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('searchCode', handleSearchCode);
+export async function handleSearchCode({ iterateCall, generateContentFn, prompt, options, }) {
+    try {
+        const searchCodeCall = await generateSearchCodeCall(generateContentFn, prompt, iterateCall, options);
+        // Extract search parameters from the call arguments
+        const args = searchCodeCall?.args;
+        if (!searchCodeCall || !args?.query) {
+            throw new Error('Missing search query in searchCode call');
+        }
+        // Set default values for optional parameters
+        const searchParams = {
+            searchInContent: true,
+            searchInFilenames: true,
+            caseSensitive: false,
+            maxResults: 50,
+            contextLines: 2,
+            ...args,
+        };
+        putSystemMessage('Searching code', searchParams);
+        // Execute the search
+        const searchResults = searchSourceCode(searchParams);
+        putSystemMessage('Search completed', searchResults);
+        // Return results in ActionResult format
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: iterateCall.args?.message ?? '',
+                        functionCalls: [searchCodeCall],
+                    },
+                    user: {
+                        type: 'user',
+                        functionResponses: [
+                            {
+                                name: 'searchCode',
+                                call_id: searchCodeCall.id,
+                                content: JSON.stringify(searchResults),
+                            },
+                        ],
+                    },
+                },
+            ],
+        };
+    }
+    catch (error) {
+        // Handle errors and return appropriate message
+        const errorMessage = error instanceof Error ? error.message : 'An unknown error occurred during search';
+        putSystemMessage(`Error during searchCode: ${errorMessage}`);
+        prompt.push({
+            type: 'assistant',
+            text: `Error performing search: ${errorMessage}\nPlease try again with valid search parameters.`,
+        });
+        prompt.push({
+            type: 'user',
+            text: 'Let me try again with different search parameters.',
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+}
+async function generateSearchCodeCall(generateContentFn, prompt, iterateCall, options) {
+    const [searchCodeCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Yes, you can search the code.',
+        },
+    ], {
+        functionDefs: getFunctionDefs(),
+        requiredFunctionName: 'searchCode',
+        temperature: 0.7,
+        modelType: ModelType.CHEAP,
+        expectedResponseType: {
+            text: false,
+            functionCall: true,
+            media: false,
+        },
+    }, options))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    return searchCodeCall;
+}
+//# sourceMappingURL=handle-search-code.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-search-code.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-search-code.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-search-code.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AAIrE,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AAEnE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAEpE,qBAAqB,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;AAEtD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,EACrC,WAAW,EACX,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,MAAM,sBAAsB,CAAC,iBAAiB,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;QAErG,oDAAoD;QACpD,MAAM,IAAI,GAAG,cAAc,EAAE,IAAI,CAAC;QAClC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QAED,6CAA6C;QAC7C,MAAM,YAAY,GAAmB;YACnC,eAAe,EAAE,IAAI;YACrB,iBAAiB,EAAE,IAAI;YACvB,aAAa,EAAE,KAAK;YACpB,UAAU,EAAE,EAAE;YACd,YAAY,EAAE,CAAC;YACf,GAAG,IAAI;SACR,CAAC;QAEF,gBAAgB,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;QAEjD,qBAAqB;QACrB,MAAM,aAAa,GAAG,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAErD,gBAAgB,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;QAEpD,wCAAwC;QACxC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;wBACrC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;6BACvC;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,+CAA+C;QAC/C,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,yCAAyC,CAAC;QAExG,gBAAgB,CAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,4BAA4B,YAAY,kDAAkD;SACjG,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,oDAAoD;SAC3D,CAAC,CAAC;QAEH,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,iBAA0C,EAC1C,MAAoB,EACpB,WAAwB,EACxB,OAAuB;IAEvB,MAAM,CAAC,cAAc,CAAC,GAAG,CACvB,MAAM,iBAAiB,CACrB;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,+BAA+B;SACtC;KACF,EACD;QACE,YAAY,EAAE,eAAe,EAAE;QAC/B,oBAAoB,EAAE,YAAY;QAClC,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,SAAS,CAAC,KAAK;QAC1B,oBAAoB,EAAE;YACpB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,IAAI;YAClB,KAAK,EAAE,KAAK;SACb;KACF,EACD,OAAO,CACR,CACF;SACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAEpC,OAAO,cAA0D,CAAC;AACpE,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-send-message.d.ts

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

package/dist/prompt/steps/step-iterate/handlers/handle-send-message.js

@@ -0,0 +1,21 @@
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('sendMessage', handleSendMessage);
+export async function handleSendMessage({ iterateCall, options }) {
+    const response = await askUserForInput('Your answer', '', options, true);
+    return {
+        breakLoop: false,
+        forceActionType: response.selectedActionType,
+        items: [
+            {
+                assistant: { type: 'assistant', text: iterateCall.args?.message ?? '' },
+                user: {
+                    type: 'user',
+                    text: response.answer,
+                    images: response.images,
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=handle-send-message.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-send-message.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-send-message.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-send-message.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE,qBAAqB,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;AAExD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAsB;IAClF,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;IACzE,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,eAAe,EAAE,QAAQ,CAAC,kBAAkB;QAC5C,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE;gBACvE,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ,CAAC,MAAM;oBACrB,MAAM,EAAE,QAAQ,CAAC,MAAM;iBACxB;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-structured-question.d.ts

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

package/dist/prompt/steps/step-iterate/handlers/handle-structured-question.js

@@ -0,0 +1,72 @@
+import { registerActionHandler } from '../step-iterate-handlers.js';
+import { askUserForStructuredQuestion } from '../../../../main/common/user-actions.js';
+import { ModelType, } from '../../../../ai-service/common-types.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+async function handleStructuredQuestion({ iterateCall, generateContentFn, prompt, options, }) {
+    const structuredQuestionCall = await generateStructuredQuestionCall(generateContentFn, prompt, iterateCall, options, ModelType.CHEAP);
+    putSystemMessage('Presenting structured question', { ...structuredQuestionCall?.args });
+    if (!structuredQuestionCall?.args) {
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    const userResponse = await askUserForStructuredQuestion(structuredQuestionCall.args.form, options);
+    putSystemMessage('Received structured question response', { ...userResponse });
+    return {
+        breakLoop: false,
+        items: [
+            {
+                assistant: {
+                    type: 'assistant',
+                    text: structuredQuestionCall.args?.message ?? 'Please complete the form.',
+                    functionCalls: [structuredQuestionCall],
+                },
+                user: {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: 'structuredQuestion',
+                            call_id: structuredQuestionCall.id,
+                            content: JSON.stringify(userResponse),
+                        },
+                    ],
+                },
+            },
+        ],
+    };
+}
+async function generateStructuredQuestionCall(generateContentFn, prompt, iterateCall, options, modelType) {
+    const req = [
+        [
+            ...prompt,
+            {
+                type: 'assistant',
+                text: iterateCall.args?.message ?? 'I need to ask a structured question.',
+            },
+            {
+                type: 'user',
+                text: 'Yes, you can ask the structured question.',
+            },
+        ],
+        {
+            functionDefs: getFunctionDefs(),
+            requiredFunctionName: 'structuredQuestion',
+            temperature: 0.7,
+            modelType,
+            expectedResponseType: {
+                text: false,
+                functionCall: true,
+                media: false,
+            },
+        },
+        options,
+    ];
+    const [structuredQuestionCall] = (await generateContentFn(...req))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    return structuredQuestionCall;
+}
+registerActionHandler('structuredQuestion', handleStructuredQuestion);
+//# sourceMappingURL=handle-structured-question.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-structured-question.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-structured-question.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-structured-question.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,4BAA4B,EAAE,MAAM,yCAAyC,CAAC;AACvF,OAAO,EAIL,SAAS,GAEV,MAAM,wCAAwC,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAE1E,KAAK,UAAU,wBAAwB,CAAC,EACtC,WAAW,EACX,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,MAAM,sBAAsB,GAAG,MAAM,8BAA8B,CACjE,iBAAiB,EACjB,MAAM,EACN,WAAW,EACX,OAAO,EACP,SAAS,CAAC,KAAK,CAChB,CAAC;IAEF,gBAAgB,CAAC,gCAAgC,EAAE,EAAE,GAAG,sBAAsB,EAAE,IAAI,EAAE,CAAC,CAAC;IAExF,IAAI,CAAC,sBAAsB,EAAE,IAAI,EAAE,CAAC;QAClC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,4BAA4B,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAEnG,gBAAgB,CAAC,uCAAuC,EAAE,EAAE,GAAG,YAAY,EAAE,CAAC,CAAC;IAE/E,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE;YACL;gBACE,SAAS,EAAE;oBACT,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,sBAAsB,CAAC,IAAI,EAAE,OAAO,IAAI,2BAA2B;oBACzE,aAAa,EAAE,CAAC,sBAAsB,CAAC;iBACxC;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,oBAAoB;4BAC1B,OAAO,EAAE,sBAAsB,CAAC,EAAE;4BAClC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;yBACtC;qBACF;iBACF;aACF;SACF;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,8BAA8B,CAC3C,iBAA0C,EAC1C,MAAoB,EACpB,WAAwB,EACxB,OAAuB,EACvB,SAAoB;IAEpB,MAAM,GAAG,GAAwB;QAC/B;YACE,GAAG,MAAM;YACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,sCAAsC;aAC1E;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,2CAA2C;aAClD;SACF;QACD;YACE,YAAY,EAAE,eAAe,EAAE;YAC/B,oBAAoB,EAAE,oBAAoB;YAC1C,WAAW,EAAE,GAAG;YAChB,SAAS;YACT,oBAAoB,EAAE;gBACpB,IAAI,EAAE,KAAK;gBACX,YAAY,EAAE,IAAI;gBAClB,KAAK,EAAE,KAAK;aACb;SACF;QACD,OAAO;KACR,CAAC;IACF,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAAC,GAAG,GAAG,CAAC,CAAC;SAC/D,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAuD,CAAC;IAE1F,OAAO,sBAAsB,CAAC;AAChC,CAAC;AAED,qBAAqB,CAAC,oBAAoB,EAAE,wBAAyC,CAAC,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-update-file.d.ts

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

package/dist/prompt/steps/step-iterate/handlers/handle-update-file.js

@@ -0,0 +1,161 @@
+import { executor as executeUpdateFile } from '../../../../operations/update-file/update-file-executor.js';
+import { getSourceCode } from '../../../../files/read-files.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+import { putSystemMessage, putUserMessage } from '../../../../main/common/content-bus.js';
+import { refreshFiles } from '../../../../files/find-files.js';
+import { generateRequestFilesContentCall } from './handle-request-files-content.js';
+import { executeStepEnsureContext } from '../../step-ensure-context.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('updateFile', handleUpdateFile);
+export async function handleUpdateFile({ iterateCall, options, prompt, generateContentFn, }) {
+    putSystemMessage('File update requested.', iterateCall);
+    // First confirmation: Ask user if they want to proceed with content generation
+    const generateConfirmation = await askUserForConfirmationWithAnswer('The assistant suggests generating new content for the file update. Do you want to proceed with content generation?', 'Generate content', 'Cancel content generation', false, options);
+    if (generateConfirmation.answer) {
+        putUserMessage(generateConfirmation.answer);
+    }
+    if (!generateConfirmation.confirmed) {
+        putSystemMessage('Content generation canceled.');
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        }, {
+            type: 'user',
+            text: 'Canceling file update.' + (generateConfirmation.answer ? ` ${generateConfirmation.answer}` : ''),
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    putSystemMessage('Content generation started.');
+    // Ensure file content is available in the context
+    const requestedFilesCall = await generateRequestFilesContentCall(generateContentFn, prompt, iterateCall, options, ModelType.CHEAP);
+    if (requestedFilesCall) {
+        await executeStepEnsureContext(prompt, requestedFilesCall, options);
+    }
+    const [updateFileCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Ok, please provide the update using `updateFile` function. Please remember to use absolute file path.' +
+                (generateConfirmation.answer ? ` ${generateConfirmation.answer}` : ''),
+        },
+    ], {
+        functionDefs: getFunctionDefs(),
+        requiredFunctionName: 'updateFile',
+        temperature: options.temperature ?? 0.7,
+        modelType: ModelType.DEFAULT,
+        expectedResponseType: {
+            text: false,
+            functionCall: true,
+            media: false,
+        },
+    }, options))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    if (!updateFileCall || !updateFileCall.args) {
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    const { filePath } = updateFileCall.args;
+    const file = getSourceCode({ filterPaths: [filePath], forceAll: true }, options)[filePath];
+    if (!file || !('content' in file) || !file.content) {
+        // TODO: In this corner case we should probably retry the operation but prefix it with getSourceCode
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        }, {
+            type: 'user',
+            text: 'The file does not exist or is empty or its content is not present in the context.',
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    putSystemMessage(`The assistant has generated new content for file ${filePath}.`, {
+        name: updateFileCall.name,
+        args: {
+            ...updateFileCall.args,
+            oldContent: file.content,
+        },
+    });
+    // Second confirmation: Ask user if they want to proceed with file update
+    const applyConfirmation = await askUserForConfirmationWithAnswer('The assistant has generated the new file content. Do you want to update the file?', 'Update file', 'Cancel update', false, options);
+    prompt.push({
+        type: 'assistant',
+        text: iterateCall.args.message,
+        functionCalls: [updateFileCall],
+    });
+    if (applyConfirmation.answer) {
+        putUserMessage(applyConfirmation.answer);
+    }
+    if (!applyConfirmation.confirmed) {
+        putSystemMessage('File update rejected.');
+        prompt.push({
+            type: 'user',
+            text: 'Rejecting file update.' + (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : ''),
+            functionResponses: [
+                {
+                    name: 'updateFile',
+                    call_id: updateFileCall.id,
+                    content: '',
+                },
+            ],
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    try {
+        putSystemMessage('Applying file update');
+        await executeUpdateFile(updateFileCall.args);
+        refreshFiles();
+        prompt.push({
+            type: 'user',
+            text: 'Accepting file update.' + (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : ''),
+            functionResponses: [
+                {
+                    name: 'updateFile',
+                    call_id: updateFileCall.id,
+                    content: '',
+                },
+            ],
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    catch {
+        putSystemMessage('File update failed.');
+        prompt.push({
+            type: 'user',
+            text: 'Accepting file update.' +
+                (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : '') +
+                ' Unfortunately, the file update failed, the new content was not saved.',
+            functionResponses: [
+                {
+                    name: 'updateFile',
+                    call_id: updateFileCall.id,
+                    content: '',
+                },
+            ],
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+}
+//# sourceMappingURL=handle-update-file.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-update-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-update-file.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-update-file.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,IAAI,iBAAiB,EAAE,MAAM,4DAA4D,CAAC;AAE3G,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAC1F,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAC/D,OAAO,EAAE,+BAA+B,EAAE,MAAM,mCAAmC,CAAC;AACpF,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAEpE,qBAAqB,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;AAEtD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,EACrC,WAAW,EACX,OAAO,EACP,MAAM,EACN,iBAAiB,GACE;IACnB,gBAAgB,CAAC,wBAAwB,EAAE,WAAW,CAAC,CAAC;IAExD,+EAA+E;IAC/E,MAAM,oBAAoB,GAAG,MAAM,gCAAgC,CACjE,oHAAoH,EACpH,kBAAkB,EAClB,2BAA2B,EAC3B,KAAK,EACL,OAAO,CACR,CAAC;IAEF,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC;QAChC,cAAc,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,CAAC;QACpC,gBAAgB,CAAC,8BAA8B,CAAC,CAAC;QAEjD,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACxG,CACF,CAAC;QAEF,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,6BAA6B,CAAC,CAAC;IAEhD,kDAAkD;IAClD,MAAM,kBAAkB,GAAG,MAAM,+BAA+B,CAC9D,iBAAiB,EACjB,MAAM,EACN,WAAW,EACX,OAAO,EACP,SAAS,CAAC,KAAK,CAChB,CAAC;IAEF,IAAI,kBAAkB,EAAE,CAAC;QACvB,MAAM,wBAAwB,CAAC,MAAM,EAAE,kBAAkB,EAAE,OAAO,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,GAAG,CACvB,MAAM,iBAAiB,CACrB;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,uGAAuG;gBACvG,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzE;KACF,EACD;QACE,YAAY,EAAE,eAAe,EAAE;QAC/B,oBAAoB,EAAE,YAAY;QAClC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;QACvC,SAAS,EAAE,SAAS,CAAC,OAAO;QAC5B,oBAAoB,EAAE;YACpB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,IAAI;YAClB,KAAK,EAAE,KAAK;SACb;KACF,EACD,OAAO,CACR,CACF;SACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAkD,CAAC;IAErF,IAAI,CAAC,cAAc,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QAC5C,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,EAAE,QAAQ,EAAE,GAAG,cAAc,CAAC,IAAI,CAAC;IAEzC,MAAM,IAAI,GAAG,aAAa,CAAC,EAAE,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC;IAE3F,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACnD,oGAAoG;QACpG,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,mFAAmF;SAC1F,CACF,CAAC;QACF,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,oDAAoD,QAAQ,GAAG,EAAE;QAChF,IAAI,EAAE,cAAc,CAAC,IAAI;QACzB,IAAI,EAAE;YACJ,GAAG,cAAc,CAAC,IAAI;YACtB,UAAU,EAAE,IAAI,CAAC,OAAO;SACzB;KACF,CAAC,CAAC;IAEH,yEAAyE;IACzE,MAAM,iBAAiB,GAAG,MAAM,gCAAgC,CAC9D,mFAAmF,EACnF,aAAa,EACb,eAAe,EACf,KAAK,EACL,OAAO,CACR,CAAC;IAEF,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;QAC/B,aAAa,EAAE,CAAC,cAAc,CAAC;KAChC,CAAC,CAAC;IAEH,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC;QAC7B,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,CAAC;QACjC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACjG,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE,cAAc,CAAC,EAAE;oBAC1B,OAAO,EAAE,EAAE;iBACZ;aACF;SACF,CAAC,CAAC;QAEH,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,gBAAgB,CAAC,sBAAsB,CAAC,CAAC;QACzC,MAAM,iBAAiB,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAC7C,YAAY,EAAE,CAAC;QACf,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACjG,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE,cAAc,CAAC,EAAE;oBAC1B,OAAO,EAAE,EAAE;iBACZ;aACF;SACF,CAAC,CAAC;QACH,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;QACxC,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,wBAAwB;gBACxB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,wEAAwE;YAC1E,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE,cAAc,CAAC,EAAE;oBAC1B,OAAO,EAAE,EAAE;iBACZ;aACF;SACF,CAAC,CAAC;QACH,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-web-search.d.ts

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

package/dist/prompt/steps/step-iterate/handlers/handle-web-search.js

@@ -0,0 +1,85 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+const handleWebSearch = async ({ prompt, iterateCall, generateContentFn, options }) => {
+    const request = [
+        [
+            ...prompt,
+            {
+                type: 'assistant',
+                text: iterateCall.args.message,
+            },
+            {
+                type: 'user',
+                text: 'Please provide an exhaustie search query',
+            },
+        ],
+        {
+            functionDefs: getFunctionDefs(),
+            requiredFunctionName: 'webSearch',
+            temperature: 0.7,
+            modelType: ModelType.CHEAP,
+            expectedResponseType: {
+                text: false,
+                functionCall: true,
+                media: false,
+            },
+        },
+        options,
+    ];
+    putSystemMessage('Inferring web search query...');
+    const response = await generateContentFn(...request);
+    const webSearchCall = response
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall).find((call) => call.name === 'webSearch');
+    if (!webSearchCall) {
+        putSystemMessage('No web search call received, skipping web search.');
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args.message,
+        }, {
+            type: 'user',
+            text: 'Web search failed: No query provided',
+        });
+        return {
+            items: [],
+            breakLoop: false,
+        };
+    }
+    putSystemMessage(`Performing web search for query: "${webSearchCall.args?.query}"`, webSearchCall.args);
+    const searchResponse = await generateContentFn([
+        {
+            type: 'user',
+            text: webSearchCall?.args?.query,
+        },
+    ], {
+        temperature: 0.7,
+        modelType: ModelType.CHEAP,
+        expectedResponseType: {
+            webSearch: true,
+        },
+    }, options);
+    const webSearchResult = searchResponse.find((item) => item.type === 'webSearch');
+    putSystemMessage('Web search completed, adding results to the conversation.', webSearchResult);
+    prompt.push({
+        type: 'assistant',
+        text: iterateCall.args.message,
+        functionCalls: [webSearchCall],
+    }, {
+        type: 'user',
+        functionResponses: [
+            {
+                name: 'webSearch',
+                content: JSON.stringify(webSearchResult),
+                call_id: webSearchCall?.id,
+            },
+        ],
+    });
+    return {
+        items: [],
+        breakLoop: false,
+    };
+};
+registerActionHandler('webSearch', handleWebSearch);
+//# sourceMappingURL=handle-web-search.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-web-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-web-search.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-web-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqC,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACtG,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE,MAAM,eAAe,GAAkB,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,iBAAiB,EAAE,OAAO,EAAE,EAAE,EAAE;IACnG,MAAM,OAAO,GAAwB;QACnC;YACE,GAAG,MAAM;YACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;aAChC;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,0CAA0C;aACjD;SACF;QACD;YACE,YAAY,EAAE,eAAe,EAAE;YAC/B,oBAAoB,EAAE,WAAW;YACjC,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,SAAS,CAAC,KAAK;YAC1B,oBAAoB,EAAE;gBACpB,IAAI,EAAE,KAAK;gBACX,YAAY,EAAE,IAAI;gBAClB,KAAK,EAAE,KAAK;aACb;SACF;QACD,OAAO;KACR,CAAC;IAEF,gBAAgB,CAAC,+BAA+B,CAAC,CAAC;IAElD,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CAAC,GAAG,OAAO,CAAC,CAAC;IACrD,MAAM,aAAa,GACjB,QAAQ;SACL,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CACnC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;IAE5C,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,gBAAgB,CAAC,mDAAmD,CAAC,CAAC;QACtE,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;SAC7C,CACF,CAAC;QACF,OAAO;YACL,KAAK,EAAE,EAAE;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,qCAAqC,aAAa,CAAC,IAAI,EAAE,KAAK,GAAG,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;IAExG,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAC5C;QACE;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,KAAK;SACjC;KACF,EACD;QACE,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,SAAS,CAAC,KAAK;QAC1B,oBAAoB,EAAE;YACpB,SAAS,EAAE,IAAI;SAChB;KACF,EACD,OAAO,CACR,CAAC;IAEF,MAAM,eAAe,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;IAEjF,gBAAgB,CAAC,2DAA2D,EAAE,eAAe,CAAC,CAAC;IAE/F,MAAM,CAAC,IAAI,CACT;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;QAC/B,aAAa,EAAE,CAAC,aAAa,CAAC;KAC/B,EACD;QACE,IAAI,EAAE,MAAM;QACZ,iBAAiB,EAAE;YACjB;gBACE,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC;gBACxC,OAAO,EAAE,aAAa,EAAE,EAAE;aAC3B;SACF;KACF,CACF,CAAC;IAEF,OAAO;QACL,KAAK,EAAE,EAAE;QACT,SAAS,EAAE,KAAK;KACjB,CAAC;AACJ,CAAC,CAAC;AAEF,qBAAqB,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC"}

package/dist/prompt/steps/step-iterate/iterate-handler.d.ts

@@ -0,0 +1,4 @@
+import { ActionHandler, ActionType } from './step-iterate-types.js';
+import './handlers/handle-structured-question.js';
+import './handlers/handle-code-execution.js';
+export declare function getActionHandler(actionType: ActionType): ActionHandler;

package/dist/prompt/steps/step-iterate/iterate-handler.js

@@ -0,0 +1,14 @@
+import { getRegisteredActionHandlers } from '../../../main/plugin-loader.js';
+import { handlers } from './step-iterate-handlers.js';
+import './handlers/handle-structured-question.js';
+import './handlers/handle-code-execution.js';
+export function getActionHandler(actionType) {
+    // First, check if there's a plugin-provided handler for this action type
+    const pluginHandler = getRegisteredActionHandlers().get(actionType);
+    if (pluginHandler) {
+        return pluginHandler;
+    }
+    // If no plugin handler is found, use the built-in handlers
+    return handlers[actionType] ?? handlers['sendMessage'];
+}
+//# sourceMappingURL=iterate-handler.js.map

package/dist/prompt/steps/step-iterate/iterate-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"iterate-handler.js","sourceRoot":"","sources":["../../../../src/prompt/steps/step-iterate/iterate-handler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,2BAA2B,EAAE,MAAM,gCAAgC,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAEtD,OAAO,0CAA0C,CAAC;AAClD,OAAO,qCAAqC,CAAC;AAE7C,MAAM,UAAU,gBAAgB,CAAC,UAAsB;IACrD,yEAAyE;IACzE,MAAM,aAAa,GAAG,2BAA2B,EAAE,CAAC,GAAG,CAAC,UAAgC,CAAC,CAAC;IAC1F,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,2DAA2D;IAC3D,OAAO,QAAQ,CAAC,UAAU,CAAC,IAAI,QAAQ,CAAC,aAAa,CAAE,CAAC;AAC1D,CAAC"}

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

@@ -0,0 +1,3 @@
+import { ActionType, ActionHandler } from './step-iterate-types.js';
+export declare const handlers: Partial<Record<ActionType, ActionHandler>>;
+export declare function registerActionHandler(type: ActionType, handler: ActionHandler): void;

package/dist/prompt/steps/step-iterate/step-iterate-handlers.js

@@ -0,0 +1,5 @@
+export const handlers = {};
+export function registerActionHandler(type, handler) {
+    handlers[type] = handler;
+}
+//# sourceMappingURL=step-iterate-handlers.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"step-iterate-handlers.js","sourceRoot":"","sources":["../../../../src/prompt/steps/step-iterate/step-iterate-handlers.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,QAAQ,GAA+C,EAAE,CAAC;AAEvE,MAAM,UAAU,qBAAqB,CAAC,IAAgB,EAAE,OAAsB;IAC5E,QAAQ,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC;AAC3B,CAAC"}