genaicode 0.14.1 → 0.15.0

package/dist/prompt/steps/step-iterate/handlers/handle-compound-action.js

@@ -0,0 +1,305 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+import { getOperationExecutor, getOperationDefs } from '../../../../operations/operations-index.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+import { putAssistantMessage, putSystemMessage, putUserMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { getCompoundActionDef } from '../../../function-defs/compound-action.js';
+import { executeStepEnsureContext } from '../../step-ensure-context.js';
+import { StepResult } from '../../steps-types.js';
+import { getSourceCode } from '../../../../files/read-files.js';
+import { executeStepVerifyPatch } from '../../step-verify-patch.js';
+/**
+ * Constructs the prompt for the initial planning phase of a compound action.
+ * @param basePrompt The current conversation history.
+ * @param assistantMessage The assistant's message initiating the compound action.
+ * @param compoundActionDef The function definition for the compound action planner.
+ * @returns The prompt items for the planning phase.
+ */
+export function constructCompoundActionPlanningPrompt(basePrompt, assistantMessage, compoundActionDef) {
+    return [
+        ...basePrompt,
+        {
+            type: 'assistant',
+            text: assistantMessage ?? '',
+        },
+        {
+            type: 'user',
+            text: `Based on the conversation history and my last message, please identify the sequence of file operations (like ${getOperationDefs()
+                .map((op) => op.name)
+                .join(', ')}). For each operation, specify the target 'filePath' where applicable. Provide an array of actions and a concise summary message for my confirmation. Please provide information about dependency between actions. Use the "${compoundActionDef.name}" function to return this information.`,
+        },
+    ];
+}
+/**
+ * Constructs the prompt for inferring parameters for a specific action within a compound action.
+ * @param basePrompt The current conversation history (which includes prior planning and parameter inferences).
+ * @param actionId The unique ID of the current action.
+ * @param actionName The name of the current action (e.g., "createFile").
+ * @param operationDef The function definition for the current action.
+ * @returns The prompt items for the parameter inference phase.
+ */
+export function constructCompoundActionParameterInferencePrompt(basePrompt, actionId, actionName, operationDef) {
+    return [
+        ...basePrompt,
+        {
+            type: 'user',
+            text: `Now, for the action "${actionId}:${actionName}", please determine the specific parameters required. Refer to its definition: ${JSON.stringify(operationDef.parameters)}. Use the "${actionName}" function to return these parameters.`,
+        },
+    ];
+}
+export const handleCompoundAction = async ({ iterateCall, options, generateContentFn, prompt, }) => {
+    const compoundActionDef = getCompoundActionDef();
+    let actions = [];
+    let summary = '';
+    let initialInferenceCall;
+    // Step 1: Infer High-Level Plan (Action Types and Summary)
+    try {
+        putSystemMessage('Inferring high-level plan for compound action...');
+        const initialInferencePrompt = constructCompoundActionPlanningPrompt(prompt, iterateCall.args?.message, compoundActionDef);
+        const initialResult = await generateContentFn(initialInferencePrompt, {
+            modelType: ModelType.CHEAP, // Use a capable model for planning
+            functionDefs: getFunctionDefs(), // getFunctionDefs() already includes compoundActionDef
+            requiredFunctionName: compoundActionDef.name,
+            temperature: 0.2,
+            expectedResponseType: { text: false, functionCall: true, media: false },
+        }, options);
+        initialInferenceCall = initialResult.find((part) => part.type === 'functionCall')?.functionCall;
+        if (!initialInferenceCall?.args?.actions || initialInferenceCall.args.actions.length === 0) {
+            putSystemMessage('AI failed to generate a list of action types or returned an empty list. Cannot proceed.');
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args?.message ?? '',
+            });
+            prompt.push({
+                type: 'user',
+                text: 'I was unable to determine the necessary actions. Could you please clarify your request or try rephrasing it?',
+            });
+            return { breakLoop: false, items: [] };
+        }
+        actions = initialInferenceCall.args.actions;
+        summary = initialInferenceCall.args.summary;
+        putSystemMessage(`Inferred actions`, { actions, summary });
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        putSystemMessage(`Error during high-level plan inference: ${errorMessage}`);
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        }, {
+            type: 'user',
+            text: `I encountered an error while planning the actions: ${errorMessage}. Please clarify your request.`,
+        });
+        return { breakLoop: false, items: [] };
+    }
+    prompt.push({
+        type: 'assistant',
+        text: iterateCall.args?.message ?? '', // This is the original assistant message that triggered compound action
+        functionCalls: [initialInferenceCall], // The AI call for planning
+    });
+    // prompt user for confirmation of the inferred actions
+    let confirmation = await askUserForConfirmationWithAnswer('The assistant is ready to start working on the inferred actions. Do you want to proceed?', 'Yes, proceed with these actions', 'No, I want to change something', true, options);
+    if (confirmation.answer) {
+        putUserMessage(confirmation.answer);
+    }
+    if (!confirmation.confirmed) {
+        putSystemMessage('User declined the proposed actions.');
+        prompt.push({
+            type: 'user',
+            text: `Declining the proposed actions. ${confirmation.answer ?? ''}`,
+            functionResponses: [
+                {
+                    name: initialInferenceCall.name,
+                    call_id: initialInferenceCall.id,
+                    content: '',
+                },
+            ],
+        });
+        return { breakLoop: false, items: [] };
+    }
+    putSystemMessage('User confirmed the proposed actions. Proceeding with parameter inference...');
+    // Ensure context for all planned files
+    const uniqueFilePaths = Array.from(new Set(actions.map((a) => a.filePath).filter((p) => !!p)));
+    if (uniqueFilePaths.length > 0) {
+        const contextAssuranceCall = {
+            name: 'contextAssurance',
+            args: { filePaths: uniqueFilePaths },
+        };
+        const contextResult = await executeStepEnsureContext(prompt, contextAssuranceCall, options);
+        if (contextResult === StepResult.BREAK) {
+            putSystemMessage('Error: Context ensuring failed during compound action. Aborting.');
+            return { breakLoop: false, items: [] };
+        }
+    }
+    // Add initial planning call to conversation history
+    prompt.push({
+        type: 'user', // User function response for the planning call (empty content as it was a planning step)
+        text: `Accepting the proposed actions. ${confirmation.answer ?? ''}`,
+        functionResponses: [
+            {
+                name: initialInferenceCall.name,
+                call_id: initialInferenceCall.id,
+                content: '',
+            },
+        ],
+    }, 
+    // Assistant message indicating planning is done, before parameter inference starts
+    {
+        type: 'assistant',
+        text: `I have inferred the actions and I am ready to determine their parameters.`,
+    });
+    // Step 2: Infer Parameters for Each Action
+    const detailedActions = [];
+    let parameterInferenceFailed = false;
+    const completedActionIds = new Set();
+    let actionsToInfer = [...actions]; // Create a mutable copy
+    while (actionsToInfer.length > 0) {
+        const processableActions = actionsToInfer.filter((action) => !action.dependsOn || action.dependsOn.every((dep) => completedActionIds.has(dep)));
+        if (processableActions.length === 0 && actionsToInfer.length > 0) {
+            putSystemMessage('Error: Circular dependency or unresolvable dependencies in compound actions.');
+            parameterInferenceFailed = true;
+            break;
+        }
+        if (processableActions.length === 0)
+            break; // All done or stuck
+        const results = await Promise.allSettled(processableActions.map(async (actionToProcess) => {
+            const { id: actionId, name: actionName, filePath } = actionToProcess;
+            const currentOperationDef = getOperationDefs().find((op) => op.name === actionName);
+            if (!currentOperationDef) {
+                putSystemMessage(`Error: Operation definition not found for action "${actionId}:${actionName}". Skipping this action.`);
+                parameterInferenceFailed = true;
+                return { success: false, actionId };
+            }
+            try {
+                putSystemMessage(`Inferring parameters for action: ${actionId}:${actionName}...`);
+                const paramInferencePrompt = constructCompoundActionParameterInferencePrompt(prompt, // Pass the *current* prompt history
+                actionId, actionName, currentOperationDef);
+                const paramResult = await generateContentFn(paramInferencePrompt, {
+                    modelType: ModelType.DEFAULT,
+                    functionDefs: getFunctionDefs(), // getFunctionDefs() includes all operation defs
+                    requiredFunctionName: actionName,
+                    temperature: 0.1,
+                    expectedResponseType: { text: false, functionCall: true, media: false },
+                }, options);
+                let paramFunctionCall = paramResult.find((part) => part.type === 'functionCall')?.functionCall;
+                if (!paramFunctionCall?.args) {
+                    putSystemMessage(`AI failed to generate parameters for action "${actionId}:${actionName}".`);
+                    parameterInferenceFailed = true;
+                    return { success: false, actionId };
+                }
+                if (actionName === 'patchFile') {
+                    paramFunctionCall = await executeStepVerifyPatch(paramFunctionCall.args, generateContentFn, prompt, getFunctionDefs(), options.temperature, false, options);
+                    if (!paramFunctionCall || !paramFunctionCall.args) {
+                        putSystemMessage(`Error verifying patch for action "${actionId}:${actionName}".`);
+                        parameterInferenceFailed = true;
+                        return { success: false, actionId };
+                    }
+                }
+                if (filePath) {
+                    const fileSource = getSourceCode({ filterPaths: [filePath], forceAll: true }, options)[filePath];
+                    if (fileSource && 'content' in fileSource) {
+                        paramFunctionCall.args.oldContent = fileSource.content;
+                    }
+                }
+                detailedActions.push({ name: actionName, args: paramFunctionCall.args });
+                putSystemMessage(`Inferred parameters for ${actionId}:${actionName}`, { ...paramFunctionCall });
+                // Add this specific parameter inference to the prompt history for subsequent inferences
+                prompt.push({
+                    type: 'assistant',
+                    text: `Inferring parameters for ${actionId}:${actionName}.`, // Short text for assistant turn
+                    functionCalls: [paramFunctionCall],
+                }, {
+                    type: 'user',
+                    functionResponses: [
+                        {
+                            name: paramFunctionCall.name,
+                            call_id: paramFunctionCall.id,
+                            content: '', // Empty content for the function response
+                        },
+                    ],
+                });
+                completedActionIds.add(actionId);
+                return { success: true, actionId };
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                putSystemMessage(`Error inferring parameters for action "${actionName}": ${errorMessage}`);
+                parameterInferenceFailed = true;
+                return { success: false, actionId };
+            }
+        }));
+        // Filter out processed actions from actionsToInfer
+        actionsToInfer = actionsToInfer.filter((action) => !processableActions.find((pa) => pa.id === action.id));
+        if (results.some((result) => result.status === 'rejected' || (result.status === 'fulfilled' && !result.value.success))) {
+            putSystemMessage('Some actions failed to infer parameters. Stopping parameter inference.');
+            parameterInferenceFailed = true;
+            break;
+        }
+    }
+    if (parameterInferenceFailed) {
+        const assistantMessage = 'I was unable to determine all the necessary parameters for the requested actions. Please review the errors and try again, perhaps with more specific instructions.';
+        putAssistantMessage(assistantMessage);
+        prompt.push({
+            type: 'assistant',
+            text: assistantMessage,
+        });
+        // No user response needed here, assistant states the failure.
+        return { breakLoop: false, items: [] };
+    }
+    // Step 3: Consolidate and Confirm with User
+    putSystemMessage('Proposed actions', { actions: detailedActions });
+    confirmation = await askUserForConfirmationWithAnswer('Code changes are generated, now what?', 'Apply code changes', 'Reject code changes', true, options);
+    if (confirmation.answer) {
+        putUserMessage(confirmation.answer);
+    }
+    if (!confirmation.confirmed) {
+        putSystemMessage('User declined code changes.');
+        prompt.push({
+            type: 'user',
+            text: `Rejecting proposed code changes. ${confirmation.answer ?? ''}`,
+        });
+        return { breakLoop: false, items: [] };
+    }
+    // Step 4: Execute Actions
+    putSystemMessage('User accepted code changes. Applying...');
+    let executionError = null;
+    let executionErrorMessage = '';
+    prompt.push({
+        type: 'user', // User confirms the plan
+        text: `Applying code changes. ${confirmation.answer ?? 'Yes'}`,
+    });
+    for (const [index, action] of detailedActions.entries()) {
+        putSystemMessage(`Executing action ${index + 1}/${detailedActions.length}: ${action.name}`);
+        const executor = getOperationExecutor(action.name);
+        if (!executor) {
+            executionErrorMessage = `Error: Operation executor not found for action "${action.name}". Skipping this action.`;
+            putSystemMessage(executionErrorMessage);
+            continue;
+        }
+        try {
+            await executor(action.args, options);
+            putSystemMessage(`Action ${action.name} completed successfully.`);
+        }
+        catch (error) {
+            executionError = error instanceof Error ? error : new Error(String(error));
+            executionErrorMessage = `Error executing action ${index + 1} (${action.name}): ${executionError.message}. Stopping batch execution.`;
+            putSystemMessage(executionErrorMessage);
+        }
+    }
+    const finalAssistantMessage = executionError
+        ? `Failed to execute the batch: ${executionErrorMessage}`
+        : 'Successfully executed all actions in the batch.';
+    putAssistantMessage(finalAssistantMessage);
+    prompt.push({
+        type: 'assistant',
+        text: finalAssistantMessage,
+    });
+    // No final user "Thank you" here, as it would be a new turn.
+    return {
+        breakLoop: false,
+        items: [],
+    };
+};
+registerActionHandler('compoundAction', handleCompoundAction);
+//# sourceMappingURL=handle-compound-action.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-compound-action.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-compound-action.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,SAAS,EAA2B,MAAM,wCAAwC,CAAC;AAE1G,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAE,MAAM,4CAA4C,CAAC;AACpG,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAC/G,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAEpE;;;;;;GAMG;AACH,MAAM,UAAU,qCAAqC,CACnD,UAAwB,EACxB,gBAAoC,EACpC,iBAA8B;IAE9B,OAAO;QACL,GAAG,UAAU;QACb;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,gBAAgB,IAAI,EAAE;SAC7B;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,gHAAgH,gBAAgB,EAAE;iBACrI,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;iBACpB,IAAI,CACH,IAAI,CACL,+NACD,iBAAiB,CAAC,IACpB,wCAAwC;SACzC;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,+CAA+C,CAC7D,UAAwB,EACxB,QAAgB,EAChB,UAAkB,EAClB,YAAyB;IAEzB,OAAO;QACL,GAAG,UAAU;QACb;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,wBAAwB,QAAQ,IAAI,UAAU,kFAAkF,IAAI,CAAC,SAAS,CAClJ,YAAY,CAAC,UAAU,CACxB,cAAc,UAAU,wCAAwC;SAClE;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAkB,KAAK,EAAE,EACxD,WAAW,EACX,OAAO,EACP,iBAAiB,EACjB,MAAM,GACa,EAAyB,EAAE;IAC9C,MAAM,iBAAiB,GAAG,oBAAoB,EAAE,CAAC;IACjD,IAAI,OAAO,GAKN,EAAE,CAAC;IACR,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,oBAUS,CAAC;IAEd,2DAA2D;IAC3D,IAAI,CAAC;QACH,gBAAgB,CAAC,kDAAkD,CAAC,CAAC;QACrE,MAAM,sBAAsB,GAAG,qCAAqC,CAClE,MAAM,EACN,WAAW,CAAC,IAAI,EAAE,OAAO,EACzB,iBAAiB,CAClB,CAAC;QAEF,MAAM,aAAa,GAAG,MAAM,iBAAiB,CAC3C,sBAAsB,EACtB;YACE,SAAS,EAAE,SAAS,CAAC,KAAK,EAAE,mCAAmC;YAC/D,YAAY,EAAE,eAAe,EAAE,EAAE,uDAAuD;YACxF,oBAAoB,EAAE,iBAAiB,CAAC,IAAI;YAC5C,WAAW,EAAE,GAAG;YAChB,oBAAoB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;SACxE,EACD,OAAO,CACR,CAAC;QAEF,oBAAoB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC,EAAE,YAQjF,CAAC;QAEH,IAAI,CAAC,oBAAoB,EAAE,IAAI,EAAE,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3F,gBAAgB,CAAC,yFAAyF,CAAC,CAAC;YAC5G,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;aACtC,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,8GAA8G;aACrH,CAAC,CAAC;YACH,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QACD,OAAO,GAAG,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC;QAC5C,OAAO,GAAG,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC;QAC5C,gBAAgB,CAAC,kBAAkB,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;IAC7D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,gBAAgB,CAAC,2CAA2C,YAAY,EAAE,CAAC,CAAC;QAC5E,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,sDAAsD,YAAY,gCAAgC;SACzG,CACF,CAAC;QACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,wEAAwE;QAC/G,aAAa,EAAE,CAAC,oBAAoB,CAAC,EAAE,2BAA2B;KACnE,CAAC,CAAC;IAEH,uDAAuD;IACvD,IAAI,YAAY,GAAG,MAAM,gCAAgC,CACvD,0FAA0F,EAC1F,iCAAiC,EACjC,gCAAgC,EAChC,IAAI,EACJ,OAAO,CACR,CAAC;IAEF,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;QACxB,cAAc,CAAC,YAAY,CAAC,MAAO,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;QAC5B,gBAAgB,CAAC,qCAAqC,CAAC,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,mCAAmC,YAAY,CAAC,MAAM,IAAI,EAAE,EAAE;YACpE,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,oBAAoB,CAAC,IAAI;oBAC/B,OAAO,EAAE,oBAAoB,CAAC,EAAE;oBAChC,OAAO,EAAE,EAAE;iBACZ;aACF;SACF,CAAC,CAAC;QACH,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IACD,gBAAgB,CAAC,6EAA6E,CAAC,CAAC;IAEhG,uCAAuC;IACvC,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5G,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,MAAM,oBAAoB,GAAiB;YACzC,IAAI,EAAE,kBAAkB;YACxB,IAAI,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE;SACrC,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,CAAC,CAAC;QAC5F,IAAI,aAAa,KAAK,UAAU,CAAC,KAAK,EAAE,CAAC;YACvC,gBAAgB,CAAC,kEAAkE,CAAC,CAAC;YACrF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,MAAM,CAAC,IAAI,CACT;QACE,IAAI,EAAE,MAAM,EAAE,yFAAyF;QACvG,IAAI,EAAE,mCAAmC,YAAY,CAAC,MAAM,IAAI,EAAE,EAAE;QACpE,iBAAiB,EAAE;YACjB;gBACE,IAAI,EAAE,oBAAoB,CAAC,IAAI;gBAC/B,OAAO,EAAE,oBAAoB,CAAC,EAAE;gBAChC,OAAO,EAAE,EAAE;aACZ;SACF;KACF;IACD,mFAAmF;IACnF;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,2EAA2E;KAClF,CACF,CAAC;IAEF,2CAA2C;IAC3C,MAAM,eAAe,GAAyB,EAAE,CAAC;IACjD,IAAI,wBAAwB,GAAG,KAAK,CAAC;IACrC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC7C,IAAI,cAAc,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,wBAAwB;IAE3D,OAAO,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,MAAM,kBAAkB,GAAG,cAAc,CAAC,MAAM,CAC9C,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAC9F,CAAC;QAEF,IAAI,kBAAkB,CAAC,MAAM,KAAK,CAAC,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjE,gBAAgB,CAAC,8EAA8E,CAAC,CAAC;YACjG,wBAAwB,GAAG,IAAI,CAAC;YAChC,MAAM;QACR,CAAC;QACD,IAAI,kBAAkB,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,CAAC,oBAAoB;QAEhE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CACtC,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,eAAe,EAAE,EAAE;YAC/C,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,eAAe,CAAC;YACrE,MAAM,mBAAmB,GAAG,gBAAgB,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAEpF,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,gBAAgB,CACd,qDAAqD,QAAQ,IAAI,UAAU,0BAA0B,CACtG,CAAC;gBACF,wBAAwB,GAAG,IAAI,CAAC;gBAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YACtC,CAAC;YAED,IAAI,CAAC;gBACH,gBAAgB,CAAC,oCAAoC,QAAQ,IAAI,UAAU,KAAK,CAAC,CAAC;gBAClF,MAAM,oBAAoB,GAAG,+CAA+C,CAC1E,MAAM,EAAE,oCAAoC;gBAC5C,QAAQ,EACR,UAAU,EACV,mBAAmB,CACpB,CAAC;gBAEF,MAAM,WAAW,GAAG,MAAM,iBAAiB,CACzC,oBAAoB,EACpB;oBACE,SAAS,EAAE,SAAS,CAAC,OAAO;oBAC5B,YAAY,EAAE,eAAe,EAAE,EAAE,gDAAgD;oBACjF,oBAAoB,EAAE,UAAU;oBAChC,WAAW,EAAE,GAAG;oBAChB,oBAAoB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;iBACxE,EACD,OAAO,CACR,CAAC;gBAEF,IAAI,iBAAiB,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC,EAAE,YAAY,CAAC;gBAE/F,IAAI,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC;oBAC7B,gBAAgB,CAAC,gDAAgD,QAAQ,IAAI,UAAU,IAAI,CAAC,CAAC;oBAC7F,wBAAwB,GAAG,IAAI,CAAC;oBAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;gBACtC,CAAC;gBAED,IAAI,UAAU,KAAK,WAAW,EAAE,CAAC;oBAC/B,iBAAiB,GAAG,MAAM,sBAAsB,CAC9C,iBAAiB,CAAC,IAAgE,EAClF,iBAAiB,EACjB,MAAM,EACN,eAAe,EAAE,EACjB,OAAO,CAAC,WAAY,EACpB,KAAK,EACL,OAAO,CACR,CAAC;oBAEF,IAAI,CAAC,iBAAiB,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;wBAClD,gBAAgB,CAAC,qCAAqC,QAAQ,IAAI,UAAU,IAAI,CAAC,CAAC;wBAClF,wBAAwB,GAAG,IAAI,CAAC;wBAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;oBACtC,CAAC;gBACH,CAAC;gBAED,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,UAAU,GAAG,aAAa,CAAC,EAAE,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC;oBACjG,IAAI,UAAU,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;wBAC1C,iBAAiB,CAAC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC;oBACzD,CAAC;gBACH,CAAC;gBAED,eAAe,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzE,gBAAgB,CAAC,2BAA2B,QAAQ,IAAI,UAAU,EAAE,EAAE,EAAE,GAAG,iBAAiB,EAAE,CAAC,CAAC;gBAEhG,wFAAwF;gBACxF,MAAM,CAAC,IAAI,CACT;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,4BAA4B,QAAQ,IAAI,UAAU,GAAG,EAAE,gCAAgC;oBAC7F,aAAa,EAAE,CAAC,iBAAiB,CAAC;iBACnC,EACD;oBACE,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,iBAAiB,CAAC,IAAI;4BAC5B,OAAO,EAAE,iBAAiB,CAAC,EAAE;4BAC7B,OAAO,EAAE,EAAE,EAAE,0CAA0C;yBACxD;qBACF;iBACF,CACF,CAAC;gBACF,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACjC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YACrC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC5E,gBAAgB,CAAC,0CAA0C,UAAU,MAAM,YAAY,EAAE,CAAC,CAAC;gBAC3F,wBAAwB,GAAG,IAAI,CAAC;gBAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QAEF,mDAAmD;QACnD,cAAc,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE1G,IACE,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAClH,CAAC;YACD,gBAAgB,CAAC,wEAAwE,CAAC,CAAC;YAC3F,wBAAwB,GAAG,IAAI,CAAC;YAChC,MAAM;QACR,CAAC;IACH,CAAC;IAED,IAAI,wBAAwB,EAAE,CAAC;QAC7B,MAAM,gBAAgB,GACpB,oKAAoK,CAAC;QACvK,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,gBAAgB;SACvB,CAAC,CAAC;QACH,8DAA8D;QAC9D,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,4CAA4C;IAC5C,gBAAgB,CAAC,kBAAkB,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC,CAAC;IAEnE,YAAY,GAAG,MAAM,gCAAgC,CACnD,uCAAuC,EACvC,oBAAoB,EACpB,qBAAqB,EACrB,IAAI,EACJ,OAAO,CACR,CAAC;IAEF,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;QACxB,cAAc,CAAC,YAAY,CAAC,MAAO,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;QAC5B,gBAAgB,CAAC,6BAA6B,CAAC,CAAC;QAChD,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,oCAAoC,YAAY,CAAC,MAAM,IAAI,EAAE,EAAE;SACtE,CAAC,CAAC;QACH,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,0BAA0B;IAC1B,gBAAgB,CAAC,yCAAyC,CAAC,CAAC;IAC5D,IAAI,cAAc,GAAiB,IAAI,CAAC;IACxC,IAAI,qBAAqB,GAAG,EAAE,CAAC;IAE/B,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,MAAM,EAAE,yBAAyB;QACvC,IAAI,EAAE,0BAA0B,YAAY,CAAC,MAAM,IAAI,KAAK,EAAE;KAC/D,CAAC,CAAC;IAEH,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,eAAe,CAAC,OAAO,EAAE,EAAE,CAAC;QACxD,gBAAgB,CAAC,oBAAoB,KAAK,GAAG,CAAC,IAAI,eAAe,CAAC,MAAM,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAC5F,MAAM,QAAQ,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,qBAAqB,GAAG,mDAAmD,MAAM,CAAC,IAAI,0BAA0B,CAAC;YACjH,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;YACxC,SAAS;QACX,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACrC,gBAAgB,CAAC,UAAU,MAAM,CAAC,IAAI,0BAA0B,CAAC,CAAC;QACpE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,cAAc,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3E,qBAAqB,GAAG,0BAA0B,KAAK,GAAG,CAAC,KAAK,MAAM,CAAC,IAAI,MACzE,cAAc,CAAC,OACjB,6BAA6B,CAAC;YAC9B,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,MAAM,qBAAqB,GAAG,cAAc;QAC1C,CAAC,CAAC,gCAAgC,qBAAqB,EAAE;QACzD,CAAC,CAAC,iDAAiD,CAAC;IAEtD,mBAAmB,CAAC,qBAAqB,CAAC,CAAC;IAE3C,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,qBAAqB;KAC5B,CAAC,CAAC;IACH,6DAA6D;IAE7D,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE;KACV,CAAC;AACJ,CAAC,CAAC;AAEF,qBAAqB,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-confirm-code-generation.d.ts

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

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

@@ -0,0 +1,40 @@
+import { putSystemMessage, putUserMessage } from '../../../../main/common/content-bus.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+import { getActionHandler } from '../iterate-handler.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('confirmCodeGeneration', handleConfirmCodeGeneration);
+export async function handleConfirmCodeGeneration({ iterateCall, options, prompt, ...props }) {
+    const userConfirmation = await askUserForConfirmationWithAnswer('The assistant is ready to start code generation. Do you want to proceed?', 'Start code generation', 'Continue conversation', true, options);
+    if (userConfirmation.answer) {
+        putUserMessage(userConfirmation.answer);
+    }
+    if (userConfirmation.confirmed) {
+        putSystemMessage('Proceeding with code generation.');
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        }, {
+            type: 'user',
+            text: 'Confirmed. Proceed with code generation.' + (userConfirmation.answer ? ` ${userConfirmation.answer}` : ''),
+        });
+        return await getActionHandler('codeGeneration')({
+            iterateCall,
+            options,
+            prompt,
+            ...props,
+        });
+    }
+    else {
+        putSystemMessage('Declined. Continuing the conversation.');
+        prompt.push({ type: 'assistant', text: iterateCall.args?.message ?? '' }, {
+            type: 'user',
+            text: 'Declined. Please continue the conversation.' +
+                (userConfirmation.answer ? ` ${userConfirmation.answer}` : ''),
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+}
+//# sourceMappingURL=handle-confirm-code-generation.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-confirm-code-generation.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-confirm-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAC1F,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE,qBAAqB,CAAC,uBAAuB,EAAE,2BAA2B,CAAC,CAAC;AAE5E,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAAC,EAChD,WAAW,EACX,OAAO,EACP,MAAM,EACN,GAAG,KAAK,EACW;IACnB,MAAM,gBAAgB,GAAG,MAAM,gCAAgC,CAC7D,0EAA0E,EAC1E,uBAAuB,EACvB,uBAAuB,EACvB,IAAI,EACJ,OAAO,CACR,CAAC;IAEF,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC;QAC5B,cAAc,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC/B,gBAAgB,CAAC,kCAAkC,CAAC,CAAC;QAErD,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,0CAA0C,GAAG,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9G,CACF,CAAC;QAEF,OAAO,MAAM,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;YAC9C,WAAW;YACX,OAAO;YACP,MAAM;YACN,GAAG,KAAK;SACT,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,gBAAgB,CAAC,wCAAwC,CAAC,CAAC;QAC3D,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,EAC5D;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,6CAA6C;gBAC7C,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACjE,CACF,CAAC;QAEF,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;AACH,CAAC"}

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

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

package/dist/prompt/steps/step-iterate/handlers/handle-context-compression.js

@@ -0,0 +1,35 @@
+import { executeStepContextCompression } from '../../step-context-compression.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { StepResult } from '../../steps-types.js';
+import { askUserForConfirmation } from '../../../../main/common/user-actions.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('contextCompression', handleContextCompression);
+export async function handleContextCompression({ iterateCall, prompt, options, generateContentFn, }) {
+    putSystemMessage('Context compression requested.');
+    // First confirmation: Ask user if they want to proceed with content generation
+    const compressionConfirm = await askUserForConfirmation('The assistant suggests compressing the context. Do you want to proceed with context compression?', false, options, 'Compress context', 'Continue conversation');
+    if (!compressionConfirm.confirmed) {
+        putSystemMessage('Context compression rejected.');
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        }, {
+            type: 'user',
+            text: 'Rejecting context compression.',
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    try {
+        // Execute compression step
+        const result = await executeStepContextCompression(generateContentFn, prompt, options);
+        return { breakLoop: result === StepResult.BREAK, items: [] };
+    }
+    catch (error) {
+        putSystemMessage('Error during context compression', { error });
+        return { breakLoop: false, items: [] };
+    }
+}
+//# sourceMappingURL=handle-context-compression.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-context-compression.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-context-compression.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,6BAA6B,EAAE,MAAM,mCAAmC,CAAC;AAElF,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AACjF,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAEpE,qBAAqB,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,CAAC;AAEtE,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,EAC7C,WAAW,EACX,MAAM,EACN,OAAO,EACP,iBAAiB,GACE;IACnB,gBAAgB,CAAC,gCAAgC,CAAC,CAAC;IAEnD,+EAA+E;IAC/E,MAAM,kBAAkB,GAAG,MAAM,sBAAsB,CACrD,kGAAkG,EAClG,KAAK,EACL,OAAO,EACP,kBAAkB,EAClB,uBAAuB,CACxB,CAAC;IAEF,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,CAAC;QAClC,gBAAgB,CAAC,+BAA+B,CAAC,CAAC;QAElD,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,gCAAgC;SACvC,CACF,CAAC;QAEF,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,2BAA2B;QAC3B,MAAM,MAAM,GAAG,MAAM,6BAA6B,CAAC,iBAAiB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAEvF,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,UAAU,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IAC/D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,gBAAgB,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAChE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;AACH,CAAC"}

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

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

package/dist/prompt/steps/step-iterate/handlers/handle-context-optimization.js

@@ -0,0 +1,34 @@
+import { refreshFiles } from '../../../../files/find-files.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { askUserForConfirmation } from '../../../../main/common/user-actions.js';
+import { executeStepContextOptimization } from '../../step-context-optimization.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('contextOptimization', handleContextOptimization);
+export async function handleContextOptimization({ iterateCall, 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, options);
+    prompt.push({
+        type: 'assistant',
+        text: iterateCall.args?.message ?? '',
+    });
+    prompt.push({
+        type: 'user',
+        text: userConfirmation.confirmed
+            ? 'Context optimization proposal accepted.'
+            : 'Context optimization proposal rejected.',
+    });
+    if (userConfirmation.confirmed) {
+        // 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, options);
+    }
+    else {
+        putSystemMessage('Context optimization rejected.');
+    }
+    return {
+        breakLoop: false,
+        items: [],
+    };
+}
+//# sourceMappingURL=handle-context-optimization.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-context-optimization.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-context-optimization.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AACjF,OAAO,EAAE,8BAA8B,EAAE,MAAM,oCAAoC,CAAC;AACpF,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE,qBAAqB,CAAC,qBAAqB,EAAE,yBAAyB,CAAC,CAAC;AAExE,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAC9C,WAAW,EACX,MAAM,EACN,OAAO,EACP,iBAAiB,GACE;IACnB,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,iHAAiH,EACjH,KAAK,EACL,OAAO,CACR,CAAC;IAEF,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;KACtC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,gBAAgB,CAAC,SAAS;YAC9B,CAAC,CAAC,yCAAyC;YAC3C,CAAC,CAAC,yCAAyC;KAC9C,CAAC,CAAC;IAEH,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC/B,4EAA4E;QAC5E,YAAY,EAAE,CAAC;QAEf,oCAAoC;QACpC,gBAAgB,CAAC,sCAAsC,CAAC,CAAC;QACzD,MAAM,8BAA8B,CAAC,iBAAiB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC3E,CAAC;SAAM,CAAC;QACN,gBAAgB,CAAC,gCAAgC,CAAC,CAAC;IACrD,CAAC;IAED,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE;KACV,CAAC;AACJ,CAAC"}

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

@@ -0,0 +1,11 @@
+import { ActionHandlerProps, ActionResult } from '../step-iterate-types.js';
+import { ConversationEdge, ConversationNode } from '../../../function-defs/conversation-graph.js';
+export declare const CONVERSATION_GRAPH_PROMPT = "# Conversation Graph Guide\n\nA conversation graph is a flexible structure for managing complex interactions. It consists of nodes (conversation states) connected by edges (transitions).\n\n## Core Components\n\n1. **Nodes (States)**\n   - Each node represents a distinct state in the conversation\n   - Contains:\n     - id: Unique identifier\n     - actionType: The action to perform\n     - instruction: Internal guidance for execution\n\n2. **Edges (Transitions)**\n   - Connect nodes to define possible conversation flows\n   - Contains:\n     - sourceNode: Starting point\n     - targetNode: Destination\n     - instruction: Condition for transition\n\n## Key Concepts\n\n- **State Management**: Nodes capture the current state and required action\n- **Flow Control**: Edges determine how the conversation can progress\n- **Flexibility**: Support for various conversation patterns and flows\n- **Context Awareness**: Each node operates with awareness of the conversation state\n- **User Interaction**: User can reject or accept some of the actions, so the graph should account for that\n\n## Implementation Notes\n\n- Nodes should be self-contained and focused\n- Edges should have clear, evaluatable conditions\n- Any node can connect to any other node if logically appropriate\n- Nodes and edges must formulate a graph structure, allowing for complex interactions\n\n## Graph Structure requirements and constraints\n\n- The graph may contain cycles, but should be designed to avoid infinite loops\n- Each node should be reachable from the entry node\n- The entry node should be clearly defined and serve as the starting point for the conversation\n\n## Technical Details\n\nNodes require:\n- Unique ID for reference\n- Valid action type\n- Clear execution instructions\n\nEdges require:\n- Valid source and target node IDs\n- Clear transition conditions\n\n# Let's get started!\n\n0. Read and understand the conersation we had so far (context).\n1. Think about the big picture of the upcoming conversation flow and the desired outcome.\n2. Define the entry node to start the upcoming conversation.\n3. Create nodes with reasoning, action type, and instructions.\n4. Connect nodes with edges to define the conversation flow.\n";
+export declare const getEdgeEvaluationPrompt: (currentNode: ConversationNode, outgoingEdges: ConversationEdge[]) => string;
+export declare const GRAPH_ANALYSIS_PROMPT_TEXT = "Please analyze the conversation graph provided by the assistant in the preceding message. Simulate diverse conversation paths mentally and identify any structural or logical problems, dead ends, unreachable nodes, missing transitions, or potential infinite loops. Explain the problems you find in detail. If no significant problems are found, state that clearly.";
+export declare const GRAPH_REVISION_PROMPT_TEXT = "Based on the analysis provided in the preceding message, generate a new, improved conversation graph using the 'conversationGraph' function. Ensure the new graph addresses the identified problems and better facilitates the original conversation goal.";
+/**
+ * Main handler for the conversationGraph action.
+ * Orchestrates confirmation, generation, analysis/revision, and traversal.
+ */
+export declare function handleConversationGraph(props: ActionHandlerProps): Promise<ActionResult>;