genaicode 0.5.1 → 0.6.0

package/dist/src/prompt/steps/step-ask-question/handlers/confirm-code-generation.js

@@ -0,0 +1,37 @@
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleConfirmCodeGeneration({ askQuestionCall }) {
+    const userConfirmation = await askUserForConfirmationWithAnswer('The assistant is ready to start code generation. Do you want to proceed?', 'Start code generation', 'Continue conversation', true);
+    if (userConfirmation.confirmed) {
+        putSystemMessage('Proceeding with code generation.');
+        return {
+            breakLoop: true,
+            stepResult: StepResult.CONTINUE,
+            items: [
+                {
+                    assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },
+                    user: { type: 'user', text: userConfirmation.answer || 'Confirmed. Proceed with code generation.' },
+                },
+            ],
+        };
+    }
+    else {
+        putSystemMessage('Declined. Continuing the conversation.');
+        return {
+            breakLoop: false,
+            stepResult: StepResult.CONTINUE,
+            items: [
+                {
+                    assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                    user: {
+                        type: 'user',
+                        text: userConfirmation.answer || 'Declined. Please continue the conversation.',
+                        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+                    },
+                },
+            ],
+        };
+    }
+}
+//# sourceMappingURL=confirm-code-generation.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/confirm-code-generation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"confirm-code-generation.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/confirm-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAAC,EAAE,eAAe,EAAsB;IACvF,MAAM,gBAAgB,GAAG,MAAM,gCAAgC,CAC7D,0EAA0E,EAC1E,uBAAuB,EACvB,uBAAuB,EACvB,IAAI,CACL,CAAC;IACF,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC/B,gBAAgB,CAAC,kCAAkC,CAAC,CAAC;QACrD,OAAO;YACL,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,UAAU,CAAC,QAAQ;YAC/B,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;oBAC9F,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,CAAC,MAAM,IAAI,0CAA0C,EAAE;iBACpG;aACF;SACF,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,gBAAgB,CAAC,wCAAwC,CAAC,CAAC;QAC3D,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;YAC/B,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;oBAC7G,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,gBAAgB,CAAC,MAAM,IAAI,6CAA6C;wBAC9E,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;qBACpG;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/context-optimization.d.ts

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

package/dist/src/prompt/steps/step-ask-question/handlers/context-optimization.js

@@ -0,0 +1,36 @@
+import { refreshFiles } from '../../../../files/find-files.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { askUserForConfirmation } from '../../../../main/common/user-actions.js';
+import { executeStepContextOptimization } from '../../step-context-optimization.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleContextOptimization({ askQuestionCall, prompt, options, generateContentFn, }) {
+    const userConfirmation = await askUserForConfirmation('The assistant suggests optimizing the context to reduce token usage, cost, and latency. Do you want to proceed?', false);
+    const user = {
+        type: 'user',
+        text: userConfirmation ? 'Context optimization applied.' : 'Context optimization not applied.',
+        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+    };
+    const assistant = {
+        type: 'assistant',
+        text: askQuestionCall.args?.content ?? '',
+        functionCalls: [askQuestionCall],
+    };
+    if (userConfirmation) {
+        // the request may be caused be an appearance of a new file, so lets refresh
+        refreshFiles();
+        // Execute context optimization step
+        putSystemMessage('Executing context optimization step.');
+        await executeStepContextOptimization(generateContentFn, [...prompt, assistant, user], options);
+    }
+    return {
+        breakLoop: false,
+        stepResult: StepResult.CONTINUE,
+        items: [
+            {
+                assistant,
+                user,
+            },
+        ],
+    };
+}
+//# sourceMappingURL=context-optimization.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/context-optimization.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-optimization.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/context-optimization.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AACjF,OAAO,EAAE,8BAA8B,EAAE,MAAM,oCAAoC,CAAC;AACpF,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAC9C,eAAe,EACf,MAAM,EACN,OAAO,EACP,iBAAiB,GACE;IACnB,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,iHAAiH,EACjH,KAAK,CACN,CAAC;IAEF,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC,mCAAmC;QAC9F,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;KACpG,CAAC;IAEF,MAAM,SAAS,GAAkB;QAC/B,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QACzC,aAAa,EAAE,CAAC,eAAe,CAAC;KACjC,CAAC;IAEF,IAAI,gBAAgB,EAAE,CAAC;QACrB,4EAA4E;QAC5E,YAAY,EAAE,CAAC;QAEf,oCAAoC;QACpC,gBAAgB,CAAC,sCAAsC,CAAC,CAAC;QACzD,MAAM,8BAA8B,CAAC,iBAAiB,EAAE,CAAC,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;IACjG,CAAC;IAED,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS;gBACT,IAAI;aACL;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/default-action.d.ts

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+export declare function handleDefaultAction({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult>;

package/dist/src/prompt/steps/step-ask-question/handlers/default-action.js

@@ -0,0 +1,18 @@
+import { StepResult } from '../../steps-types.js';
+export async function handleDefaultAction({ askQuestionCall }) {
+    return {
+        breakLoop: false,
+        stepResult: StepResult.CONTINUE,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                user: {
+                    type: 'user',
+                    text: "I don't want to start the code generation yet, let's talk a bit more.",
+                    functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=default-action.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/default-action.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"default-action.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/default-action.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EAAE,eAAe,EAAsB;IAC/E,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;gBAC7G,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,uEAAuE;oBAC7E,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;iBACpG;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/handle-request-answer.d.ts

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+export declare function handleRequestAnswer({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult>;

package/dist/src/prompt/steps/step-ask-question/handlers/handle-request-answer.js

@@ -0,0 +1,20 @@
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleRequestAnswer({ askQuestionCall }) {
+    const userText = await askUserForInput('Your answer', askQuestionCall.args?.content ?? '');
+    return {
+        breakLoop: false,
+        stepResult: StepResult.CONTINUE,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                user: {
+                    type: 'user',
+                    text: userText,
+                    functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=handle-request-answer.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/handle-request-answer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-request-answer.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/handle-request-answer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EAAE,eAAe,EAAsB;IAC/E,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;IAC3F,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;gBAC7G,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ;oBACd,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;iBACpG;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/remove-files-from-context.d.ts

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+export declare function handleRemoveFilesFromContext({ askQuestionCall, prompt, }: ActionHandlerProps): Promise<ActionResult>;

package/dist/src/prompt/steps/step-ask-question/handlers/remove-files-from-context.js

@@ -0,0 +1,43 @@
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { StepResult } from '../../steps-types.js';
+import { getSourceCodeResponse } from '../../steps-utils.js';
+export async function handleRemoveFilesFromContext({ askQuestionCall, prompt, }) {
+    const filesToRemove = askQuestionCall.args?.removeFilesFromContext ?? [];
+    let userText = '';
+    if (filesToRemove.length > 0) {
+        removeFileContentsFromPrompt(prompt, filesToRemove);
+        putSystemMessage('Context reduction applied', filesToRemove);
+        userText = 'Context reduction applied';
+    }
+    else {
+        userText = 'No specific files were provided for context reduction. The context remains unchanged.';
+    }
+    return {
+        breakLoop: false,
+        stepResult: StepResult.CONTINUE,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                user: {
+                    type: 'user',
+                    text: userText,
+                    functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+                },
+            },
+        ],
+    };
+}
+function removeFileContentsFromPrompt(prompt, filesToRemove) {
+    const response = getSourceCodeResponse(prompt);
+    if (!response || !response.content) {
+        throw new Error('Could not find source code response');
+    }
+    const contentObj = JSON.parse(response.content);
+    filesToRemove.forEach((file) => {
+        if (contentObj[file]) {
+            delete contentObj[file].content;
+        }
+    });
+    response.content = JSON.stringify(contentObj);
+}
+//# sourceMappingURL=remove-files-from-context.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/remove-files-from-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"remove-files-from-context.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/remove-files-from-context.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAG7D,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACjD,eAAe,EACf,MAAM,GACa;IACnB,MAAM,aAAa,GAAG,eAAe,CAAC,IAAI,EAAE,sBAAsB,IAAI,EAAE,CAAC;IACzE,IAAI,QAAQ,GAAG,EAAE,CAAC;IAElB,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,4BAA4B,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACpD,gBAAgB,CAAC,2BAA2B,EAAE,aAAa,CAAC,CAAC;QAC7D,QAAQ,GAAG,2BAA2B,CAAC;IACzC,CAAC;SAAM,CAAC;QACN,QAAQ,GAAG,uFAAuF,CAAC;IACrG,CAAC;IAED,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;gBAC7G,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ;oBACd,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;iBACpG;aACF;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAoB,EAAE,aAAuB;IACjF,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC/C,IAAI,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAChD,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC7B,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACrB,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;QAClC,CAAC;IACH,CAAC,CAAC,CAAC;IACH,QAAQ,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;AAChD,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/request-files-content.d.ts

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

package/dist/src/prompt/steps/step-ask-question/handlers/request-files-content.js

@@ -0,0 +1,49 @@
+import { getSourceFiles, refreshFiles } from '../../../../files/find-files.js';
+import { getSourceCode } from '../../../../files/read-files.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleRequestFilesContent({ askQuestionCall, options, }) {
+    const requestedFiles = askQuestionCall.args?.requestFilesContent ?? [];
+    // the request may be caused be an appearance of a new file, so lets refresh
+    refreshFiles();
+    const { legitimateFiles, illegitimateFiles } = categorizeLegitimateFiles(requestedFiles);
+    const sourceCallId = (askQuestionCall.id ?? '') + '_source';
+    const assistant = {
+        type: 'assistant',
+        text: askQuestionCall.args?.content ?? '',
+        functionCalls: [askQuestionCall, { name: 'getSourceCode', id: sourceCallId, args: { filePaths: legitimateFiles } }],
+    };
+    const sourceCode = getSourceCode({ filterPaths: legitimateFiles, forceAll: true }, options);
+    const user = {
+        type: 'user',
+        text: illegitimateFiles.length > 0
+            ? 'Some files are not legitimate and their content cannot be provided'
+            : 'All requested file contents have been provided.',
+        functionResponses: [
+            { name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined },
+            {
+                name: 'getSourceCode',
+                call_id: sourceCallId,
+                content: JSON.stringify(sourceCode),
+            },
+        ],
+        cache: true,
+    };
+    return { breakLoop: false, stepResult: StepResult.CONTINUE, items: [{ assistant, user }] };
+}
+function categorizeLegitimateFiles(requestedFiles) {
+    const legitimateFiles = [];
+    const illegitimateFiles = [];
+    requestedFiles.forEach((filePath) => {
+        if (isFilePathLegitimate(filePath)) {
+            legitimateFiles.push(filePath);
+        }
+        else {
+            illegitimateFiles.push(filePath);
+        }
+    });
+    return { legitimateFiles, illegitimateFiles };
+}
+function isFilePathLegitimate(filePath) {
+    return getSourceFiles().includes(filePath);
+}
+//# sourceMappingURL=request-files-content.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/request-files-content.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-files-content.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/request-files-content.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAC/E,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAC9C,eAAe,EACf,OAAO,GACY;IACnB,MAAM,cAAc,GAAG,eAAe,CAAC,IAAI,EAAE,mBAAmB,IAAI,EAAE,CAAC;IACvE,4EAA4E;IAC5E,YAAY,EAAE,CAAC;IAEf,MAAM,EAAE,eAAe,EAAE,iBAAiB,EAAE,GAAG,yBAAyB,CAAC,cAAc,CAAC,CAAC;IAEzF,MAAM,YAAY,GAAG,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,SAAS,CAAC;IAC5D,MAAM,SAAS,GAAkB;QAC/B,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QACzC,aAAa,EAAE,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE,EAAE,CAAC;KACpH,CAAC;IAEF,MAAM,UAAU,GAAG,aAAa,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;IAC5F,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EACF,iBAAiB,CAAC,MAAM,GAAG,CAAC;YAC1B,CAAC,CAAC,oEAAoE;YACtE,CAAC,CAAC,iDAAiD;QACvD,iBAAiB,EAAE;YACjB,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE;YAC9E;gBACE,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,YAAY;gBACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;aACpC;SACF;QACD,KAAK,EAAE,IAAI;KACZ,CAAC;IAEF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;AAC7F,CAAC;AAED,SAAS,yBAAyB,CAAC,cAAwB;IAIzD,MAAM,eAAe,GAAa,EAAE,CAAC;IACrC,MAAM,iBAAiB,GAAa,EAAE,CAAC;IAEvC,cAAc,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;QAClC,IAAI,oBAAoB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,CAAC;AAChD,CAAC;AAED,SAAS,oBAAoB,CAAC,QAAgB;IAC5C,OAAO,cAAc,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AAC7C,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/request-permissions.d.ts

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

package/dist/src/prompt/steps/step-ask-question/handlers/request-permissions.js

@@ -0,0 +1,46 @@
+import { askUserForConfirmation } from '../../../../main/common/user-actions.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleRequestPermissions({ askQuestionCall, options, }) {
+    const userConfirmation = await askUserForConfirmation('The assistant is requesting additional permissions. Do you want to grant them?', false);
+    const user = {
+        type: 'user',
+        text: userConfirmation ? 'Permissions granted.' : 'Permission request denied.',
+        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+    };
+    if (userConfirmation) {
+        updatePermissions(askQuestionCall, options);
+    }
+    return {
+        breakLoop: false,
+        stepResult: StepResult.CONTINUE,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                user,
+            },
+        ],
+    };
+}
+function updatePermissions(askQuestionCall, options) {
+    if (askQuestionCall.args &&
+        typeof askQuestionCall.args === 'object' &&
+        'requestPermissions' in askQuestionCall.args) {
+        const permissions = askQuestionCall.args.requestPermissions;
+        if (permissions && typeof permissions === 'object') {
+            if ('enableImagen' in permissions && permissions.enableImagen) {
+                options.imagen = options.aiService === 'chat-gpt' ? 'dall-e' : 'vertex-ai';
+            }
+            if ('enableVision' in permissions && permissions.enableVision)
+                options.vision = true;
+            if ('allowDirectoryCreate' in permissions && permissions.allowDirectoryCreate)
+                options.allowDirectoryCreate = true;
+            if ('allowFileCreate' in permissions && permissions.allowFileCreate)
+                options.allowFileCreate = true;
+            if ('allowFileDelete' in permissions && permissions.allowFileDelete)
+                options.allowFileDelete = true;
+            if ('allowFileMove' in permissions && permissions.allowFileMove)
+                options.allowFileMove = true;
+        }
+    }
+}
+//# sourceMappingURL=request-permissions.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/request-permissions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-permissions.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/request-permissions.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AACjF,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,EAC7C,eAAe,EACf,OAAO,GACY;IACnB,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,gFAAgF,EAChF,KAAK,CACN,CAAC;IAEF,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,4BAA4B;QAC9E,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;KACpG,CAAC;IAEF,IAAI,gBAAgB,EAAE,CAAC;QACrB,iBAAiB,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;gBAC7G,IAAI;aACL;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,eAAgC,EAAE,OAAuB;IAClF,IACE,eAAe,CAAC,IAAI;QACpB,OAAO,eAAe,CAAC,IAAI,KAAK,QAAQ;QACxC,oBAAoB,IAAI,eAAe,CAAC,IAAI,EAC5C,CAAC;QACD,MAAM,WAAW,GAAG,eAAe,CAAC,IAAI,CAAC,kBAAkB,CAAC;QAC5D,IAAI,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;YACnD,IAAI,cAAc,IAAI,WAAW,IAAI,WAAW,CAAC,YAAY,EAAE,CAAC;gBAC9D,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC;YAC7E,CAAC;YACD,IAAI,cAAc,IAAI,WAAW,IAAI,WAAW,CAAC,YAAY;gBAAE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;YACrF,IAAI,sBAAsB,IAAI,WAAW,IAAI,WAAW,CAAC,oBAAoB;gBAC3E,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACtC,IAAI,iBAAiB,IAAI,WAAW,IAAI,WAAW,CAAC,eAAe;gBAAE,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;YACpG,IAAI,iBAAiB,IAAI,WAAW,IAAI,WAAW,CAAC,eAAe;gBAAE,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;YACpG,IAAI,eAAe,IAAI,WAAW,IAAI,WAAW,CAAC,aAAa;gBAAE,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;QAChG,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/src/prompt/steps/step-ask-question/handlers/start-code-generation.d.ts

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+export declare function handleStartCodeGeneration({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult>;

package/dist/src/prompt/steps/step-ask-question/handlers/start-code-generation.js

@@ -0,0 +1,16 @@
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleStartCodeGeneration({ askQuestionCall }) {
+    putSystemMessage('Proceeding with code generation.');
+    return {
+        breakLoop: true,
+        stepResult: StepResult.CONTINUE,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },
+                user: { type: 'user', text: 'Proceeding with code generation.' },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=start-code-generation.js.map

package/dist/src/prompt/steps/step-ask-question/handlers/start-code-generation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"start-code-generation.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-ask-question/handlers/start-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAAE,eAAe,EAAsB;IACrF,gBAAgB,CAAC,kCAAkC,CAAC,CAAC;IACrD,OAAO;QACL,SAAS,EAAE,IAAI;QACf,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9F,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kCAAkC,EAAE;aACjE;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/src/prompt/steps/step-ask-question/step-ask-question-handlers.js

@@ -0,0 +1,2 @@
+"use strict";
+//# sourceMappingURL=step-ask-question-handlers.js.map

package/dist/src/prompt/steps/step-ask-question/step-ask-question-handlers.js.map

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

package/dist/src/prompt/steps/step-ask-question/step-ask-question-reflect.d.ts

@@ -0,0 +1,4 @@
+import { FunctionDef, GenerateContentFunction, PromptItem } from '../../../ai-service/common.js';
+import { CodegenOptions } from '../../../main/codegen-types.js';
+import { AskQuestionCall, EscalationDecision, SelfReflectionContext } from './step-ask-question-types.js';
+export declare function performSelfReflection(askQuestionCall: AskQuestionCall, context: SelfReflectionContext, prompt: PromptItem[], functionDefs: FunctionDef[], options: CodegenOptions, generateContentFn: GenerateContentFunction): Promise<EscalationDecision>;

package/dist/src/prompt/steps/step-ask-question/step-ask-question-reflect.js

@@ -0,0 +1,91 @@
+import { validateAndRecoverSingleResult } from '../step-validate-recover.js';
+const SELF_REFLECT_PROMPT = `Please reflect on your last response provided in \`askQuestion\` function call in the context of our conversation by considering these questions:
+
+- Did you fully understand and address the user's request?
+- Was your response clear, accurate, and helpful?
+- Are you confident that no further clarification or escalation is needed?
+- Is your response too similar to previous questions or responses in the conversation?
+- Is your response adding value to the conversation or moving it forward?
+- Are you maintaining your own identity and perspective as an AI assistant?
+- Are you directly addressing the user's points without deflecting or prolonging the conversation?
+- Are you providing clear, concise, and helpful responses without being overly cautious or evasive?
+- Are you demonstrating confidence and competence in your abilities without expressing unnecessary uncertainty?
+
+Common problems/pitfalls:
+
+- There is an in-depth question to the user, but actionType is set to confirmCodeGeneration, while it should be requestAnswer
+- Question was already answered before, but the assistant is asking the same question again
+- The question content is incomplete, for example the question content ends with a colon
+- Saying that an analysis will be done, instead of actually doing the analysis
+- The response is too similar to a previous question or response
+- The assistant takes on the user's voice or persona instead of maintaining its own identity
+
+Specific examples of problematic responses:
+
+1. User: "What's the weather like today?"
+   Assistant: "To provide information about today's weather, I'll need to know your location. What city are you in?"
+   User: "New York"
+   Assistant: "To give you accurate weather information for New York, I'll need to know which specific area of New York you're interested in. Could you please provide more details about your location?"
+   Problem: The assistant is being overly cautious and asking for unnecessary details, prolonging the conversation instead of providing a helpful response based on the information already given.
+
+2. User: "Can you explain how to implement a binary search algorithm?"
+   Assistant: "Certainly! I'd be happy to explain how to implement a binary search algorithm. Before we begin, could you tell me what programming language you'd like the explanation in?"
+   User: "Python"
+   Assistant: "Great choice! Python is an excellent language for implementing a binary search algorithm. Before we dive into the Python implementation, could you let me know your level of experience with Python programming?"
+   Problem: The assistant is asking for excessive context before providing the requested explanation, which may frustrate the user who was expecting a more direct response.
+
+3. User: "I'm not convinced if those responses are really problematic"
+   Assistant: "I'm not convinced if those responses in the "Specific examples of problematic responses" section are really problematic. Could you provide more context or clarification on why you consider them to be problematic examples?"
+   Problem: The assistant is repeating the user's statement and asking for clarification instead of engaging with the user's perspective. This response fails to maintain the assistant's identity and doesn't add value to the conversation.
+
+If your response meets the criteria for a clear, helpful, and non-repetitive answer, continue the conversation as usual. If you believe that escalating to a more advanced model would significantly improve assistance or if the response is too similar to previous interactions, please indicate that using the \`shouldEscalate\` parameter and provide a \`reason\`.`;
+export async function performSelfReflection(askQuestionCall, context, prompt, functionDefs, options, generateContentFn) {
+    if (!options.selfReflectionEnabled) {
+        return {
+            shouldEscalate: false,
+            reason: 'Self-reflection is disabled by user configuration.',
+        };
+    }
+    const currentTime = Date.now();
+    const { shouldEscalate, reason } = await runSelfReflect(askQuestionCall, prompt, functionDefs, options, generateContentFn);
+    if (shouldEscalate) {
+        context.escalationCount++;
+        context.lastEscalationTime = currentTime;
+    }
+    return {
+        shouldEscalate,
+        reason,
+    };
+}
+async function runSelfReflect(askQuestionCall, prompt, functionDefs, options, generateContentFn) {
+    const currentResponse = askQuestionCall.args?.content ?? '';
+    const reflectRequest = [
+        [
+            ...prompt,
+            { type: 'assistant', text: currentResponse, functionCalls: [askQuestionCall] },
+            {
+                type: 'user',
+                text: SELF_REFLECT_PROMPT,
+                functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+            },
+        ],
+        functionDefs,
+        'askQuestionReflect',
+        0.2,
+        true,
+        options,
+    ];
+    let result = await generateContentFn(...reflectRequest);
+    result = await validateAndRecoverSingleResult(reflectRequest, result, generateContentFn);
+    const reflectionResult = result.find((call) => call.name === 'askQuestionReflect')?.args;
+    const shouldEscalate = (reflectionResult?.shouldEscalate ?? 0) > 50;
+    const reason = reflectionResult?.reason || '';
+    return {
+        shouldEscalate,
+        reason: reason ||
+            (shouldEscalate
+                ? 'Response requires escalation to a more advanced model'
+                : 'Response does not require escalation'),
+    };
+}
+//# sourceMappingURL=step-ask-question-reflect.js.map

package/dist/src/prompt/steps/step-ask-question/step-ask-question-reflect.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-ask-question-reflect.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/step-ask-question-reflect.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,8BAA8B,EAAE,MAAM,6BAA6B,CAAC;AAG7E,MAAM,mBAAmB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;0WAuC8U,CAAC;AAE3W,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,eAAgC,EAChC,OAA8B,EAC9B,MAAoB,EACpB,YAA2B,EAC3B,OAAuB,EACvB,iBAA0C;IAE1C,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;QACnC,OAAO;YACL,cAAc,EAAE,KAAK;YACrB,MAAM,EAAE,oDAAoD;SAC7D,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE/B,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,GAAG,MAAM,cAAc,CACrD,eAAe,EACf,MAAM,EACN,YAAY,EACZ,OAAO,EACP,iBAAiB,CAClB,CAAC;IAEF,IAAI,cAAc,EAAE,CAAC;QACnB,OAAO,CAAC,eAAe,EAAE,CAAC;QAC1B,OAAO,CAAC,kBAAkB,GAAG,WAAW,CAAC;IAC3C,CAAC;IAED,OAAO;QACL,cAAc;QACd,MAAM;KACP,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,eAAgC,EAChC,MAAoB,EACpB,YAA2B,EAC3B,OAAuB,EACvB,iBAA0C;IAE1C,MAAM,eAAe,GAAG,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,CAAC;IAE5D,MAAM,cAAc,GAAwB;QAC1C;YACE,GAAG,MAAM;YACT,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;YAC9E;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,mBAAmB;gBACzB,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;aACpG;SACF;QACD,YAAY;QACZ,oBAAoB;QACpB,GAAG;QACH,IAAI;QACJ,OAAO;KACR,CAAC;IAEF,IAAI,MAAM,GAAG,MAAM,iBAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;IACxD,MAAM,GAAG,MAAM,8BAA8B,CAAC,cAAc,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC;IAEzF,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,oBAAoB,CAAC,EAAE,IAAI,CAAC;IACzF,MAAM,cAAc,GAAG,CAAE,gBAAgB,EAAE,cAAqC,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC;IAC5F,MAAM,MAAM,GAAI,gBAAgB,EAAE,MAAiB,IAAI,EAAE,CAAC;IAE1D,OAAO;QACL,cAAc;QACd,MAAM,EACJ,MAAM;YACN,CAAC,cAAc;gBACb,CAAC,CAAC,uDAAuD;gBACzD,CAAC,CAAC,sCAAsC,CAAC;KAC9C,CAAC;AACJ,CAAC"}

package/dist/src/prompt/steps/step-ask-question/step-ask-question-types.d.ts

@@ -0,0 +1,55 @@
+import { FunctionCall, GenerateContentFunction, PromptItem } from '../../../ai-service/common.js';
+import { CodegenOptions } from '../../../main/codegen-types.js';
+import { StepResult } from '../steps-types.js';
+export type ActionType = 'requestAnswer' | 'requestPermissions' | 'requestFilesContent' | 'removeFilesFromContext' | 'confirmCodeGeneration' | 'startCodeGeneration' | 'cancelCodeGeneration' | 'contextOptimization';
+type AskQuestionArgs = {
+    actionType: ActionType;
+    content: string;
+    requestFilesContent?: string[];
+    requestPermissions?: Record<'allowDirectoryCreate' | 'allowFileCreate' | 'allowFileDelete' | 'allowFileMove' | 'enableVision' | 'enableImagen', boolean>;
+    removeFilesFromContext?: string[];
+};
+export type AskQuestionCall = FunctionCall<AskQuestionArgs>;
+export interface AssistantItem {
+    type: 'assistant';
+    text: string;
+    functionCalls: FunctionCall[];
+    cache?: true;
+}
+export interface UserItem {
+    type: 'user';
+    text: string;
+    functionResponses?: Array<{
+        name: string;
+        call_id: string;
+        content: string | undefined;
+    }>;
+    cache?: true;
+}
+export interface ActionResult {
+    breakLoop: boolean;
+    stepResult: StepResult;
+    items: Array<{
+        assistant: AssistantItem;
+        user: UserItem;
+    }>;
+}
+export type ActionHandlerProps = {
+    askQuestionCall: AskQuestionCall;
+    prompt: PromptItem[];
+    options: CodegenOptions;
+    generateContentFn: GenerateContentFunction;
+    messages: {
+        contextSourceCode: (paths: string[], pathsOnly: boolean) => string;
+    };
+};
+export type ActionHandler = (props: ActionHandlerProps) => Promise<ActionResult>;
+export interface EscalationDecision {
+    shouldEscalate: boolean;
+    reason: string;
+}
+export interface SelfReflectionContext {
+    escalationCount: number;
+    lastEscalationTime: number;
+}
+export {};

package/dist/src/prompt/steps/step-ask-question/step-ask-question-types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=step-ask-question-types.js.map

package/dist/src/prompt/steps/step-ask-question/step-ask-question-types.js.map

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

package/dist/src/prompt/steps/step-ask-question/step-ask-question.d.ts

@@ -0,0 +1,6 @@
+import { FunctionDef, GenerateContentFunction, PromptItem } from '../../../ai-service/common.js';
+import { StepResult } from '../steps-types.js';
+import { CodegenOptions } from '../../../main/codegen-types.js';
+export declare function executeStepAskQuestion(generateContentFn: GenerateContentFunction, prompt: PromptItem[], functionDefs: FunctionDef[], temperature: number, messages: {
+    contextSourceCode: (paths: string[], pathsOnly: boolean) => string;
+}, options: CodegenOptions): Promise<StepResult>;

package/dist/src/prompt/steps/step-ask-question/step-ask-question.js

@@ -0,0 +1,88 @@
+import { StepResult } from '../steps-types.js';
+import { putAssistantMessage, putSystemMessage, putUserMessage } from '../../../main/common/content-bus.js';
+import { abortController } from '../../../main/interactive/codegen-worker.js';
+import { validateAndRecoverSingleResult } from '../step-validate-recover.js';
+import { handleRequestFilesContent } from './handlers/request-files-content.js';
+import { handleContextOptimization } from './handlers/context-optimization.js';
+import { handleRemoveFilesFromContext } from './handlers/remove-files-from-context.js';
+import { handleRequestPermissions } from './handlers/request-permissions.js';
+import { handleDefaultAction } from './handlers/default-action.js';
+import { handleRequestAnswer } from './handlers/handle-request-answer.js';
+import { handleStartCodeGeneration } from './handlers/start-code-generation.js';
+import { handleConfirmCodeGeneration } from './handlers/confirm-code-generation.js';
+import { handleCancelCodeGeneration } from './handlers/cancel-code-generation.js';
+import { performSelfReflection } from './step-ask-question-reflect.js';
+export async function executeStepAskQuestion(generateContentFn, prompt, functionDefs, temperature, messages, options) {
+    console.log('Allowing the assistant to ask a question...');
+    const selfReflectionContext = {
+        escalationCount: 0,
+        lastEscalationTime: 0,
+    };
+    while (!abortController?.signal.aborted) {
+        try {
+            const askQuestionCall = await getAskQuestionCall(generateContentFn, prompt, functionDefs, temperature, options, selfReflectionContext);
+            if (!askQuestionCall) {
+                break;
+            }
+            console.log('Assistant asks:', askQuestionCall.args);
+            if (askQuestionCall.args?.content) {
+                putAssistantMessage(askQuestionCall.args.content, askQuestionCall.args);
+            }
+            const actionType = askQuestionCall.args?.actionType;
+            if (actionType) {
+                const actionHandler = getActionHandler(actionType);
+                const result = await actionHandler({ askQuestionCall, prompt, options, messages, generateContentFn });
+                // This is important to display the content to the user interface (ui or interactive cli)
+                putUserMessage(result.items.slice(-1)[0].user.text);
+                if (result.breakLoop) {
+                    return result.stepResult;
+                }
+                prompt.push(...result.items.map(({ assistant, user }) => [assistant, user]).flat());
+                console.log('The question was answered');
+            }
+            else {
+                console.error('Invalid action type received');
+                break;
+            }
+        }
+        catch (error) {
+            console.error('Error in executeStepAskQuestion:', error);
+            putSystemMessage(`An error occurred: ${error instanceof Error ? error.message : String(error)}`);
+            return StepResult.BREAK;
+        }
+    }
+    putSystemMessage('Assistant did not ask a question. This unexpected, we need to abort.');
+    return StepResult.BREAK;
+}
+async function getAskQuestionCall(generateContentFn, prompt, functionDefs, temperature, options, selfReflectionContext) {
+    const askQuestionRequest = [prompt, functionDefs, 'askQuestion', temperature, true, options];
+    let askQuestionResult = await generateContentFn(...askQuestionRequest);
+    askQuestionResult = await validateAndRecoverSingleResult(askQuestionRequest, askQuestionResult, generateContentFn);
+    const cheapModelResponse = askQuestionResult.find((call) => call.name === 'askQuestion');
+    if (cheapModelResponse) {
+        const escalationDecision = await performSelfReflection(cheapModelResponse, selfReflectionContext, prompt, functionDefs, options, generateContentFn);
+        if (escalationDecision.shouldEscalate) {
+            console.log('Self-reflection suggests escalating to non-cheap model.');
+            // Re-run with non-cheap model
+            const nonCheapRequest = [prompt, functionDefs, 'askQuestion', temperature, false, options];
+            let nonCheapResult = await generateContentFn(...nonCheapRequest);
+            nonCheapResult = await validateAndRecoverSingleResult(nonCheapRequest, nonCheapResult, generateContentFn);
+            return nonCheapResult.find((call) => call.name === 'askQuestion');
+        }
+    }
+    return cheapModelResponse;
+}
+function getActionHandler(actionType) {
+    const handlers = {
+        cancelCodeGeneration: handleCancelCodeGeneration,
+        confirmCodeGeneration: handleConfirmCodeGeneration,
+        startCodeGeneration: handleStartCodeGeneration,
+        requestFilesContent: handleRequestFilesContent,
+        requestPermissions: handleRequestPermissions,
+        removeFilesFromContext: handleRemoveFilesFromContext,
+        requestAnswer: handleRequestAnswer,
+        contextOptimization: handleContextOptimization,
+    };
+    return handlers[actionType] || handleDefaultAction;
+}
+//# sourceMappingURL=step-ask-question.js.map