genaicode 0.2.1 → 0.4.1

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

@@ -0,0 +1,265 @@
+import { StepResult } from './steps-types.js';
+import { askUserForConfirmation, askUserForInput } from '../../main/common/user-actions.js';
+import { putSystemMessage } from '../../main/common/content-bus.js';
+import { getSourceFiles, refreshFiles } from '../../files/find-files.js';
+import { executeStepContextOptimization } from './step-context-optimization.js';
+import { getSourceCodeResponse } from './steps-utils.js';
+import { getSourceCode } from '../../files/read-files.js';
+export async function handleCancelCodeGeneration({ askQuestionCall }) {
+    putSystemMessage('Assistant requested to stop code generation. Exiting...');
+    return {
+        breakLoop: true,
+        stepResult: StepResult.BREAK,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },
+                user: { type: 'user', text: 'Code generation cancelled.' },
+            },
+        ],
+    };
+}
+export async function handleConfirmCodeGeneration({ askQuestionCall }) {
+    const userConfirmation = await askUserForConfirmation('The assistant is ready to start code generation. Do you want to proceed?', true);
+    if (userConfirmation) {
+        putSystemMessage('Proceeding with code generation.');
+        return {
+            breakLoop: true,
+            stepResult: StepResult.CONTINUE,
+            items: [
+                {
+                    assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },
+                    user: { type: 'user', text: '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: 'Declined. Please continue the conversation.',
+                        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+                    },
+                },
+            ],
+        };
+    }
+}
+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.' },
+            },
+        ],
+    };
+}
+export async function handleRequestFilesContent({ askQuestionCall, prompt, 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 assistant = {
+        type: 'assistant',
+        text: askQuestionCall.args?.content ?? '',
+        functionCalls: [askQuestionCall],
+    };
+    const user = {
+        type: 'user',
+        text: '',
+        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+    };
+    if (legitimateFiles.length > 0) {
+        addFileContentsToPrompt(prompt, legitimateFiles, options);
+    }
+    user.text =
+        illegitimateFiles.length > 0
+            ? 'Some files are not legitimate and their content cannot be provided'
+            : 'All requested file contents have been provided automatically.';
+    return { breakLoop: false, stepResult: StepResult.CONTINUE, items: [{ assistant, user }] };
+}
+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,
+            },
+        ],
+    };
+}
+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 }],
+                },
+            },
+        ],
+    };
+}
+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,
+            },
+        ],
+    };
+}
+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 }],
+                },
+            },
+        ],
+    };
+}
+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 }],
+                },
+            },
+        ],
+    };
+}
+function categorizeLegitimateFiles(requestedFiles) {
+    const legitimateFiles = [];
+    const illegitimateFiles = [];
+    requestedFiles.forEach((filePath) => {
+        if (isFilePathLegitimate(filePath)) {
+            legitimateFiles.push(filePath);
+        }
+        else {
+            illegitimateFiles.push(filePath);
+        }
+    });
+    return { legitimateFiles, illegitimateFiles };
+}
+function addFileContentsToPrompt(prompt, filesToAdd, options) {
+    putSystemMessage('Automatically providing content for legitimate files', filesToAdd);
+    const response = getSourceCodeResponse(prompt);
+    if (!response || !response.content) {
+        throw new Error('Could not find source code response');
+    }
+    const sourceCode = getSourceCode({ filterPaths: filesToAdd, forceAll: true }, options);
+    const contentObj = JSON.parse(response.content);
+    filesToAdd.forEach((file) => {
+        contentObj[file] = sourceCode[file];
+    });
+    response.content = JSON.stringify(contentObj);
+}
+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;
+        }
+    }
+}
+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);
+}
+function isFilePathLegitimate(filePath) {
+    return getSourceFiles().includes(filePath);
+}
+//# sourceMappingURL=step-ask-question-handlers.js.map

package/dist/prompt/steps/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-handlers.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AAC5F,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAQzE,OAAO,EAAE,8BAA8B,EAAE,MAAM,gCAAgC,CAAC;AAChF,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAE,eAAe,EAAsB;IACtF,gBAAgB,CAAC,yDAAyD,CAAC,CAAC;IAC5E,OAAO;QACL,SAAS,EAAE,IAAI;QACf,UAAU,EAAE,UAAU,CAAC,KAAK;QAC5B,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,4BAA4B,EAAE;aAC3D;SACF;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAAC,EAAE,eAAe,EAAsB;IACvF,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,0EAA0E,EAC1E,IAAI,CACL,CAAC;IACF,IAAI,gBAAgB,EAAE,CAAC;QACrB,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,0CAA0C,EAAE;iBACzE;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,6CAA6C;wBACnD,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;AAED,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;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAC9C,eAAe,EACf,MAAM,EACN,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,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;IACF,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,EAAE;QACR,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,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,uBAAuB,CAAC,MAAM,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,CAAC,IAAI;QACP,iBAAiB,CAAC,MAAM,GAAG,CAAC;YAC1B,CAAC,CAAC,oEAAoE;YACtE,CAAC,CAAC,+DAA+D,CAAC;IAEtE,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,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,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,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;AAED,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;AAED,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;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,uBAAuB,CAAC,MAAoB,EAAE,UAAoB,EAAE,OAAuB;IAClG,gBAAgB,CAAC,sDAAsD,EAAE,UAAU,CAAC,CAAC;IACrF,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,aAAa,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;IACvF,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAChD,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC1B,UAAU,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IACH,QAAQ,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;AAChD,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;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;AAED,SAAS,oBAAoB,CAAC,QAAgB;IAC5C,OAAO,cAAc,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AAC7C,CAAC"}

package/dist/prompt/steps/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/prompt/steps/step-ask-question-reflect.js

@@ -0,0 +1,40 @@
+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 = await runSelfReflect(askQuestionCall, prompt, functionDefs, options, generateContentFn);
+    if (shouldEscalate) {
+        context.escalationCount++;
+        context.lastEscalationTime = currentTime;
+    }
+    return {
+        shouldEscalate,
+        reason: shouldEscalate
+            ? 'Response must be escalated to non cheap model'
+            : 'Response should not be escalated to non cheap model',
+    };
+}
+async function runSelfReflect(askQuestionCall, prompt, functionDefs, options, generateContentFn) {
+    const result = await generateContentFn([
+        ...prompt,
+        { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+        {
+            type: 'user',
+            text: `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?
+
+If your response meets these criteria, continue the conversation as usual. If you believe that escalating to a more advanced model would significantly improve assistance, please indicate that using the \`shouldEscalate\` parameter.`,
+            functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+        },
+    ], functionDefs, 'askQuestionReflect', 0.2, true, options);
+    const shouldEscalate = result.find((call) => call.name === 'askQuestionReflect')?.args?.shouldEscalate;
+    return shouldEscalate > 50;
+}
+//# sourceMappingURL=step-ask-question-reflect.js.map

package/dist/prompt/steps/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-reflect.ts"],"names":[],"mappings":"AAIA,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,cAAc,GAAG,MAAM,cAAc,CAAC,eAAe,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAC;IAE/G,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,EAAE,cAAc;YACpB,CAAC,CAAC,+CAA+C;YACjD,CAAC,CAAC,qDAAqD;KAC1D,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,eAAgC,EAChC,MAAoB,EACpB,YAA2B,EAC3B,OAAuB,EACvB,iBAA0C;IAE1C,MAAM,MAAM,GAAG,MAAM,iBAAiB,CACpC;QACE,GAAG,MAAM;QAET,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;QAClG;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE;;;;;;wOAM0N;YAChO,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;SACpG;KACF,EACD,YAAY,EACZ,oBAAoB,EACpB,GAAG,EACH,IAAI,EACJ,OAAO,CACR,CAAC;IACF,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,oBAAoB,CAAC,EAAE,IAAI,EAAE,cAAwB,CAAC;IACjH,OAAO,cAAc,GAAG,EAAE,CAAC;AAC7B,CAAC"}

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

@@ -0,0 +1,60 @@
+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';
+export 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[];
+}
+export interface UserItem {
+    type: 'user';
+    text: string;
+    functionResponses?: Array<{
+        name: string;
+        call_id: string;
+        content: string | undefined;
+    }>;
+}
+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 Messages {
+    contextSourceCode: (paths: string[], pathsOnly: boolean) => string;
+}
+export interface EscalationDecision {
+    shouldEscalate: boolean;
+    reason: string;
+}
+export interface SelfReflectionContext {
+    escalationCount: number;
+    lastEscalationTime: number;
+}
+export type SelfReflectionFunction = (response: AskQuestionCall, context: SelfReflectionContext) => Promise<EscalationDecision>;
+export interface SelfReflectionResult {
+    decision: EscalationDecision;
+    updatedContext: SelfReflectionContext;
+}

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

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

package/dist/prompt/steps/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-types.ts"],"names":[],"mappings":""}

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

@@ -1,6 +1,6 @@
-import { FunctionCall, FunctionDef, PromptItem } from '../../ai-service/common.js';
+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: (prompt: PromptItem[], functionDefs: FunctionDef[], requiredFunctionName: string, temperature: number, cheap: boolean, options: CodegenOptions) => Promise<FunctionCall[]>, prompt: PromptItem[], functionDefs: FunctionDef[], temperature: number, cheap: boolean, messages: {
-    contextSourceCode: (paths: string[]) => string;
+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/prompt/steps/step-ask-question.js

@@ -1,80 +1,80 @@
-import { input, confirm } from '@inquirer/prompts';
 import { StepResult } from './steps-types.js';
-export async function executeStepAskQuestion(generateContentFn, prompt, functionDefs, temperature, cheap, messages, options) {
+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 { handleCancelCodeGeneration, handleConfirmCodeGeneration, handleStartCodeGeneration, handleRequestFilesContent, handleRequestPermissions, handleRemoveFilesFromContext, handleRequestAnswer, handleDefaultAction, handleContextOptimization, } from './step-ask-question-handlers.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 questionAsked = false;
-    while (!questionAsked) {
-        const askQuestionResult = await generateContentFn(prompt, functionDefs, 'askQuestion', temperature, cheap, options);
-        const askQuestionCall = askQuestionResult.find((call) => call.name === 'askQuestion');
-        if (askQuestionCall) {
-            console.log('Assistant asks:', askQuestionCall.args);
-            if (askQuestionCall.args?.stopCodegen) {
-                console.log('Assistant requested to stop code generation. Exiting...');
-                return StepResult.BREAK;
-            }
-            else if (!askQuestionCall.args?.shouldPrompt) {
-                console.log('Proceeding with code generation.');
+    const selfReflectionContext = {
+        escalationCount: 0,
+        lastEscalationTime: 0,
+    };
+    while (!abortController?.signal.aborted) {
+        try {
+            const askQuestionCall = await getAskQuestionCall(generateContentFn, prompt, functionDefs, temperature, options, selfReflectionContext);
+            if (!askQuestionCall) {
                 break;
             }
-            const fileContentRequested = (askQuestionCall.args?.requestFilesContent?.length ?? 0) > 0;
-            const permissionsRequested = Object.entries(askQuestionCall.args?.requestPermissions ?? {}).filter(([, enabled]) => enabled).length > 0;
-            const userAnswer = askQuestionCall.args?.shouldPrompt
-                ? fileContentRequested
-                    ? (await confirm({
-                        message: 'The assistant is requesting file contents. Do you want to provide them?',
-                        default: true,
-                    }))
-                        ? 'Providing requested files content.'
-                        : 'Request for file contents denied.'
-                    : permissionsRequested
-                        ? (await confirm({
-                            message: 'The assistant is requesting additional permissions. Do you want to grant them?',
-                            default: false,
-                        }))
-                            ? 'Permissions granted.'
-                            : 'Permission request denied.'
-                        : await input({ message: 'Your answer' })
-                : "Let's proceed with code generation.";
-            if (permissionsRequested && askQuestionCall.args?.requestPermissions && userAnswer === 'Permissions granted.') {
-                if (askQuestionCall.args?.requestPermissions.enableImagen) {
-                    options.imagen = options.aiService === 'chat-gpt' ? 'dall-e' : 'vertex-ai';
-                }
-                if (askQuestionCall.args?.requestPermissions.enableVision) {
-                    options.vision = true;
-                }
-                if (askQuestionCall.args?.requestPermissions.allowDirectoryCreate) {
-                    options.allowDirectoryCreate = true;
-                }
-                if (askQuestionCall.args?.requestPermissions.allowFileCreate) {
-                    options.allowFileCreate = true;
-                }
-                if (askQuestionCall.args?.requestPermissions.allowFileDelete) {
-                    options.allowFileDelete = true;
-                }
-                if (askQuestionCall.args?.requestPermissions.allowFileMove) {
-                    options.allowFileMove = true;
+            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;
             }
-            prompt.push({ type: 'assistant', functionCalls: [askQuestionCall] }, {
-                type: 'user',
-                text: userAnswer,
-                functionResponses: [
-                    {
-                        name: 'askQuestion',
-                        call_id: askQuestionCall.id,
-                        content: fileContentRequested && userAnswer === 'Providing requested files content.'
-                            ? messages.contextSourceCode(askQuestionCall.args.requestFilesContent)
-                            : undefined,
-                    },
-                ],
-            });
-            console.log('The question was answered', userAnswer);
         }
-        else {
-            console.log('Assistant did not ask a question. Proceeding with code generation.');
-            break;
+        catch (error) {
+            console.error('Error in executeStepAskQuestion:', error);
+            putSystemMessage(`An error occurred: ${error instanceof Error ? error.message : String(error)}`);
+            return StepResult.BREAK;
         }
     }
-    return StepResult.CONTINUE;
+    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

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

@@ -1 +1 @@
-{"version":3,"file":"step-ask-question.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-ask-question.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEnD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAG9C,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,iBAO4B,EAC5B,MAAoB,EACpB,YAA2B,EAC3B,WAAmB,EACnB,KAAc,EACd,QAEC,EACD,OAAuB;IAEvB,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,KAAK,CAAC;IAC5B,OAAO,CAAC,aAAa,EAAE,CAAC;QACtB,MAAM,iBAAiB,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACpH,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAevE,CAAC;QACd,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;YAErD,IAAI,eAAe,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC;gBACtC,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;gBACvE,OAAO,UAAU,CAAC,KAAK,CAAC;YAC1B,CAAC;iBAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,YAAY,EAAE,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;gBAChD,MAAM;YACR,CAAC;YAED,MAAM,oBAAoB,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,mBAAmB,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAC1F,MAAM,oBAAoB,GACxB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,EAAE,kBAAkB,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;YAC7G,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,EAAE,YAAY;gBACnD,CAAC,CAAC,oBAAoB;oBACpB,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC;wBACb,OAAO,EAAE,yEAAyE;wBAClF,OAAO,EAAE,IAAI;qBACd,CAAC,CAAC;wBACH,CAAC,CAAC,oCAAoC;wBACtC,CAAC,CAAC,mCAAmC;oBACvC,CAAC,CAAC,oBAAoB;wBACpB,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC;4BACb,OAAO,EAAE,gFAAgF;4BACzF,OAAO,EAAE,KAAK;yBACf,CAAC,CAAC;4BACH,CAAC,CAAC,sBAAsB;4BACxB,CAAC,CAAC,4BAA4B;wBAChC,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;gBAC7C,CAAC,CAAC,qCAAqC,CAAC;YAE1C,IAAI,oBAAoB,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,IAAI,UAAU,KAAK,sBAAsB,EAAE,CAAC;gBAC9G,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,YAAY,EAAE,CAAC;oBAC1D,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC;gBAC7E,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,YAAY,EAAE,CAAC;oBAC1D,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;gBACxB,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,oBAAoB,EAAE,CAAC;oBAClE,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC;gBACtC,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,eAAe,EAAE,CAAC;oBAC7D,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;gBACjC,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,eAAe,EAAE,CAAC;oBAC7D,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;gBACjC,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,aAAa,EAAE,CAAC;oBAC3D,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;gBAC/B,CAAC;YACH,CAAC;YAED,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE,EACvD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,UAAU;gBAChB,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,aAAa;wBACnB,OAAO,EAAE,eAAe,CAAC,EAAE;wBAC3B,OAAO,EACL,oBAAoB,IAAI,UAAU,KAAK,oCAAoC;4BACzE,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC,eAAe,CAAC,IAAK,CAAC,mBAAoB,CAAC;4BACxE,CAAC,CAAC,SAAS;qBAChB;iBACF;aACF,CACF,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC;YAClF,MAAM;QACR,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC,QAAQ,CAAC;AAC7B,CAAC"}
+{"version":3,"file":"step-ask-question.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-ask-question.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AACzG,OAAO,EAAE,eAAe,EAAE,MAAM,0CAA0C,CAAC;AAC3E,OAAO,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AAE5E,OAAO,EACL,0BAA0B,EAC1B,2BAA2B,EAC3B,yBAAyB,EACzB,yBAAyB,EACzB,wBAAwB,EACxB,4BAA4B,EAC5B,mBAAmB,EACnB,mBAAmB,EACnB,yBAAyB,GAC1B,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAEvE,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,WAAmB,EACnB,QAEC,EACD,OAAuB;IAEvB,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAE3D,MAAM,qBAAqB,GAA0B;QACnD,eAAe,EAAE,CAAC;QAClB,kBAAkB,EAAE,CAAC;KACtB,CAAC;IAEF,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QACxC,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,MAAM,kBAAkB,CAC9C,iBAAiB,EACjB,MAAM,EACN,YAAY,EACZ,WAAW,EACX,OAAO,EACP,qBAAqB,CACtB,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,MAAM;YACR,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,eAAe,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;gBAClC,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;YAC1E,CAAC;YAED,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,EAAE,UAAU,CAAC;YACpD,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,aAAa,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC;gBACnD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,EAAE,eAAe,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,CAAC,CAAC;gBAEtG,yFAAyF;gBACzF,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAEpD,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,OAAO,MAAM,CAAC,UAAU,CAAC;gBAC3B,CAAC;gBAED,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBACpF,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAC3C,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;gBAC9C,MAAM;YACR,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;YACzD,gBAAgB,CAAC,sBAAsB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACjG,OAAO,UAAU,CAAC,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,sEAAsE,CAAC,CAAC;IACzF,OAAO,UAAU,CAAC,KAAK,CAAC;AAC1B,CAAC;AAED,KAAK,UAAU,kBAAkB,CAC/B,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,WAAmB,EACnB,OAAuB,EACvB,qBAA4C;IAE5C,MAAM,kBAAkB,GAAwB,CAAC,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAClH,IAAI,iBAAiB,GAAG,MAAM,iBAAiB,CAAC,GAAG,kBAAkB,CAAC,CAAC;IACvE,iBAAiB,GAAG,MAAM,8BAA8B,CAAC,kBAAkB,EAAE,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;IAEnH,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAE1E,CAAC;IAEd,IAAI,kBAAkB,EAAE,CAAC;QACvB,MAAM,kBAAkB,GAAG,MAAM,qBAAqB,CACpD,kBAAkB,EAClB,qBAAqB,EACrB,MAAM,EACN,YAAY,EACZ,OAAO,EACP,iBAAiB,CAClB,CAAC;QAEF,IAAI,kBAAkB,CAAC,cAAc,EAAE,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;YACvE,8BAA8B;YAC9B,MAAM,eAAe,GAAwB,CAAC,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YAChH,IAAI,cAAc,GAAG,MAAM,iBAAiB,CAAC,GAAG,eAAe,CAAC,CAAC;YACjE,cAAc,GAAG,MAAM,8BAA8B,CAAC,eAAe,EAAE,cAAc,EAAE,iBAAiB,CAAC,CAAC;YAC1G,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAAgC,CAAC;QACnG,CAAC;IACH,CAAC;IAED,OAAO,kBAAkB,CAAC;AAC5B,CAAC;AAED,SAAS,gBAAgB,CAAC,UAAsB;IAC9C,MAAM,QAAQ,GAAsC;QAClD,oBAAoB,EAAE,0BAA0B;QAChD,qBAAqB,EAAE,2BAA2B;QAClD,mBAAmB,EAAE,yBAAyB;QAC9C,mBAAmB,EAAE,yBAAyB;QAC9C,kBAAkB,EAAE,wBAAwB;QAC5C,sBAAsB,EAAE,4BAA4B;QACpD,aAAa,EAAE,mBAAmB;QAClC,mBAAmB,EAAE,yBAAyB;KAC/C,CAAC;IAEF,OAAO,QAAQ,CAAC,UAAU,CAAC,IAAI,mBAAmB,CAAC;AACrD,CAAC"}

package/dist/prompt/steps/step-context-optimization.d.ts

@@ -0,0 +1,4 @@
+import { GenerateContentFunction, PromptItem } from '../../ai-service/common.js';
+import { CodegenOptions } from '../../main/codegen-types.js';
+import { StepResult } from './steps-types.js';
+export declare function executeStepContextOptimization(generateContentFn: GenerateContentFunction, prompt: PromptItem[], options: CodegenOptions): Promise<StepResult>;