genaicode 0.14.1 → 0.15.0

package/dist/prompt/steps/step-iterate/handlers/handle-generate-image.js

@@ -0,0 +1,110 @@
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+import { putAssistantMessage, putSystemMessage } from '../../../../main/common/content-bus.js';
+import { executeStepGenerateImage } from '../../step-generate-image.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { getTempBuffer } from '../../../../files/temp-buffer.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('generateImage', handleGenerateImage);
+export async function handleGenerateImage({ iterateCall, options, generateContentFn, prompt, generateImageFn, }) {
+    if (!options.imagen) {
+        putSystemMessage('Image generation is not enabled. Please enable it using --imagen parameter.');
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    const [generateImageCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Yes, you can generate an image.',
+        },
+    ], {
+        functionDefs: getFunctionDefs(),
+        requiredFunctionName: 'generateImage',
+        temperature: 0.7,
+        modelType: ModelType.CHEAP,
+        expectedResponseType: {
+            text: false,
+            functionCall: true,
+            media: false,
+        },
+    }, options))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    if (!generateImageCall?.args) {
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    // Get the image generation request from the args
+    if (!generateImageCall.args.prompt) {
+        putSystemMessage('Image generation request with prompt is required for sendMessageWithImage action.');
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    // Create a temporary file path for the generated image
+    generateImageCall.args.filePath = `/tmp/question-image-${Date.now()}.png`;
+    // Execute image generation
+    const downloadFileCall = await executeStepGenerateImage(generateImageFn, generateImageCall);
+    // If image generation failed, the downloadFileCall will be an explanation
+    if (!downloadFileCall.args?.downloadUrl) {
+        putSystemMessage('Failed to generate image for the question.');
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    let buffer;
+    const downloadUrl = downloadFileCall.args.downloadUrl;
+    if (downloadUrl.startsWith('temp://')) {
+        buffer = getTempBuffer(downloadUrl);
+    }
+    else {
+        const response = await fetch(downloadFileCall.args?.downloadUrl);
+        const arrayBuffer = await response.arrayBuffer();
+        buffer = Buffer.from(arrayBuffer);
+    }
+    if (!buffer) {
+        putSystemMessage('Failed to generate image for the question.');
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    const imageData = {
+        base64url: buffer.toString('base64'),
+        mediaType: 'image/png',
+    };
+    putAssistantMessage(iterateCall.args?.message ?? '', generateImageCall.args, [], [imageData]);
+    // Get user's response with the image displayed
+    const inputResponse = await askUserForInput('Your answer', '', options);
+    // Return the conversation items with the generated image included in the context
+    return {
+        breakLoop: false,
+        items: [
+            {
+                assistant: {
+                    type: 'assistant',
+                    text: iterateCall.args?.message ?? '',
+                    functionCalls: [generateImageCall],
+                },
+                user: {
+                    type: 'user',
+                    text: inputResponse.answer,
+                    images: [imageData],
+                    functionResponses: [{ name: generateImageCall.name, call_id: generateImageCall.id, content: undefined }],
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=handle-generate-image.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-generate-image.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-generate-image.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAE/F,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAEpE,qBAAqB,CAAC,eAAe,EAAE,mBAAmB,CAAC,CAAC;AAE5D,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EACxC,WAAW,EACX,OAAO,EACP,iBAAiB,EACjB,MAAM,EACN,eAAe,GACI;IACnB,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACpB,gBAAgB,CAAC,6EAA6E,CAAC,CAAC;QAChG,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAC1B,MAAM,iBAAiB,CACrB;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,iCAAiC;SACxC;KACF,EACD;QACE,YAAY,EAAE,eAAe,EAAE;QAC/B,oBAAoB,EAAE,eAAe;QACrC,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,SAAS,CAAC,KAAK;QAC1B,oBAAoB,EAAE;YACpB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,IAAI;YAClB,KAAK,EAAE,KAAK;SACb;KACF,EACD,OAAO,CACR,CACF;SACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAkD,CAAC;IAErF,IAAI,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC;QAC7B,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,iDAAiD;IACjD,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACnC,gBAAgB,CAAC,mFAAmF,CAAC,CAAC;QACtG,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,uDAAuD;IACvD,iBAAiB,CAAC,IAAI,CAAC,QAAQ,GAAG,uBAAuB,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;IAE1E,2BAA2B;IAC3B,MAAM,gBAAgB,GAAG,MAAM,wBAAwB,CAAC,eAAe,EAAE,iBAAiB,CAAC,CAAC;IAE5F,0EAA0E;IAC1E,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC;QACxC,gBAAgB,CAAC,4CAA4C,CAAC,CAAC;QAC/D,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,IAAI,MAA0B,CAAC;IAC/B,MAAM,WAAW,GAAG,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC;IAEtD,IAAI,WAAW,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACtC,MAAM,GAAG,aAAa,CAAC,WAAW,CAAC,CAAC;IACtC,CAAC;SAAM,CAAC;QACN,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACjE,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACpC,CAAC;IAED,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,gBAAgB,CAAC,4CAA4C,CAAC,CAAC;QAC/D,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,SAAS,GAAG;QAChB,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACpC,SAAS,EAAE,WAAW;KACd,CAAC;IAEX,mBAAmB,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,iBAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAE9F,+CAA+C;IAC/C,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IAExE,iFAAiF;IACjF,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE;YACL;gBACE,SAAS,EAAE;oBACT,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;oBACrC,aAAa,EAAE,CAAC,iBAAiB,CAAC;iBACnC;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,aAAa,CAAC,MAAM;oBAC1B,MAAM,EAAE,CAAC,SAAS,CAAC;oBACnB,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,CAAC,IAAI,EAAE,OAAO,EAAE,iBAAiB,CAAC,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;iBACzG;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

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

@@ -0,0 +1,6 @@
+import { ActionHandlerProps, ActionResult } from '../step-iterate-types.js';
+/**
+ * Handles the performAnalysis action by executing analysis with enhanced context
+ * and processing the results.
+ */
+export declare function handlePerformAnalysis({ iterateCall, prompt, options, generateContentFn, waitIfPaused, }: ActionHandlerProps): Promise<ActionResult>;

package/dist/prompt/steps/step-iterate/handlers/handle-perform-analysis.js

@@ -0,0 +1,110 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { putAssistantMessage, putSystemMessage } from '../../../../main/common/content-bus.js';
+import { performAnalysis } from '../../../function-defs/perform-analysis.js';
+import { analysisResult } from '../../../function-defs/analysis-result.js';
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('performAnalysis', handlePerformAnalysis);
+/**
+ * Handles the performAnalysis action by executing analysis with enhanced context
+ * and processing the results.
+ */
+export async function handlePerformAnalysis({ iterateCall, prompt, options, generateContentFn, waitIfPaused, }) {
+    const functionDefs = [performAnalysis, analysisResult];
+    try {
+        // Wait if paused
+        await waitIfPaused();
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        });
+        // Get the performAnalysis call from the assistant
+        const performAnalysisRequest = await getPerformAnalysisCall(generateContentFn, prompt, functionDefs, options);
+        if (!performAnalysisRequest?.args) {
+            putSystemMessage('Failed to get valid performAnalysis request');
+            return { breakLoop: false, items: [] };
+        }
+        prompt.slice(-1)[0].functionCalls = [performAnalysisRequest];
+        // Prepare enhanced context
+        prompt.push({
+            type: 'user',
+            text: performAnalysisRequest.args.prompt,
+            functionResponses: [{ name: 'performAnalysis', call_id: performAnalysisRequest.id, content: '' }],
+        });
+        // Execute analysis
+        const analysisResultCall = await executeAnalysis(generateContentFn, prompt, functionDefs, options);
+        if (!analysisResultCall) {
+            putSystemMessage('Failed to get valid analysis results');
+            return { breakLoop: false, items: [] };
+        }
+        putAssistantMessage(analysisResultCall.args?.message ?? '', {
+            analysisRequest: performAnalysisRequest.args,
+            analysisResult: analysisResultCall.args,
+        });
+        const response = await askUserForInput('Your answer', '', options);
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: analysisResultCall.args?.message ?? '',
+                        functionCalls: [analysisResultCall],
+                    },
+                    user: {
+                        type: 'user',
+                        text: response.answer,
+                        functionResponses: [{ name: 'analysisResult', call_id: analysisResultCall.id, content: '' }],
+                    },
+                },
+            ],
+        };
+    }
+    catch (error) {
+        putSystemMessage(`Error during analysis: ${error instanceof Error ? error.message : String(error)}`);
+        return { breakLoop: false, items: [] };
+    }
+}
+/**
+ * Gets the performAnalysis call from the assistant
+ */
+async function getPerformAnalysisCall(generateContentFn, prompt, functionDefs, options) {
+    const performAnalysisRequest = [
+        prompt,
+        {
+            functionDefs,
+            requiredFunctionName: 'performAnalysis',
+            modelType: ModelType.CHEAP,
+            temperature: 0.7,
+            expectedResponseType: { text: false, functionCall: true, media: false },
+        },
+        options,
+    ];
+    const performAnalysisResult = await generateContentFn(...performAnalysisRequest);
+    return performAnalysisResult
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall)
+        .find((call) => call.name === 'performAnalysis');
+}
+/**
+ * Executes the analysis with the enhanced context
+ */
+async function executeAnalysis(generateContentFn, prompt, functionDefs, options) {
+    const analysisRequest = [
+        prompt,
+        {
+            functionDefs,
+            requiredFunctionName: 'analysisResult',
+            modelType: ModelType.DEFAULT,
+            temperature: 0.7,
+            expectedResponseType: { text: false, functionCall: true, media: false },
+        },
+        options,
+    ];
+    const analysisResult = await generateContentFn(...analysisRequest);
+    return analysisResult
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall)
+        .find((call) => call.name === 'analysisResult');
+}
+//# sourceMappingURL=handle-perform-analysis.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-perform-analysis.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-perform-analysis.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AAEnE,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC/F,OAAO,EAAE,eAAe,EAAE,MAAM,4CAA4C,CAAC;AAC7E,OAAO,EAAE,cAAc,EAAE,MAAM,2CAA2C,CAAC;AAE3E,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAEpE,qBAAqB,CAAC,iBAAiB,EAAE,qBAAqB,CAAC,CAAC;AAEhE;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,EAC1C,WAAW,EACX,MAAM,EACN,OAAO,EACP,iBAAiB,EACjB,YAAY,GACO;IACnB,MAAM,YAAY,GAAkB,CAAC,eAAe,EAAE,cAAc,CAAC,CAAC;IAEtE,IAAI,CAAC;QACH,iBAAiB;QACjB,MAAM,YAAY,EAAE,CAAC;QAErB,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC,CAAC,CAAC;QAEH,kDAAkD;QAClD,MAAM,sBAAsB,GAAG,MAAM,sBAAsB,CAAC,iBAAiB,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;QAE9G,IAAI,CAAC,sBAAsB,EAAE,IAAI,EAAE,CAAC;YAClC,gBAAgB,CAAC,6CAA6C,CAAC,CAAC;YAChE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAE7D,2BAA2B;QAC3B,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,sBAAsB,CAAC,IAAI,CAAC,MAAM;YACxC,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,sBAAsB,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;SAClG,CAAC,CAAC;QAEH,mBAAmB;QACnB,MAAM,kBAAkB,GAAG,MAAM,eAAe,CAAC,iBAAiB,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;QAEnG,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxB,gBAAgB,CAAC,sCAAsC,CAAC,CAAC;YACzD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,mBAAmB,CAAC,kBAAkB,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE;YAC1D,eAAe,EAAE,sBAAsB,CAAC,IAAI;YAC5C,cAAc,EAAE,kBAAkB,CAAC,IAAI;SACxC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;QAEnE,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,kBAAkB,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;wBAC5C,aAAa,EAAE,CAAC,kBAAkB,CAAC;qBACpC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,QAAQ,CAAC,MAAM;wBACrB,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,kBAAkB,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;qBAC7F;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,gBAAgB,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACrG,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,sBAAsB,CACnC,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,OAAuB;IAEvB,MAAM,sBAAsB,GAAwB;QAClD,MAAM;QACN;YACE,YAAY;YACZ,oBAAoB,EAAE,iBAAiB;YACvC,SAAS,EAAE,SAAS,CAAC,KAAK;YAC1B,WAAW,EAAE,GAAG;YAChB,oBAAoB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;SACxE;QACD,OAAO;KACR,CAAC;IACF,MAAM,qBAAqB,GAAG,MAAM,iBAAiB,CAAC,GAAG,sBAAsB,CAAC,CAAC;IACjF,OAAO,qBAAqB;SACzB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC;SAChC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,iBAAiB,CAAoC,CAAC;AACxF,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,eAAe,CAC5B,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,OAAuB;IAEvB,MAAM,eAAe,GAAwB;QAC3C,MAAM;QACN;YACE,YAAY;YACZ,oBAAoB,EAAE,gBAAgB;YACtC,SAAS,EAAE,SAAS,CAAC,OAAO;YAC5B,WAAW,EAAE,GAAG;YAChB,oBAAoB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;SACxE;QACD,OAAO;KACR,CAAC;IACF,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAAC,GAAG,eAAe,CAAC,CAAC;IACnE,OAAO,cAAc;SAClB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC;SAChC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAmC,CAAC;AACtF,CAAC"}

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

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

package/dist/prompt/steps/step-iterate/handlers/handle-project-command.js

@@ -0,0 +1,280 @@
+import { exec } from 'child_process';
+import { promisify } from 'util';
+import { putSystemMessage, putUserMessage } from '../../../../main/common/content-bus.js';
+import { rcConfig, getProjectCommand, getProjectCommands } from '../../../../main/config.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+const execPromise = promisify(exec);
+export async function handleRunProjectCommand({ generateContentFn, iterateCall, prompt, options, }) {
+    const runProjectCommandRequest = [
+        [
+            ...prompt,
+            {
+                type: 'assistant',
+                text: iterateCall.args?.message ?? '',
+            },
+            {
+                type: 'user',
+                text: JSON.stringify(rcConfig.projectCommands),
+            },
+        ],
+        {
+            functionDefs: getFunctionDefs(),
+            requiredFunctionName: 'runProjectCommand',
+            modelType: ModelType.CHEAP,
+            temperature: 0.7,
+            expectedResponseType: { text: false, functionCall: true, media: false },
+        },
+        options,
+    ];
+    const runProjectCommandResult = await generateContentFn(...runProjectCommandRequest);
+    const RunProjectCommandCall = runProjectCommandResult
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall)
+        .find((call) => call.name === 'runProjectCommand');
+    if (!RunProjectCommandCall?.args) {
+        putSystemMessage('No runProjectCommand call found in the assistant response.');
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args.message,
+        }, {
+            type: 'user',
+            text: 'No runProjectCommand call found in the assistant response.',
+        });
+        return { breakLoop: false, items: [] };
+    }
+    const { name, args = [], env: envOverride, workingDirOverride, truncMode = 'summarize' } = RunProjectCommandCall.args;
+    const command = getProjectCommand(name);
+    if (!command) {
+        const errorMessage = `Project command "${name}" not found. Available commands are: ${[
+            ...getProjectCommands().keys(),
+        ].join(', ')}`;
+        putSystemMessage(errorMessage);
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args.message,
+            functionCalls: [{ name: 'runProjectCommand', id: iterateCall.id + '_runProjectCommand', args: { name } }],
+        }, {
+            type: 'user',
+            functionResponses: [
+                {
+                    name: 'runProjectCommand',
+                    call_id: iterateCall.id + '_runProjectCommand',
+                    content: JSON.stringify({ error: errorMessage }),
+                    isError: true,
+                },
+            ],
+        });
+        return { breakLoop: false, items: [] };
+    }
+    // Auto-approval logic
+    let skipConfirmation = false;
+    try {
+        if (command.autoApprove === true) {
+            skipConfirmation = true;
+            putSystemMessage(`Auto-approving project command "${name}" (autoApprove: true).`);
+        }
+        else if (typeof command.autoApprove === 'string') {
+            const approved = await evaluateAutoApproveCondition(prompt, generateContentFn, options, command.autoApprove, name, args, command.description);
+            if (approved) {
+                skipConfirmation = true;
+                putSystemMessage(`Auto-approving project command "${name}" based on condition: ${command.autoApprove}`);
+            }
+            else {
+                putSystemMessage(`Auto-approval condition did not pass for command "${name}". Proceeding with user confirmation.`);
+            }
+        }
+    }
+    catch (error) {
+        putSystemMessage(`Failed to evaluate auto-approval condition for command "${name}". Falling back to user confirmation.`, { error: error.message });
+    }
+    if (!skipConfirmation) {
+        const confirmation = await askUserForConfirmationWithAnswer(`Do you want to run the project command "${name}"?`, 'Run command', 'Reject', true, options);
+        if (confirmation.answer) {
+            putUserMessage(confirmation.answer);
+        }
+        if (!confirmation.confirmed) {
+            putSystemMessage(`Project command "${name}" cancelled by user.`);
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+                functionCalls: [
+                    { name: 'runProjectCommand', id: iterateCall.id + '_runProjectCommand', args: { name, args } },
+                ],
+            }, {
+                type: 'user',
+                text: `I reject running the project command "${name}".` + (confirmation.answer ? ` ${confirmation.answer}` : ''),
+                functionResponses: [
+                    {
+                        name: 'runProjectCommand',
+                        call_id: iterateCall.id + '_runProjectCommand',
+                        content: JSON.stringify({ error: 'User rejected command execution.' }),
+                        isError: true,
+                    },
+                ],
+            });
+            return { breakLoop: false, items: [] };
+        }
+    }
+    // Proceed with execution if confirmed/auto-approved
+    const finalArgs = [...(command.defaultArgs || []), ...args];
+    const fullCommand = [command.command, ...finalArgs].join(' ');
+    const cwd = workingDirOverride || command.workingDir || rcConfig.rootDir;
+    const env = { ...process.env, ...(command.env || {}), ...envOverride };
+    putSystemMessage(`Executing project command "${name}": ${fullCommand}`, {
+        cwd,
+        envOverride,
+        args: finalArgs,
+        truncMode,
+        workingDirOverride,
+        autoApproved: skipConfirmation,
+    });
+    let result;
+    try {
+        const { stdout, stderr } = await execPromise(fullCommand, { cwd, env });
+        result = { success: true, exitCode: 0, stdout, stderr };
+        putSystemMessage(`Project command "${name}" executed successfully.`, { stdout, stderr });
+    }
+    catch (error) {
+        const { code, stdout, stderr } = error;
+        result = {
+            success: false,
+            exitCode: code ?? 1,
+            stdout,
+            stderr,
+        };
+        putSystemMessage(`Project command "${name}" failed with exit code ${result.exitCode}.`, {
+            stdout: result.stdout,
+            stderr: result.stderr,
+        });
+    }
+    // Handle output truncation
+    if (truncMode === 'first') {
+        result.stdout = truncateOutput(result.stdout, 'first');
+        result.stderr = truncateOutput(result.stderr, 'first');
+    }
+    else if (truncMode === 'last') {
+        result.stdout = truncateOutput(result.stdout, 'last');
+        result.stderr = truncateOutput(result.stderr, 'last');
+    }
+    else if (truncMode === 'summarize') {
+        if (result.stdout.length > 0) {
+            result.stdout = await summarizeOutput(result.stdout, generateContentFn, options);
+        }
+        if (result.stderr.length > 0) {
+            result.stderr = await summarizeOutput(result.stderr, generateContentFn, options);
+        }
+    }
+    putSystemMessage(`Command output sent to AI`, {
+        truncMode,
+        stdout: result.stdout,
+        stderr: result.stderr,
+    });
+    prompt.push({
+        type: 'assistant',
+        functionCalls: [{ name: 'runProjectCommand', id: iterateCall.id + '_runProjectCommand', args: { name, args } }],
+    }, {
+        type: 'user',
+        text: `Command "${name}" finished with exit code ${result.exitCode}. Please analyze the output.`,
+        functionResponses: [
+            {
+                name: 'runProjectCommand',
+                call_id: iterateCall.id + '_runProjectCommand',
+                content: JSON.stringify(result),
+            },
+        ],
+    });
+    return {
+        breakLoop: false,
+        items: [],
+    };
+}
+async function evaluateAutoApproveCondition(prompt, generateContentFn, options, condition, name, args, description) {
+    const evalPromptText = `You are a strict decision engine. Evaluate the following natural-language condition against the provided project command context. Reply with ONLY YES or NO (uppercase), without any explanation.
+
+Condition: ${condition}
+
+Command:
+- name: ${name}
+- description: ${description ?? 'n/a'}
+- args: ${JSON.stringify(args)}
+`;
+    const evalPrompt = [...prompt, { type: 'user', text: evalPromptText }];
+    const evalRequest = [
+        evalPrompt,
+        {
+            modelType: ModelType.LITE,
+            temperature: 0,
+            expectedResponseType: { text: true, functionCall: false, media: false },
+        },
+        options,
+    ];
+    try {
+        const evalResult = await generateContentFn(...evalRequest);
+        const text = evalResult
+            .find((p) => p.type === 'text')
+            ?.text?.trim()
+            .toUpperCase();
+        if (text === 'YES')
+            return true;
+        if (text === 'NO')
+            return false;
+        // Any other unexpected output -> not approved
+        return false;
+    }
+    catch (e) {
+        // On error, do not auto-approve
+        return false;
+    }
+}
+const MAX_OUTPUT_LENGTH = 2000;
+const TRUNCATE_LINES = 200;
+const TRUNCATE_CHARS = 1000;
+function truncateOutput(output, mode) {
+    if (output.length <= MAX_OUTPUT_LENGTH) {
+        return output;
+    }
+    const lines = output.split('\n');
+    if (mode === 'first') {
+        const truncatedLines = lines.slice(0, TRUNCATE_LINES).join('\n');
+        return truncatedLines.length > TRUNCATE_CHARS ? truncatedLines.substring(0, TRUNCATE_CHARS) : truncatedLines;
+    }
+    else {
+        // mode === 'last'
+        const truncatedLines = lines.slice(-TRUNCATE_LINES).join('\n');
+        return truncatedLines.length > TRUNCATE_CHARS
+            ? truncatedLines.substring(truncatedLines.length - TRUNCATE_CHARS)
+            : truncatedLines;
+    }
+}
+async function summarizeOutput(output, generateContentFn, options) {
+    if (output.length <= MAX_OUTPUT_LENGTH) {
+        return output;
+    }
+    putSystemMessage('Summarizing command output...');
+    const summaryPrompt = [
+        {
+            type: 'user',
+            text: `Please summarize the following command output:\n\n---\n\n${output}`,
+        },
+    ];
+    const summaryRequest = [
+        summaryPrompt,
+        {
+            modelType: ModelType.LITE,
+            expectedResponseType: { text: true, functionCall: false, media: false },
+        },
+        options,
+    ];
+    const summaryResult = await generateContentFn(...summaryRequest);
+    const summaryText = summaryResult.find((part) => part.type === 'text')?.text;
+    if (!summaryText) {
+        putSystemMessage('Failed to summarize output, returning truncated version.');
+        return truncateOutput(output, 'first');
+    }
+    return summaryText;
+}
+registerActionHandler('runProjectCommand', handleRunProjectCommand);
+//# sourceMappingURL=handle-project-command.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-project-command.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-project-command.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AACjC,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAC1F,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAO7F,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAqC,SAAS,EAAc,MAAM,wCAAwC,CAAC;AAClH,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAE3F,MAAM,WAAW,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;AAEpC,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,EAC5C,iBAAiB,EACjB,WAAW,EACX,MAAM,EACN,OAAO,GACY;IACnB,MAAM,wBAAwB,GAAwB;QACpD;YACE,GAAG,MAAM;YACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;aACtC;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,eAAe,CAAC;aAC/C;SACF;QACD;YACE,YAAY,EAAE,eAAe,EAAE;YAC/B,oBAAoB,EAAE,mBAAmB;YACzC,SAAS,EAAE,SAAS,CAAC,KAAK;YAC1B,WAAW,EAAE,GAAG;YAChB,oBAAoB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;SACxE;QACD,OAAO;KACR,CAAC;IACF,MAAM,uBAAuB,GAAG,MAAM,iBAAiB,CAAC,GAAG,wBAAwB,CAAC,CAAC;IACrF,MAAM,qBAAqB,GAAG,uBAAuB;SAClD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC;SAChC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,mBAAmB,CAAoD,CAAC;IAExG,IAAI,CAAC,qBAAqB,EAAE,IAAI,EAAE,CAAC;QACjC,gBAAgB,CAAC,4DAA4D,CAAC,CAAC;QAC/E,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;SAChC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,4DAA4D;SACnE,CACF,CAAC;QACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,MAAM,EAAE,IAAI,EAAE,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,WAAW,EAAE,kBAAkB,EAAE,SAAS,GAAG,WAAW,EAAE,GAAG,qBAAqB,CAAC,IAAI,CAAC;IAEtH,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAExC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,YAAY,GAAG,oBAAoB,IAAI,wCAAwC;YACnF,GAAG,kBAAkB,EAAE,CAAC,IAAI,EAAE;SAC/B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACf,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAC/B,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;YAC/B,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE,EAAE,WAAW,CAAC,EAAE,GAAG,oBAAoB,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC;SAC1G,EACD;YACE,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,mBAAmB;oBACzB,OAAO,EAAE,WAAW,CAAC,EAAE,GAAG,oBAAoB;oBAC9C,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;oBAChD,OAAO,EAAE,IAAI;iBACd;aACF;SACF,CACF,CAAC;QACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,sBAAsB;IACtB,IAAI,gBAAgB,GAAG,KAAK,CAAC;IAC7B,IAAI,CAAC;QACH,IAAI,OAAO,CAAC,WAAW,KAAK,IAAI,EAAE,CAAC;YACjC,gBAAgB,GAAG,IAAI,CAAC;YACxB,gBAAgB,CAAC,mCAAmC,IAAI,wBAAwB,CAAC,CAAC;QACpF,CAAC;aAAM,IAAI,OAAO,OAAO,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;YACnD,MAAM,QAAQ,GAAG,MAAM,4BAA4B,CACjD,MAAM,EACN,iBAAiB,EACjB,OAAO,EACP,OAAO,CAAC,WAAW,EACnB,IAAI,EACJ,IAAI,EACJ,OAAO,CAAC,WAAW,CACpB,CAAC;YACF,IAAI,QAAQ,EAAE,CAAC;gBACb,gBAAgB,GAAG,IAAI,CAAC;gBACxB,gBAAgB,CAAC,mCAAmC,IAAI,yBAAyB,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;YAC1G,CAAC;iBAAM,CAAC;gBACN,gBAAgB,CACd,qDAAqD,IAAI,uCAAuC,CACjG,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,gBAAgB,CACd,2DAA2D,IAAI,uCAAuC,EACtG,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CACpC,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,MAAM,YAAY,GAAG,MAAM,gCAAgC,CACzD,2CAA2C,IAAI,IAAI,EACnD,aAAa,EACb,QAAQ,EACR,IAAI,EACJ,OAAO,CACR,CAAC;QAEF,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;YACxB,cAAc,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;YAC5B,gBAAgB,CAAC,oBAAoB,IAAI,sBAAsB,CAAC,CAAC;YACjE,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;gBAC/B,aAAa,EAAE;oBACb,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE,EAAE,WAAW,CAAC,EAAE,GAAG,oBAAoB,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;iBAC/F;aACF,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EACF,yCAAyC,IAAI,IAAI,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5G,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,mBAAmB;wBACzB,OAAO,EAAE,WAAW,CAAC,EAAE,GAAG,oBAAoB;wBAC9C,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,kCAAkC,EAAE,CAAC;wBACtE,OAAO,EAAE,IAAI;qBACd;iBACF;aACF,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,MAAM,SAAS,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;IAC5D,MAAM,WAAW,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9D,MAAM,GAAG,GAAG,kBAAkB,IAAI,OAAO,CAAC,UAAU,IAAI,QAAQ,CAAC,OAAO,CAAC;IACzE,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,EAAE,CAAC;IAEvE,gBAAgB,CAAC,8BAA8B,IAAI,MAAM,WAAW,EAAE,EAAE;QACtE,GAAG;QACH,WAAW;QACX,IAAI,EAAE,SAAS;QACf,SAAS;QACT,kBAAkB;QAClB,YAAY,EAAE,gBAAgB;KAC/B,CAAC,CAAC;IAEH,IAAI,MAA4B,CAAC;IAEjC,IAAI,CAAC;QACH,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,WAAW,CAAC,WAAW,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;QACxE,MAAM,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QACxD,gBAAgB,CAAC,oBAAoB,IAAI,0BAA0B,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAC3F,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAAyD,CAAC;QAC3F,MAAM,GAAG;YACP,OAAO,EAAE,KAAK;YACd,QAAQ,EAAE,IAAI,IAAI,CAAC;YACnB,MAAM;YACN,MAAM;SACP,CAAC;QACF,gBAAgB,CAAC,oBAAoB,IAAI,2BAA2B,MAAM,CAAC,QAAQ,GAAG,EAAE;YACtF,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;IACL,CAAC;IAED,2BAA2B;IAC3B,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzD,CAAC;SAAM,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QAChC,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxD,CAAC;SAAM,IAAI,SAAS,KAAK,WAAW,EAAE,CAAC;QACrC,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,MAAM,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;QACnF,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,MAAM,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IACD,gBAAgB,CAAC,2BAA2B,EAAE;QAC5C,SAAS;QACT,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CACT;QACE,IAAI,EAAE,WAAW;QACjB,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE,EAAE,WAAW,CAAC,EAAE,GAAG,oBAAoB,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC;KAChH,EACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,YAAY,IAAI,6BAA6B,MAAM,CAAC,QAAQ,8BAA8B;QAChG,iBAAiB,EAAE;YACjB;gBACE,IAAI,EAAE,mBAAmB;gBACzB,OAAO,EAAE,WAAW,CAAC,EAAE,GAAG,oBAAoB;gBAC9C,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAChC;SACF;KACF,CACF,CAAC;IAEF,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE;KACV,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,4BAA4B,CACzC,MAAoB,EACpB,iBAA0D,EAC1D,OAAsC,EACtC,SAAiB,EACjB,IAAY,EACZ,IAAc,EACd,WAAoB;IAEpB,MAAM,cAAc,GAAG;;aAEZ,SAAS;;;UAGZ,IAAI;iBACG,WAAW,IAAI,KAAK;UAC3B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;CAC7B,CAAC;IAEA,MAAM,UAAU,GAAiB,CAAC,GAAG,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;IAErF,MAAM,WAAW,GAAwB;QACvC,UAAU;QACV;YACE,SAAS,EAAE,SAAS,CAAC,IAAI;YACzB,WAAW,EAAE,CAAC;YACd,oBAAoB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;SACxE;QACD,OAAO;KACR,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,GAAG,WAAW,CAAC,CAAC;QAC3D,MAAM,IAAI,GAAG,UAAU;aACpB,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;YAC/B,EAAE,IAAI,EAAE,IAAI,EAAE;aACb,WAAW,EAAE,CAAC;QACjB,IAAI,IAAI,KAAK,KAAK;YAAE,OAAO,IAAI,CAAC;QAChC,IAAI,IAAI,KAAK,IAAI;YAAE,OAAO,KAAK,CAAC;QAChC,8CAA8C;QAC9C,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,gCAAgC;QAChC,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAC/B,MAAM,cAAc,GAAG,GAAG,CAAC;AAC3B,MAAM,cAAc,GAAG,IAAI,CAAC;AAE5B,SAAS,cAAc,CAAC,MAAc,EAAE,IAAsB;IAC5D,IAAI,MAAM,CAAC,MAAM,IAAI,iBAAiB,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QACrB,MAAM,cAAc,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,OAAO,cAAc,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC;IAC/G,CAAC;SAAM,CAAC;QACN,kBAAkB;QAClB,MAAM,cAAc,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,OAAO,cAAc,CAAC,MAAM,GAAG,cAAc;YAC3C,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,cAAc,CAAC,MAAM,GAAG,cAAc,CAAC;YAClE,CAAC,CAAC,cAAc,CAAC;IACrB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,MAAc,EACd,iBAA0D,EAC1D,OAAsC;IAEtC,IAAI,MAAM,CAAC,MAAM,IAAI,iBAAiB,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,gBAAgB,CAAC,+BAA+B,CAAC,CAAC;IAClD,MAAM,aAAa,GAAiB;QAClC;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,4DAA4D,MAAM,EAAE;SAC3E;KACF,CAAC;IACF,MAAM,cAAc,GAAwB;QAC1C,aAAa;QACb;YACE,SAAS,EAAE,SAAS,CAAC,IAAI;YACzB,oBAAoB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;SACxE;QACD,OAAO;KACR,CAAC;IACF,MAAM,aAAa,GAAG,MAAM,iBAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;IACjE,MAAM,WAAW,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC;IAC7E,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,gBAAgB,CAAC,0DAA0D,CAAC,CAAC;QAC7E,OAAO,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzC,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,qBAAqB,CAAC,mBAAmB,EAAE,uBAAuB,CAAC,CAAC"}

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

@@ -0,0 +1,9 @@
+import { ActionResult, ActionHandlerProps } from '../step-iterate-types.js';
+/**
+ * Handler for pullAppContext action
+ * Retrieves context value from the content-bus system and includes it in the conversation
+ *
+ * @param props Action handler properties containing the function call
+ * @returns ActionResult with the context value and conversation items
+ */
+export declare function handlePullAppContext({ generateContentFn, prompt, options, }: ActionHandlerProps): Promise<ActionResult>;

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

@@ -0,0 +1,94 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { getContextValue } from '../../../../main/common/app-context-bus.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('pullAppContext', handlePullAppContext);
+/**
+ * Handler for pullAppContext action
+ * Retrieves context value from the content-bus system and includes it in the conversation
+ *
+ * @param props Action handler properties containing the function call
+ * @returns ActionResult with the context value and conversation items
+ */
+export async function handlePullAppContext({ generateContentFn, prompt, options, }) {
+    putSystemMessage('Retrieving context...');
+    const [pullAppContextCall] = (await generateContentFn(prompt, {
+        functionDefs: getFunctionDefs(),
+        requiredFunctionName: 'pullAppContext',
+        temperature: 0.7,
+        modelType: ModelType.CHEAP,
+        expectedResponseType: {
+            text: false,
+            functionCall: true,
+            media: false,
+        },
+    }, options))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    if (!pullAppContextCall?.args) {
+        putSystemMessage('Failed to get valid pullAppContext request');
+        return { breakLoop: false, items: [] };
+    }
+    const { key, reason } = pullAppContextCall.args;
+    try {
+        // Validate input
+        if (!key) {
+            throw new Error('Context key is required');
+        }
+        // Get context value from content-bus
+        const contextValue = await getContextValue(key);
+        // Format the response for conversation
+        const contextInfo = contextValue
+            ? `Retrieved context value for key "${key}": ${JSON.stringify(contextValue, null, 2)}`
+            : `No context value found for key "${key}"`;
+        const reasonInfo = reason ? `\nReason for retrieval: ${reason}` : '';
+        putSystemMessage('Context retrieved successfully', {
+            contextKey: key,
+            contextValue: contextValue,
+            reason,
+            operation: 'get',
+        });
+        prompt.push({
+            type: 'assistant',
+            text: `${contextInfo}${reasonInfo}`,
+            functionCalls: [pullAppContextCall],
+        }, {
+            type: 'user',
+            functionResponses: [
+                {
+                    name: 'pullAppContext',
+                    call_id: pullAppContextCall.id,
+                    content: JSON.stringify({ contextKey: key, contextValue, reason, operation: 'get' }),
+                },
+            ],
+        });
+        // Return the action result with context information
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    catch (error) {
+        // Handle any errors that occur during context retrieval
+        const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
+        putSystemMessage(`Error retrieving context`, {
+            contextKey: key,
+            error: errorMessage,
+            reason,
+            operation: 'get',
+        });
+        prompt.push({
+            type: 'assistant',
+            text: `Failed to retrieve context for key "${key}": ${errorMessage}`,
+        }, {
+            type: 'user',
+            text: 'Context retrieval failed.',
+        });
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+}
+//# sourceMappingURL=handle-pull-app-context.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-pull-app-context.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/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;AAC/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE,qBAAqB,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,CAAC;AAE9D;;;;;;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,CAC3B,MAAM,iBAAiB,CACrB,MAAM,EACN;QACE,YAAY,EAAE,eAAe,EAAE;QAC/B,oBAAoB,EAAE,gBAAgB;QACtC,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,SAAS,CAAC,KAAK;QAC1B,oBAAoB,EAAE;YACpB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,IAAI;YAClB,KAAK,EAAE,KAAK;SACb;KACF,EACD,OAAO,CACR,CACF;SACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAmD,CAAC;IAEtF,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,gBAAgB,CAAC,gCAAgC,EAAE;YACjD,UAAU,EAAE,GAAG;YACf,YAAY,EAAE,YAAY;YAC1B,MAAM;YACN,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,GAAG,WAAW,GAAG,UAAU,EAAE;YACnC,aAAa,EAAE,CAAC,kBAAkB,CAAC;SACpC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,gBAAgB;oBACtB,OAAO,EAAE,kBAAkB,CAAC,EAAE;oBAC9B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,YAAY,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;iBACrF;aACF;SACF,CACF,CAAC;QAEF,oDAAoD;QACpD,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,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,0BAA0B,EAAE;YAC3C,UAAU,EAAE,GAAG;YACf,KAAK,EAAE,YAAY;YACnB,MAAM;YACN,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,uCAAuC,GAAG,MAAM,YAAY,EAAE;SACrE,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,2BAA2B;SAClC,CACF,CAAC;QAEF,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;AACH,CAAC"}