genaicode 0.6.1 → 0.8.0

package/dist/prompt/static-prompts.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Static prompts used in the prompt-service module.
+ * These prompts are used in the initial conversation with the AI service.
+ */
+/** Initial greeting from the user */
+export declare const INITIAL_GREETING = "Hello, GenAIcode!";
+/** Assistant's response asking for source code and other assets */
+export declare const REQUEST_SOURCE_CODE = "Hello there! I guess you have a task for me today. Before we start, could you please provide me with: \n      - the current source code of your application\n      - the image assets (if available)\n      - and conversational history (if available)\n      \n      Thanks";
+/** User's response when providing source code and assets */
+export declare const SOURCE_CODE_RESPONSE = "Sure, here is the application source code, image assets, and the history.";
+/** Assistant's acknowledgment after receiving source code */
+export declare const READY_TO_ASSIST = "Thank you, I'm ready to assist you with your request.";
+/** Assistant's response when generating code */
+export declare const CODEGEN_SUMMARY_PROMPT = "Planning phase completed. Now please create the code generation summary.";
+/** Template for partial prompt when requesting changes for a specific file */
+export declare const getPartialPromptTemplate: (path: string) => string;

package/dist/prompt/static-prompts.js

@@ -0,0 +1,22 @@
+/**
+ * Static prompts used in the prompt-service module.
+ * These prompts are used in the initial conversation with the AI service.
+ */
+/** Initial greeting from the user */
+export const INITIAL_GREETING = 'Hello, GenAIcode!';
+/** Assistant's response asking for source code and other assets */
+export const REQUEST_SOURCE_CODE = `Hello there! I guess you have a task for me today. Before we start, could you please provide me with: 
+      - the current source code of your application
+      - the image assets (if available)
+      - and conversational history (if available)
+      
+      Thanks`;
+/** User's response when providing source code and assets */
+export const SOURCE_CODE_RESPONSE = 'Sure, here is the application source code, image assets, and the history.';
+/** Assistant's acknowledgment after receiving source code */
+export const READY_TO_ASSIST = "Thank you, I'm ready to assist you with your request.";
+/** Assistant's response when generating code */
+export const CODEGEN_SUMMARY_PROMPT = 'Planning phase completed. Now please create the code generation summary.';
+/** Template for partial prompt when requesting changes for a specific file */
+export const getPartialPromptTemplate = (path) => `Please now generate update for the \`${path}\` file using appropriate tools according to code generation summary.`;
+//# sourceMappingURL=static-prompts.js.map

package/dist/prompt/static-prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"static-prompts.js","sourceRoot":"","sources":["../../src/prompt/static-prompts.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,qCAAqC;AACrC,MAAM,CAAC,MAAM,gBAAgB,GAAG,mBAAmB,CAAC;AAEpD,mEAAmE;AACnE,MAAM,CAAC,MAAM,mBAAmB,GAAG;;;;;aAKtB,CAAC;AAEd,4DAA4D;AAC5D,MAAM,CAAC,MAAM,oBAAoB,GAAG,2EAA2E,CAAC;AAEhH,6DAA6D;AAC7D,MAAM,CAAC,MAAM,eAAe,GAAG,uDAAuD,CAAC;AAEvF,gDAAgD;AAChD,MAAM,CAAC,MAAM,sBAAsB,GAAG,0EAA0E,CAAC;AAEjH,8EAA8E;AAC9E,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,IAAY,EAAU,EAAE,CAC/D,wCAAwC,IAAI,uEAAuE,CAAC"}

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

@@ -1,13 +1,11 @@
 import { putSystemMessage } from '../../../../main/common/content-bus.js';
-import { StepResult } from '../../steps-types.js';
 export async function handleCancelCodeGeneration({ askQuestionCall }) {
     putSystemMessage('Assistant requested to stop code generation. Exiting...');
     return {
         breakLoop: true,
-        stepResult: StepResult.BREAK,
         items: [
             {
-                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },
+                assistant: { type: 'assistant', text: askQuestionCall.args?.message ?? '' },
                 user: { type: 'user', text: 'Code generation cancelled.' },
             },
         ],

package/dist/prompt/steps/step-ask-question/handlers/cancel-code-generation.js.map

@@ -1 +1 @@
-{"version":3,"file":"cancel-code-generation.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/cancel-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAE,eAAe,EAAsB;IACtF,gBAAgB,CAAC,yDAAyD,CAAC,CAAC;IAC5E,OAAO;QACL,SAAS,EAAE,IAAI;QACf,UAAU,EAAE,UAAU,CAAC,KAAK;QAC5B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9F,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,4BAA4B,EAAE;aAC3D;SACF;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"cancel-code-generation.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/cancel-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAG1E,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAE,eAAe,EAAsB;IACtF,gBAAgB,CAAC,yDAAyD,CAAC,CAAC;IAC5E,OAAO;QACL,SAAS,EAAE,IAAI;QACf,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE;gBAC3E,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,4BAA4B,EAAE;aAC3D;SACF;KACF,CAAC;AACJ,CAAC"}

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

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

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

@@ -0,0 +1,58 @@
+import { putAssistantMessage, putUserMessage } from '../../../../main/common/content-bus.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { executeStepCodegenPlanning } from '../../step-codegen-planning.js';
+import { executeStepCodegenSummary } from '../../step-codegen-summary.js';
+import { StepResult } from '../../steps-types.js';
+export async function handleCodeGeneration({ generateContentFn, generateImageFn, waitIfPaused, prompt, options, }) {
+    const planningResult = await executeStepCodegenPlanning(generateContentFn, prompt, options);
+    if (planningResult === StepResult.BREAK) {
+        return {
+            breakLoop: true,
+            items: [],
+        };
+    }
+    // Ask user to confirm the planning result
+    const planningConfirmation = await askUserForConfirmationWithAnswer('Planning phase completed. Would you like to proceed with the planned changes?', 'Accept planning and continue', 'Reject planning and return to conversation', true);
+    if (!planningConfirmation.confirmed) {
+        // User rejected the planning, return to conversation
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: 'Planning phase completed. Would you like to proceed with the planned changes?',
+                        functionCalls: [],
+                    },
+                    user: {
+                        type: 'user',
+                        text: planningConfirmation.answer || 'Planning rejected. Returning to conversation.',
+                    },
+                },
+            ],
+        };
+    }
+    putAssistantMessage('Planning phase completed. Would you like to proceed with the planned changes?');
+    putUserMessage('Accept planning and continue');
+    // Execute the codegen summary step
+    const result = await executeStepCodegenSummary(generateContentFn, prompt, getFunctionDefs(), options, waitIfPaused, generateImageFn);
+    const confirmed = await askUserForConfirmationWithAnswer('Code changes are generated, now what?', 'Apply code changes', 'Continue conversation', true);
+    return {
+        breakLoop: confirmed.confirmed ?? true,
+        stepResult: result,
+        items: [
+            {
+                assistant: {
+                    type: 'assistant',
+                    text: 'Code changes are generated, now what?',
+                },
+                user: {
+                    type: 'user',
+                    text: confirmed.answer || 'Apply code changes.',
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=code-generation.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"code-generation.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAC7F,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,0BAA0B,EAAE,MAAM,gCAAgC,CAAC;AAC5E,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,EACzC,iBAAiB,EACjB,eAAe,EACf,YAAY,EACZ,MAAM,EACN,OAAO,GACY;IACnB,MAAM,cAAc,GAAG,MAAM,0BAA0B,CAAC,iBAAiB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC5F,IAAI,cAAc,KAAK,UAAU,CAAC,KAAK,EAAE,CAAC;QACxC,OAAO;YACL,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,0CAA0C;IAC1C,MAAM,oBAAoB,GAAG,MAAM,gCAAgC,CACjE,+EAA+E,EAC/E,8BAA8B,EAC9B,4CAA4C,EAC5C,IAAI,CACL,CAAC;IAEF,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,CAAC;QACpC,qDAAqD;QACrD,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,+EAA+E;wBACrF,aAAa,EAAE,EAAE;qBAClB;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,oBAAoB,CAAC,MAAM,IAAI,+CAA+C;qBACrF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,+EAA+E,CAAC,CAAC;IACrG,cAAc,CAAC,8BAA8B,CAAC,CAAC;IAE/C,mCAAmC;IACnC,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAC5C,iBAAiB,EACjB,MAAM,EACN,eAAe,EAAE,EACjB,OAAO,EACP,YAAY,EACZ,eAAe,CAChB,CAAC;IAEF,MAAM,SAAS,GAAG,MAAM,gCAAgC,CACtD,uCAAuC,EACvC,oBAAoB,EACpB,uBAAuB,EACvB,IAAI,CACL,CAAC;IAEF,OAAO;QACL,SAAS,EAAE,SAAS,CAAC,SAAS,IAAI,IAAI;QACtC,UAAU,EAAE,MAAM;QAClB,KAAK,EAAE;YACL;gBACE,SAAS,EAAE;oBACT,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,uCAAuC;iBAC9C;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,SAAS,CAAC,MAAM,IAAI,qBAAqB;iBAChD;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/prompt/steps/step-ask-question/handlers/confirm-code-generation.d.ts

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

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

@@ -1,16 +1,19 @@
 import { putSystemMessage } from '../../../../main/common/content-bus.js';
 import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
-import { StepResult } from '../../steps-types.js';
-export async function handleConfirmCodeGeneration({ askQuestionCall }) {
+export async function handleConfirmCodeGeneration({ askQuestionCall, options, }) {
     const userConfirmation = await askUserForConfirmationWithAnswer('The assistant is ready to start code generation. Do you want to proceed?', 'Start code generation', 'Continue conversation', true);
+    if (userConfirmation.options?.aiService) {
+        options.aiService = userConfirmation.options.aiService;
+        options.cheap = userConfirmation.options.cheap;
+    }
     if (userConfirmation.confirmed) {
         putSystemMessage('Proceeding with code generation.');
         return {
             breakLoop: true,
-            stepResult: StepResult.CONTINUE,
+            executeCodegen: true,
             items: [
                 {
-                    assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },
+                    assistant: { type: 'assistant', text: askQuestionCall.args?.message ?? '', functionCalls: [] },
                     user: { type: 'user', text: userConfirmation.answer || 'Confirmed. Proceed with code generation.' },
                 },
             ],
@@ -20,14 +23,12 @@ export async function handleConfirmCodeGeneration({ askQuestionCall }) {
         putSystemMessage('Declined. Continuing the conversation.');
         return {
             breakLoop: false,
-            stepResult: StepResult.CONTINUE,
             items: [
                 {
-                    assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                    assistant: { type: 'assistant', text: askQuestionCall.args?.message ?? '' },
                     user: {
                         type: 'user',
                         text: userConfirmation.answer || 'Declined. Please continue the conversation.',
-                        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
                     },
                 },
             ],

package/dist/prompt/steps/step-ask-question/handlers/confirm-code-generation.js.map

@@ -1 +1 @@
-{"version":3,"file":"confirm-code-generation.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/confirm-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAAC,EAAE,eAAe,EAAsB;IACvF,MAAM,gBAAgB,GAAG,MAAM,gCAAgC,CAC7D,0EAA0E,EAC1E,uBAAuB,EACvB,uBAAuB,EACvB,IAAI,CACL,CAAC;IACF,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC/B,gBAAgB,CAAC,kCAAkC,CAAC,CAAC;QACrD,OAAO;YACL,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,UAAU,CAAC,QAAQ;YAC/B,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;oBAC9F,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,CAAC,MAAM,IAAI,0CAA0C,EAAE;iBACpG;aACF;SACF,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,gBAAgB,CAAC,wCAAwC,CAAC,CAAC;QAC3D,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;YAC/B,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;oBAC7G,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,gBAAgB,CAAC,MAAM,IAAI,6CAA6C;wBAC9E,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;qBACpG;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}
+{"version":3,"file":"confirm-code-generation.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/confirm-code-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAG3F,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAAC,EAChD,eAAe,EACf,OAAO,GACY;IACnB,MAAM,gBAAgB,GAAG,MAAM,gCAAgC,CAC7D,0EAA0E,EAC1E,uBAAuB,EACvB,uBAAuB,EACvB,IAAI,CACL,CAAC;IACF,IAAI,gBAAgB,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC;QACxC,OAAO,CAAC,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC;QACvD,OAAO,CAAC,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC;IACjD,CAAC;IACD,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC/B,gBAAgB,CAAC,kCAAkC,CAAC,CAAC;QACrD,OAAO;YACL,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,IAAI;YACpB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;oBAC9F,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,CAAC,MAAM,IAAI,0CAA0C,EAAE;iBACpG;aACF;SACF,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,gBAAgB,CAAC,wCAAwC,CAAC,CAAC;QAC3D,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE;oBAC3E,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,gBAAgB,CAAC,MAAM,IAAI,6CAA6C;qBAC/E;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

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

@@ -2,20 +2,21 @@ 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 { StepResult } from '../../steps-types.js';
 export async function handleContextOptimization({ askQuestionCall, prompt, options, generateContentFn, }) {
     const userConfirmation = await askUserForConfirmation('The assistant suggests optimizing the context to reduce token usage, cost, and latency. Do you want to proceed?', false);
+    if (userConfirmation.options?.aiService) {
+        options.aiService = userConfirmation.options.aiService;
+        options.cheap = userConfirmation.options.cheap;
+    }
     const user = {
         type: 'user',
         text: userConfirmation ? 'Context optimization applied.' : 'Context optimization not applied.',
-        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
     };
     const assistant = {
         type: 'assistant',
-        text: askQuestionCall.args?.content ?? '',
-        functionCalls: [askQuestionCall],
+        text: askQuestionCall.args?.message ?? '',
     };
-    if (userConfirmation) {
+    if (userConfirmation.confirmed) {
         // the request may be caused be an appearance of a new file, so lets refresh
         refreshFiles();
         // Execute context optimization step
@@ -24,7 +25,6 @@ export async function handleContextOptimization({ askQuestionCall, prompt, optio
     }
     return {
         breakLoop: false,
-        stepResult: StepResult.CONTINUE,
         items: [
             {
                 assistant,

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

@@ -1 +1 @@
-{"version":3,"file":"context-optimization.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/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,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAC9C,eAAe,EACf,MAAM,EACN,OAAO,EACP,iBAAiB,GACE;IACnB,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,iHAAiH,EACjH,KAAK,CACN,CAAC;IAEF,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC,mCAAmC;QAC9F,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;KACpG,CAAC;IAEF,MAAM,SAAS,GAAkB;QAC/B,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QACzC,aAAa,EAAE,CAAC,eAAe,CAAC;KACjC,CAAC;IAEF,IAAI,gBAAgB,EAAE,CAAC;QACrB,4EAA4E;QAC5E,YAAY,EAAE,CAAC;QAEf,oCAAoC;QACpC,gBAAgB,CAAC,sCAAsC,CAAC,CAAC;QACzD,MAAM,8BAA8B,CAAC,iBAAiB,EAAE,CAAC,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;IACjG,CAAC;IAED,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS;gBACT,IAAI;aACL;SACF;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"context-optimization.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/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;AAGpF,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAC9C,eAAe,EACf,MAAM,EACN,OAAO,EACP,iBAAiB,GACE;IACnB,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,iHAAiH,EACjH,KAAK,CACN,CAAC;IAEF,IAAI,gBAAgB,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC;QACxC,OAAO,CAAC,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC;QACvD,OAAO,CAAC,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC;IACjD,CAAC;IAED,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC,mCAAmC;KAC/F,CAAC;IAEF,MAAM,SAAS,GAAkB;QAC/B,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;KAC1C,CAAC;IAEF,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,CAAC,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;IACjG,CAAC;IAED,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE;YACL;gBACE,SAAS;gBACT,IAAI;aACL;SACF;KACF,CAAC;AACJ,CAAC"}

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

@@ -1,15 +1,12 @@
-import { StepResult } from '../../steps-types.js';
 export async function handleDefaultAction({ askQuestionCall }) {
     return {
         breakLoop: false,
-        stepResult: StepResult.CONTINUE,
         items: [
             {
-                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                assistant: { type: 'assistant', text: askQuestionCall.args?.message ?? '' },
                 user: {
                     type: 'user',
                     text: "I don't want to start the code generation yet, let's talk a bit more.",
-                    functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
                 },
             },
         ],

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

@@ -1 +1 @@
-{"version":3,"file":"default-action.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/default-action.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGlD,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EAAE,eAAe,EAAsB;IAC/E,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE;gBAC7G,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,uEAAuE;oBAC7E,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;iBACpG;aACF;SACF;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"default-action.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/default-action.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EAAE,eAAe,EAAsB;IAC/E,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE;YACL;gBACE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE;gBAC3E,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,uEAAuE;iBAC9E;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/prompt/steps/step-ask-question/handlers/handle-request-answer-with-image.d.ts

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

package/dist/prompt/steps/step-ask-question/handlers/handle-request-answer-with-image.js

@@ -0,0 +1,86 @@
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+import { putAssistantMessage, putSystemMessage } from '../../../../main/common/content-bus.js';
+import { StepResult } from '../../steps-types.js';
+import { executeStepGenerateImage } from '../../step-generate-image.js';
+export async function handleRequestAnswerWithImage({ askQuestionCall, options, generateImageFn, }) {
+    if (!options.imagen) {
+        putSystemMessage('Image generation is not enabled. Please enable it using --imagen parameter.');
+        return {
+            breakLoop: true,
+            stepResult: StepResult.BREAK,
+            items: [],
+        };
+    }
+    // Get the image generation request from the args
+    const imageGenerationRequest = askQuestionCall.args?.imageGenerationRequest;
+    if (!imageGenerationRequest?.prompt) {
+        putSystemMessage('Image generation request with prompt is required for requestAnswerWithImage action.');
+        return {
+            breakLoop: true,
+            stepResult: StepResult.BREAK,
+            items: [],
+        };
+    }
+    // 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 = {
+        name: 'generateImage',
+        args: {
+            prompt: imageGenerationRequest.prompt,
+            filePath: tempImagePath,
+            width: 1024,
+            height: 1024,
+            explanation: 'Generating image to support the question',
+            ...(imageGenerationRequest.contextImage && { contextImagePath: imageGenerationRequest.contextImage }),
+        },
+    };
+    // 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: true,
+            stepResult: StepResult.BREAK,
+            items: [],
+        };
+    }
+    const response = await fetch(downloadFileCall.args?.downloadUrl);
+    const arrayBuffer = await response.arrayBuffer();
+    const buffer = Buffer.from(arrayBuffer);
+    putAssistantMessage('Here is the image to support the question:', generateImageCall.args, [], [
+        {
+            base64url: buffer.toString('base64'),
+            mediaType: 'image/png',
+        },
+    ]);
+    // Get user's response with the image displayed
+    const userText = await askUserForInput('Your answer', askQuestionCall.args?.content ?? '');
+    // Return the conversation items with the generated image
+    return {
+        breakLoop: false,
+        stepResult: StepResult.CONTINUE,
+        items: [
+            {
+                assistant: {
+                    type: 'assistant',
+                    text: askQuestionCall.args?.content ?? '',
+                    functionCalls: [askQuestionCall, generateImageCall, downloadFileCall],
+                },
+                user: {
+                    type: 'user',
+                    text: userText,
+                    functionResponses: [
+                        {
+                            name: 'askQuestion',
+                            call_id: askQuestionCall.id ?? '',
+                            content: undefined,
+                        },
+                    ],
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=handle-request-answer-with-image.js.map

package/dist/prompt/steps/step-ask-question/handlers/handle-request-answer-with-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-request-answer-with-image.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-request-answer-with-image.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC/F,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAElD,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AAExE,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACjD,eAAe,EACf,OAAO,EACP,eAAe,GACI;IACnB,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACpB,gBAAgB,CAAC,6EAA6E,CAAC,CAAC;QAChG,OAAO;YACL,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,UAAU,CAAC,KAAK;YAC5B,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,iDAAiD;IACjD,MAAM,sBAAsB,GAAG,eAAe,CAAC,IAAI,EAAE,sBAAsB,CAAC;IAC5E,IAAI,CAAC,sBAAsB,EAAE,MAAM,EAAE,CAAC;QACpC,gBAAgB,CAAC,qFAAqF,CAAC,CAAC;QACxG,OAAO;YACL,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,UAAU,CAAC,KAAK;YAC5B,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,uDAAuD;IACvD,MAAM,aAAa,GAAG,uBAAuB,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;IAE9D,uDAAuD;IACvD,MAAM,iBAAiB,GAAG;QACxB,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE;YACJ,MAAM,EAAE,sBAAsB,CAAC,MAAM;YACrC,QAAQ,EAAE,aAAa;YACvB,KAAK,EAAE,IAAI;YACX,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE,0CAA0C;YACvD,GAAG,CAAC,sBAAsB,CAAC,YAAY,IAAI,EAAE,gBAAgB,EAAE,sBAAsB,CAAC,YAAY,EAAE,CAAC;SACtG;KACF,CAAC;IAEF,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,UAAU,EAAE,UAAU,CAAC,KAAK;YAC5B,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IACjE,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;IACjD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAExC,mBAAmB,CACjB,4CAA4C,EAC5C,iBAAiB,CAAC,IAAI,EACtB,EAAE,EACF;QACE;YACE,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACpC,SAAS,EAAE,WAAW;SACvB;KACF,CACF,CAAC;IAEF,+CAA+C;IAC/C,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;IAE3F,yDAAyD;IACzD,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;QAC/B,KAAK,EAAE;YACL;gBACE,SAAS,EAAE;oBACT,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;oBACzC,aAAa,EAAE,CAAC,eAAe,EAAE,iBAAiB,EAAE,gBAAgB,CAAC;iBACtE;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ;oBACd,iBAAiB,EAAE;wBACjB;4BACE,IAAI,EAAE,aAAa;4BACnB,OAAO,EAAE,eAAe,CAAC,EAAE,IAAI,EAAE;4BACjC,OAAO,EAAE,SAAS;yBACnB;qBACF;iBACF;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/prompt/steps/step-ask-question/handlers/handle-send-message-with-image.d.ts

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

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

@@ -0,0 +1,97 @@
+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({ askQuestionCall, options, generateContentFn, prompt, generateImageFn, }) {
+    if (!options.imagen) {
+        putSystemMessage('Image generation is not enabled. Please enable it using --imagen parameter.');
+        return {
+            breakLoop: true,
+            items: [],
+        };
+    }
+    const [sendMessageWithImageCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: askQuestionCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Yes, you can generate an image.',
+        },
+    ], getFunctionDefs(), 'sendMessageWithImage', 0.7, true, options));
+    if (!sendMessageWithImageCall) {
+        return {
+            breakLoop: true,
+            items: [],
+        };
+    }
+    // Get the image generation request from the args
+    const imageGenerationRequest = sendMessageWithImageCall.args;
+    if (!imageGenerationRequest?.prompt) {
+        putSystemMessage('Image generation request with prompt is required for sendMessageWithImage action.');
+        return {
+            breakLoop: true,
+            items: [],
+        };
+    }
+    // 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 }),
+        },
+    };
+    // 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: true,
+            items: [],
+        };
+    }
+    const response = await fetch(downloadFileCall.args?.downloadUrl);
+    const arrayBuffer = await response.arrayBuffer();
+    const buffer = Buffer.from(arrayBuffer);
+    const imageData = {
+        base64url: buffer.toString('base64'),
+        mediaType: 'image/png',
+    };
+    putAssistantMessage('Here is the image to support the question:', generateImageCall.args, [], [imageData]);
+    // Get user's response with the image displayed
+    const inputResponse = await askUserForInput('Your answer', askQuestionCall.args?.message ?? '');
+    if (inputResponse.options?.aiService) {
+        options.aiService = inputResponse.options.aiService;
+    }
+    // Return the conversation items with the generated image included in the context
+    return {
+        breakLoop: false,
+        items: [
+            {
+                assistant: {
+                    type: 'assistant',
+                    text: askQuestionCall.args?.message ?? '',
+                    functionCalls: [generateImageCall],
+                },
+                user: {
+                    type: 'user',
+                    text: inputResponse.answer,
+                    images: [imageData],
+                    functionResponses: [{ name: generateImageCall.name, call_id: generateImageCall.id, content: undefined }],
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=handle-send-message-with-image.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-send-message-with-image.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-send-message-with-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;AAG/D,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAC/C,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,wBAAwB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CACzD;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,sBAAsB,EACtB,GAAG,EACH,IAAI,EACJ,OAAO,CACR,CAAyD,CAAC;IAE3D,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAC9B,OAAO;YACL,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,iDAAiD;IACjD,MAAM,sBAAsB,GAAG,wBAAwB,CAAC,IAAI,CAAC;IAC7D,IAAI,CAAC,sBAAsB,EAAE,MAAM,EAAE,CAAC;QACpC,gBAAgB,CAAC,mFAAmF,CAAC,CAAC;QACtG,OAAO;YACL,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,uDAAuD;IACvD,MAAM,aAAa,GAAG,uBAAuB,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;IAE9D,uDAAuD;IACvD,MAAM,iBAAiB,GAAG;QACxB,EAAE,EAAE,iBAAiB,GAAG,wBAAwB,CAAC,EAAE;QACnD,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE;YACJ,MAAM,EAAE,sBAAsB,CAAC,MAAM;YACrC,QAAQ,EAAE,aAAa;YACvB,KAAK,EAAE,IAAI;YACX,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE,0CAA0C;YACvD,GAAG,CAAC,sBAAsB,CAAC,YAAY,IAAI,EAAE,gBAAgB,EAAE,sBAAsB,CAAC,YAAY,EAAE,CAAC;SACtG;KACF,CAAC;IAEF,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,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IACjE,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;IACjD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACxC,MAAM,SAAS,GAAG;QAChB,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACpC,SAAS,EAAE,WAAW;KACd,CAAC;IAEX,mBAAmB,CAAC,4CAA4C,EAAE,iBAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAE3G,+CAA+C;IAC/C,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;IAChG,IAAI,aAAa,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC;QACrC,OAAO,CAAC,SAAS,GAAG,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,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-send-message.d.ts

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

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

@@ -0,0 +1,20 @@
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+export async function handleSendMessage({ askQuestionCall, options }) {
+    const response = await askUserForInput('Your answer', askQuestionCall.args?.message ?? '');
+    if (response.options?.aiService) {
+        options.aiService = response.options.aiService;
+    }
+    return {
+        breakLoop: false,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.message ?? '' },
+                user: {
+                    type: 'user',
+                    text: response.answer,
+                },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=handle-send-message.js.map

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

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

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

@@ -0,0 +1,2 @@
+"use strict";
+//# sourceMappingURL=handler-utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handler-utils.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handler-utils.ts"],"names":[],"mappings":""}

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

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

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

@@ -1,8 +1,25 @@
+import { getSourceCodeTree, parseSourceCodeTree } from '../../../../files/source-code-tree.js';
 import { putSystemMessage } from '../../../../main/common/content-bus.js';
-import { StepResult } from '../../steps-types.js';
-import { getSourceCodeResponse } from '../../steps-utils.js';
-export async function handleRemoveFilesFromContext({ askQuestionCall, prompt, }) {
-    const filesToRemove = askQuestionCall.args?.removeFilesFromContext ?? [];
+import { getFunctionDefs } from '../../../function-calling.js';
+export async function handleRemoveFilesFromContext({ askQuestionCall, generateContentFn, prompt, options, }) {
+    const [removeFilesFromContextCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: askQuestionCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Yes, you can request the removal of files from context.',
+        },
+    ], getFunctionDefs(), 'removeFilesFromContext', 0.7, true, options));
+    if (!removeFilesFromContextCall) {
+        return {
+            breakLoop: true,
+            items: [],
+        };
+    }
+    const filesToRemove = removeFilesFromContextCall.args?.filePaths ?? [];
     let userText = '';
     if (filesToRemove.length > 0) {
         removeFileContentsFromPrompt(prompt, filesToRemove);
@@ -14,30 +31,55 @@ export async function handleRemoveFilesFromContext({ askQuestionCall, prompt, })
     }
     return {
         breakLoop: false,
-        stepResult: StepResult.CONTINUE,
         items: [
             {
-                assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },
+                assistant: {
+                    type: 'assistant',
+                    text: askQuestionCall.args?.message ?? '',
+                    functionCalls: [removeFilesFromContextCall],
+                },
                 user: {
                     type: 'user',
                     text: userText,
-                    functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id ?? '', content: undefined }],
+                    functionResponses: [
+                        { name: 'removeFilesFromContext', call_id: removeFilesFromContextCall.id, content: undefined },
+                    ],
                 },
             },
         ],
     };
 }
 function removeFileContentsFromPrompt(prompt, filesToRemove) {
-    const response = getSourceCodeResponse(prompt);
-    if (!response || !response.content) {
-        throw new Error('Could not find source code response');
+    // Find all source code responses in the prompt history
+    const sourceCodeResponses = prompt.filter((item) => item.type === 'user' && item.functionResponses?.some((fr) => fr.name === 'getSourceCode'));
+    if (sourceCodeResponses.length === 0) {
+        console.warn('No source code responses found in prompt history');
+        return;
     }
-    const contentObj = JSON.parse(response.content);
-    filesToRemove.forEach((file) => {
-        if (contentObj[file]) {
-            delete contentObj[file].content;
+    sourceCodeResponses.forEach((response) => {
+        const sourceCodeResponse = response.functionResponses?.find((fr) => fr.name === 'getSourceCode');
+        if (!sourceCodeResponse || !sourceCodeResponse.content) {
+            console.warn('Skipping invalid source code response');
+            return;
+        }
+        try {
+            const contentObj = parseSourceCodeTree(JSON.parse(sourceCodeResponse.content));
+            let modifiedFiles = 0;
+            filesToRemove.forEach((file) => {
+                if (contentObj[file] && 'content' in contentObj[file]) {
+                    contentObj[file].content = null;
+                    modifiedFiles++;
+                }
+            });
+            // Only update if files were actually modified
+            if (modifiedFiles > 0) {
+                sourceCodeResponse.content = JSON.stringify(getSourceCodeTree(contentObj));
+                console.log(`Removed content from ${modifiedFiles} files in source code response`);
+            }
+        }
+        catch (error) {
+            console.error('Error processing source code response:', error);
         }
     });
-    response.content = JSON.stringify(contentObj);
 }
 //# sourceMappingURL=remove-files-from-context.js.map