genaicode 0.0.37 → 0.2.0

package/dist/prompt/prompt-service-ask-question.test.js

@@ -0,0 +1,142 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { promptService } from './prompt-service.js';
+import * as vertexAi from '../ai-service/vertex-ai.js';
+import * as prompts from '@inquirer/prompts';
+import { CancelablePromise } from '@inquirer/type';
+import '../cli/cli-params.js';
+import '../files/read-files.js';
+import '../files/find-files.js';
+import { getCodeGenPrompt } from './prompt-codegen.js';
+vi.mock('../ai-service/vertex-ai.js', () => ({ generateContent: vi.fn() }));
+vi.mock('@inquirer/prompts', () => ({
+    input: vi.fn(),
+}));
+vi.mock('../cli/cli-params.js', () => ({
+    requireExplanations: false,
+    considerAllFiles: false,
+    dependencyTree: false,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    disableContextOptimization: false,
+    vision: false,
+    imagen: false,
+    temperature: 0.7,
+    cheap: false,
+    askQuestion: true,
+}));
+// Mock find-files module
+vi.mock('../files/find-files.js', () => ({
+    getSourceFiles: () => [],
+    getImageAssetFiles: () => [],
+}));
+// Mock read-files module
+vi.mock('../files/read-files.js', () => ({
+    getSourceCode: () => ({}),
+    getImageAssets: vi.fn(() => ({})),
+}));
+vi.mock('../main/config.js', () => ({
+    rootDir: '/mocked/root/dir',
+    rcConfig: {
+        rootDir: '.',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+    importantContext: {},
+}));
+const GENERATE_CONTENT_FNS = { 'vertex-ai': vertexAi.generateContent };
+const GENERATE_IMAGE_FNS = {};
+describe('promptService with askQuestion', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+        console.log = vi.fn();
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should handle askQuestion when enabled', async () => {
+        const mockAskQuestionCall = [
+            {
+                name: 'askQuestion',
+                args: {
+                    content: 'Do you want to proceed with code generation?',
+                    shouldPrompt: true,
+                    promptNecessity: 80,
+                    stopCodegen: false,
+                },
+            },
+        ];
+        const mockAskQuestionCall2 = [
+            {
+                name: 'askQuestion',
+                args: {
+                    content: 'Ok lets go',
+                    shouldPrompt: false,
+                    promptNecessity: 80,
+                    stopCodegen: false,
+                },
+            },
+        ];
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [],
+                    contextPaths: [],
+                    explanation: 'No updates needed',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent)
+            .mockResolvedValueOnce(mockAskQuestionCall)
+            .mockResolvedValueOnce(mockAskQuestionCall2)
+            .mockResolvedValueOnce(mockCodegenSummary);
+        vi.mocked(prompts.input).mockImplementationOnce(() => CancelablePromise.resolve('Yes'));
+        await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ aiService: 'vertex-ai', interactive: true, askQuestion: true }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+        expect(prompts.input).toHaveBeenCalledWith({ message: 'Your answer' });
+        expect(console.log).toHaveBeenCalledWith('Assistant asks:', expect.any(Object));
+        expect(console.log).toHaveBeenCalledWith('The question was answered');
+    });
+    it('should stop code generation when askQuestion returns stopCodegen: true', async () => {
+        const mockAskQuestionCall = [
+            {
+                name: 'askQuestion',
+                args: {
+                    content: 'Stopping code generation as requested.',
+                    shouldPrompt: false,
+                    promptNecessity: 100,
+                    stopCodegen: true,
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockAskQuestionCall);
+        const result = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ aiService: 'vertex-ai', interactive: true }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(1);
+        expect(result).toEqual([]);
+        expect(console.log).toHaveBeenCalledWith('Assistant requested to stop code generation. Exiting...');
+    });
+    it('should proceed with code generation when askQuestion returns shouldPrompt: false', async () => {
+        const mockAskQuestionCall = [];
+        const mockCodegenSummary = [
+            {
+                name: 'codegenSummary',
+                args: {
+                    fileUpdates: [],
+                    contextPaths: [],
+                    explanation: 'No updates needed',
+                },
+            },
+        ];
+        vi.mocked(vertexAi.generateContent)
+            .mockResolvedValueOnce(mockAskQuestionCall)
+            .mockResolvedValueOnce(mockCodegenSummary);
+        await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt({ aiService: 'vertex-ai', interactive: true }));
+        expect(vertexAi.generateContent).toHaveBeenCalledTimes(2);
+        expect(prompts.input).not.toHaveBeenCalled();
+        expect(console.log).toHaveBeenCalledWith('Assistant did not ask a question. Proceeding with code generation.');
+    });
+});
+//# sourceMappingURL=prompt-service-ask-question.test.js.map

package/dist/prompt/prompt-service-ask-question.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-service-ask-question.test.js","sourceRoot":"","sources":["../../src/prompt/prompt-service-ask-question.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AAEvD,OAAO,KAAK,OAAO,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,sBAAsB,CAAC;AAC9B,OAAO,wBAAwB,CAAC;AAChC,OAAO,wBAAwB,CAAC;AAChC,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAGvD,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;CACf,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,mBAAmB,EAAE,KAAK;IAC1B,gBAAgB,EAAE,KAAK;IACvB,cAAc,EAAE,KAAK;IACrB,cAAc,EAAE,KAAK;IACrB,eAAe,EAAE,KAAK;IACtB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,KAAK;IAC3B,aAAa,EAAE,KAAK;IACpB,aAAa,EAAE,KAAK;IACpB,0BAA0B,EAAE,KAAK;IACjC,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,WAAW,EAAE,IAAI;CAClB,CAAC,CAAC,CAAC;AAEJ,yBAAyB;AACzB,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;CAC7B,CAAC,CAAC,CAAC;AAEJ,yBAAyB;AACzB,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;IACzB,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;CAClC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE;QACR,OAAO,EAAE,GAAG;QACZ,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;IACD,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,MAAM,oBAAoB,GAAG,EAAE,WAAW,EAAE,QAAQ,CAAC,eAAe,EAGnE,CAAC;AACF,MAAM,kBAAkB,GAAG,EAA+C,CAAC;AAE3E,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,mBAAmB,GAAG;YAC1B;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,OAAO,EAAE,8CAA8C;oBACvD,YAAY,EAAE,IAAI;oBAClB,eAAe,EAAE,EAAE;oBACnB,WAAW,EAAE,KAAK;iBACnB;aACF;SACF,CAAC;QACF,MAAM,oBAAoB,GAAG;YAC3B;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,OAAO,EAAE,YAAY;oBACrB,YAAY,EAAE,KAAK;oBACnB,eAAe,EAAE,EAAE;oBACnB,WAAW,EAAE,KAAK;iBACnB;aACF;SACF,CAAC;QACF,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,EAAE;oBACf,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,mBAAmB;iBACjC;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC;aAChC,qBAAqB,CAAC,mBAAmB,CAAC;aAC1C,qBAAqB,CAAC,oBAAoB,CAAC;aAC3C,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAE7C,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAC7C,GAAG,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAA8B,CACpE,CAAC;QAEF,MAAM,aAAa,CACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CACnF,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;QACvE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAChF,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,mBAAmB,GAAG;YAC1B;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,OAAO,EAAE,wCAAwC;oBACjD,YAAY,EAAE,KAAK;oBACnB,eAAe,EAAE,GAAG;oBACpB,WAAW,EAAE,IAAI;iBAClB;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;QAE/E,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAChE,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,yDAAyD,CAAC,CAAC;IACtG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,mBAAmB,GAAmB,EAAE,CAAC;QAC/C,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,EAAE;oBACf,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,mBAAmB;iBACjC;aACF;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC;aAChC,qBAAqB,CAAC,mBAAmB,CAAC;aAC1C,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAE7C,MAAM,aAAa,CACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAChE,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,oEAAoE,CAAC,CAAC;IACjH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/prompt/prompt-service.d.ts

@@ -0,0 +1,28 @@
+import { FunctionCall, GenerateContentFunction } from '../ai-service/common.js';
+import { AiServiceType, ImagenType } from '../main/codegen-types.js';
+interface GenerateImageFunction {
+    (prompt: string, contextImagePath: string | undefined, size: {
+        width: number;
+        height: number;
+    }, cheap: boolean): Promise<string>;
+}
+import { CodegenPrompt } from './prompt-codegen.js';
+/** A function that communicates with model using */
+export declare function promptService(generateContentFns: Record<AiServiceType, GenerateContentFunction>, generateImageFns: Record<ImagenType, GenerateImageFunction>, codegenPrompt: CodegenPrompt): Promise<FunctionCall[]>;
+export type PromptMessages = ReturnType<typeof prepareMessages>;
+/**
+ * Function to prepare messages for AI services
+ */
+declare function prepareMessages(codegen: CodegenPrompt): {
+    suggestSourceCode: string;
+    requestSourceCode: string;
+    suggestImageAssets: string;
+    requestImageAssets: string;
+    prompt: string;
+    sourceCode: string;
+    contextSourceCode: (paths: string[]) => string;
+    imageAssets: string;
+    partialPromptTemplate(path: string): string;
+    invalidFunctionCall: string;
+};
+export {};

package/dist/prompt/prompt-service.js

@@ -0,0 +1,163 @@
+import assert from 'node:assert';
+import fs from 'fs';
+import mime from 'mime-types';
+import { getSystemPrompt } from './systemprompt.js';
+import { functionDefs } from './function-calling.js';
+import { getSourceCode, getImageAssets } from '../files/read-files.js';
+import { importantContext } from '../main/config.js';
+import { executeStepAskQuestion } from './steps/step-ask-question.js';
+import { validateAndRecoverSingleResult } from './steps/step-validate-recover.js';
+import { executeStepVerifyPatch } from './steps/step-verify-patch.js';
+import { executeStepGenerateImage } from './steps/step-generate-image.js';
+import { StepResult } from './steps/steps-types.js';
+/** A function that communicates with model using */
+export async function promptService(generateContentFns, generateImageFns, codegenPrompt) {
+    const { vision } = codegenPrompt.options;
+    const messages = prepareMessages(codegenPrompt);
+    const generateContentFn = (...args) => generateContentFns[codegenPrompt.options.aiService](...args);
+    const generateImageFn = (...args) => generateImageFns[codegenPrompt.options.imagen](...args);
+    // First stage: generate code generation summary, which should not take a lot of output tokens
+    const getSourceCodeRequest = { name: 'getSourceCode' };
+    const prompt = [
+        { type: 'systemPrompt', systemPrompt: getSystemPrompt(codegenPrompt.options) },
+        { type: 'user', text: messages.suggestSourceCode },
+        { type: 'assistant', text: messages.requestSourceCode, functionCalls: [getSourceCodeRequest] },
+    ];
+    const getSourceCodeResponse = {
+        type: 'user',
+        functionResponses: [{ name: 'getSourceCode', content: messages.sourceCode }],
+        cache: true,
+    };
+    prompt.push(getSourceCodeResponse);
+    if (vision) {
+        prompt.slice(-1)[0].text = messages.suggestImageAssets;
+        prompt.push({ type: 'assistant', text: messages.requestImageAssets, functionCalls: [{ name: 'getImageAssets' }] }, { type: 'user', functionResponses: [{ name: 'getImageAssets', content: messages.imageAssets }] });
+    }
+    prompt.slice(-1)[0].text = messages.prompt;
+    // Execute the ask question step
+    if (codegenPrompt.options.askQuestion !== false &&
+        codegenPrompt.options.interactive &&
+        (await executeStepAskQuestion(generateContentFn, prompt, functionDefs, codegenPrompt.options.temperature ?? 0.7, codegenPrompt.options.cheap ?? false, messages, codegenPrompt.options)) === StepResult.BREAK) {
+        return [];
+    }
+    else if (codegenPrompt.options.askQuestion === false) {
+        console.log('Ask question is not enabled.');
+    }
+    const baseRequest = [
+        prompt,
+        functionDefs,
+        'codegenSummary',
+        codegenPrompt.options.temperature ?? 0.7,
+        codegenPrompt.options.cheap ?? false,
+        codegenPrompt.options,
+    ];
+    let baseResult = await generateContentFn(...baseRequest);
+    let codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');
+    if (codegenSummaryRequest) {
+        // Second stage: for each file request the actual code updates
+        console.log('Received codegen summary, will collect partial updates', codegenSummaryRequest.args);
+        baseResult = await validateAndRecoverSingleResult(baseRequest, baseResult, messages, generateContentFn);
+        codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');
+        // Sometimes the result happens to be a string
+        assert(Array.isArray(codegenSummaryRequest?.args?.fileUpdates), 'fileUpdates is not an array');
+        assert(Array.isArray(codegenSummaryRequest?.args.contextPaths), 'contextPaths is not an array');
+        if (!codegenPrompt.options.disableContextOptimization) {
+            console.log('Optimize with context paths.');
+            // Monkey patch the initial getSourceCode, do not send parts of source code that are consider irrelevant
+            getSourceCodeRequest.args = {
+                filePaths: [
+                    ...codegenSummaryRequest.args.fileUpdates.map((file) => file.path),
+                    ...codegenSummaryRequest.args.contextPaths,
+                    ...(importantContext.files ?? []),
+                ],
+            };
+            getSourceCodeResponse.functionResponses.find((item) => item.name === 'getSourceCode').content =
+                messages.contextSourceCode(getSourceCodeRequest.args?.filePaths);
+        }
+        // Store the first stage response entirely in conversation history
+        prompt.push({ type: 'assistant', functionCalls: baseResult });
+        prompt.push({
+            type: 'user',
+            functionResponses: baseResult.map((call) => ({ name: call.name, call_id: call.id })),
+            cache: true,
+        });
+        const result = [];
+        for (const file of codegenSummaryRequest.args.fileUpdates) {
+            console.log('Collecting partial update for: ' + file.path + ' using tool: ' + file.updateToolName);
+            console.log('- Prompt:', file.prompt);
+            console.log('- Temperature', file.temperature);
+            console.log('- Cheap', file.cheap);
+            if (vision) {
+                console.log('- Context image assets', file.contextImageAssets);
+            }
+            // this is needed, otherwise we will get an error
+            if (prompt.slice(-1)[0].type === 'user') {
+                prompt.slice(-1)[0].text = file.prompt ?? messages.partialPromptTemplate(file.path);
+            }
+            else {
+                prompt.push({ type: 'user', text: file.prompt ?? messages.partialPromptTemplate(file.path) });
+            }
+            if (vision && file.contextImageAssets) {
+                prompt.slice(-1)[0].images = file.contextImageAssets.map((path) => ({
+                    path,
+                    base64url: fs.readFileSync(path, 'base64'),
+                    mediaType: mime.lookup(path) || '',
+                }));
+            }
+            const partialRequest = [
+                prompt,
+                functionDefs,
+                file.updateToolName,
+                file.temperature ?? codegenPrompt.options.temperature,
+                file.cheap === true,
+                codegenPrompt.options,
+            ];
+            let partialResult = await generateContentFn(...partialRequest);
+            // Validate if function call is compliant with the schema
+            partialResult = await validateAndRecoverSingleResult(partialRequest, partialResult, messages, generateContentFn);
+            // Handle image generation requests
+            const generateImageCall = partialResult.find((call) => call.name === 'generateImage');
+            if (generateImageCall) {
+                partialResult.push(await executeStepGenerateImage(generateImageFn, generateImageCall));
+            }
+            // Verify if patchFile is one of the functions called, and test if patch is valid and can be applied successfully
+            const patchFileCall = partialResult.find((call) => call.name === 'patchFile');
+            if (patchFileCall) {
+                partialResult = await executeStepVerifyPatch(patchFileCall.args, generateContentFn, prompt, functionDefs, file.temperature ?? codegenPrompt.options.temperature, file.cheap === true, messages, codegenPrompt.options);
+            }
+            // add the code gen result to the context, as the subsequent code gen may depend on the result
+            prompt.push({ type: 'assistant', functionCalls: partialResult }, {
+                type: 'user',
+                functionResponses: partialResult.map((call) => ({ name: call.name, call_id: call.id })),
+            });
+            result.push(...partialResult);
+        }
+        return result;
+    }
+    else {
+        // This is unexpected, if happens probably means no code updates.
+        console.log('Did not receive codegen summary, returning result.');
+        return baseResult;
+    }
+}
+/**
+ * Function to prepare messages for AI services
+ */
+function prepareMessages(codegen) {
+    return {
+        suggestSourceCode: 'I should provide you with application source code.',
+        requestSourceCode: 'Please provide application source code.',
+        suggestImageAssets: 'I should also provide you with a summary of application image assets',
+        requestImageAssets: 'Please provide summary of application image assets.',
+        prompt: codegen.prompt +
+            '\n Start from generating codegen summary, this summary will be used as a context to generate updates, so make sure that it contains useful information.',
+        sourceCode: JSON.stringify(getSourceCode({ taskFile: codegen.options.taskFile }, codegen.options)),
+        contextSourceCode: (paths) => JSON.stringify(getSourceCode({ filterPaths: paths, taskFile: codegen.options.taskFile, forceAll: true }, codegen.options)),
+        imageAssets: JSON.stringify(getImageAssets()),
+        partialPromptTemplate(path) {
+            return `Thank you for providing the summary, now suggest changes for the \`${path}\` file using appropriate tools.`;
+        },
+        invalidFunctionCall: 'Function call was invalid, please analyze the error and respond with corrected function call.',
+    };
+}
+//# sourceMappingURL=prompt-service.js.map

package/dist/prompt/prompt-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-service.js","sourceRoot":"","sources":["../../src/prompt/prompt-service.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAE9B,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAEvE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAYrD,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,8BAA8B,EAAE,MAAM,kCAAkC,CAAC;AAClF,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAGpD,oDAAoD;AACpD,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,kBAAkE,EAClE,gBAA2D,EAC3D,aAA4B;IAE5B,MAAM,EAAE,MAAM,EAAE,GAAG,aAAa,CAAC,OAAO,CAAC;IACzC,MAAM,QAAQ,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;IAEhD,MAAM,iBAAiB,GAA4B,CAAC,GAAG,IAAI,EAAE,EAAE,CAC7D,kBAAkB,CAAC,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAE/D,MAAM,eAAe,GAA0B,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,aAAa,CAAC,OAAO,CAAC,MAAO,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAErH,8FAA8F;IAC9F,MAAM,oBAAoB,GAAiB,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;IAErE,MAAM,MAAM,GAAiB;QAC3B,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,eAAe,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE;QAC9E,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,iBAAiB,EAAE;QAClD,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,CAAC,iBAAiB,EAAE,aAAa,EAAE,CAAC,oBAAoB,CAAC,EAAE;KAC/F,CAAC;IAEF,MAAM,qBAAqB,GAAe;QACxC,IAAI,EAAE,MAAM;QACZ,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC;QAC5E,KAAK,EAAE,IAAI;KACZ,CAAC;IACF,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IAEnC,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,QAAQ,CAAC,kBAAkB,CAAC;QACvD,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,CAAC,kBAAkB,EAAE,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,EAAE,EACrG,EAAE,IAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,EAAE,CACjG,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC;IAE3C,gCAAgC;IAChC,IACE,aAAa,CAAC,OAAO,CAAC,WAAW,KAAK,KAAK;QAC3C,aAAa,CAAC,OAAO,CAAC,WAAW;QACjC,CAAC,MAAM,sBAAsB,CAC3B,iBAAiB,EACjB,MAAM,EACN,YAAY,EACZ,aAAa,CAAC,OAAO,CAAC,WAAW,IAAI,GAAG,EACxC,aAAa,CAAC,OAAO,CAAC,KAAK,IAAI,KAAK,EACpC,QAAQ,EACR,aAAa,CAAC,OAAO,CACtB,CAAC,KAAK,UAAU,CAAC,KAAK,EACvB,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;SAAM,IAAI,aAAa,CAAC,OAAO,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,WAAW,GAA2E;QAC1F,MAAM;QACN,YAAY;QACZ,gBAAgB;QAChB,aAAa,CAAC,OAAO,CAAC,WAAW,IAAI,GAAG;QACxC,aAAa,CAAC,OAAO,CAAC,KAAK,IAAI,KAAK;QACpC,aAAa,CAAC,OAAO;KACtB,CAAC;IACF,IAAI,UAAU,GAAG,MAAM,iBAAiB,CAAC,GAAG,WAAW,CAAC,CAAC;IAEzD,IAAI,qBAAqB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;IAEtF,IAAI,qBAAqB,EAAE,CAAC;QAC1B,8DAA8D;QAC9D,OAAO,CAAC,GAAG,CAAC,wDAAwD,EAAE,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAElG,UAAU,GAAG,MAAM,8BAA8B,CAAC,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QACxG,qBAAqB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;QAElF,8CAA8C;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,6BAA6B,CAAC,CAAC;QAC/F,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,CAAC,YAAY,CAAC,EAAE,8BAA8B,CAAC,CAAC;QAEhG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,0BAA0B,EAAE,CAAC;YACtD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;YAC5C,wGAAwG;YACxG,oBAAoB,CAAC,IAAI,GAAG;gBAC1B,SAAS,EAAE;oBACT,GAAG,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;oBACpF,GAAG,qBAAqB,CAAC,IAAI,CAAC,YAAY;oBAC1C,GAAG,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC;iBAClC;aACF,CAAC;YACF,qBAAqB,CAAC,iBAAkB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAE,CAAC,OAAO;gBAC7F,QAAQ,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,IAAI,EAAE,SAAqB,CAAC,CAAC;QACjF,CAAC;QAED,kEAAkE;QAClE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YACpF,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAmB,EAAE,CAAC;QAElC,KAAK,MAAM,IAAI,IAAI,qBAAsB,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YAC3D,OAAO,CAAC,GAAG,CAAC,iCAAiC,GAAG,IAAI,CAAC,IAAI,GAAG,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;YACnG,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACjE,CAAC;YAED,iDAAiD;YACjD,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtF,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChG,CAAC;YAED,IAAI,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACtC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC;oBAC1E,IAAI;oBACJ,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC;oBAC1C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE;iBACnC,CAAC,CAAC,CAAC;YACN,CAAC;YAED,MAAM,cAAc,GAA2E;gBAC7F,MAAM;gBACN,YAAY;gBACZ,IAAI,CAAC,cAAc;gBACnB,IAAI,CAAC,WAAW,IAAI,aAAa,CAAC,OAAO,CAAC,WAAW;gBACrD,IAAI,CAAC,KAAK,KAAK,IAAI;gBACnB,aAAa,CAAC,OAAO;aACtB,CAAC;YACF,IAAI,aAAa,GAAG,MAAM,iBAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;YAE/D,yDAAyD;YACzD,aAAa,GAAG,MAAM,8BAA8B,CAAC,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAEjH,mCAAmC;YACnC,MAAM,iBAAiB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;YACtF,IAAI,iBAAiB,EAAE,CAAC;gBACtB,aAAa,CAAC,IAAI,CAAC,MAAM,wBAAwB,CAAC,eAAgB,EAAE,iBAAiB,CAAC,CAAC,CAAC;YAC1F,CAAC;YAED,iHAAiH;YACjH,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;YAC9E,IAAI,aAAa,EAAE,CAAC;gBAClB,aAAa,GAAG,MAAM,sBAAsB,CAC1C,aAAa,CAAC,IAA2C,EACzD,iBAAiB,EACjB,MAAM,EACN,YAAY,EACZ,IAAI,CAAC,WAAW,IAAI,aAAa,CAAC,OAAO,CAAC,WAAW,EACrD,IAAI,CAAC,KAAK,KAAK,IAAI,EACnB,QAAQ,EACR,aAAa,CAAC,OAAO,CACtB,CAAC;YACJ,CAAC;YAED,8FAA8F;YAC9F,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,EAAE,EACnD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;aACxF,CACF,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;SAAM,CAAC;QACN,iEAAiE;QACjE,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;QAClE,OAAO,UAAU,CAAC;IACpB,CAAC;AACH,CAAC;AAID;;GAEG;AACH,SAAS,eAAe,CAAC,OAAsB;IAC7C,OAAO;QACL,iBAAiB,EAAE,oDAAoD;QACvE,iBAAiB,EAAE,yCAAyC;QAC5D,kBAAkB,EAAE,sEAAsE;QAC1F,kBAAkB,EAAE,qDAAqD;QACzE,MAAM,EACJ,OAAO,CAAC,MAAM;YACd,yJAAyJ;QAC3J,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAClG,iBAAiB,EAAE,CAAC,KAAe,EAAE,EAAE,CACrC,IAAI,CAAC,SAAS,CACZ,aAAa,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,OAAO,CAAC,CAC3G;QACH,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,CAAC;QAC7C,qBAAqB,CAAC,IAAY;YAChC,OAAO,sEAAsE,IAAI,kCAAkC,CAAC;QACtH,CAAC;QACD,mBAAmB,EACjB,+FAA+F;KAClG,CAAC;AACJ,CAAC"}

package/dist/prompt/prompt-service.test.d.ts

@@ -0,0 +1 @@
+import '../files/find-files.js';