genaicode 0.0.37 → 0.2.0

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

@@ -0,0 +1 @@
+{"version":3,"file":"step-ask-question.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-ask-question.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEnD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAG9C,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,iBAO4B,EAC5B,MAAoB,EACpB,YAA2B,EAC3B,WAAmB,EACnB,KAAc,EACd,QAEC,EACD,OAAuB;IAEvB,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,KAAK,CAAC;IAC5B,OAAO,CAAC,aAAa,EAAE,CAAC;QACtB,MAAM,iBAAiB,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACpH,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAevE,CAAC;QACd,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;YAErD,IAAI,eAAe,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC;gBACtC,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;gBACvE,OAAO,UAAU,CAAC,KAAK,CAAC;YAC1B,CAAC;iBAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,YAAY,EAAE,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;gBAChD,MAAM;YACR,CAAC;YAED,MAAM,oBAAoB,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,mBAAmB,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAC1F,MAAM,oBAAoB,GACxB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,EAAE,kBAAkB,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;YAC7G,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,EAAE,YAAY;gBACnD,CAAC,CAAC,oBAAoB;oBACpB,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC;wBACb,OAAO,EAAE,yEAAyE;wBAClF,OAAO,EAAE,IAAI;qBACd,CAAC,CAAC;wBACH,CAAC,CAAC,oCAAoC;wBACtC,CAAC,CAAC,mCAAmC;oBACvC,CAAC,CAAC,oBAAoB;wBACpB,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC;4BACb,OAAO,EAAE,gFAAgF;4BACzF,OAAO,EAAE,KAAK;yBACf,CAAC,CAAC;4BACH,CAAC,CAAC,sBAAsB;4BACxB,CAAC,CAAC,4BAA4B;wBAChC,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;gBAC7C,CAAC,CAAC,qCAAqC,CAAC;YAE1C,IACE,oBAAoB;gBACpB,eAAe,CAAC,IAAI,EAAE,kBAAkB;gBACxC,UAAU,KAAK,4BAA4B,EAC3C,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,YAAY,EAAE,CAAC;oBAC1D,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC;gBAC7E,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,YAAY,EAAE,CAAC;oBAC1D,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;gBACxB,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,oBAAoB,EAAE,CAAC;oBAClE,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC;gBACtC,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,eAAe,EAAE,CAAC;oBAC7D,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;gBACjC,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,eAAe,EAAE,CAAC;oBAC7D,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;gBACjC,CAAC;gBACD,IAAI,eAAe,CAAC,IAAI,EAAE,kBAAkB,CAAC,aAAa,EAAE,CAAC;oBAC3D,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;gBAC/B,CAAC;YACH,CAAC;YAED,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE,EACvD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,UAAU;gBAChB,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,aAAa;wBACnB,OAAO,EAAE,eAAe,CAAC,EAAE;wBAC3B,OAAO,EACL,oBAAoB,IAAI,UAAU,KAAK,oCAAoC;4BACzE,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC,eAAe,CAAC,IAAK,CAAC,mBAAoB,CAAC;4BACxE,CAAC,CAAC,SAAS;qBAChB;iBACF;aACF,CACF,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC;YAClF,MAAM;QACR,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC,QAAQ,CAAC;AAC7B,CAAC"}

package/dist/prompt/steps/step-generate-image.d.ts

@@ -0,0 +1,2 @@
+import { FunctionCall, GenerateImageFunction } from '../../ai-service/common.js';
+export declare function executeStepGenerateImage(generateImageFn: GenerateImageFunction, generateImageCall: FunctionCall): Promise<FunctionCall>;

package/dist/prompt/steps/step-generate-image.js

@@ -0,0 +1,29 @@
+import assert from 'node:assert';
+export async function executeStepGenerateImage(generateImageFn, generateImageCall) {
+    assert(!!generateImageFn, 'Image generation requested, but an image generation service was not provided');
+    console.log('Processing image generation request:', generateImageCall.args);
+    try {
+        const { prompt: imagePrompt, filePath, contextImagePath, width, height, cheap, } = generateImageCall.args;
+        const generatedImageUrl = await generateImageFn(imagePrompt, contextImagePath, { width, height }, cheap === true);
+        // Add a downloadFile call to the result to ensure the generated image is tracked
+        return {
+            name: 'downloadFile',
+            args: {
+                filePath: filePath,
+                downloadUrl: generatedImageUrl,
+                explanation: `Downloading generated image`,
+            },
+        };
+    }
+    catch (error) {
+        console.error('Error generating image:', error);
+        // Add an explanation about the failed image generation
+        return {
+            name: 'explanation',
+            args: {
+                text: `Failed to generate image: ${error.message}`,
+            },
+        };
+    }
+}
+//# sourceMappingURL=step-generate-image.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"step-generate-image.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-generate-image.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AAGjC,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,eAAsC,EACtC,iBAA+B;IAE/B,MAAM,CAAC,CAAC,CAAC,eAAe,EAAE,8EAA8E,CAAC,CAAC;IAE1G,OAAO,CAAC,GAAG,CAAC,sCAAsC,EAAE,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAE5E,IAAI,CAAC;QACH,MAAM,EACJ,MAAM,EAAE,WAAW,EACnB,QAAQ,EACR,gBAAgB,EAChB,KAAK,EACL,MAAM,EACN,KAAK,GACN,GAAG,iBAAiB,CAAC,IAOrB,CAAC;QACF,MAAM,iBAAiB,GAAG,MAAM,eAAe,CAAC,WAAW,EAAE,gBAAgB,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC;QAElH,iFAAiF;QACjF,OAAO;YACL,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE;gBACJ,QAAQ,EAAE,QAAQ;gBAClB,WAAW,EAAE,iBAAiB;gBAC9B,WAAW,EAAE,6BAA6B;aAC3C;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QAChD,uDAAuD;QACvD,OAAO;YACL,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE;gBACJ,IAAI,EAAE,6BAA8B,KAAe,CAAC,OAAO,EAAE;aAC9D;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-validate-recover.d.ts

@@ -0,0 +1,13 @@
+import { PromptItem, FunctionDef, FunctionCall, GenerateContentFunction } from '../../ai-service/common.js';
+import { CodegenOptions } from '../../main/codegen-types.js';
+export declare function validateAndRecoverSingleResult([prompt, functionDefs, requiredFunctionName, temperature, cheap, options]: [
+    PromptItem[],
+    FunctionDef[],
+    string,
+    number,
+    boolean,
+    CodegenOptions
+], result: FunctionCall[], messages: {
+    invalidFunctionCall: string;
+    partialPromptTemplate: (path: string) => string;
+}, generateContentFn: GenerateContentFunction): Promise<FunctionCall[]>;

package/dist/prompt/steps/step-validate-recover.js

@@ -0,0 +1,51 @@
+import assert from 'node:assert';
+import { validateFunctionCall } from '../function-calling-validate.js';
+export async function validateAndRecoverSingleResult([prompt, functionDefs, requiredFunctionName, temperature, cheap, options], result, messages, generateContentFn) {
+    if (result.length > 1) {
+        // quite unexpected
+        return result;
+    }
+    let call = result[0];
+    const validatorError = validateFunctionCall(call, requiredFunctionName);
+    if (validatorError) {
+        console.log('Invalid function call', call, validatorError);
+        if (!call) {
+            call = { name: requiredFunctionName };
+            if (requiredFunctionName === 'patchFile') {
+                console.log('Switching patchFile to updateFile');
+                requiredFunctionName = 'updateFile';
+            }
+        }
+        prompt.push({ type: 'assistant', functionCalls: [call] }, {
+            type: 'user',
+            text: messages.invalidFunctionCall,
+            functionResponses: [
+                {
+                    name: call.name,
+                    call_id: call.id,
+                    content: JSON.stringify({ args: call.args, error: validatorError }),
+                    isError: true,
+                },
+            ],
+        });
+        console.log('Trying to recover...');
+        if (cheap) {
+            console.log('Disabling --cheap for recovery.');
+        }
+        result = await generateContentFn(prompt, functionDefs, requiredFunctionName, temperature, false, options);
+        console.log('Recover result:', result);
+        if (result?.length === 1) {
+            const recoveryError = validateFunctionCall(result[0], requiredFunctionName);
+            assert(!recoveryError, 'Recovery failed');
+            console.log('Recovery was successful');
+        }
+        else if (result?.length === 0) {
+            throw new Error('Did not receive any function calls unexpectedly.');
+        }
+        else {
+            console.log('Unexpected number of function calls', result);
+        }
+    }
+    return result;
+}
+//# sourceMappingURL=step-validate-recover.js.map

package/dist/prompt/steps/step-validate-recover.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-validate-recover.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-validate-recover.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AAEjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AAGvE,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAClD,CAAC,MAAM,EAAE,YAAY,EAAE,oBAAoB,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,CAOvE,EACD,MAAsB,EACtB,QAGC,EACD,iBAA0C;IAE1C,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,mBAAmB;QACnB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,IAAI,GAA6B,MAAM,CAAC,CAAC,CAAC,CAAC;IAC/C,MAAM,cAAc,GAAG,oBAAoB,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC;IACxE,IAAI,cAAc,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;QAC3D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;YACtC,IAAI,oBAAoB,KAAK,WAAW,EAAE,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;gBACjD,oBAAoB,GAAG,YAAY,CAAC;YACtC,CAAC;QACH,CAAC;QAED,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC,IAAI,CAAC,EAAE,EAC5C;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ,CAAC,mBAAmB;YAClC,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,OAAO,EAAE,IAAI,CAAC,EAAE;oBAChB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;oBACnE,OAAO,EAAE,IAAI;iBACd;aACF;SACF,CACF,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QACpC,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,YAAY,EAAE,oBAAoB,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC1G,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAEvC,IAAI,MAAM,EAAE,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAC;YAC5E,MAAM,CAAC,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QACzC,CAAC;aAAM,IAAI,MAAM,EAAE,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,qCAAqC,EAAE,MAAM,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/prompt/steps/step-verify-patch.d.ts

@@ -0,0 +1,7 @@
+import { PromptItem, FunctionDef, FunctionCall, GenerateContentFunction } from '../../ai-service/common.js';
+import { PromptMessages } from '../prompt-service.js';
+import { CodegenOptions } from '../../main/codegen-types.js';
+export declare function executeStepVerifyPatch({ filePath, patch }: {
+    filePath: string;
+    patch: string;
+}, generateContentFn: GenerateContentFunction, prompt: PromptItem[], functionDefs: FunctionDef[], temperature: number, cheap: boolean, messages: PromptMessages, options: CodegenOptions): Promise<FunctionCall[]>;

package/dist/prompt/steps/step-verify-patch.js

@@ -0,0 +1,41 @@
+import fs from 'fs';
+import * as diff from 'diff';
+import { validateAndRecoverSingleResult } from './step-validate-recover.js';
+export async function executeStepVerifyPatch({ filePath, patch }, generateContentFn, prompt, functionDefs, temperature, cheap, messages, options) {
+    console.log('Verification of patch for file:', filePath);
+    let updatedContent = false;
+    const currentContent = fs.readFileSync(filePath, 'utf-8');
+    try {
+        updatedContent = diff.applyPatch(currentContent, patch);
+    }
+    catch (e) {
+        console.log('Error when applying patch', e);
+    }
+    if (!updatedContent) {
+        console.log(`Patch could not be applied for ${filePath}. Retrying without patchFile function.`);
+        // Rerun content generation without patchFile function
+        const partialRequest = [
+            prompt,
+            functionDefs,
+            'updateFile',
+            temperature,
+            cheap,
+            options,
+        ];
+        let partialResult = await generateContentFn(...partialRequest);
+        partialResult = await validateAndRecoverSingleResult(partialRequest, partialResult, messages, generateContentFn);
+        const getSourceCodeCall = partialResult.find((call) => call.name === 'getSourceCode');
+        if (getSourceCodeCall) {
+            throw new Error('Unexpected getSourceCode: ' + JSON.stringify(getSourceCodeCall));
+        }
+        if (partialResult.find((call) => call.name === 'patchFile')) {
+            throw new Error('Unexpected patchFile in retry response');
+        }
+        return partialResult;
+    }
+    else {
+        console.log('Patch verified successfully');
+        return [{ name: 'patchFile', args: { filePath, patch } }];
+    }
+}
+//# sourceMappingURL=step-verify-patch.js.map

package/dist/prompt/steps/step-verify-patch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-verify-patch.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-verify-patch.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAE7B,OAAO,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AAI5E,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,EAAE,QAAQ,EAAE,KAAK,EAAuC,EACxD,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,WAAmB,EACnB,KAAc,EACd,QAAwB,EACxB,OAAuB;IAEvB,OAAO,CAAC,GAAG,CAAC,iCAAiC,EAAE,QAAQ,CAAC,CAAC;IAEzD,IAAI,cAAc,GAAqB,KAAK,CAAC;IAC7C,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAE1D,IAAI,CAAC;QACH,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;IAC1D,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,CAAC,GAAG,CAAC,kCAAkC,QAAQ,wCAAwC,CAAC,CAAC;QAEhG,sDAAsD;QACtD,MAAM,cAAc,GAA2E;YAC7F,MAAM;YACN,YAAY;YACZ,YAAY;YACZ,WAAW;YACX,KAAK;YACL,OAAO;SACR,CAAC;QACF,IAAI,aAAa,GAAG,MAAM,iBAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;QAE/D,aAAa,GAAG,MAAM,8BAA8B,CAAC,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAEjH,MAAM,iBAAiB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;QACtF,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,4BAA4B,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACpF,CAAC;QACD,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,EAAE,CAAC;YAC5D,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;QAC3C,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC"}

package/dist/prompt/steps/steps-types.d.ts

@@ -0,0 +1,4 @@
+export declare enum StepResult {
+    CONTINUE = 0,
+    BREAK = 1
+}

package/dist/prompt/steps/steps-types.js

@@ -0,0 +1,6 @@
+export var StepResult;
+(function (StepResult) {
+    StepResult[StepResult["CONTINUE"] = 0] = "CONTINUE";
+    StepResult[StepResult["BREAK"] = 1] = "BREAK";
+})(StepResult || (StepResult = {}));
+//# sourceMappingURL=steps-types.js.map

package/dist/prompt/steps/steps-types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"steps-types.js","sourceRoot":"","sources":["../../../src/prompt/steps/steps-types.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,UAGX;AAHD,WAAY,UAAU;IACpB,mDAAQ,CAAA;IACR,6CAAK,CAAA;AACP,CAAC,EAHW,UAAU,KAAV,UAAU,QAGrB"}

package/dist/prompt/systemprompt.d.ts

@@ -0,0 +1,3 @@
+import { CodegenOptions } from '../main/codegen-types.js';
+/** Generates a system prompt */
+export declare function getSystemPrompt({ verbose, askQuestion, interactive }: CodegenOptions): string;

package/dist/prompt/systemprompt.js

@@ -0,0 +1,47 @@
+import { CODEGEN_TRIGGER } from './prompt-consts.js';
+import { verifySystemPromptLimit } from './limits.js';
+import { rcConfig } from '../main/config.js';
+/** Generates a system prompt */
+export function getSystemPrompt({ verbose, askQuestion, interactive }) {
+    console.log('Generate system prompt');
+    let systemPrompt = `
+  You are a code generation assistant. I want you to help me generate code for my ideas in my application source code.
+
+  You can generate new code, or modify the existing one. You will receive instructions on what is the goal of requested code modification.
+
+  Instructions will be passed to you either directly via message, with a file, or using the ${CODEGEN_TRIGGER} comment in the code.
+
+  You should parse my application source code and then suggest changes using appropriate tools.
+
+  The root directory of my application is \`${rcConfig.rootDir}\` and you should limit the changes only to this path.
+
+  Additional suggestions, or corners cases to think about:
+  - When suggesting changes always use absolute file paths, exactly as you have been provided. Modified paths will cause errors and we want to avoid that.
+  - Always aim to return working code
+  - Do not leave out commented out fragments like "// ... (keep other existing functions)"
+  - For large files prefer to use \`patchFile\` function
+  - Suggest user to split large files if it makes sense for current task
+  - At the start of conversation the user will express what you are allowed to do (for example: create files, move files, generate images etc.). At the start of conversation you should ensure you have necessary allowances for the current task.
+  - Consider failing the task with explanation if instructions are not clear enough, or something feels completely wrong.
+  - Do not produce unnecessary code, ensure the code you generate will be used once all changes are done.
+  - Always ask for sufficient context paths in codegen summary, ask for files in context that is needed to completed the task
+
+  `;
+    if (askQuestion && interactive) {
+        systemPrompt +=
+            '\nYou have the ability to ask the user a question at the beginning of the conversation if you need more information or clarification. ' +
+                'Use this feature wisely to gather any crucial information that would help you better understand the task or provide more accurate code generation.\n' +
+                "To ask a question, use the 'askQuestion' function. This function allows you to:" +
+                `- inform the user about your thoughts regarding the task
+      - ask questions, and give suggestions to the user regarding the task
+      - request access to content of files if they are not provided in the conversation so far, but are important for the task
+      - request enablement of permissions for operations that were restricted on the start of conversation, but are important for completion of the task`;
+    }
+    if (verbose) {
+        console.log('System prompt:');
+        console.log(systemPrompt);
+    }
+    verifySystemPromptLimit(systemPrompt);
+    return systemPrompt;
+}
+//# sourceMappingURL=systemprompt.js.map

package/dist/prompt/systemprompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"systemprompt.js","sourceRoot":"","sources":["../../src/prompt/systemprompt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,uBAAuB,EAAE,MAAM,aAAa,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAG7C,gCAAgC;AAChC,MAAM,UAAU,eAAe,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,EAAkB;IACnF,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IAEtC,IAAI,YAAY,GAAG;;;;;8FAKyE,eAAe;;;;8CAI/D,QAAQ,CAAC,OAAO;;;;;;;;;;;;;GAa3D,CAAC;IAEF,IAAI,WAAW,IAAI,WAAW,EAAE,CAAC;QAC/B,YAAY;YACV,wIAAwI;gBACxI,sJAAsJ;gBACtJ,iFAAiF;gBACjF;;;yJAGmJ,CAAC;IACxJ,CAAC;IAED,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC9B,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAC5B,CAAC;IAED,uBAAuB,CAAC,YAAY,CAAC,CAAC;IAEtC,OAAO,YAAY,CAAC;AACtB,CAAC"}

package/dist/prompt/systemprompt.test.d.ts

@@ -0,0 +1,3 @@
+import '../cli/cli-params.js';
+import '../files/find-files.js';
+import '../main/config.js';

package/dist/prompt/systemprompt.test.js

@@ -0,0 +1,46 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { getSystemPrompt } from './systemprompt.js';
+import '../cli/cli-params.js';
+import '../files/find-files.js';
+import '../main/config.js';
+vi.mock('../cli/cli-params.js', () => ({
+    requireExplanations: false,
+    considerAllFiles: false,
+    dependencyTree: false,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    askQuestion: false,
+}));
+vi.mock('../files/find-files.js', () => ({
+    getSourceFiles: vi.fn(),
+}));
+vi.mock('../main/config.js', () => ({
+    rcConfig: { rootDir: '/mocked/root/dir' },
+}));
+describe('getSystemPrompt', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    it('generates correct system prompt', () => {
+        const systemPrompt = getSystemPrompt({ aiService: 'vertex-ai' });
+        expect(systemPrompt).toContain('You are a code generation assistant. I want you to help me generate code for my ideas in my application source code.');
+        expect(systemPrompt).toContain('You can generate new code, or modify the existing one.');
+        expect(systemPrompt).toContain('Instructions will be passed to you either directly via message, with a file, or using the ' +
+            '@' +
+            'CODEGEN comment in the code.');
+        expect(systemPrompt).toContain('The root directory of my application is `/mocked/root/dir` and you should limit the changes only to this path.');
+        expect(systemPrompt).toContain('When suggesting changes always use absolute file paths');
+    });
+    it('verifies system prompt limit', () => {
+        const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => { });
+        getSystemPrompt({ aiService: 'vertex-ai', verbose: true });
+        expect(consoleSpy).toHaveBeenCalledWith('System prompt:');
+        expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining('You are a code generation assistant. I want you to help me generate code for my ideas'));
+        consoleSpy.mockRestore();
+    });
+});
+//# sourceMappingURL=systemprompt.test.js.map

package/dist/prompt/systemprompt.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"systemprompt.test.js","sourceRoot":"","sources":["../../src/prompt/systemprompt.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,sBAAsB,CAAC;AAC9B,OAAO,wBAAwB,CAAC;AAChC,OAAO,mBAAmB,CAAC;AAE3B,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,WAAW,EAAE,KAAK;CACnB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;CACxB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE;CAC1C,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,YAAY,GAAG,eAAe,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;QAEjE,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAC5B,sHAAsH,CACvH,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,wDAAwD,CAAC,CAAC;QACzF,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAC5B,4FAA4F;YAC1F,GAAG;YACH,8BAA8B,CACjC,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAC5B,gHAAgH,CACjH,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,wDAAwD,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAEzE,eAAe,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAE3D,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAC1D,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC,uFAAuF,CAAC,CACjH,CAAC;QAEF,UAAU,CAAC,WAAW,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "genaicode",
-  "version": "0.0.37",
+  "version": "0.2.0",
   "author": "Grzegorz Tańczyk",
   "repository": {
     "type": "git",
@@ -19,17 +19,19 @@
   "license": "Apache-2.0",
   "files": [
     "bin",
-    "src",
+    "dist",
     "media",
     "README.md",
     "LICENSE"
   ],
   "scripts": {
-    "lint": "eslint . --ext js --report-unused-disable-directives --max-warnings 0",
+    "lint": "eslint . --ext js,ts --report-unused-disable-directives --max-warnings 0",
     "test": "vitest",
     "coverage": "vitest run --coverage",
     "prepare": "husky install",
-    "postinstall": "node ./bin/vertex-monkey-patch.cjs"
+    "type-check": "tsc --noEmit",
+    "build": "tsc",
+    "prepublishOnly": "npm run build"
   },
   "devDependencies": {
     "@typescript-eslint/eslint-plugin": "^7.17.0",
@@ -40,17 +42,25 @@
     "husky": "^9.1.1",
     "lint-staged": "^15.2.7",
     "prettier": "^3.1.0",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.0.0",
     "vitest": "^2.0.4"
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.26.1",
     "@anthropic-ai/vertex-sdk": "^0.4.1",
     "@google-cloud/aiplatform": "^3.25.0",
-    "@google-cloud/vertexai": "^1.3.0",
+    "@google-cloud/vertexai": "^1.6.0",
+    "@google/generative-ai": "^0.17.1",
     "@imgly/background-removal-node": "^1.4.5",
+    "@inquirer/prompts": "^5.3.8",
+    "@types/diff": "^5.2.1",
+    "@types/mime-types": "^2.1.4",
+    "@types/node": "^18.0.0",
     "diff": "^5.2.0",
     "glob-regex": "^0.3.2",
     "image-size": "^1.1.1",
+    "inquirer-file-selector": "^0.4.0",
     "jsonschema": "^1.4.1",
     "mime-types": "^2.1.35",
     "openai": "^4.52.7",

package/bin/genaicode.js

@@ -1,5 +0,0 @@
-#!/usr/bin/env node
-
-import { runCodegen } from '../src/main/codegen.js';
-
-await runCodegen();

package/bin/vertex-monkey-patch.cjs

@@ -1,33 +0,0 @@
-#!/usr/bin/env node
-
-const fs = require('fs');
-
-// Add node version check
-const pkg = require('../package.json');
-const pleaseUpgradeNode = require('please-upgrade-node');
-pleaseUpgradeNode(pkg);
-
-const MONKEY_PATCH_FILE = '@google-cloud/vertexai/build/src/functions/generate_content.js';
-const MONKEY_PATCH_TOOL_CONFIG = `// MONKEY PATCH TOOL_CONFIG`;
-
-console.log('Apply vertex monkey patch');
-const path = require.resolve(MONKEY_PATCH_FILE);
-const content = fs.readFileSync(path, 'utf-8');
-
-if (!content) {
-  console.log('Could not find file:', path);
-  return;
-}
-
-if (content.includes(MONKEY_PATCH_TOOL_CONFIG)) {
-  console.log('Patch already applied');
-  return;
-}
-
-const newContent = content.replaceAll(
-  'data: generateContentRequest,',
-  `// MONKEY PATCH TOOL_CONFIG
-        data: {...generateContentRequest,tool_config: {function_calling_config: { mode: request.toolConfig?.functionCallingConfig?.mode, allowed_function_names: request.toolConfig?.functionCallingConfig.allowedFunctionNames }}},`,
-);
-fs.writeFileSync(path, newContent, 'utf-8');
-console.log('Vertex monkey patch applied');

package/src/ai-service/anthropic.js

@@ -1,126 +0,0 @@
-import Anthropic from '@anthropic-ai/sdk';
-import { printTokenUsageAndCost, processFunctionCalls } from './common.js';
-import { disableCache } from '../cli/cli-params.js';
-
-/**
- * This function generates content using the Anthropic Claude model.
- */
-export async function generateContent(prompt, functionDefs, requiredFunctionName, temperature, cheap = false) {
-  const anthropic = new Anthropic({
-    defaultHeaders: {
-      'anthropic-beta': 'max-tokens-3-5-sonnet-2024-07-15' + (!disableCache ? ',prompt-caching-2024-07-31' : ''),
-    },
-  });
-
-  const messages = prompt
-    .filter((item) => item.type !== 'systemPrompt')
-    .map((item) => {
-      if (item.type === 'user') {
-        return {
-          role: 'user',
-          content: [
-            ...(item.functionResponses ?? []).map((response) => ({
-              tool_use_id: response.call_id ?? response.name,
-              content: response.content,
-              type: 'tool_result',
-              is_error: response.isError === true,
-            })),
-            ...(item.images ?? []).map((image) => ({
-              type: 'image',
-              source: {
-                type: 'base64',
-                media_type: image.mediaType,
-                data: image.base64url,
-              },
-            })),
-            {
-              type: 'text',
-              text: item.text,
-              ...(item.cache && !disableCache ? { cache_control: { type: 'ephemeral' } } : {}),
-            },
-          ],
-        };
-      } else if (item.type === 'assistant') {
-        return {
-          role: 'assistant',
-          content: [
-            ...(item.text ? [{ type: 'text', text: item.text }] : []),
-            ...item.functionCalls.map((call) => ({
-              id: call.id ?? call.name,
-              name: call.name,
-              input: call.args ?? {},
-              type: 'tool_use',
-            })),
-          ],
-        };
-      }
-    });
-
-  const model = cheap ? 'claude-3-haiku-20240307' : 'claude-3-5-sonnet-20240620';
-  console.log(`Using Anthropic model: ${model}`);
-
-  let retryCount = 0;
-  let response;
-  while (retryCount < 3) {
-    try {
-      response = await anthropic.beta.promptCaching.messages.create({
-        model: model,
-        system: prompt.find((item) => item.type === 'systemPrompt').systemPrompt,
-        messages,
-        tools: functionDefs.map((fd) => ({
-          name: fd.name,
-          description: fd.description,
-          input_schema: fd.parameters,
-        })),
-        tool_choice: requiredFunctionName ? { type: 'tool', name: requiredFunctionName } : { type: 'any' },
-        max_tokens: cheap ? 4096 : 8192,
-        temperature: temperature,
-      });
-      break; // Exit loop if successful
-    } catch (error) {
-      if (error.headers?.['retry-after']) {
-        let retryAfter;
-        if (error.headers['retry-after'] === '0') {
-          retryAfter = (new Date(error.headers['anthropic-ratelimit-tokens-reset']).getTime() - Date.now()) / 1000;
-        } else {
-          retryAfter = Math.max(parseInt(error.headers['retry-after'], 10), 10);
-        }
-        console.log(`Rate limited. Retrying after ${retryAfter} seconds. Attempt ${retryCount + 1} of 3.`);
-        await new Promise((resolve) => setTimeout(resolve, retryAfter * 1000));
-        retryCount++;
-      } else {
-        console.error('An error occurred:', error);
-        throw error; // Re-throw the error if it's not a rate limit error
-      }
-    }
-  }
-
-  if (retryCount === 3) {
-    console.error('Failed to complete request after 3 attempts due to rate limiting.');
-    throw new Error('Rate limit exceeded. Operation aborted.');
-  }
-
-  // Print token usage for Anthropic
-  const usage = {
-    cacheCreateTokens: response.usage.cache_creation_input_tokens,
-    cacheReadTokens: response.usage.cache_read_input_tokens,
-    inputTokens: response.usage.input_tokens,
-    outputTokens: response.usage.output_tokens,
-    totalTokens: response.usage.input_tokens + response.usage.output_tokens,
-  };
-  printTokenUsageAndCost(usage, 3 / 1000 / 1000, 15 / 1000 / 1000);
-
-  const responseMessages = response.content.filter((item) => item.type !== 'tool_use');
-  if (responseMessages.length > 0) {
-    console.log('Response messages', responseMessages);
-  }
-
-  const functionCalls = response.content
-    .filter((item) => item.type === 'tool_use')
-    .map((item) => ({
-      name: item.name,
-      args: item.input,
-    }));
-
-  return processFunctionCalls(functionCalls);
-}