genaicode 0.0.35 → 0.1.1

package/dist/prompt/function-defs/imgly-remove-background.js

@@ -0,0 +1,26 @@
+/**
+ * Function definition for imglyRemoveBackground
+ */
+export const imglyRemoveBackground = {
+    name: 'imglyRemoveBackground',
+    description: 'Removes background from an image using @imgly/background-removal-node',
+    parameters: {
+        type: 'object',
+        properties: {
+            inputFilePath: {
+                type: 'string',
+                description: 'The file path of the input image.',
+            },
+            outputFilePath: {
+                type: 'string',
+                description: 'The file path to save the output image with removed background.',
+            },
+            explanation: {
+                type: 'string',
+                description: 'The explanation of the reasoning behind removing the background from this image',
+            },
+        },
+        required: ['inputFilePath', 'outputFilePath'],
+    },
+};
+//# sourceMappingURL=imgly-remove-background.js.map

package/dist/prompt/function-defs/imgly-remove-background.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"imgly-remove-background.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/imgly-remove-background.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,IAAI,EAAE,uBAAuB;IAC7B,WAAW,EAAE,uEAAuE;IACpF,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,aAAa,EAAE;gBACb,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,mCAAmC;aACjD;YACD,cAAc,EAAE;gBACd,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iEAAiE;aAC/E;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iFAAiF;aAC/F;SACF;QACD,QAAQ,EAAE,CAAC,eAAe,EAAE,gBAAgB,CAAC;KAC9C;CACO,CAAC"}

package/dist/prompt/function-defs/move-file.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Function definition for moveFile
+ */
+export declare const moveFile: {
+    readonly name: "moveFile";
+    readonly description: "Move a file from one location to another";
+    readonly parameters: {
+        readonly type: "object";
+        readonly properties: {
+            readonly source: {
+                readonly type: "string";
+                readonly description: "The current file path.";
+            };
+            readonly destination: {
+                readonly type: "string";
+                readonly description: "The new file path.";
+            };
+            readonly explanation: {
+                readonly type: "string";
+                readonly description: "The explanation of the reasoning behind moving this file";
+            };
+        };
+        readonly required: readonly ["source", "destination"];
+    };
+};

package/dist/prompt/function-defs/move-file.js

@@ -0,0 +1,26 @@
+/**
+ * Function definition for moveFile
+ */
+export const moveFile = {
+    name: 'moveFile',
+    description: 'Move a file from one location to another',
+    parameters: {
+        type: 'object',
+        properties: {
+            source: {
+                type: 'string',
+                description: 'The current file path.',
+            },
+            destination: {
+                type: 'string',
+                description: 'The new file path.',
+            },
+            explanation: {
+                type: 'string',
+                description: 'The explanation of the reasoning behind moving this file',
+            },
+        },
+        required: ['source', 'destination'],
+    },
+};
+//# sourceMappingURL=move-file.js.map

package/dist/prompt/function-defs/move-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"move-file.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/move-file.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG;IACtB,IAAI,EAAE,UAAU;IAChB,WAAW,EAAE,0CAA0C;IACvD,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,wBAAwB;aACtC;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,oBAAoB;aAClC;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0DAA0D;aACxE;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,EAAE,aAAa,CAAC;KACpC;CACO,CAAC"}

package/dist/prompt/function-defs/patch-file.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Function definition for patchFile
+ */
+export declare const patchFile: {
+    readonly name: "patchFile";
+    readonly description: "Partially update a file content. The file must already exists in the application source code. The function should be called only if there is a need to actually change something.";
+    readonly parameters: {
+        readonly type: "object";
+        readonly properties: {
+            readonly filePath: {
+                readonly type: "string";
+                readonly description: "The file path to patch.";
+            };
+            readonly patch: {
+                readonly type: "string";
+                readonly description: "Modification to the file expressed in patch format. Example patch:\n\n```\nIndex: filename.js\n===================================================================\n--- filename.js\n+++ filename.js\n@@ -1,2 +1,3 @@\n line1\n+line3\n line2\n\\n```\n          ";
+            };
+            readonly explanation: {
+                readonly type: "string";
+                readonly description: "The explanation of the reasoning behind the suggested code changes for this file";
+            };
+        };
+        readonly required: readonly ["filePath", "patch"];
+    };
+};

package/dist/prompt/function-defs/patch-file.js

@@ -0,0 +1,39 @@
+/**
+ * Function definition for patchFile
+ */
+export const patchFile = {
+    name: 'patchFile',
+    description: 'Partially update a file content. The file must already exists in the application source code. The function should be called only if there is a need to actually change something.',
+    parameters: {
+        type: 'object',
+        properties: {
+            filePath: {
+                type: 'string',
+                description: 'The file path to patch.',
+            },
+            patch: {
+                type: 'string',
+                description: `Modification to the file expressed in patch format. Example patch:
+
+\`\`\`
+Index: filename.js
+===================================================================
+--- filename.js
++++ filename.js
+@@ -1,2 +1,3 @@
+ line1
++line3
+ line2
+\
+\`\`\`
+          `,
+            },
+            explanation: {
+                type: 'string',
+                description: 'The explanation of the reasoning behind the suggested code changes for this file',
+            },
+        },
+        required: ['filePath', 'patch'],
+    },
+};
+//# sourceMappingURL=patch-file.js.map

package/dist/prompt/function-defs/patch-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"patch-file.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/patch-file.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG;IACvB,IAAI,EAAE,WAAW;IACjB,WAAW,EACT,mLAAmL;IACrL,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yBAAyB;aACvC;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE;;;;;;;;;;;;;WAaV;aACJ;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kFAAkF;aAChG;SACF;QACD,QAAQ,EAAE,CAAC,UAAU,EAAE,OAAO,CAAC;KAChC;CACO,CAAC"}

package/dist/prompt/function-defs/resize-image.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Function definition for resizeImage
+ */
+export declare const resizeImage: {
+    readonly name: "resizeImage";
+    readonly description: "Resize image to the desired size";
+    readonly parameters: {
+        readonly type: "object";
+        readonly properties: {
+            readonly filePath: {
+                readonly type: "string";
+                readonly description: "The file path of the image.";
+            };
+            readonly size: {
+                readonly type: "object";
+                readonly properties: {
+                    readonly width: {
+                        readonly type: "number";
+                        readonly description: "width of the image";
+                    };
+                    readonly height: {
+                        readonly type: "number";
+                        readonly description: "height of the image";
+                    };
+                };
+                readonly required: readonly ["width", "height"];
+                readonly description: "The size of the image to generate.";
+            };
+            readonly explanation: {
+                readonly type: "string";
+                readonly description: "The explanation of the reasoning behind removing the background from this image";
+            };
+        };
+        readonly required: readonly ["filePath", "size"];
+    };
+};

package/dist/prompt/function-defs/resize-image.js

@@ -0,0 +1,37 @@
+/**
+ * Function definition for resizeImage
+ */
+export const resizeImage = {
+    name: 'resizeImage',
+    description: 'Resize image to the desired size',
+    parameters: {
+        type: 'object',
+        properties: {
+            filePath: {
+                type: 'string',
+                description: 'The file path of the image.',
+            },
+            size: {
+                type: 'object',
+                properties: {
+                    width: {
+                        type: 'number',
+                        description: 'width of the image',
+                    },
+                    height: {
+                        type: 'number',
+                        description: 'height of the image',
+                    },
+                },
+                required: ['width', 'height'],
+                description: 'The size of the image to generate.',
+            },
+            explanation: {
+                type: 'string',
+                description: 'The explanation of the reasoning behind removing the background from this image',
+            },
+        },
+        required: ['filePath', 'size'],
+    },
+};
+//# sourceMappingURL=resize-image.js.map

package/dist/prompt/function-defs/resize-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resize-image.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/resize-image.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,kCAAkC;IAC/C,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,6BAA6B;aAC3C;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,KAAK,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,oBAAoB;qBAClC;oBACD,MAAM,EAAE;wBACN,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,qBAAqB;qBACnC;iBACF;gBACD,QAAQ,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC;gBAC7B,WAAW,EAAE,oCAAoC;aAClD;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iFAAiF;aAC/F;SACF;QACD,QAAQ,EAAE,CAAC,UAAU,EAAE,MAAM,CAAC;KAC/B;CACO,CAAC"}

package/dist/prompt/function-defs/split-image.d.ts

@@ -0,0 +1,57 @@
+/**
+ * Function definition for splitImage
+ */
+export declare const splitImage: {
+    readonly name: "splitImage";
+    readonly description: "Split an image into multiple parts and save them as separate files.";
+    readonly parameters: {
+        readonly type: "object";
+        readonly properties: {
+            readonly inputFilePath: {
+                readonly type: "string";
+                readonly description: "The file path of the input image to be split.";
+            };
+            readonly parts: {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "object";
+                    readonly properties: {
+                        readonly rect: {
+                            readonly type: "object";
+                            readonly properties: {
+                                readonly x: {
+                                    readonly type: "number";
+                                    readonly description: "The x-coordinate of the top-left corner of the rectangle.";
+                                };
+                                readonly y: {
+                                    readonly type: "number";
+                                    readonly description: "The y-coordinate of the top-left corner of the rectangle.";
+                                };
+                                readonly width: {
+                                    readonly type: "number";
+                                    readonly description: "The width of the rectangle.";
+                                };
+                                readonly height: {
+                                    readonly type: "number";
+                                    readonly description: "The height of the rectangle.";
+                                };
+                            };
+                            readonly required: readonly ["x", "y", "width", "height"];
+                        };
+                        readonly outputFilePath: {
+                            readonly type: "string";
+                            readonly description: "The file path to save the extracted part of the image.";
+                        };
+                    };
+                    readonly required: readonly ["rect", "outputFilePath"];
+                };
+                readonly description: "An array of parts to extract from the image, each with a rectangle and output file path.";
+            };
+            readonly explanation: {
+                readonly type: "string";
+                readonly description: "The explanation of the reasoning behind splitting this image";
+            };
+        };
+        readonly required: readonly ["inputFilePath", "parts"];
+    };
+};

package/dist/prompt/function-defs/split-image.js

@@ -0,0 +1,46 @@
+/**
+ * Function definition for splitImage
+ */
+export const splitImage = {
+    name: 'splitImage',
+    description: 'Split an image into multiple parts and save them as separate files.',
+    parameters: {
+        type: 'object',
+        properties: {
+            inputFilePath: {
+                type: 'string',
+                description: 'The file path of the input image to be split.',
+            },
+            parts: {
+                type: 'array',
+                items: {
+                    type: 'object',
+                    properties: {
+                        rect: {
+                            type: 'object',
+                            properties: {
+                                x: { type: 'number', description: 'The x-coordinate of the top-left corner of the rectangle.' },
+                                y: { type: 'number', description: 'The y-coordinate of the top-left corner of the rectangle.' },
+                                width: { type: 'number', description: 'The width of the rectangle.' },
+                                height: { type: 'number', description: 'The height of the rectangle.' },
+                            },
+                            required: ['x', 'y', 'width', 'height'],
+                        },
+                        outputFilePath: {
+                            type: 'string',
+                            description: 'The file path to save the extracted part of the image.',
+                        },
+                    },
+                    required: ['rect', 'outputFilePath'],
+                },
+                description: 'An array of parts to extract from the image, each with a rectangle and output file path.',
+            },
+            explanation: {
+                type: 'string',
+                description: 'The explanation of the reasoning behind splitting this image',
+            },
+        },
+        required: ['inputFilePath', 'parts'],
+    },
+};
+//# sourceMappingURL=split-image.js.map

package/dist/prompt/function-defs/split-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"split-image.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/split-image.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,IAAI,EAAE,YAAY;IAClB,WAAW,EAAE,qEAAqE;IAClF,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,aAAa,EAAE;gBACb,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,+CAA+C;aAC7D;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,IAAI,EAAE;4BACJ,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2DAA2D,EAAE;gCAC/F,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2DAA2D,EAAE;gCAC/F,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,6BAA6B,EAAE;gCACrE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,8BAA8B,EAAE;6BACxE;4BACD,QAAQ,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,CAAC;yBACxC;wBACD,cAAc,EAAE;4BACd,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,wDAAwD;yBACtE;qBACF;oBACD,QAAQ,EAAE,CAAC,MAAM,EAAE,gBAAgB,CAAC;iBACrC;gBACD,WAAW,EAAE,0FAA0F;aACxG;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,8DAA8D;aAC5E;SACF;QACD,QAAQ,EAAE,CAAC,eAAe,EAAE,OAAO,CAAC;KACrC;CACO,CAAC"}

package/dist/prompt/function-defs/update-file.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Function definition for updateFile
+ */
+export declare const updateFile: {
+    readonly name: "updateFile";
+    readonly description: "Update a file with new content. The file must already exists in the application source code. The function should be called only if there is a need to actually change something.";
+    readonly parameters: {
+        readonly type: "object";
+        readonly properties: {
+            readonly filePath: {
+                readonly type: "string";
+                readonly description: "The file path to update.";
+            };
+            readonly newContent: {
+                readonly type: "string";
+                readonly description: "The content to update the file with. Must not be empty.";
+            };
+            readonly explanation: {
+                readonly type: "string";
+                readonly description: "The explanation of the reasoning behind the suggested code changes for this file";
+            };
+        };
+        readonly required: readonly ["filePath", "newContent"];
+    };
+};

package/dist/prompt/function-defs/update-file.js

@@ -0,0 +1,26 @@
+/**
+ * Function definition for updateFile
+ */
+export const updateFile = {
+    name: 'updateFile',
+    description: 'Update a file with new content. The file must already exists in the application source code. The function should be called only if there is a need to actually change something.',
+    parameters: {
+        type: 'object',
+        properties: {
+            filePath: {
+                type: 'string',
+                description: 'The file path to update.',
+            },
+            newContent: {
+                type: 'string',
+                description: 'The content to update the file with. Must not be empty.',
+            },
+            explanation: {
+                type: 'string',
+                description: 'The explanation of the reasoning behind the suggested code changes for this file',
+            },
+        },
+        required: ['filePath', 'newContent'],
+    },
+};
+//# sourceMappingURL=update-file.js.map

package/dist/prompt/function-defs/update-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-file.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/update-file.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,IAAI,EAAE,YAAY;IAClB,WAAW,EACT,kLAAkL;IACpL,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0BAA0B;aACxC;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yDAAyD;aACvE;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kFAAkF;aAChG;SACF;QACD,QAAQ,EAAE,CAAC,UAAU,EAAE,YAAY,CAAC;KACrC;CACO,CAAC"}

package/dist/prompt/limits.d.ts

@@ -0,0 +1,3 @@
+export declare function verifySystemPromptLimit(systemPrompt: string): void;
+export declare function verifyCodegenPromptLimit(codeGenPrompt: string): void;
+export declare function verifySourceCodeLimit(sourceCode: string): void;

package/dist/prompt/limits.js

@@ -0,0 +1,19 @@
+import assert from 'node:assert';
+const SYSTEM_PROMPT_LIMIT = 300;
+const CODEGEN_PROMPT_LIMIT = 500;
+const SOURCE_CODE_LIMIT = 20000;
+function verifyPromptLimit(promptType, prompt, limit) {
+    const tokenCount = prompt.split(/\s+/).length;
+    console.log(`${promptType} prompt token count: ${tokenCount}`);
+    assert(tokenCount <= limit, `Token limit exceeded: ${tokenCount} > ${limit}`);
+}
+export function verifySystemPromptLimit(systemPrompt) {
+    verifyPromptLimit('system', systemPrompt, SYSTEM_PROMPT_LIMIT);
+}
+export function verifyCodegenPromptLimit(codeGenPrompt) {
+    verifyPromptLimit('codegen', codeGenPrompt, CODEGEN_PROMPT_LIMIT);
+}
+export function verifySourceCodeLimit(sourceCode) {
+    verifyPromptLimit('sourceCode', sourceCode, SOURCE_CODE_LIMIT);
+}
+//# sourceMappingURL=limits.js.map

package/dist/prompt/limits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"limits.js","sourceRoot":"","sources":["../../src/prompt/limits.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AAEjC,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAChC,MAAM,oBAAoB,GAAG,GAAG,CAAC;AACjC,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAEhC,SAAS,iBAAiB,CAAC,UAAkB,EAAE,MAAc,EAAE,KAAa;IAC1E,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;IAC9C,OAAO,CAAC,GAAG,CAAC,GAAG,UAAU,wBAAwB,UAAU,EAAE,CAAC,CAAC;IAC/D,MAAM,CAAC,UAAU,IAAI,KAAK,EAAE,yBAAyB,UAAU,MAAM,KAAK,EAAE,CAAC,CAAC;AAChF,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,YAAoB;IAC1D,iBAAiB,CAAC,QAAQ,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,aAAqB;IAC5D,iBAAiB,CAAC,SAAS,EAAE,aAAa,EAAE,oBAAoB,CAAC,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,UAAkB;IACtD,iBAAiB,CAAC,YAAY,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAAC;AACjE,CAAC"}

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

@@ -0,0 +1 @@
+export {};

package/dist/prompt/limits.test.js

@@ -0,0 +1,35 @@
+import { describe, it, expect } from 'vitest';
+import { verifySystemPromptLimit, verifyCodegenPromptLimit, verifySourceCodeLimit } from './limits.js';
+describe('Prompt Limits', () => {
+    describe('verifySystemPromptLimit', () => {
+        it('should not throw for a prompt within the limit', () => {
+            const validPrompt = 'A '.repeat(200 - 1);
+            expect(() => verifySystemPromptLimit(validPrompt)).not.toThrow();
+        });
+        it('should throw for a prompt exceeding the limit', () => {
+            const invalidPrompt = 'A '.repeat(301);
+            expect(() => verifySystemPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 302 > 300');
+        });
+    });
+    describe('verifyCodegenPromptLimit', () => {
+        it('should not throw for a prompt within the limit', () => {
+            const validPrompt = 'B '.repeat(200 - 1);
+            expect(() => verifyCodegenPromptLimit(validPrompt)).not.toThrow();
+        });
+        it('should throw for a prompt exceeding the limit', () => {
+            const invalidPrompt = 'B '.repeat(501);
+            expect(() => verifyCodegenPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 502 > 500');
+        });
+    });
+    describe('verifySourceCodeLimit', () => {
+        it('should not throw for source code within the limit', () => {
+            const validSourceCode = 'C '.repeat(20000 - 1);
+            expect(() => verifySourceCodeLimit(validSourceCode)).not.toThrow();
+        });
+        it('should throw for source code exceeding the limit', () => {
+            const invalidSourceCode = 'C '.repeat(20001);
+            expect(() => verifySourceCodeLimit(invalidSourceCode)).toThrow('Token limit exceeded: 20002 > 20000');
+        });
+    });
+});
+//# sourceMappingURL=limits.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"limits.test.js","sourceRoot":"","sources":["../../src/prompt/limits.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAEvG,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,GAAG,EAAE,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACvC,MAAM,CAAC,GAAG,EAAE,CAAC,uBAAuB,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,iCAAiC,CAAC,CAAC;QAClG,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACvC,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,iCAAiC,CAAC,CAAC;QACnG,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,GAAG,EAAE,CAAC,qBAAqB,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,CAAC,GAAG,EAAE,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;QACxG,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,4 @@
+/** Get codegen prompt */
+export declare function getCodeGenPrompt(): string;
+/** Get lint fix prompt */
+export declare function getLintFixPrompt(command: string, stdout: string, stderr: string): string;

package/dist/prompt/prompt-codegen.js

@@ -0,0 +1,81 @@
+import assert from 'node:assert';
+import { getSourceCode } from '../files/read-files.js';
+import { CODEGEN_TRIGGER } from './prompt-consts.js';
+import { considerAllFiles, allowFileCreate, allowFileDelete, allowDirectoryCreate, allowFileMove, explicitPrompt, dependencyTree, verbosePrompt, vision, imagen, } from '../cli/cli-params.js';
+import { getDependencyList } from '../files/find-files.js';
+import { verifyCodegenPromptLimit } from './limits.js';
+/** Get codegen prompt */
+export function getCodeGenPrompt() {
+    let codeGenFiles;
+    if (considerAllFiles) {
+        codeGenFiles = Object.keys(getSourceCode());
+    }
+    else {
+        codeGenFiles = Object.entries(getSourceCode())
+            .filter(([, { content }]) => content?.match(new RegExp(`([^'^\`]+)${CODEGEN_TRIGGER}`)))
+            .map(([path]) => path);
+    }
+    // Add logic to consider dependency tree
+    if (dependencyTree) {
+        assert(codeGenFiles.length > 0, `You must use ${CODEGEN_TRIGGER} together with --dependency-tree`);
+        const dependencyTreeFiles = new Set();
+        codeGenFiles
+            .map(getDependencyList)
+            .flat()
+            .forEach((key) => dependencyTreeFiles.add(key));
+        codeGenFiles = Array.from(dependencyTreeFiles);
+    }
+    const codeGenPrompt = (explicitPrompt ? explicitPrompt + '\n\n' : '') +
+        `${considerAllFiles
+            ? codeGenFiles.length > 0
+                ? `I have marked some files with the ${CODEGEN_TRIGGER} fragments:\n${codeGenFiles.join('\n')}`
+                : `No files are marked with ${CODEGEN_TRIGGER} fragment, so you can consider doing changes in any file.`
+            : codeGenFiles.length > 0
+                ? `Generate updates only for the following files:\n${codeGenFiles.join('\n')}`
+                : ''}
+
+${considerAllFiles
+            ? 'You are allowed to modify all files in the application regardless if they contain codegen fragments or not.'
+            : 'Do not modify files which do not contain the fragments.'}
+${allowFileCreate ? 'You are allowed to create new files.' : 'Do not create new files.'}
+${allowFileDelete
+            ? 'You are allowed to delete files, in such case add empty string as content.'
+            : 'Do not delete files.'}
+${allowDirectoryCreate ? 'You are allowed to create new directories.' : 'Do not create new directories.'}
+${allowFileMove ? 'You are allowed to move files.' : 'Do not move files.'}
+${vision ? 'You are allowed to analyze image assets.' : 'Do not analyze image assets.'}
+${imagen ? 'You are allowed to generate images.' : 'You are not allowed to generate images.'}
+`;
+    if (verbosePrompt) {
+        console.log('Code gen prompt:');
+        console.log(codeGenPrompt);
+    }
+    verifyCodegenPromptLimit(codeGenPrompt);
+    return codeGenPrompt;
+}
+/** Get lint fix prompt */
+export function getLintFixPrompt(command, stdout, stderr) {
+    const lintFixPrompt = `The following lint errors were encountered after the initial code generation:
+
+Lint command: ${command}
+Lint command stdout:
+
+\`\`\`
+${stdout}
+\`\`\`
+
+Lint command stderr:
+
+\`\`\`
+${stderr}
+\`\`\`
+        
+Please suggest changes to fix these lint errors.`;
+    if (verbosePrompt) {
+        console.log('Lint fix prompt:');
+        console.log(lintFixPrompt);
+    }
+    verifyCodegenPromptLimit(lintFixPrompt);
+    return lintFixPrompt;
+}
+//# sourceMappingURL=prompt-codegen.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-codegen.js","sourceRoot":"","sources":["../../src/prompt/prompt-codegen.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,eAAe,EACf,oBAAoB,EACpB,aAAa,EACb,cAAc,EACd,cAAc,EACd,aAAa,EACb,MAAM,EACN,MAAM,GACP,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAMvD,yBAAyB;AACzB,MAAM,UAAU,gBAAgB;IAC9B,IAAI,YAAsB,CAAC;IAC3B,IAAI,gBAAgB,EAAE,CAAC;QACrB,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;IAC9C,CAAC;SAAM,CAAC;QACN,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,aAAa,EAAqC,CAAC;aAC9E,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,aAAa,eAAe,EAAE,CAAC,CAAC,CAAC;aACvF,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC;IAED,wCAAwC;IACxC,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,gBAAgB,eAAe,kCAAkC,CAAC,CAAC;QAEnG,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC9C,YAAY;aACT,GAAG,CAAC,iBAAiB,CAAC;aACtB,IAAI,EAAE;aACN,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAClD,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,aAAa,GACjB,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,GACE,gBAAgB;YACd,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC;gBACvB,CAAC,CAAC,qCAAqC,eAAe,gBAAgB,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBAC/F,CAAC,CAAC,4BAA4B,eAAe,2DAA2D;YAC1G,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC;gBACvB,CAAC,CAAC,mDAAmD,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBAC9E,CAAC,CAAC,EACR;;EAGF,gBAAgB;YACd,CAAC,CAAC,6GAA6G;YAC/G,CAAC,CAAC,yDACN;EACE,eAAe,CAAC,CAAC,CAAC,sCAAsC,CAAC,CAAC,CAAC,0BAA0B;EAErF,eAAe;YACb,CAAC,CAAC,4EAA4E;YAC9E,CAAC,CAAC,sBACN;EACE,oBAAoB,CAAC,CAAC,CAAC,4CAA4C,CAAC,CAAC,CAAC,gCAAgC;EACtG,aAAa,CAAC,CAAC,CAAC,gCAAgC,CAAC,CAAC,CAAC,oBAAoB;EACvE,MAAM,CAAC,CAAC,CAAC,0CAA0C,CAAC,CAAC,CAAC,8BAA8B;EACpF,MAAM,CAAC,CAAC,CAAC,qCAAqC,CAAC,CAAC,CAAC,yCAAyC;CAC3F,CAAC;IAEA,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC7B,CAAC;IAED,wBAAwB,CAAC,aAAa,CAAC,CAAC;IAExC,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,0BAA0B;AAC1B,MAAM,UAAU,gBAAgB,CAAC,OAAe,EAAE,MAAc,EAAE,MAAc;IAC9E,MAAM,aAAa,GAAG;;gBAER,OAAO;;;;EAIrB,MAAM;;;;;;EAMN,MAAM;;;iDAGyC,CAAC;IAEhD,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC7B,CAAC;IAED,wBAAwB,CAAC,aAAa,CAAC,CAAC;IAExC,OAAO,aAAa,CAAC;AACvB,CAAC"}

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

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