genaicode 0.8.6 → 0.9.1

package/dist/prompt/steps/step-ask-question/handlers/handle-pull-app-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-pull-app-context.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-pull-app-context.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,4CAA4C,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAG/D;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,EACzC,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,gBAAgB,CAAC,uBAAuB,CAAC,CAAC;IAC1C,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CACnD,MAAM,EACN,eAAe,EAAE,EACjB,gBAAgB,EAChB,GAAG,EACH,SAAS,CAAC,KAAK,EACf,OAAO,CACR,CAAmD,CAAC;IAErD,IAAI,CAAC,kBAAkB,EAAE,IAAI,EAAE,CAAC;QAC9B,gBAAgB,CAAC,4CAA4C,CAAC,CAAC;QAC/D,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,kBAAkB,CAAC,IAAI,CAAC;IAEhD,IAAI,CAAC;QACH,iBAAiB;QACjB,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,qCAAqC;QACrC,MAAM,YAAY,GAAG,MAAM,eAAe,CAAC,GAAG,CAAC,CAAC;QAEhD,uCAAuC;QACvC,MAAM,WAAW,GAAG,YAAY;YAC9B,CAAC,CAAC,oCAAoC,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;YACtF,CAAC,CAAC,mCAAmC,GAAG,GAAG,CAAC;QAE9C,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,2BAA2B,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAErE,oDAAoD;QACpD,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,GAAG,WAAW,GAAG,UAAU,EAAE;wBACnC,aAAa,EAAE,CAAC,kBAAkB,CAAC;qBACpC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,iCAAiC;wBACvC,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,gBAAgB;gCACtB,OAAO,EAAE,kBAAkB,CAAC,EAAE;gCAC9B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,YAAY,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;6BACrF;yBACF;wBACD,IAAI,EAAE;4BACJ,UAAU,EAAE,GAAG;4BACf,YAAY,EAAE,YAAY;4BAC1B,MAAM;4BACN,SAAS,EAAE,KAAK;yBACjB;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,wDAAwD;QACxD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC;QACvF,gBAAgB,CAAC,6BAA6B,YAAY,EAAE,CAAC,CAAC;QAE9D,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,uCAAuC,GAAG,MAAM,YAAY,EAAE;qBACrE;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,2BAA2B;wBACjC,IAAI,EAAE;4BACJ,UAAU,EAAE,GAAG;4BACf,KAAK,EAAE,YAAY;4BACnB,MAAM;4BACN,SAAS,EAAE,KAAK;yBACjB;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-ask-question/handlers/handle-push-app-context.d.ts

@@ -0,0 +1,9 @@
+import { ActionResult, ActionHandlerProps } from '../step-ask-question-types.js';
+/**
+ * Handler for pushAppContext action
+ * Updates context value in the content-bus system with proper validation and error handling
+ *
+ * @param props Action handler properties containing the function call
+ * @returns ActionResult with the updated context value and conversation items
+ */
+export declare function handlePushAppContext({ generateContentFn, prompt, options, }: ActionHandlerProps): Promise<ActionResult>;

package/dist/prompt/steps/step-ask-question/handlers/handle-push-app-context.js

@@ -0,0 +1,91 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { setContextValue } from '../../../../main/common/app-context-bus.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+/**
+ * Handler for pushAppContext action
+ * Updates context value in the content-bus system with proper validation and error handling
+ *
+ * @param props Action handler properties containing the function call
+ * @returns ActionResult with the updated context value and conversation items
+ */
+export async function handlePushAppContext({ generateContentFn, prompt, options, }) {
+    putSystemMessage('Updating context...');
+    const [pushAppContextCall] = (await generateContentFn(prompt, getFunctionDefs(), 'pushAppContext', 0.7, ModelType.CHEAP, options));
+    if (!pushAppContextCall?.args) {
+        putSystemMessage('Failed to get valid pushAppContext request');
+        return { breakLoop: false, items: [] };
+    }
+    const { key, reason } = pushAppContextCall.args;
+    let { value } = pushAppContextCall.args;
+    try {
+        // Validate input
+        if (!key) {
+            throw new Error('Context key is required');
+        }
+        if (value === undefined) {
+            throw new Error('Context value cannot be undefined');
+        }
+        try {
+            value = JSON.parse(value);
+        }
+        catch {
+            // ignore error
+        }
+        // Set context value using content-bus
+        // Note: setContextValue includes value validation internally
+        await setContextValue(key, value);
+        // Format success message
+        const updateInfo = `Successfully updated context key "${key}" with value: ${JSON.stringify(value, null, 2)}`;
+        const reasonInfo = reason ? `\nReason for update: ${reason}` : '';
+        // Return the action result with update confirmation
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: `${updateInfo}${reasonInfo}`,
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Context updated successfully.',
+                        data: {
+                            contextKey: key,
+                            contextValue: value,
+                            reason,
+                            operation: 'set',
+                        },
+                    },
+                },
+            ],
+        };
+    }
+    catch (error) {
+        // Handle any errors that occur during context update
+        const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
+        putSystemMessage(`Error updating context: ${errorMessage}`);
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: `Failed to update context for key "${key}": ${errorMessage}`,
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Context update failed.',
+                        data: {
+                            contextKey: key,
+                            error: errorMessage,
+                            reason,
+                            operation: 'set',
+                        },
+                    },
+                },
+            ],
+        };
+    }
+}
+//# sourceMappingURL=handle-push-app-context.js.map

package/dist/prompt/steps/step-ask-question/handlers/handle-push-app-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-push-app-context.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-push-app-context.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,4CAA4C,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAG/D;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,EACzC,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;IAExC,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CACnD,MAAM,EACN,eAAe,EAAE,EACjB,gBAAgB,EAChB,GAAG,EACH,SAAS,CAAC,KAAK,EACf,OAAO,CACR,CAAuC,CAAC;IAEzC,IAAI,CAAC,kBAAkB,EAAE,IAAI,EAAE,CAAC;QAC9B,gBAAgB,CAAC,4CAA4C,CAAC,CAAC;QAC/D,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,kBAAkB,CAAC,IAAI,CAAC;IAChD,IAAI,EAAE,KAAK,EAAE,GAAG,kBAAkB,CAAC,IAAI,CAAC;IAExC,IAAI,CAAC;QACH,iBAAiB;QACjB,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC;YACH,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;QAED,sCAAsC;QACtC,6DAA6D;QAC7D,MAAM,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAElC,yBAAyB;QACzB,MAAM,UAAU,GAAG,qCAAqC,GAAG,iBAAiB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC7G,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,wBAAwB,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAElE,oDAAoD;QACpD,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,GAAG,UAAU,GAAG,UAAU,EAAE;qBACnC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,+BAA+B;wBACrC,IAAI,EAAE;4BACJ,UAAU,EAAE,GAAG;4BACf,YAAY,EAAE,KAAK;4BACnB,MAAM;4BACN,SAAS,EAAE,KAAK;yBACjB;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,qDAAqD;QACrD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC;QACvF,gBAAgB,CAAC,2BAA2B,YAAY,EAAE,CAAC,CAAC;QAE5D,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,qCAAqC,GAAG,MAAM,YAAY,EAAE;qBACnE;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,wBAAwB;wBAC9B,IAAI,EAAE;4BACJ,UAAU,EAAE,GAAG;4BACf,KAAK,EAAE,YAAY;4BACnB,MAAM;4BACN,SAAS,EAAE,KAAK;yBACjB;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-ask-question/handlers/handle-reasoning-inference.d.ts

@@ -0,0 +1,7 @@
+import { ActionResult, ActionHandlerProps } from '../step-ask-question-types.js';
+/**
+ * Handler for the reasoningInference action.
+ * This handler is designed to work with models that don't support function calling
+ * but provide reasoning tokens along with their responses.
+ */
+export declare function handleReasoningInference({ askQuestionCall, prompt, generateContentFn, options, }: ActionHandlerProps): Promise<ActionResult>;

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

@@ -0,0 +1,97 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { validateAndRecoverSingleResult } from '../../step-validate-recover.js';
+import { getContextSourceCode } from '../../../../files/source-code-utils.js';
+/**
+ * Handler for the reasoningInference action.
+ * This handler is designed to work with models that don't support function calling
+ * but provide reasoning tokens along with their responses.
+ */
+export async function handleReasoningInference({ askQuestionCall, prompt, generateContentFn, options, }) {
+    try {
+        putSystemMessage('Reasoning inference: generating prompt');
+        const request = [
+            [
+                ...prompt,
+                {
+                    type: 'assistant',
+                    text: askQuestionCall.args.message,
+                },
+            ],
+            getFunctionDefs(),
+            'reasoningInference',
+            0.7,
+            ModelType.CHEAP,
+            options,
+        ];
+        const result = await generateContentFn(...request);
+        const [reasoningInferenceCall] = (await validateAndRecoverSingleResult(request, result, generateContentFn));
+        if (!reasoningInferenceCall?.args?.prompt) {
+            putSystemMessage('Failed to get valid reasoningInference request');
+            prompt.push({
+                type: 'assistant',
+                text: askQuestionCall.args.message,
+            }, {
+                type: 'user',
+                text: 'Failed to get valid reasoningInference request',
+            });
+            return { breakLoop: false, items: [] };
+        }
+        putSystemMessage('Reasoning inference: calling reasoning model', reasoningInferenceCall.args);
+        // Get expanded context using getContextSourceCode
+        const contextPaths = reasoningInferenceCall.args.contextPaths || [];
+        const expandedContextSourceCodeMap = getContextSourceCode(contextPaths, options);
+        // Call the reasoning model with appropriate settings
+        const [reasoningInferenceResponseCall] = (await generateContentFn([
+            {
+                type: 'user',
+                text: reasoningInferenceCall.args.prompt +
+                    (Object.keys(expandedContextSourceCodeMap).length > 0
+                        ? '\n\nContents of relevant files:\n' +
+                            Object.entries(expandedContextSourceCodeMap)
+                                .map(([path, file]) => 'content' in file && file.content ? `File: ${path}\n\`\`\`${file.content}\`\`\`` : '')
+                                .filter(Boolean)
+                                .join('\n\n')
+                        : ''),
+            },
+        ], [], 'reasoningInferenceResponse', 0.7, ModelType.REASONING, options));
+        putSystemMessage('Reasoning inference: received response', reasoningInferenceResponseCall.args);
+        prompt.push({
+            type: 'assistant',
+            text: askQuestionCall.args.message,
+            functionCalls: [reasoningInferenceCall],
+        }, {
+            type: 'user',
+            functionResponses: [
+                {
+                    name: 'reasoningInference',
+                    call_id: reasoningInferenceCall.id,
+                    content: reasoningInferenceResponseCall.args?.response,
+                },
+            ],
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    catch (error) {
+        // Handle errors gracefully
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        putSystemMessage(`Error during reasoning inference: ${errorMessage}`);
+        prompt.push({
+            type: 'assistant',
+            text: askQuestionCall.args.message,
+        }, {
+            type: 'user',
+            text: `Error during reasoning inference.`,
+        });
+        return {
+            // TODO: rather do a retry here?
+            breakLoop: true,
+            items: [],
+        };
+    }
+}
+//# sourceMappingURL=handle-reasoning-inference.js.map

package/dist/prompt/steps/step-ask-question/handlers/handle-reasoning-inference.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-reasoning-inference.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-reasoning-inference.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,8BAA8B,EAAE,MAAM,gCAAgC,CAAC;AAOhF,OAAO,EAAE,oBAAoB,EAAE,MAAM,wCAAwC,CAAC;AAE9E;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,EAC7C,eAAe,EACf,MAAM,EACN,iBAAiB,EACjB,OAAO,GACY;IACnB,IAAI,CAAC;QACH,gBAAgB,CAAC,wCAAwC,CAAC,CAAC;QAE3D,MAAM,OAAO,GAAwB;YACnC;gBACE,GAAG,MAAM;gBACT;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;iBACpC;aACF;YACD,eAAe,EAAE;YACjB,oBAAoB;YACpB,GAAG;YACH,SAAS,CAAC,KAAK;YACf,OAAO;SACR,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,GAAG,OAAO,CAAC,CAAC;QACnD,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,MAAM,8BAA8B,CAAC,OAAO,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAEzG,CAAC;QAEF,IAAI,CAAC,sBAAsB,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAC1C,gBAAgB,CAAC,gDAAgD,CAAC,CAAC;YACnE,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;aACpC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,gDAAgD;aACvD,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,gBAAgB,CAAC,8CAA8C,EAAE,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAE9F,kDAAkD;QAClD,MAAM,YAAY,GAAG,sBAAsB,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC;QACpE,MAAM,4BAA4B,GAAG,oBAAoB,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAEjF,qDAAqD;QACrD,MAAM,CAAC,8BAA8B,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAC/D;YACE;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EACF,sBAAsB,CAAC,IAAI,CAAC,MAAM;oBAClC,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC,MAAM,GAAG,CAAC;wBACnD,CAAC,CAAC,mCAAmC;4BACnC,MAAM,CAAC,OAAO,CAAC,4BAA4B,CAAC;iCACzC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CACpB,SAAS,IAAI,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,IAAI,WAAW,IAAI,CAAC,OAAO,QAAQ,CAAC,CAAC,CAAC,EAAE,CACtF;iCACA,MAAM,CAAC,OAAO,CAAC;iCACf,IAAI,CAAC,MAAM,CAAC;wBACjB,CAAC,CAAC,EAAE,CAAC;aACV;SACF,EACD,EAAE,EACF,4BAA4B,EAC5B,GAAG,EACH,SAAS,CAAC,SAAS,EACnB,OAAO,CACR,CAAmD,CAAC;QAErD,gBAAgB,CAAC,wCAAwC,EAAE,8BAA8B,CAAC,IAAI,CAAC,CAAC;QAEhG,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;YACnC,aAAa,EAAE,CAAC,sBAAsB,CAAC;SACxC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,oBAAoB;oBAC1B,OAAO,EAAE,sBAAsB,CAAC,EAAE;oBAClC,OAAO,EAAE,8BAA8B,CAAC,IAAI,EAAE,QAAQ;iBACvD;aACF;SACF,CACF,CAAC;QAEF,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,2BAA2B;QAC3B,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,gBAAgB,CAAC,qCAAqC,YAAY,EAAE,CAAC,CAAC;QAEtE,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;SACpC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,mCAAmC;SAC1C,CACF,CAAC;QAEF,OAAO;YACL,gCAAgC;YAChC,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;AACH,CAAC"}

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

@@ -1,4 +1,5 @@
 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';
 /**
@@ -6,7 +7,7 @@ import { putSystemMessage } from '../../../../main/common/content-bus.js';
  */
 export async function handleSearchCode({ askQuestionCall, generateContentFn, prompt, options, }) {
     try {
-        const searchCodeCall = await generateSearchCodeCall(generateContentFn, prompt, askQuestionCall, options, false);
+        const searchCodeCall = await generateSearchCodeCall(generateContentFn, prompt, askQuestionCall, options);
         // Extract search parameters from the call arguments
         const args = searchCodeCall?.args;
         if (!searchCodeCall || !args?.query) {
@@ -37,7 +38,6 @@ export async function handleSearchCode({ askQuestionCall, generateContentFn, pro
                     },
                     user: {
                         type: 'user',
-                        text: 'Search completed, lets continue the conversation.',
                         functionResponses: [
                             {
                                 name: 'searchCode',
@@ -70,7 +70,7 @@ export async function handleSearchCode({ askQuestionCall, generateContentFn, pro
         };
     }
 }
-async function generateSearchCodeCall(generateContentFn, prompt, askQuestionCall, options, cheap) {
+async function generateSearchCodeCall(generateContentFn, prompt, askQuestionCall, options) {
     const [searchCodeCall] = await generateContentFn([
         ...prompt,
         {
@@ -81,7 +81,7 @@ async function generateSearchCodeCall(generateContentFn, prompt, askQuestionCall
             type: 'user',
             text: 'Yes, you can search the code.',
         },
-    ], getFunctionDefs(), 'searchCode', 0.7, cheap, options);
+    ], getFunctionDefs(), 'searchCode', 0.7, ModelType.CHEAP, options);
     return searchCodeCall;
 }
 //# sourceMappingURL=handle-search-code.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"handle-search-code.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-search-code.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AAGrE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAE1E;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,EACrC,eAAe,EACf,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,MAAM,sBAAsB,CAAC,iBAAiB,EAAE,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAEhH,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,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;wBACzC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,mDAAmD;wBACzD,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,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,4BAA4B,YAAY,kDAAkD;qBACjG;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,oDAAoD;qBAC3D;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,iBAA0C,EAC1C,MAAoB,EACpB,eAAgC,EAChC,OAAuB,EACvB,KAAc;IAEd,MAAM,CAAC,cAAc,CAAC,GAAG,MAAM,iBAAiB,CAC9C;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,+BAA+B;SACtC;KACF,EACD,eAAe,EAAE,EACjB,YAAY,EACZ,GAAG,EACH,KAAK,EACL,OAAO,CACR,CAAC;IAEF,OAAO,cAA0D,CAAC;AACpE,CAAC"}
+{"version":3,"file":"handle-search-code.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/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;AAE1E;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,EACrC,eAAe,EACf,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,MAAM,sBAAsB,CAAC,iBAAiB,EAAE,MAAM,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;QAEzG,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,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;wBACzC,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,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,4BAA4B,YAAY,kDAAkD;qBACjG;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,oDAAoD;qBAC3D;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,iBAA0C,EAC1C,MAAoB,EACpB,eAAgC,EAChC,OAAuB;IAEvB,MAAM,CAAC,cAAc,CAAC,GAAG,MAAM,iBAAiB,CAC9C;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,+BAA+B;SACtC;KACF,EACD,eAAe,EAAE,EACjB,YAAY,EACZ,GAAG,EACH,SAAS,CAAC,KAAK,EACf,OAAO,CACR,CAAC;IAEF,OAAO,cAA0D,CAAC;AACpE,CAAC"}

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

@@ -1,9 +1,6 @@
 import { askUserForInput } from '../../../../main/common/user-actions.js';
 export async function handleSendMessage({ askQuestionCall, options }) {
-    const response = await askUserForInput('Your answer', askQuestionCall.args?.message ?? '');
-    if (response.options?.aiService) {
-        options.aiService = response.options.aiService;
-    }
+    const response = await askUserForInput('Your answer', askQuestionCall.args?.message ?? '', options);
     return {
         breakLoop: false,
         items: [

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

@@ -1 +1 @@
-{"version":3,"file":"handle-send-message.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-send-message.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAG1E,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,EAAE,eAAe,EAAE,OAAO,EAAsB;IACtF,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;IAC3F,IAAI,QAAQ,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC;QAChC,OAAO,CAAC,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC;IACjD,CAAC;IACD,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE;gBAC3E,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ,CAAC,MAAM;iBACtB;aACF;SACF;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"handle-send-message.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-send-message.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAG1E,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,EAAE,eAAe,EAAE,OAAO,EAAsB;IACtF,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;IACpG,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE;gBAC3E,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ,CAAC,MAAM;iBACtB;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

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

@@ -1,11 +1,42 @@
 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 } from '../../../../main/common/content-bus.js';
+import { putSystemMessage, putUserMessage } from '../../../../main/common/content-bus.js';
 import { refreshFiles } from '../../../../files/find-files.js';
+import { generateRequestFilesContentCall } from './request-files-content.js';
+import { executeStepEnsureContext } from '../../step-ensure-context.js';
 export const handleUpdateFile = async ({ askQuestionCall, options, prompt, generateContentFn, }) => {
-    putSystemMessage('File update requested, new content generation started.', askQuestionCall);
+    putSystemMessage('File update requested.', askQuestionCall);
+    // 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.confirmed) {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Canceling file update.' + (generateConfirmation.answer ? ` ${generateConfirmation.answer}` : ''),
+                    },
+                },
+            ],
+        };
+    }
+    if (generateConfirmation.answer) {
+        putUserMessage(generateConfirmation.answer);
+    }
+    putSystemMessage('Content generation started.', askQuestionCall);
+    // Ensure file content is available in the context
+    const requestedFilesCall = await generateRequestFilesContentCall(generateContentFn, prompt, askQuestionCall, options, ModelType.CHEAP);
+    if (requestedFilesCall) {
+        await executeStepEnsureContext(prompt, requestedFilesCall, options);
+    }
     const [updateFileCall] = (await generateContentFn([
         ...prompt,
         {
@@ -14,9 +45,10 @@ export const handleUpdateFile = async ({ askQuestionCall, options, prompt, gener
         },
         {
             type: 'user',
-            text: 'Ok, please provide the update using `updateFile` function. Please remember to use absolute file path.',
+            text: 'Ok, please provide the update using `updateFile` function. Please remember to use absolute file path.' +
+                (generateConfirmation.answer ? ` ${generateConfirmation.answer}` : ''),
         },
-    ], getFunctionDefs(), 'updateFile', options.temperature ?? 0.7, false, options));
+    ], getFunctionDefs(), 'updateFile', options.temperature ?? 0.7, ModelType.DEFAULT, options));
     if (!updateFileCall || !updateFileCall.args) {
         return {
             breakLoop: true,
@@ -43,15 +75,43 @@ export const handleUpdateFile = async ({ askQuestionCall, options, prompt, gener
             ],
         };
     }
-    putSystemMessage(`The assistant suggests updating the file ${filePath}.`, {
+    putSystemMessage(`The assistant has generated new content for file ${filePath}.`, {
         name: updateFileCall.name,
         args: {
             ...updateFileCall.args,
             oldContent: file.content,
         },
     });
-    const userConfirmation = await askUserForConfirmationWithAnswer('The assistant suggests updating the file. Do you want to proceed?', 'Accept file update', 'Reject file update', false);
-    if (!userConfirmation.confirmed) {
+    // 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);
+    if (!applyConfirmation.confirmed) {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                        functionCalls: [updateFileCall],
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Rejecting file update.' + (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : ''),
+                        functionResponses: [
+                            {
+                                name: 'updateFile',
+                                call_id: updateFileCall.id,
+                                content: '',
+                            },
+                        ],
+                    },
+                },
+            ],
+        };
+    }
+    try {
+        await executeUpdateFile(updateFileCall.args);
+        refreshFiles();
         return {
             breakLoop: false,
             items: [
@@ -63,7 +123,7 @@ export const handleUpdateFile = async ({ askQuestionCall, options, prompt, gener
                     },
                     user: {
                         type: 'user',
-                        text: 'Rejecting file update.' + (userConfirmation.answer ? ` ${userConfirmation.answer}` : ''),
+                        text: 'Accepting file update.' + (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : ''),
                         functionResponses: [
                             {
                                 name: 'updateFile',
@@ -76,61 +136,32 @@ export const handleUpdateFile = async ({ askQuestionCall, options, prompt, gener
             ],
         };
     }
-    else {
-        try {
-            await executeUpdateFile(updateFileCall.args);
-            refreshFiles();
-            return {
-                breakLoop: false,
-                items: [
-                    {
-                        assistant: {
-                            type: 'assistant',
-                            text: askQuestionCall.args.message,
-                            functionCalls: [updateFileCall],
-                        },
-                        user: {
-                            type: 'user',
-                            text: 'Accepting file update.' + (userConfirmation.answer ? ` ${userConfirmation.answer}` : ''),
-                            functionResponses: [
-                                {
-                                    name: 'updateFile',
-                                    call_id: updateFileCall.id,
-                                    content: '',
-                                },
-                            ],
-                        },
+    catch {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                        functionCalls: [updateFileCall],
                     },
-                ],
-            };
-        }
-        catch {
-            return {
-                breakLoop: false,
-                items: [
-                    {
-                        assistant: {
-                            type: 'assistant',
-                            text: askQuestionCall.args.message,
-                            functionCalls: [updateFileCall],
-                        },
-                        user: {
-                            type: 'user',
-                            text: 'Accepting file update.' +
-                                (userConfirmation.answer ? ` ${userConfirmation.answer}` : '') +
-                                ' Unfortunately, the file update failed, the new content was not saved.',
-                            functionResponses: [
-                                {
-                                    name: 'updateFile',
-                                    call_id: updateFileCall.id,
-                                    content: '',
-                                },
-                            ],
-                        },
+                    user: {
+                        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: '',
+                            },
+                        ],
                     },
-                ],
-            };
-        }
+                },
+            ],
+        };
     }
 };
 //# sourceMappingURL=handle-update-file.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"handle-update-file.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/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,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAE/D,MAAM,CAAC,MAAM,gBAAgB,GAAkB,KAAK,EAAE,EACpD,eAAe,EACf,OAAO,EACP,MAAM,EACN,iBAAiB,GAClB,EAAyB,EAAE;IAC1B,gBAAgB,CAAC,wDAAwD,EAAE,eAAe,CAAC,CAAC;IAE5F,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAC/C;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,uGAAuG;SAC9G;KACF,EACD,eAAe,EAAE,EACjB,YAAY,EACZ,OAAO,CAAC,WAAW,IAAI,GAAG,EAC1B,KAAK,EACL,OAAO,CACR,CAAkD,CAAC;IAEpD,IAAI,CAAC,cAAc,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QAC5C,OAAO;YACL,SAAS,EAAE,IAAI;YACf,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;QAEpG,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;qBACpC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,mFAAmF;qBAC1F;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,4CAA4C,QAAQ,GAAG,EAAE;QACxE,IAAI,EAAE,cAAc,CAAC,IAAI;QACzB,IAAI,EAAE;YACJ,GAAG,cAAc,CAAC,IAAI;YACtB,UAAU,EAAE,IAAI,CAAC,OAAO;SACzB;KACF,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,MAAM,gCAAgC,CAC7D,mEAAmE,EACnE,oBAAoB,EACpB,oBAAoB,EACpB,KAAK,CACN,CAAC;IAEF,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAChC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;wBACnC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC/F,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,IAAI,CAAC;YACH,MAAM,iBAAiB,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAC7C,YAAY,EAAE,CAAC;YACf,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE;oBACL;wBACE,SAAS,EAAE;4BACT,IAAI,EAAE,WAAW;4BACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;4BACnC,aAAa,EAAE,CAAC,cAAc,CAAC;yBAChC;wBACD,IAAI,EAAE;4BACJ,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;4BAC/F,iBAAiB,EAAE;gCACjB;oCACE,IAAI,EAAE,YAAY;oCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;oCAC1B,OAAO,EAAE,EAAE;iCACZ;6BACF;yBACF;qBACF;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE;oBACL;wBACE,SAAS,EAAE;4BACT,IAAI,EAAE,WAAW;4BACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;4BACnC,aAAa,EAAE,CAAC,cAAc,CAAC;yBAChC;wBACD,IAAI,EAAE;4BACJ,IAAI,EAAE,MAAM;4BACZ,IAAI,EACF,wBAAwB;gCACxB,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gCAC9D,wEAAwE;4BAC1E,iBAAiB,EAAE;gCACjB;oCACE,IAAI,EAAE,YAAY;oCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;oCAC1B,OAAO,EAAE,EAAE;iCACZ;6BACF;yBACF;qBACF;iBACF;aACF,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC,CAAC"}
+{"version":3,"file":"handle-update-file.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/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,4BAA4B,CAAC;AAC7E,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AAExE,MAAM,CAAC,MAAM,gBAAgB,GAAkB,KAAK,EAAE,EACpD,eAAe,EACf,OAAO,EACP,MAAM,EACN,iBAAiB,GAClB,EAAyB,EAAE;IAC1B,gBAAgB,CAAC,wBAAwB,EAAE,eAAe,CAAC,CAAC;IAE5D,+EAA+E;IAC/E,MAAM,oBAAoB,GAAG,MAAM,gCAAgC,CACjE,oHAAoH,EACpH,kBAAkB,EAClB,2BAA2B,EAC3B,KAAK,EACL,OAAO,CACR,CAAC;IAEF,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,CAAC;QACpC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;qBACpC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBACxG;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC;QAChC,cAAc,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,gBAAgB,CAAC,6BAA6B,EAAE,eAAe,CAAC,CAAC;IAEjE,kDAAkD;IAClD,MAAM,kBAAkB,GAAG,MAAM,+BAA+B,CAC9D,iBAAiB,EACjB,MAAM,EACN,eAAe,EACf,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,CAAC,MAAM,iBAAiB,CAC/C;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C;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,eAAe,EAAE,EACjB,YAAY,EACZ,OAAO,CAAC,WAAW,IAAI,GAAG,EAC1B,SAAS,CAAC,OAAO,EACjB,OAAO,CACR,CAAkD,CAAC;IAEpD,IAAI,CAAC,cAAc,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QAC5C,OAAO;YACL,SAAS,EAAE,IAAI;YACf,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,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;qBACpC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,mFAAmF;qBAC1F;iBACF;aACF;SACF,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,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,CAAC;QACjC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;wBACnC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjG,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,iBAAiB,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAC7C,YAAY,EAAE,CAAC;QACf,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;wBACnC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,wBAAwB,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjG,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;wBACnC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EACF,wBAAwB;4BACxB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;4BAChE,wEAAwE;wBAC1E,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC,CAAC"}

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

@@ -1,4 +1,4 @@
-import { getSourceCodeTree, parseSourceCodeTree } from '../../../../files/source-code-tree.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
 import { putSystemMessage } from '../../../../main/common/content-bus.js';
 import { getFunctionDefs } from '../../../function-calling.js';
 export async function handleRemoveFilesFromContext({ askQuestionCall, generateContentFn, prompt, options, }) {
@@ -12,7 +12,7 @@ export async function handleRemoveFilesFromContext({ askQuestionCall, generateCo
             type: 'user',
             text: 'Yes, you can request the removal of files from context.',
         },
-    ], getFunctionDefs(), 'removeFilesFromContext', 0.7, true, options));
+    ], getFunctionDefs(), 'removeFilesFromContext', 0.7, ModelType.CHEAP, options));
     if (!removeFilesFromContextCall) {
         return {
             breakLoop: true,
@@ -27,26 +27,23 @@ export async function handleRemoveFilesFromContext({ askQuestionCall, generateCo
         userText = 'Context reduction applied';
     }
     else {
+        putSystemMessage('No specific files were provided for context reduction. The context remains unchanged.');
         userText = 'No specific files were provided for context reduction. The context remains unchanged.';
     }
+    prompt.push({
+        type: 'assistant',
+        text: askQuestionCall.args?.message ?? '',
+        functionCalls: [removeFilesFromContextCall],
+    }, {
+        type: 'user',
+        text: userText,
+        functionResponses: [
+            { name: 'removeFilesFromContext', call_id: removeFilesFromContextCall.id, content: undefined },
+        ],
+    });
     return {
         breakLoop: false,
-        items: [
-            {
-                assistant: {
-                    type: 'assistant',
-                    text: askQuestionCall.args?.message ?? '',
-                    functionCalls: [removeFilesFromContextCall],
-                },
-                user: {
-                    type: 'user',
-                    text: userText,
-                    functionResponses: [
-                        { name: 'removeFilesFromContext', call_id: removeFilesFromContextCall.id, content: undefined },
-                    ],
-                },
-            },
-        ],
+        items: [],
     };
 }
 function removeFileContentsFromPrompt(prompt, filesToRemove) {
@@ -63,7 +60,7 @@ function removeFileContentsFromPrompt(prompt, filesToRemove) {
             return;
         }
         try {
-            const contentObj = parseSourceCodeTree(JSON.parse(sourceCodeResponse.content));
+            const contentObj = JSON.parse(sourceCodeResponse.content);
             let modifiedFiles = 0;
             filesToRemove.forEach((file) => {
                 if (contentObj[file] && 'content' in contentObj[file]) {
@@ -73,7 +70,7 @@ function removeFileContentsFromPrompt(prompt, filesToRemove) {
             });
             // Only update if files were actually modified
             if (modifiedFiles > 0) {
-                sourceCodeResponse.content = JSON.stringify(getSourceCodeTree(contentObj));
+                sourceCodeResponse.content = JSON.stringify(contentObj);
                 console.log(`Removed content from ${modifiedFiles} files in source code response`);
             }
         }

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

@@ -1 +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,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AAC/F,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAG/D,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACjD,eAAe,EACf,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,MAAM,CAAC,0BAA0B,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAC3D;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,yDAAyD;SAChE;KACF,EACD,eAAe,EAAE,EACjB,wBAAwB,EACxB,GAAG,EACH,IAAI,EACJ,OAAO,CACR,CAA2D,CAAC;IAE7D,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAChC,OAAO;YACL,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,GAAG,0BAA0B,CAAC,IAAI,EAAE,SAAS,IAAI,EAAE,CAAC;IACvE,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,KAAK,EAAE;YACL;gBACE,SAAS,EAAE;oBACT,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;oBACzC,aAAa,EAAE,CAAC,0BAA0B,CAAC;iBAC5C;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ;oBACd,iBAAiB,EAAE;wBACjB,EAAE,IAAI,EAAE,wBAAwB,EAAE,OAAO,EAAE,0BAA0B,CAAC,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE;qBAC/F;iBACF;aACF;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAoB,EAAE,aAAuB;IACjF,uDAAuD;IACvD,MAAM,mBAAmB,GAAG,MAAM,CAAC,MAAM,CACvC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,eAAe,CAAC,CACpG,CAAC;IAEF,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;QACjE,OAAO;IACT,CAAC;IAED,mBAAmB,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;QACvC,MAAM,kBAAkB,GAAG,QAAQ,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;QAEjG,IAAI,CAAC,kBAAkB,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC;YACvD,OAAO,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,mBAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/E,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC7B,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,SAAS,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;oBACtD,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;oBAChC,aAAa,EAAE,CAAC;gBAClB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,8CAA8C;YAC9C,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;gBACtB,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,CAAC;gBAC3E,OAAO,CAAC,GAAG,CAAC,wBAAwB,aAAa,gCAAgC,CAAC,CAAC;YACrF,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QACjE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}
+{"version":3,"file":"remove-files-from-context.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/remove-files-from-context.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAG/D,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACjD,eAAe,EACf,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,MAAM,CAAC,0BAA0B,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAC3D;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,yDAAyD;SAChE;KACF,EACD,eAAe,EAAE,EACjB,wBAAwB,EACxB,GAAG,EACH,SAAS,CAAC,KAAK,EACf,OAAO,CACR,CAA2D,CAAC;IAE7D,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAChC,OAAO;YACL,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,GAAG,0BAA0B,CAAC,IAAI,EAAE,SAAS,IAAI,EAAE,CAAC;IACvE,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,gBAAgB,CAAC,uFAAuF,CAAC,CAAC;QAC1G,QAAQ,GAAG,uFAAuF,CAAC;IACrG,CAAC;IAED,MAAM,CAAC,IAAI,CACT;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QACzC,aAAa,EAAE,CAAC,0BAA0B,CAAC;KAC5C,EACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,QAAQ;QACd,iBAAiB,EAAE;YACjB,EAAE,IAAI,EAAE,wBAAwB,EAAE,OAAO,EAAE,0BAA0B,CAAC,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE;SAC/F;KACF,CACF,CAAC;IAEF,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE;KACV,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAoB,EAAE,aAAuB;IACjF,uDAAuD;IACvD,MAAM,mBAAmB,GAAG,MAAM,CAAC,MAAM,CACvC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,eAAe,CAAC,CACpG,CAAC;IAEF,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;QACjE,OAAO;IACT,CAAC;IAED,mBAAmB,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;QACvC,MAAM,kBAAkB,GAAG,QAAQ,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;QAEjG,IAAI,CAAC,kBAAkB,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC;YACvD,OAAO,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAE1D,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC7B,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,SAAS,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;oBACtD,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;oBAChC,aAAa,EAAE,CAAC;gBAClB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,8CAA8C;YAC9C,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;gBACtB,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;gBACxD,OAAO,CAAC,GAAG,CAAC,wBAAwB,aAAa,gCAAgC,CAAC,CAAC;YACrF,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QACjE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

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

@@ -1,2 +1,8 @@
-import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+import { GenerateContentFunction } from '../../../../ai-service/common-types.js';
+import { PromptItem } from '../../../../ai-service/common-types.js';
+import { FunctionCall } from '../../../../ai-service/common-types.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { CodegenOptions } from '../../../../main/codegen-types.js';
+import { ActionHandlerProps, ActionResult, AskQuestionCall, RequestFilesContentArgs } from '../step-ask-question-types.js';
 export declare function handleRequestFilesContent({ askQuestionCall, options, prompt, generateContentFn, }: ActionHandlerProps): Promise<ActionResult>;
+export declare function generateRequestFilesContentCall(generateContentFn: GenerateContentFunction, prompt: PromptItem[], askQuestionCall: AskQuestionCall, options: CodegenOptions, modelType: ModelType): Promise<FunctionCall<RequestFilesContentArgs> | undefined>;