genaicode 0.5.1 → 0.6.1

package/dist/src/prompt/function-defs/codegen-summary.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen-summary.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/codegen-summary.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,uBAAuB,EAAE,MAAM,6BAA6B,CAAC;AAEtE;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,GAAgB,EAAE,CAAC,CAAC;IACtD,IAAI,EAAE,gBAAgB;IACtB,WAAW,EACT,+DAA+D;QAC/D,mHAAmH;QACnH,iHAAiH;QACjH,0GAA0G;QAC1G,mEAAmE;IACrE,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0FAA0F;aACxG;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,sFAAsF;gBACnG,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EACT,8JAA8J;oBAChK,UAAU,EAAE;wBACV,QAAQ,EAAE;4BACR,IAAI,EAAE,QAAQ;4BACd,WAAW,EACT,gGAAgG;yBACnG;wBACD,MAAM,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,WAAW,EACT,wJAAwJ;yBAC3J;wBACD,cAAc,EAAE;4BACd,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE;gCACJ,YAAY;gCACZ,YAAY;gCACZ,WAAW;gCACX,YAAY;gCACZ,iBAAiB;gCACjB,UAAU;gCACV,eAAe;gCACf,cAAc;gCACd,YAAY;gCACZ,aAAa;gCACb,uBAAuB;gCACvB,GAAG,uBAAuB,EAAE,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC;6BACpE;4BACD,WAAW,EAAE,+DAA+D;yBAC7E;wBACD,WAAW,EAAE;4BACX,IAAI,EAAE,QAAQ;4BACd,WAAW,EACT,oIAAoI;4BACtI,OAAO,EAAE,GAAG;4BACZ,OAAO,EAAE,GAAG;yBACb;wBACD,KAAK,EAAE;4BACL,IAAI,EAAE,SAAS;4BACf,WAAW,EACT,4JAA4J;yBAC/J;wBACD,kBAAkB,EAAE;4BAClB,IAAI,EAAE,OAAO;4BACb,WAAW,EACT,8JAA8J;4BAChK,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;yBAC1B;qBACF;oBACD,QAAQ,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,gBAAgB,CAAC;iBACnD;aACF;YACD,YAAY,EAAE;gBACZ,IAAI,EAAE,OAAO;gBACb,WAAW,EACT,gLAAgL;gBAClL,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;iBACf;aACF;SACF;QACD,QAAQ,EAAE,CAAC,aAAa,EAAE,aAAa,EAAE,cAAc,CAAC;KACzD;CACF,CAAC,CAAC"}

package/dist/src/prompt/function-defs/explanation.d.ts

@@ -0,0 +1,5 @@
+import { FunctionDef } from '../../ai-service/common';
+/**
+ * Function definition for explanation
+ */
+export declare const explanation: FunctionDef;

package/dist/src/prompt/function-defs/explanation.js

@@ -0,0 +1,18 @@
+/**
+ * Function definition for explanation
+ */
+export const explanation = {
+    name: 'explanation',
+    description: 'Explain the reasoning behind the suggested code changes or reasoning for lack of code changes',
+    parameters: {
+        type: 'object',
+        properties: {
+            text: {
+                type: 'string',
+                description: 'The explanation text',
+            },
+        },
+        required: ['text'],
+    },
+};
+//# sourceMappingURL=explanation.js.map

package/dist/src/prompt/function-defs/explanation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"explanation.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/explanation.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAgB;IACtC,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,+FAA+F;IAC5G,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,sBAAsB;aACpC;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB;CACF,CAAC"}

package/dist/src/prompt/function-defs/generate-image.d.ts

@@ -0,0 +1,5 @@
+import { FunctionDef } from '../../ai-service/common';
+/**
+ * Function definition for generateImage
+ */
+export declare const generateImage: FunctionDef;

package/dist/src/prompt/function-defs/generate-image.js

@@ -0,0 +1,42 @@
+/**
+ * Function definition for generateImage
+ */
+export const generateImage = {
+    name: 'generateImage',
+    description: 'Generate an image using AI service and save it as a file.',
+    parameters: {
+        type: 'object',
+        properties: {
+            explanation: {
+                type: 'string',
+                description: 'The explanation of the reasoning behind generating this image',
+            },
+            prompt: {
+                type: 'string',
+                description: 'The prompt that will be used to generate the image. This prompt must be detailed, it will be used by image generation model.',
+            },
+            filePath: {
+                type: 'string',
+                description: 'The file path to save the generated image.',
+            },
+            contextImagePath: {
+                type: 'string',
+                description: 'Path to a image file that will be used as a context for image generation. It is useful if there is a need to edit an image with genAI.',
+            },
+            width: {
+                type: 'number',
+                description: 'width of the image',
+            },
+            height: {
+                type: 'number',
+                description: 'height of the image',
+            },
+            cheap: {
+                type: 'boolean',
+                description: 'true value means that the prompt will be executed with cheaper model, which work faster, but provides lower quality results, so please use it only in situation when lower quality results are acceptable for the prompt.',
+            },
+        },
+        required: ['prompt', 'filePath', 'width', 'height'],
+    },
+};
+//# sourceMappingURL=generate-image.js.map

package/dist/src/prompt/function-defs/generate-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-image.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/generate-image.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAgB;IACxC,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,2DAA2D;IACxE,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,+DAA+D;aAC7E;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EACT,8HAA8H;aACjI;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4CAA4C;aAC1D;YACD,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,WAAW,EACT,wIAAwI;aAC3I;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,oBAAoB;aAClC;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,qBAAqB;aACnC;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,SAAS;gBACf,WAAW,EACT,2NAA2N;aAC9N;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,CAAC;KACpD;CACF,CAAC"}

package/dist/src/prompt/function-defs/get-image-assets.d.ts

@@ -0,0 +1,5 @@
+import { FunctionDef } from '../../ai-service/common';
+/**
+ * Function definition for getImageAssets
+ */
+export declare const getImageAssets: FunctionDef;

package/dist/src/prompt/function-defs/get-image-assets.js

@@ -0,0 +1,21 @@
+/**
+ * Function definition for getImageAssets
+ */
+export const getImageAssets = {
+    name: 'getImageAssets',
+    description: 'This function returns a map of application image assets. This map contains absolute file path, and basic metadata information. It does not contain contents. Contents must be requested using dedicated tool.',
+    parameters: {
+        type: 'object',
+        properties: {
+            filePaths: {
+                type: 'array',
+                description: 'An array of absolute paths of files that should be used to provided context.',
+                items: {
+                    type: 'string',
+                },
+            },
+        },
+        required: [],
+    },
+};
+//# sourceMappingURL=get-image-assets.js.map

package/dist/src/prompt/function-defs/get-image-assets.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-image-assets.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/get-image-assets.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAgB;IACzC,IAAI,EAAE,gBAAgB;IACtB,WAAW,EACT,+MAA+M;IACjN,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,8EAA8E;gBAC3F,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;iBACf;aACF;SACF;QACD,QAAQ,EAAE,EAAE;KACb;CACF,CAAC"}

package/dist/src/prompt/function-defs/get-source-code.d.ts

@@ -0,0 +1,5 @@
+import { FunctionDef } from '../../ai-service/common';
+/**
+ * Function definition for getSourceCode
+ */
+export declare const getSourceCode: FunctionDef;

package/dist/src/prompt/function-defs/get-source-code.js

@@ -0,0 +1,22 @@
+/**
+ * Function definition for getSourceCode
+ */
+export const getSourceCode = {
+    name: 'getSourceCode',
+    description: 'This function returns source code of the application in Map format, where absolute file path is the key, and the value is an object, where one of the properties may be the content of the file.' +
+        'Some keys may not provide content. Some keys may provide a short summary of content. This function can be called only once during the conversation, and only if suggested by the user.',
+    parameters: {
+        type: 'object',
+        properties: {
+            filePaths: {
+                type: 'array',
+                description: 'An array of absolute paths of files that should be used to provided context.',
+                items: {
+                    type: 'string',
+                },
+            },
+        },
+        required: [],
+    },
+};
+//# sourceMappingURL=get-source-code.js.map

package/dist/src/prompt/function-defs/get-source-code.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-source-code.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/get-source-code.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAgB;IACxC,IAAI,EAAE,eAAe;IACrB,WAAW,EACT,kMAAkM;QAClM,wLAAwL;IAC1L,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,8EAA8E;gBAC3F,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;iBACf;aACF;SACF;QACD,QAAQ,EAAE,EAAE;KACb;CACF,CAAC"}

package/dist/src/prompt/function-defs/optimize-context.d.ts

@@ -0,0 +1,5 @@
+import { FunctionDef } from '../../ai-service/common';
+/**
+ * Function definition for optimizeContext
+ */
+export declare const optimizeContext: FunctionDef;

package/dist/src/prompt/function-defs/optimize-context.js

@@ -0,0 +1,47 @@
+/**
+ * Function definition for optimizeContext
+ */
+export const optimizeContext = {
+    name: 'optimizeContext',
+    description: 'This function narrows the context of code generation to a list of relevant files, including their estimated token counts and relevance scores. It helps prioritize files based on their importance to the user prompt and manages token usage.',
+    parameters: {
+        type: 'object',
+        properties: {
+            userPrompt: {
+                type: 'string',
+                description: "The user's original prompt.",
+            },
+            optimizedContext: {
+                type: 'array',
+                description: 'An array of objects representing relevant files, their relevance scores, and token counts.',
+                items: {
+                    type: 'object',
+                    properties: {
+                        filePath: {
+                            type: 'string',
+                            description: 'The absolute file path of a relevant file.',
+                        },
+                        relevance: {
+                            type: 'number',
+                            description: 'A score from 0 to 1 indicating the relevance of the file to the user prompt.',
+                            minimum: 0,
+                            maximum: 1,
+                        },
+                        tokenCount: {
+                            type: 'integer',
+                            description: 'The estimated token count for the file content.',
+                            minimum: 0,
+                        },
+                    },
+                    required: ['filePath', 'relevance', 'tokenCount'],
+                },
+            },
+            totalTokenCount: {
+                type: 'number',
+                description: 'The estimated token count for the entire context.',
+            },
+        },
+        required: ['userPrompt', 'optimizedContext', 'totalTokenCount'],
+    },
+};
+//# sourceMappingURL=optimize-context.js.map

package/dist/src/prompt/function-defs/optimize-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"optimize-context.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/optimize-context.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAgB;IAC1C,IAAI,EAAE,iBAAiB;IACvB,WAAW,EACT,gPAAgP;IAClP,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,6BAA6B;aAC3C;YACD,gBAAgB,EAAE;gBAChB,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,4FAA4F;gBACzG,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,QAAQ,EAAE;4BACR,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,4CAA4C;yBAC1D;wBACD,SAAS,EAAE;4BACT,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,8EAA8E;4BAC3F,OAAO,EAAE,CAAC;4BACV,OAAO,EAAE,CAAC;yBACX;wBACD,UAAU,EAAE;4BACV,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,iDAAiD;4BAC9D,OAAO,EAAE,CAAC;yBACX;qBACF;oBACD,QAAQ,EAAE,CAAC,UAAU,EAAE,WAAW,EAAE,YAAY,CAAC;iBAClD;aACF;YACD,eAAe,EAAE;gBACf,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,mDAAmD;aACjE;SACF;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,kBAAkB,EAAE,iBAAiB,CAAC;KAChE;CACF,CAAC"}

package/dist/src/prompt/function-defs/read-history.d.ts

@@ -0,0 +1,2 @@
+import { FunctionDef } from '../../ai-service/common';
+export declare const readHistory: FunctionDef;

package/dist/src/prompt/function-defs/read-history.js

@@ -0,0 +1,14 @@
+export const readHistory = {
+    name: 'readHistory',
+    description: 'This function reads the history from the cache and returns it as a string.',
+    parameters: {
+        type: 'object',
+        properties: {
+            dateTime: {
+                type: 'string',
+            },
+        },
+        required: [],
+    },
+};
+//# sourceMappingURL=read-history.js.map

package/dist/src/prompt/function-defs/read-history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"read-history.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/read-history.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,WAAW,GAAgB;IACtC,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,4EAA4E;IACzF,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;aACf;SACF;QACD,QAAQ,EAAE,EAAE;KACb;CACF,CAAC"}

package/dist/src/prompt/function-defs/set-summaries.d.ts

@@ -0,0 +1,5 @@
+import { FunctionDef } from '../../ai-service/common';
+/**
+ * Function definition for setSummaries
+ */
+export declare const setSummaries: FunctionDef;

package/dist/src/prompt/function-defs/set-summaries.js

@@ -0,0 +1,32 @@
+/**
+ * Function definition for setSummaries
+ */
+export const setSummaries = {
+    name: 'setSummaries',
+    description: 'Use this function to save summaries of files',
+    parameters: {
+        type: 'object',
+        properties: {
+            summaries: {
+                type: 'array',
+                description: 'An array of results, each corresponding to an analyzed file, containing the absolute file path, and a brief summary.',
+                items: {
+                    type: 'object',
+                    properties: {
+                        filePath: {
+                            type: 'string',
+                            description: 'The absolute file path of the analyzed file.',
+                        },
+                        summary: {
+                            type: 'string',
+                            description: "A summary of the file's content, highlighting its main purpose, functionality, or details which may be useful for context optimization.",
+                        },
+                    },
+                    required: ['filePath', 'summary'],
+                },
+            },
+        },
+        required: ['summaries'],
+    },
+};
+//# sourceMappingURL=set-summaries.js.map

package/dist/src/prompt/function-defs/set-summaries.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"set-summaries.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/set-summaries.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAgB;IACvC,IAAI,EAAE,cAAc;IACpB,WAAW,EAAE,8CAA8C;IAC3D,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,OAAO;gBACb,WAAW,EACT,sHAAsH;gBACxH,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,QAAQ,EAAE;4BACR,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,8CAA8C;yBAC5D;wBACD,OAAO,EAAE;4BACP,IAAI,EAAE,QAAQ;4BACd,WAAW,EACT,yIAAyI;yBAC5I;qBACF;oBACD,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC;iBAClC;aACF;SACF;QACD,QAAQ,EAAE,CAAC,WAAW,CAAC;KACxB;CACF,CAAC"}

package/dist/src/prompt/function-defs/update-history.d.ts

@@ -0,0 +1,5 @@
+import { FunctionDef } from '../../ai-service/common';
+/**
+ * Function definition for updateHistory
+ */
+export declare const updateHistory: FunctionDef;

package/dist/src/prompt/function-defs/update-history.js

@@ -0,0 +1,22 @@
+/**
+ * Function definition for updateHistory
+ */
+export const updateHistory = {
+    name: 'updateHistory',
+    description: 'Update the history.',
+    parameters: {
+        type: 'object',
+        properties: {
+            recentConversationSummary: {
+                type: 'string',
+                description: 'Summary of recent conversation',
+            },
+            newHistoryContent: {
+                type: 'string',
+                description: 'The new history content, combination of recent conversation summary, and current history',
+            },
+        },
+        required: ['recentConversationSummary', 'newHistoryContent'],
+    },
+};
+//# sourceMappingURL=update-history.js.map

package/dist/src/prompt/function-defs/update-history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-history.js","sourceRoot":"","sources":["../../../../src/prompt/function-defs/update-history.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAgB;IACxC,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,qBAAqB;IAClC,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,yBAAyB,EAAE;gBACzB,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,gCAAgC;aAC9C;YACD,iBAAiB,EAAE;gBACjB,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0FAA0F;aACxG;SACF;QACD,QAAQ,EAAE,CAAC,2BAA2B,EAAE,mBAAmB,CAAC;KAC7D;CACF,CAAC"}

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

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

package/dist/src/prompt/limits.js

@@ -0,0 +1,21 @@
+import assert from 'node:assert';
+import { estimateTokenCount } from './token-estimator.js';
+const SYSTEM_PROMPT_LIMIT = 2800;
+const CODEGEN_PROMPT_LIMIT = 850;
+const SOURCE_CODE_LIMIT = 100000;
+function verifyPromptLimit(promptType, prompt, limit) {
+    const tokenCount = estimateTokenCount(prompt);
+    console.log(`${promptType} prompt estimated token count: ${tokenCount}`);
+    assert(tokenCount <= limit, `Token limit exceeded: ${tokenCount} > ${limit}`);
+    return tokenCount;
+}
+export function verifySystemPromptLimit(systemPrompt) {
+    return verifyPromptLimit('system', systemPrompt, SYSTEM_PROMPT_LIMIT);
+}
+export function verifyCodegenPromptLimit(codeGenPrompt) {
+    return verifyPromptLimit('codegen', codeGenPrompt, CODEGEN_PROMPT_LIMIT);
+}
+export function verifySourceCodeLimit(sourceCode) {
+    return verifyPromptLimit('sourceCode', sourceCode, SOURCE_CODE_LIMIT);
+}
+//# sourceMappingURL=limits.js.map

package/dist/src/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;AACjC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D,MAAM,mBAAmB,GAAG,IAAI,CAAC;AACjC,MAAM,oBAAoB,GAAG,GAAG,CAAC;AACjC,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAEjC,SAAS,iBAAiB,CAAC,UAAkB,EAAE,MAAc,EAAE,KAAa;IAC1E,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAC9C,OAAO,CAAC,GAAG,CAAC,GAAG,UAAU,kCAAkC,UAAU,EAAE,CAAC,CAAC;IACzE,MAAM,CAAC,UAAU,IAAI,KAAK,EAAE,yBAAyB,UAAU,MAAM,KAAK,EAAE,CAAC,CAAC;IAC9E,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,YAAoB;IAC1D,OAAO,iBAAiB,CAAC,QAAQ,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,aAAqB;IAC5D,OAAO,iBAAiB,CAAC,SAAS,EAAE,aAAa,EAAE,oBAAoB,CAAC,CAAC;AAC3E,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,UAAkB;IACtD,OAAO,iBAAiB,CAAC,YAAY,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAAC;AACxE,CAAC"}

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

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

package/dist/src/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(5000);
+            expect(() => verifySystemPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 3250 > 2800');
+        });
+    });
+    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(10000);
+            expect(() => verifyCodegenPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 6500 > 850');
+        });
+    });
+    describe('verifySourceCodeLimit', () => {
+        it('should not throw for source code within the limit', () => {
+            const validSourceCode = 'C '.repeat(40000 - 1);
+            expect(() => verifySourceCodeLimit(validSourceCode)).not.toThrow();
+        });
+        it('should throw for source code exceeding the limit', () => {
+            const invalidSourceCode = 'C '.repeat(400001);
+            expect(() => verifySourceCodeLimit(invalidSourceCode)).toThrow('Token limit exceeded: 260001 > 100000');
+        });
+    });
+});
+//# sourceMappingURL=limits.test.js.map

package/dist/src/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,IAAI,CAAC,CAAC;YACxC,MAAM,CAAC,GAAG,EAAE,CAAC,uBAAuB,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;QACpG,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,KAAK,CAAC,CAAC;YACzC,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;QACpG,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,MAAM,CAAC,CAAC;YAC9C,MAAM,CAAC,GAAG,EAAE,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,9 @@
+import { CodegenOptions } from '../main/codegen-types.js';
+export interface CodegenPrompt {
+    prompt: string;
+    options: CodegenOptions;
+}
+/** Get codegen prompt */
+export declare function getCodeGenPrompt(options: CodegenOptions): CodegenPrompt;
+/** Get lint fix prompt */
+export declare function getLintFixPrompt(command: string, { verbose }: CodegenOptions, stdout: string, stderr: string): string;

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

@@ -0,0 +1,117 @@
+import assert from 'node:assert';
+import { getSourceCode } from '../files/read-files.js';
+import { CODEGEN_TRIGGER } from './prompt-consts.js';
+import { getDependencyList } from '../files/find-files.js';
+import { verifyCodegenPromptLimit } from './limits.js';
+import { importantContext } from '../main/config.js';
+/** Get codegen prompt */
+export function getCodeGenPrompt(options) {
+    let { explicitPrompt } = options;
+    const { taskFile, considerAllFiles, allowFileCreate, allowFileDelete, allowDirectoryCreate, allowFileMove, vision, imagen, dependencyTree, verbose, askQuestion, interactive, ui, } = options;
+    assert(!explicitPrompt || !taskFile, 'Both taskFile and explicitPrompt are not allowed');
+    if (taskFile) {
+        explicitPrompt = `I want you to perform a coding task. The task is described in the ${taskFile} file. Use those instructions.`;
+    }
+    let codeGenFiles = [];
+    if (!considerAllFiles) {
+        codeGenFiles = Object.entries(getSourceCode({ taskFile, forceAll: true }, options))
+            .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 importantTextPrompts = importantContext.textPrompts?.map((prompt) => prompt.content).join('\n\n') || '';
+    const requestPermissionText = askQuestion && (interactive || ui) ? ' (request permission via `askQuestion`)' : '';
+    const taskPrompt = explicitPrompt
+        ? 'This is your task: ' + explicitPrompt + '\n\n'
+        : codeGenFiles.length > 0
+            ? `I have marked some files with the ${CODEGEN_TRIGGER} fragments:\n${codeGenFiles.join('\n')}\nYour task is to generate updates for those files according to the ${CODEGEN_TRIGGER} comments/context.`
+            : `No files are marked with the ${CODEGEN_TRIGGER} fragment, so you can consider making changes in any file${askQuestion ? ', but you should consult me by asking a question first.' : ''}.`;
+    const codeGenPrompt = `${taskPrompt}
+
+Before proceeding with code generation, please:
+
+1. **Analyze the task and requirements**.
+
+2. **Use the \`askQuestion\` function** to seek clarification if needed. For analysis requests, use the 'requestAnswer' actionType.
+
+3. **When you believe code changes are necessary**, use the 'confirmCodeGeneration' actionType to confirm with the user before proceeding.
+
+4. **Only after confirmation**, summarize the plan of updates by calling the \`codegenSummary\` function with the appropriate arguments.
+
+   - Ensure that you include:
+     - **\`explanation\`**: A brief description of the planned changes or reasoning for no changes.
+     - **\`fileUpdates\`**: A list of proposed file updates, each with required properties:
+       - **\`path\`**: Absolute file path to be updated.
+       - **\`updateToolName\`**: The tool to be used for the update (e.g., \`createFile\`, \`updateFile\`).
+       - **\`prompt\`**: A detailed prompt summarizing the planned changes for this file.
+       - **Other Properties**: Include any other necessary properties as per the \`codegenSummary\` function definition.
+     - **\`contextPaths\`**: A list of file paths that should be used as context for the code generation requests.
+
+5. **Context Optimization**: Utilize the \`contextOptimization\` actionType to manage and optimize the context during code generation tasks. Generate a prompt using the \`contextOptimization\` property to guide the LLM in determining which parts of the context are most relevant to keep, enhancing efficiency and focus.
+
+6. **Check ActionType Selection**: Always verify that the correct \`actionType\` is being used, especially before requesting file content, to ensure the 'requestFilesContent' actionType is applied. This helps maintain vigilance and accuracy in task execution.
+
+## My Requirements for Task Execution:
+
+${importantTextPrompts ? `${importantTextPrompts}\n` : ''}
+${considerAllFiles
+        ? 'You are allowed to modify all files in the application regardless of whether they contain codegen fragments.'
+        : 'Do not modify files which do not contain the fragments.'}
+${allowFileCreate ? 'You are allowed to create new files.' : 'Do not create new files. (request permission via `askQuestion`)'}
+${allowFileDelete
+        ? 'You are allowed to delete files; in such cases, add an empty string as content.'
+        : 'Do not delete files. (request permission via `askQuestion`)'}
+${allowDirectoryCreate
+        ? 'You are allowed to create new directories.'
+        : `Do not create new directories.${requestPermissionText}`}
+${allowFileMove ? 'You are allowed to move files.' : 'Do not move files. (request permission via `askQuestion`)'}
+${vision ? 'You are allowed to analyze image assets.' : 'Do not analyze image assets. (request permission via `askQuestion`)'}
+${imagen
+        ? 'You are allowed to generate images.'
+        : 'You are not allowed to generate images. (request permission via `askQuestion`)'}
+
+Remember to optimize token usage by reducing context when appropriate during askQuestion conversations. This will help avoid hitting rate limits and improve overall efficiency.
+
+ Start from generating codegen summary, this summary will be used as a context to generate updates, so make sure that it contains useful information.`;
+    if (verbose) {
+        console.log('Code gen prompt:');
+        console.log(codeGenPrompt);
+    }
+    verifyCodegenPromptLimit(codeGenPrompt);
+    return { prompt: codeGenPrompt, options };
+}
+/** Get lint fix prompt */
+export function getLintFixPrompt(command, { verbose }, 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. Use the 'requestAnswer' actionType if you need any clarifications before proposing fixes.`;
+    if (verbose) {
+        console.log('Lint fix prompt:');
+        console.log(lintFixPrompt);
+    }
+    verifyCodegenPromptLimit(lintFixPrompt);
+    return lintFixPrompt;
+}
+//# sourceMappingURL=prompt-codegen.js.map

package/dist/src/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,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAYrD,yBAAyB;AACzB,MAAM,UAAU,gBAAgB,CAAC,OAAuB;IACtD,IAAI,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;IACjC,MAAM,EACJ,QAAQ,EACR,gBAAgB,EAChB,eAAe,EACf,eAAe,EACf,oBAAoB,EACpB,aAAa,EACb,MAAM,EACN,MAAM,EACN,cAAc,EACd,OAAO,EACP,WAAW,EACX,WAAW,EACX,EAAE,GACH,GAAG,OAAO,CAAC;IAEZ,MAAM,CAAC,CAAC,cAAc,IAAI,CAAC,QAAQ,EAAE,kDAAkD,CAAC,CAAC;IAEzF,IAAI,QAAQ,EAAE,CAAC;QACb,cAAc,GAAG,qEAAqE,QAAQ,gCAAgC,CAAC;IACjI,CAAC;IAED,IAAI,YAAY,GAAa,EAAE,CAAC;IAChC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,YAAY,GAAG,MAAM,CAAC,OAAO,CAC3B,aAAa,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,OAAO,CAAoC,CACxF;aACE,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,oBAAoB,GAAG,gBAAgB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IAC9G,MAAM,qBAAqB,GAAG,WAAW,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,yCAAyC,CAAC,CAAC,CAAC,EAAE,CAAC;IAElH,MAAM,UAAU,GAAG,cAAc;QAC/B,CAAC,CAAC,qBAAqB,GAAG,cAAc,GAAG,MAAM;QACjD,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC;YACvB,CAAC,CAAC,qCAAqC,eAAe,gBAAgB,YAAY,CAAC,IAAI,CACnF,IAAI,CACL,uEAAuE,eAAe,oBAAoB;YAC7G,CAAC,CAAC,gCAAgC,eAAe,4DAC7C,WAAW,CAAC,CAAC,CAAC,yDAAyD,CAAC,CAAC,CAAC,EAC5E,GAAG,CAAC;IAEV,MAAM,aAAa,GAAG,GAAG,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2BnC,oBAAoB,CAAC,CAAC,CAAC,GAAG,oBAAoB,IAAI,CAAC,CAAC,CAAC,EAAE;EAEvD,gBAAgB;QACd,CAAC,CAAC,8GAA8G;QAChH,CAAC,CAAC,yDACN;EACE,eAAe,CAAC,CAAC,CAAC,sCAAsC,CAAC,CAAC,CAAC,iEAAiE;EAE5H,eAAe;QACb,CAAC,CAAC,iFAAiF;QACnF,CAAC,CAAC,6DACN;EAEE,oBAAoB;QAClB,CAAC,CAAC,4CAA4C;QAC9C,CAAC,CAAC,iCAAiC,qBAAqB,EAC5D;EACE,aAAa,CAAC,CAAC,CAAC,gCAAgC,CAAC,CAAC,CAAC,2DAA2D;EAC9G,MAAM,CAAC,CAAC,CAAC,0CAA0C,CAAC,CAAC,CAAC,qEAAqE;EAE3H,MAAM;QACJ,CAAC,CAAC,qCAAqC;QACvC,CAAC,CAAC,gFACN;;;;sJAIsJ,CAAC;IAErJ,IAAI,OAAO,EAAE,CAAC;QACZ,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,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC;AAC5C,CAAC;AAED,0BAA0B;AAC1B,MAAM,UAAU,gBAAgB,CAAC,OAAe,EAAE,EAAE,OAAO,EAAkB,EAAE,MAAc,EAAE,MAAc;IAC3G,MAAM,aAAa,GAAG;;gBAER,OAAO;;;;EAIrB,MAAM;;;;;;EAMN,MAAM;;;2IAGmI,CAAC;IAE1I,IAAI,OAAO,EAAE,CAAC;QACZ,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/src/prompt/prompt-codegen.test.d.ts

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