genaicode 0.8.6 → 0.9.1

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

@@ -0,0 +1,157 @@
+import { executor as executeCreateFile } from '../../../../operations/create-file/create-file-executor.js';
+import { getSourceCode } from '../../../../files/read-files.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+import { putSystemMessage, putUserMessage } from '../../../../main/common/content-bus.js';
+import { refreshFiles } from '../../../../files/find-files.js';
+export const handleCreateFile = async ({ askQuestionCall, options, prompt, generateContentFn, }) => {
+    putSystemMessage('File creation requested.', askQuestionCall);
+    // First confirmation: Ask user if they want to proceed with content generation
+    const generateConfirmation = await askUserForConfirmationWithAnswer('The assistant suggests generating content for a new file. Do you want to proceed with content generation?', 'Generate content', 'Cancel content generation', false, options);
+    if (!generateConfirmation.confirmed) {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Canceling file creation.' + (generateConfirmation.answer ? ` ${generateConfirmation.answer}` : ''),
+                    },
+                },
+            ],
+        };
+    }
+    if (generateConfirmation.answer) {
+        putUserMessage(generateConfirmation.answer);
+    }
+    putSystemMessage('Content generation started.', askQuestionCall);
+    const [createFileCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: askQuestionCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Ok, please provide the file creation using `createFile` function. Please remember to use absolute file path.' +
+                (generateConfirmation.answer ? ` ${generateConfirmation.answer}` : ''),
+        },
+    ], getFunctionDefs(), 'createFile', options.temperature ?? 0.7, ModelType.DEFAULT, options));
+    if (!createFileCall || !createFileCall.args) {
+        return {
+            breakLoop: true,
+            items: [],
+        };
+    }
+    const { filePath } = createFileCall.args;
+    const file = getSourceCode({ filterPaths: [filePath], forceAll: true }, options)[filePath];
+    if (file && 'content' in file && file.content) {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'The file already exists. Please use updateFile function to modify existing files.',
+                    },
+                },
+            ],
+        };
+    }
+    putSystemMessage(`The assistant has generated content for new file ${filePath}.`, {
+        name: createFileCall.name,
+        args: createFileCall.args,
+    });
+    // Second confirmation: Ask user if they want to proceed with file creation
+    const applyConfirmation = await askUserForConfirmationWithAnswer('The assistant has generated the file content. Do you want to create the file?', 'Create file', 'Cancel file creation', false, options);
+    if (!applyConfirmation.confirmed) {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                        functionCalls: [createFileCall],
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Rejecting file creation.' + (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : ''),
+                        functionResponses: [
+                            {
+                                name: 'createFile',
+                                call_id: createFileCall.id,
+                                content: '',
+                            },
+                        ],
+                    },
+                },
+            ],
+        };
+    }
+    try {
+        await executeCreateFile(createFileCall.args, options);
+        refreshFiles();
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                        functionCalls: [createFileCall],
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Accepting file creation.' + (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : ''),
+                        functionResponses: [
+                            {
+                                name: 'createFile',
+                                call_id: createFileCall.id,
+                                content: '',
+                            },
+                        ],
+                    },
+                },
+            ],
+        };
+    }
+    catch (error) {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args.message,
+                        functionCalls: [createFileCall],
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Accepting file creation.' +
+                            (applyConfirmation.answer ? ` ${applyConfirmation.answer}` : '') +
+                            ' Unfortunately, the file creation failed: ' +
+                            (error instanceof Error ? error.message : 'unknown error'),
+                        functionResponses: [
+                            {
+                                name: 'createFile',
+                                call_id: createFileCall.id,
+                                content: '',
+                            },
+                        ],
+                    },
+                },
+            ],
+        };
+    }
+};
+//# sourceMappingURL=handle-create-file.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"handle-create-file.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-create-file.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,IAAI,iBAAiB,EAAE,MAAM,4DAA4D,CAAC;AAE3G,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAC3F,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAC1F,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAE/D,MAAM,CAAC,MAAM,gBAAgB,GAAkB,KAAK,EAAE,EACpD,eAAe,EACf,OAAO,EACP,MAAM,EACN,iBAAiB,GAClB,EAAyB,EAAE;IAC1B,gBAAgB,CAAC,0BAA0B,EAAE,eAAe,CAAC,CAAC;IAE9D,+EAA+E;IAC/E,MAAM,oBAAoB,GAAG,MAAM,gCAAgC,CACjE,2GAA2G,EAC3G,kBAAkB,EAClB,2BAA2B,EAC3B,KAAK,EACL,OAAO,CACR,CAAC;IAEF,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,CAAC;QACpC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;qBACpC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,0BAA0B,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC1G;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC;QAChC,cAAc,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,gBAAgB,CAAC,6BAA6B,EAAE,eAAe,CAAC,CAAC;IAEjE,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAC/C;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SAC1C;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,8GAA8G;gBAC9G,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzE;KACF,EACD,eAAe,EAAE,EACjB,YAAY,EACZ,OAAO,CAAC,WAAW,IAAI,GAAG,EAC1B,SAAS,CAAC,OAAO,EACjB,OAAO,CACR,CAA+C,CAAC;IAEjD,IAAI,CAAC,cAAc,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QAC5C,OAAO;YACL,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,EAAE,QAAQ,EAAE,GAAG,cAAc,CAAC,IAAI,CAAC;IAEzC,MAAM,IAAI,GAAG,aAAa,CAAC,EAAE,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC;IAE3F,IAAI,IAAI,IAAI,SAAS,IAAI,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAC9C,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;qBACpC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,mFAAmF;qBAC1F;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,oDAAoD,QAAQ,GAAG,EAAE;QAChF,IAAI,EAAE,cAAc,CAAC,IAAI;QACzB,IAAI,EAAE,cAAc,CAAC,IAAI;KAC1B,CAAC,CAAC;IAEH,2EAA2E;IAC3E,MAAM,iBAAiB,GAAG,MAAM,gCAAgC,CAC9D,+EAA+E,EAC/E,aAAa,EACb,sBAAsB,EACtB,KAAK,EACL,OAAO,CACR,CAAC;IAEF,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,CAAC;QACjC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;wBACnC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,0BAA0B,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACnG,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,iBAAiB,CAAC,cAAc,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACtD,YAAY,EAAE,CAAC;QACf,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;wBACnC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,0BAA0B,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACnG,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,eAAe,CAAC,IAAK,CAAC,OAAO;wBACnC,aAAa,EAAE,CAAC,cAAc,CAAC;qBAChC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EACF,0BAA0B;4BAC1B,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;4BAChE,4CAA4C;4BAC5C,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;wBAC5D,iBAAiB,EAAE;4BACjB;gCACE,IAAI,EAAE,YAAY;gCAClB,OAAO,EAAE,cAAc,CAAC,EAAE;gCAC1B,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC,CAAC"}

package/dist/prompt/steps/step-ask-question/handlers/handle-end-conversation.d.ts

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

package/dist/prompt/steps/step-ask-question/handlers/handle-end-conversation.js

@@ -0,0 +1,30 @@
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';
+export async function handleEndConversation({ askQuestionCall, options }) {
+    const userConfirmation = await askUserForConfirmationWithAnswer(askQuestionCall.args?.message ?? 'Do you want to end the conversation?', 'End conversation', 'Continue conversation', true, options);
+    if (!userConfirmation.confirmed) {
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: { type: 'assistant', text: askQuestionCall.args?.message ?? '' },
+                    user: {
+                        type: 'user',
+                        text: userConfirmation.answer || 'Declined. Please continue the conversation.',
+                    },
+                },
+            ],
+        };
+    }
+    putSystemMessage('Assistant requested to stop code generation. Exiting...');
+    return {
+        breakLoop: true,
+        items: [
+            {
+                assistant: { type: 'assistant', text: askQuestionCall.args?.message ?? '' },
+                user: { type: 'user', text: 'Code generation cancelled.' },
+            },
+        ],
+    };
+}
+//# sourceMappingURL=handle-end-conversation.js.map

package/dist/prompt/steps/step-ask-question/handlers/handle-end-conversation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-end-conversation.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-end-conversation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,gCAAgC,EAAE,MAAM,yCAAyC,CAAC;AAG3F,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,EAAE,eAAe,EAAE,OAAO,EAAsB;IAC1F,MAAM,gBAAgB,GAAG,MAAM,gCAAgC,CAC7D,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,sCAAsC,EACvE,kBAAkB,EAClB,uBAAuB,EACvB,IAAI,EACJ,OAAO,CACR,CAAC;IAEF,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAChC,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;IAED,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/handle-genaicode-help.d.ts

@@ -0,0 +1,8 @@
+import { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';
+/**
+ * Handler for the help action. This handler is responsible for:
+ * 1. Reading the documentation file
+ * 2. Finding relevant sections based on the query
+ * 3. Returning formatted results to the user
+ */
+export declare function handleGenaicodeHelp({ generateContentFn, askQuestionCall, prompt, options, }: ActionHandlerProps): Promise<ActionResult>;

package/dist/prompt/steps/step-ask-question/handlers/handle-genaicode-help.js

@@ -0,0 +1,91 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { putAssistantMessage } from '../../../../main/common/content-bus.js';
+import { askUserForInput } from '../../../../main/common/user-actions.js';
+import { GENAICODE_HELP_DOCUMENT } from '../../../../help-docs/genaicode-help-document.js';
+/**
+ * Handler for the help action. This handler is responsible for:
+ * 1. Reading the documentation file
+ * 2. Finding relevant sections based on the query
+ * 3. Returning formatted results to the user
+ */
+export async function handleGenaicodeHelp({ generateContentFn, askQuestionCall, prompt, options, }) {
+    try {
+        // Get help content
+        const [genaicodeHelpCall] = await getGenaicodeHelpCall(generateContentFn, prompt);
+        if (!genaicodeHelpCall?.args) {
+            return {
+                breakLoop: false,
+                items: [
+                    {
+                        assistant: {
+                            type: 'assistant',
+                            text: askQuestionCall.args?.message || 'How can I help you?',
+                        },
+                        user: {
+                            type: 'user',
+                            text: 'Failed to get valid genaicodeHelp request',
+                        },
+                    },
+                ],
+            };
+        }
+        putAssistantMessage(genaicodeHelpCall.args?.message ?? '', genaicodeHelpCall.args);
+        const response = await askUserForInput('Your answer', genaicodeHelpCall.args?.message ?? '', options);
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: genaicodeHelpCall.args?.message ?? '',
+                        functionCalls: [genaicodeHelpCall],
+                    },
+                    user: {
+                        type: 'user',
+                        text: response.answer,
+                        functionResponses: [{ name: 'genaicodeHelp', call_id: genaicodeHelpCall.id, content: '' }],
+                    },
+                },
+            ],
+        };
+    }
+    catch (error) {
+        console.error('Error in help handler:', error);
+        return {
+            breakLoop: false,
+            items: [
+                {
+                    assistant: {
+                        type: 'assistant',
+                        text: askQuestionCall.args?.message || 'How can I help you?',
+                    },
+                    user: {
+                        type: 'user',
+                        text: 'Sorry, I encountered an error while retrieving the help documentation. Please try again or check if the documentation file exists.',
+                    },
+                },
+            ],
+        };
+    }
+}
+async function getGenaicodeHelpCall(generateContentFn, prompt) {
+    const [genaicodeHelpCall] = (await generateContentFn([
+        {
+            type: 'assistant',
+            text: 'I can help you with questions about GenAIcode, but first I need access to genaicode documentation, could you please provide it?',
+        },
+        {
+            type: 'user',
+            text: `Of course, here is the documentation:
+
+      \`\`\`
+      ${GENAICODE_HELP_DOCUMENT}
+      \`\`\`
+`,
+        },
+        ...prompt,
+    ], getFunctionDefs(), 'genaicodeHelp', 0.7, ModelType.CHEAP));
+    return [genaicodeHelpCall];
+}
+//# sourceMappingURL=handle-genaicode-help.js.map

package/dist/prompt/steps/step-ask-question/handlers/handle-genaicode-help.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-genaicode-help.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-ask-question/handlers/handle-genaicode-help.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,uBAAuB,EAAE,MAAM,kDAAkD,CAAC;AAE3F;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EACxC,iBAAiB,EACjB,eAAe,EACf,MAAM,EACN,OAAO,GACY;IACnB,IAAI,CAAC;QACH,mBAAmB;QACnB,MAAM,CAAC,iBAAiB,CAAC,GAAG,MAAM,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAElF,IAAI,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC;YAC7B,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE;oBACL;wBACE,SAAS,EAAE;4BACT,IAAI,EAAE,WAAW;4BACjB,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,IAAI,qBAAqB;yBAC7D;wBACD,IAAI,EAAE;4BACJ,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,2CAA2C;yBAClD;qBACF;iBACF;aACF,CAAC;QACJ,CAAC;QAED,mBAAmB,CAAC,iBAAiB,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEnF,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,iBAAiB,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAEtG,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE;gBACL;oBACE,SAAS,EAAE;wBACT,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,iBAAiB,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;wBAC3C,aAAa,EAAE,CAAC,iBAAiB,CAAC;qBACnC;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,QAAQ,CAAC,MAAM;wBACrB,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,iBAAiB,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;qBAC3F;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QAC/C,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,qBAAqB;qBAC7D;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,oIAAoI;qBAC3I;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,iBAA0D,EAAE,MAAoB;IAClH,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAClD;QACE;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,iIAAiI;SACxI;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE;;;QAGN,uBAAuB;;CAE9B;SACM;QACD,GAAG,MAAM;KACV,EACD,eAAe,EAAE,EACjB,eAAe,EACf,GAAG,EACH,SAAS,CAAC,KAAK,CAChB,CAAkD,CAAC;IAEpD,OAAO,CAAC,iBAAiB,CAAC,CAAC;AAC7B,CAAC"}

package/dist/prompt/steps/step-ask-question/handlers/handle-generate-image.d.ts

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

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({ askQuestionCall, 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,7 +12,7 @@ export async function handleSendMessageWithImage({ askQuestionCall, options, gen
             items: [],
         };
     }
-    const [sendMessageWithImageCall] = (await generateContentFn([
+    const [generateImageCall] = (await generateContentFn([
         ...prompt,
         {
             type: 'assistant',
@@ -20,16 +22,15 @@ export async function handleSendMessageWithImage({ askQuestionCall, options, gen
             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({ askQuestionCall, options, gen
         };
     }
     // 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({ askQuestionCall, options, gen
             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', askQuestionCall.args?.message ?? '');
-    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,
@@ -94,4 +93,4 @@ export async function handleSendMessageWithImage({ askQuestionCall, options, gen
         ],
     };
 }
-//# 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