genaicode 0.8.5 → 0.9.0

package/dist/prompt/steps/step-ask-question/handlers/{handle-send-message-with-image.js → handle-generate-image.js}

@@ -2,7 +2,9 @@ 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';
-export async function handleSendMessageWithImage({ askQuestionMessage, options, generateContentFn, prompt, generateImageFn, }) {
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { getTempBuffer } from '../../../../files/temp-buffer.js';
+export async function handleGenerateImage({ askQuestionCall, options, generateContentFn, prompt, generateImageFn, }) {
     if (!options.imagen) {
         putSystemMessage('Image generation is not enabled. Please enable it using --imagen parameter.');
         return {
@@ -10,26 +12,25 @@ export async function handleSendMessageWithImage({ askQuestionMessage, options,
             items: [],
         };
     }
-    const [sendMessageWithImageCall] = (await generateContentFn([
+    const [generateImageCall] = (await generateContentFn([
         ...prompt,
         {
             type: 'assistant',
-            text: askQuestionMessage ?? '',
+            text: askQuestionCall.args?.message ?? '',
         },
         {
             type: 'user',
             text: 'Yes, you can generate an image.',
         },
-    ], getFunctionDefs(), 'sendMessageWithImage', 0.7, true, options));
-    if (!sendMessageWithImageCall) {
+    ], getFunctionDefs(), 'generateImage', 0.7, ModelType.CHEAP, options));
+    if (!generateImageCall?.args) {
         return {
             breakLoop: true,
             items: [],
         };
     }
     // Get the image generation request from the args
-    const imageGenerationRequest = sendMessageWithImageCall.args;
-    if (!imageGenerationRequest?.prompt) {
+    if (!generateImageCall.args.prompt) {
         putSystemMessage('Image generation request with prompt is required for sendMessageWithImage action.');
         return {
             breakLoop: true,
@@ -37,20 +38,7 @@ export async function handleSendMessageWithImage({ askQuestionMessage, options,
         };
     }
     // Create a temporary file path for the generated image
-    const tempImagePath = `/tmp/question-image-${Date.now()}.png`;
-    // Generate the image using generateImage function call
-    const generateImageCall = {
-        id: 'generate_image_' + sendMessageWithImageCall.id,
-        name: 'generateImage',
-        args: {
-            prompt: imageGenerationRequest.prompt,
-            filePath: tempImagePath,
-            width: 1024,
-            height: 1024,
-            explanation: 'Generating image to support the question',
-            ...(imageGenerationRequest.contextImage && { contextImagePath: imageGenerationRequest.contextImage }),
-        },
-    };
+    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
@@ -61,19 +49,30 @@ export async function handleSendMessageWithImage({ askQuestionMessage, options,
             items: [],
         };
     }
-    const response = await fetch(downloadFileCall.args?.downloadUrl);
-    const arrayBuffer = await response.arrayBuffer();
-    const buffer = Buffer.from(arrayBuffer);
+    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: true,
+            items: [],
+        };
+    }
     const imageData = {
         base64url: buffer.toString('base64'),
         mediaType: 'image/png',
     };
-    putAssistantMessage('Here is the image to support the question:', generateImageCall.args, [], [imageData]);
+    putAssistantMessage('', generateImageCall.args, [], [imageData]);
     // Get user's response with the image displayed
-    const inputResponse = await askUserForInput('Your answer', askQuestionMessage ?? '');
-    if (inputResponse.options?.aiService) {
-        options.aiService = inputResponse.options.aiService;
-    }
+    const inputResponse = await askUserForInput('Your answer', askQuestionCall.args?.message ?? '', options);
     // Return the conversation items with the generated image included in the context
     return {
         breakLoop: false,
@@ -81,7 +80,7 @@ export async function handleSendMessageWithImage({ askQuestionMessage, options,
             {
                 assistant: {
                     type: 'assistant',
-                    text: askQuestionMessage ?? '',
+                    text: askQuestionCall.args?.message ?? '',
                     functionCalls: [generateImageCall],
                 },
                 user: {
@@ -94,4 +93,4 @@ export async function handleSendMessageWithImage({ askQuestionMessage, options,
         ],
     };
 }
-//# sourceMappingURL=handle-send-message-with-image.js.map
+//# sourceMappingURL=handle-generate-image.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-generate-image.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/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;AAEjE,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EACxC,eAAe,EACf,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,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAClD;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,iCAAiC;SACxC;KACF,EACD,eAAe,EAAE,EACjB,eAAe,EACf,GAAG,EACH,SAAS,CAAC,KAAK,EACf,OAAO,CACR,CAAkD,CAAC;IAEpD,IAAI,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC;QAC7B,OAAO;YACL,SAAS,EAAE,IAAI;YACf,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,IAAI;YACf,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,IAAI;YACf,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,IAAI;YACf,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,EAAE,EAAE,iBAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEjE,+CAA+C;IAC/C,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;IAEzG,iFAAiF;IACjF,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,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-ask-question/handlers/handle-perform-analysis.d.ts

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

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

@@ -0,0 +1,99 @@
+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 { validateAndRecoverSingleResult } from '../../step-validate-recover.js';
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+/**
+ * Handles the performAnalysis action by executing analysis with enhanced context
+ * and processing the results.
+ */
+export async function handlePerformAnalysis({ askQuestionCall, prompt, options, generateContentFn, waitIfPaused, }) {
+    const functionDefs = [performAnalysis, analysisResult];
+    try {
+        // Wait if paused
+        await waitIfPaused();
+        prompt.push({
+            type: 'assistant',
+            text: askQuestionCall.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: true, 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: true, items: [] };
+        }
+        putAssistantMessage(analysisResultCall.args?.message ?? '', {
+            analysisRequest: performAnalysisRequest.args,
+            analysisResult: analysisResultCall.args,
+        });
+        const response = await askUserForInput('Your answer', analysisResultCall.args?.message ?? '', 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: true, items: [] };
+    }
+}
+/**
+ * Gets the performAnalysis call from the assistant
+ */
+async function getPerformAnalysisCall(generateContentFn, prompt, functionDefs, options) {
+    const performAnalysisRequest = [
+        prompt,
+        functionDefs,
+        'performAnalysis',
+        0.7,
+        ModelType.CHEAP,
+        options,
+    ];
+    let performAnalysisResult = await generateContentFn(...performAnalysisRequest);
+    performAnalysisResult = await validateAndRecoverSingleResult(performAnalysisRequest, performAnalysisResult, generateContentFn);
+    return performAnalysisResult.find((call) => call.name === 'performAnalysis');
+}
+/**
+ * Executes the analysis with the enhanced context
+ */
+async function executeAnalysis(generateContentFn, prompt, functionDefs, options) {
+    const analysisRequest = [
+        prompt,
+        functionDefs,
+        'analysisResult',
+        0.7,
+        ModelType.DEFAULT,
+        options,
+    ];
+    let analysisResult = await generateContentFn(...analysisRequest);
+    analysisResult = await validateAndRecoverSingleResult(analysisRequest, analysisResult, generateContentFn);
+    return analysisResult.find((call) => call.name === 'analysisResult');
+}
+//# sourceMappingURL=handle-perform-analysis.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-perform-analysis.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-perform-analysis.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AAOnE,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;AAC3E,OAAO,EAAE,8BAA8B,EAAE,MAAM,gCAAgC,CAAC;AAEhF,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAE1E;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,EAC1C,eAAe,EACf,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,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C,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,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACxC,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,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACxC,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,kBAAkB,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAEvG,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,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACxC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,sBAAsB,CACnC,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,OAAuB;IAEvB,MAAM,sBAAsB,GAAwB;QAClD,MAAM;QACN,YAAY;QACZ,iBAAiB;QACjB,GAAG;QACH,SAAS,CAAC,KAAK;QACf,OAAO;KACR,CAAC;IACF,IAAI,qBAAqB,GAAG,MAAM,iBAAiB,CAAC,GAAG,sBAAsB,CAAC,CAAC;IAC/E,qBAAqB,GAAG,MAAM,8BAA8B,CAC1D,sBAAsB,EACtB,qBAAqB,EACrB,iBAAiB,CAClB,CAAC;IAEF,OAAO,qBAAqB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,iBAAiB,CAAoC,CAAC;AAClH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,eAAe,CAC5B,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,OAAuB;IAEvB,MAAM,eAAe,GAAwB;QAC3C,MAAM;QACN,YAAY;QACZ,gBAAgB;QAChB,GAAG;QACH,SAAS,CAAC,OAAO;QACjB,OAAO;KACR,CAAC;IACF,IAAI,cAAc,GAAG,MAAM,iBAAiB,CAAC,GAAG,eAAe,CAAC,CAAC;IACjE,cAAc,GAAG,MAAM,8BAA8B,CAAC,eAAe,EAAE,cAAc,EAAE,iBAAiB,CAAC,CAAC;IAE1G,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAmC,CAAC;AACzG,CAAC"}

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

@@ -0,0 +1,9 @@
+import { ActionResult, ActionHandlerProps } from '../step-ask-question-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-ask-question/handlers/handle-pull-app-context.js

@@ -0,0 +1,90 @@
+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';
+/**
+ * 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, getFunctionDefs(), 'pullAppContext', 0.7, ModelType.CHEAP, options));
+    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}` : '';
+        // Return the action result with context information
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: `${contextInfo}${reasonInfo}`,
+                        functionCalls: [pullAppContextCall],
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Context retrieved successfully.',
+                        functionResponses: [
+                            {
+                                name: 'pullAppContext',
+                                call_id: pullAppContextCall.id,
+                                content: JSON.stringify({ contextKey: key, contextValue, reason, operation: 'get' }),
+                            },
+                        ],
+                        data: {
+                            contextKey: key,
+                            contextValue: contextValue,
+                            reason,
+                            operation: 'get',
+                        },
+                    },
+                },
+            ],
+        };
+    }
+    catch (error) {
+        // Handle any errors that occur during context retrieval
+        const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
+        putSystemMessage(`Error retrieving context: ${errorMessage}`);
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: `Failed to retrieve context for key "${key}": ${errorMessage}`,
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Context retrieval failed.',
+                        data: {
+                            contextKey: key,
+                            error: errorMessage,
+                            reason,
+                            operation: 'get',
+                        },
+                    },
+                },
+            ],
+        };
+    }
+}
+//# sourceMappingURL=handle-pull-app-context.js.map

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,87 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { getSourceCode } from '../../../../files/read-files.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.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 [reasoningInferenceCall] = (await generateContentFn([
+            ...prompt,
+            {
+                type: 'assistant',
+                text: askQuestionCall.args.message,
+            },
+        ], getFunctionDefs(), 'reasoningInference', 0.7, ModelType.CHEAP, options));
+        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);
+        // Call the reasoning model with appropriate settings
+        const [reasoningInferenceResponseCall] = (await generateContentFn([
+            {
+                type: 'user',
+                text: reasoningInferenceCall.args.prompt +
+                    (reasoningInferenceCall.args.contextPaths?.length > 0
+                        ? '\n\nContents of relevant files:\n' +
+                            Object.entries(getSourceCode({
+                                filterPaths: reasoningInferenceCall.args.contextPaths,
+                                forceAll: true,
+                                ignoreImportantFiles: true,
+                            }, options))
+                                .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 {
+            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,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAQ/D;;;;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;QAC3D,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CACvD;YACE,GAAG,MAAM;YACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;aACpC;SACF,EACD,eAAe,EAAE,EACjB,oBAAoB,EACpB,GAAG,EACH,SAAS,CAAC,KAAK,EACf,OAAO,CACR,CAAuD,CAAC;QAEzD,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;QAC9F,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,sBAAsB,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,GAAG,CAAC;wBACnD,CAAC,CAAC,mCAAmC;4BACnC,MAAM,CAAC,OAAO,CACZ,aAAa,CACX;gCACE,WAAW,EAAE,sBAAsB,CAAC,IAAI,CAAC,YAAY;gCACrD,QAAQ,EAAE,IAAI;gCACd,oBAAoB,EAAE,IAAI;6BAC3B,EACD,OAAO,CACR,CACF;iCACE,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,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;AACH,CAAC"}

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

@@ -2,4 +2,4 @@ import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js'
 /**
  * Handler for the searchCode action
  */
-export declare function handleSearchCode({ askQuestionMessage, generateContentFn, prompt, options, }: ActionHandlerProps): Promise<ActionResult>;
+export declare function handleSearchCode({ askQuestionCall, generateContentFn, prompt, options, }: ActionHandlerProps): Promise<ActionResult>;

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

@@ -1,12 +1,13 @@
 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';
 /**
  * Handler for the searchCode action
  */
-export async function handleSearchCode({ askQuestionMessage, generateContentFn, prompt, options, }) {
+export async function handleSearchCode({ askQuestionCall, generateContentFn, prompt, options, }) {
     try {
-        const searchCodeCall = await generateSearchCodeCall(generateContentFn, prompt, askQuestionMessage, 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) {
@@ -32,12 +33,11 @@ export async function handleSearchCode({ askQuestionMessage, generateContentFn,
                 {
                     assistant: {
                         type: 'assistant',
-                        text: askQuestionMessage ?? '',
+                        text: askQuestionCall.args?.message ?? '',
                         functionCalls: [searchCodeCall],
                     },
                     user: {
                         type: 'user',
-                        text: 'Search completed, lets continue the conversation.',
                         functionResponses: [
                             {
                                 name: 'searchCode',
@@ -70,18 +70,18 @@ export async function handleSearchCode({ askQuestionMessage, generateContentFn,
         };
     }
 }
-async function generateSearchCodeCall(generateContentFn, prompt, askQuestionMessage, options, cheap) {
+async function generateSearchCodeCall(generateContentFn, prompt, askQuestionCall, options) {
     const [searchCodeCall] = await generateContentFn([
         ...prompt,
         {
             type: 'assistant',
-            text: askQuestionMessage ?? '',
+            text: askQuestionCall.args?.message ?? '',
         },
         {
             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,kBAAkB,EAClB,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,MAAM,sBAAsB,CAAC,iBAAiB,EAAE,MAAM,EAAE,kBAAkB,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAEnH,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,kBAAkB,IAAI,EAAE;wBAC9B,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,kBAAsC,EACtC,OAAuB,EACvB,KAAc;IAEd,MAAM,CAAC,cAAc,CAAC,GAAG,MAAM,iBAAiB,CAC9C;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,kBAAkB,IAAI,EAAE;SAC/B;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.d.ts

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