npm - genaicode - Versions diffs - 0.6.0 → 0.7.0 - Mend

genaicode 0.6.0 → 0.7.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (217) hide show

package/dist/main/ui/frontend/index.html CHANGED Viewed

@@ -7,7 +7,7 @@
     <meta name="keywords" content="GenAIcode, AI, code generation, programming" />
     <title>GenAIcode - AI-Powered Code Generation</title>
     <link rel="icon" href="/favicon.ico" type="image/x-icon" />
-    <script type="module" crossorigin src="/assets/index-Dk_qaEQg.js"></script>
+    <script type="module" crossorigin src="/assets/index-5e3ykhy6.js"></script>
     <link rel="stylesheet" crossorigin href="/assets/index-S_ocA1xG.css">
   </head>
   <body data-security-token="__SECURITY_TOKEN__">

package/dist/main/ui/user-action-handlers.js CHANGED Viewed

@@ -1,6 +1,6 @@
 import { registerConfirmHandler, registerInputHandler } from '../common/user-actions.js';
 export function registerUserActionHandlers(service) {
-    registerInputHandler((_, message) => service.askQuestion(message, undefined).then((response) => response.answer));
+    registerInputHandler((_, message) => service.askQuestion(message, undefined));
     registerConfirmHandler(async (props) => {
         const response = await service.askQuestion(props.prompt, {
             defaultValue: props.defaultValue,

package/dist/main/ui/user-action-handlers.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"user-action-handlers.js","sourceRoot":"","sources":["../../../src/main/ui/user-action-handlers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAGzF,MAAM,UAAU,0BAA0B,CAAC,OAAgB;IACzD,oBAAoB,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,~~IAAI,~~CAAC~~,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC~~;~~IAClH~~,sBAAsB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QACrC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE;YACvD,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,YAAY,EAAE,KAAK,CAAC,YAAY;SACjC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC"}
1	+ {"version":3,"file":"user-action-handlers.js","sourceRoot":"","sources":["../../../src/main/ui/user-action-handlers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAGzF,MAAM,UAAU,0BAA0B,CAAC,OAAgB;IACzD,oBAAoB,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;IAC9E,sBAAsB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QACrC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE;YACvD,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,YAAY,EAAE,KAAK,CAAC,YAAY;SACjC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/prompt/function-calling.js CHANGED Viewed

@@ -4,13 +4,13 @@ import { getImageAssets } from './function-defs/get-image-assets.js';
 import { getCodegenSummaryDef } from './function-defs/codegen-summary.js';
 import { explanation } from './function-defs/explanation.js';
 import { generateImage } from './function-defs/generate-image.js';
-import { getAskQuestionDef } from './function-defs/ask-question.js';
+import { getAskQuestionDef, sendMessageWithImage, requestPermissions, requestFilesContent, removeFilesFromContext, contextOptimization, } from './function-defs/ask-question.js';
 import { optimizeContext } from './function-defs/optimize-context.js';
 import { setSummaries } from './function-defs/set-summaries.js';
 import { updateHistory } from './function-defs/update-history.js';
 import { readHistory } from './function-defs/read-history.js';
-import { askQuestionReflect } from './function-defs/ask-question-reflect.js';
 import { getOperationDefs } from '../operations/operations-index.js';
+import { getCodegenPlanningDef } from './function-defs/codegen-planning.js';
 /**
  * Function definitions for function calling feature
  */
@@ -19,14 +19,19 @@ export function getFunctionDefs() {
         getSourceCode,
         getImageAssets,
         getCodegenSummaryDef(),
+        getCodegenPlanningDef(),
         explanation,
         generateImage,
         getAskQuestionDef(),
+        sendMessageWithImage,
+        requestPermissions,
+        requestFilesContent,
+        removeFilesFromContext,
+        contextOptimization,
         optimizeContext,
         setSummaries,
         updateHistory,
         readHistory,
-        askQuestionReflect,
         ...getOperationDefs(),
     ].map((fd) => {
         if (!disableExplanations &&

package/dist/prompt/function-calling.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"function-calling.js","sourceRoot":"","sources":["../../src/prompt/function-calling.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAE3D,OAAO,EAAE,aAAa,EAAE,MAAM,oCAAoC,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAC1E,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAClE,OAAO,~~EAAE~~,iBAAiB,~~EAAE~~,MAAM,iCAAiC,CAAC;~~AACpE~~,OAAO,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AACtE,OAAO,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EAAE,~~kBAAkB~~,EAAE,MAAM,~~yCAAyC~~,CAAC;~~AAC7E~~,OAAO,EAAE,~~gBAAgB~~,EAAE,MAAM,~~mCAAmC~~,CAAC;~~AAErE~~;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,OAAO;QACL,aAAa;QACb,cAAc;QACd,oBAAoB,EAAE;QACtB,WAAW;QACX,aAAa;QACb,iBAAiB,EAAE;QACnB,eAAe;QACf,YAAY;QACZ,aAAa;QACb,WAAW;QACX,~~kBAAkB;QAClB,~~GAAG,gBAAgB,EAAE;KACtB,CAAC,GAAG,CAAC,CAAC,EAAe,EAAE,EAAE;QACxB,IACE,CAAC,mBAAmB;YACpB,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,WAAW;YACpC,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,EAC/C,CAAC;YACD,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,mBAAmB,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC;QAC9C,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC"}
1	+ {"version":3,"file":"function-calling.js","sourceRoot":"","sources":["../../src/prompt/function-calling.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAE3D,OAAO,EAAE,aAAa,EAAE,MAAM,oCAAoC,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAC1E,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAClE,OAAO,EACL,iBAAiB,EACjB,oBAAoB,EACpB,kBAAkB,EAClB,mBAAmB,EACnB,sBAAsB,EACtB,mBAAmB,GACpB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AACtE,OAAO,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qCAAqC,CAAC;AAE5E;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,OAAO;QACL,aAAa;QACb,cAAc;QACd,oBAAoB,EAAE;QACtB,qBAAqB,EAAE;QACvB,WAAW;QACX,aAAa;QACb,iBAAiB,EAAE;QACnB,oBAAoB;QACpB,kBAAkB;QAClB,mBAAmB;QACnB,sBAAsB;QACtB,mBAAmB;QACnB,eAAe;QACf,YAAY;QACZ,aAAa;QACb,WAAW;QACX,GAAG,gBAAgB,EAAE;KACtB,CAAC,GAAG,CAAC,CAAC,EAAe,EAAE,EAAE;QACxB,IACE,CAAC,mBAAmB;YACpB,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,WAAW;YACpC,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,EAC/C,CAAC;YACD,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,mBAAmB,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC;QAC9C,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/prompt/function-defs/ask-question.d.ts CHANGED Viewed

@@ -6,3 +6,8 @@ import { FunctionDef } from '../../ai-service/common';
  * Each actionType serves a specific purpose, ensuring clarity and proper task execution.
  */
 export declare const getAskQuestionDef: () => FunctionDef;
+export declare const requestFilesContent: FunctionDef;
+export declare const sendMessageWithImage: FunctionDef;
+export declare const removeFilesFromContext: FunctionDef;
+export declare const contextOptimization: FunctionDef;
+export declare const requestPermissions: FunctionDef;

package/dist/prompt/function-defs/ask-question.js CHANGED Viewed

@@ -1,24 +1,18 @@
 import { getRegisteredActionHandlerDescriptions, getRegisteredActionHandlers } from '../../main/plugin-loader.js';
-/**
- * Function definition for askQuestion
- *
- * Use this function to ask questions, seek clarification, request file permissions, or manage the flow of the conversation.
- * Each actionType serves a specific purpose, ensuring clarity and proper task execution.
- */
 function getActionTypeDescription() {
     const pluginDescriptions = Array.from(getRegisteredActionHandlerDescriptions().entries())
         .map(([actionType, description]) => ` - ${actionType}: ${description}`)
         .join('\n');
-    return `This value instructs the program on what should happen next. Use "requestAnswer" for analysis requests or clarifications.,
+    return `This value instructs the program on what should happen next.
 Detailed Explanation of actionTypes:
-- requestAnswer: Use for general information, clarifications, or when no specific code is needed.
-- requestPermissions: Use when additional permissions are required for actions like creating or deleting files.
-- requestFilesContent: Use specifically when needing to access or review the contents of files.
+- sendMessage: Use for general information, clarifications, or when no specific code is needed.
+- sendMessageWithImage: Use when an image is needed to provide context or additional information.
+- requestPermissions: Use **only when you lack necessary permissions** for actions like creating, deleting, or moving files, and need to request them from the user.
+- requestFilesContent: Use specifically when needing to access or review the contents of files, and it was not provided yet in any of preceeding \`getSourceCode\` function responses.
 - removeFilesFromContext: Use to remove unnecessary file contents from context, optimizing token usage.
 - confirmCodeGeneration: Use to confirm with the user before starting code generation tasks.
-- startCodeGeneration: Use only after receiving confirmation to begin code generation.
-- cancelCodeGeneration: Use if code generation should be stopped or canceled.
+- cancelCodeGeneration: Use to stop the session, and the conversation.
 - contextOptimization: Use to manage and optimize context during code generation tasks, allowing the LLM to provide guidance on what parts of the context are most relevant to keep.
 ${pluginDescriptions}`;
 }
@@ -30,85 +24,171 @@ ${pluginDescriptions}`;
  */
 export const getAskQuestionDef = () => ({
     name: 'askQuestion',
-    description: 'Use this function to ask a question, seek clarification, or manage the flow of the conversation. For analysis requests, use actionType "requestAnswer". Only proceed to code generation when explicitly instructed or after confirmation.',
+    description: `Use this function to interact with the user for various purposes.
+  The \`decisionMakingProcess\` must be provided as first parameter to ensure clarity in decision-making, and impact on selection of \`actionType\` and \`message\`.
+  The \`message\` property must align with the chosen \`actionType\`.
+  The desired format of parameters is as follows:
+  \`\`\`
+  {
+    "decisionMakingProcess": "...", // A detailed decision-making framework the assistant followed before selecting an action.",
+    "actionType": "...", // The type of action to perform.
+    "message": "..." // The message to display to the user.
+  }
+  \`\`\`
+  **IMPORTANT**: Mind the order of the parameters, as the decision-making process must be provided first to ensure clarity in decision-making.
+  `,
     parameters: {
         type: 'object',
         properties: {
+            decisionMakingProcess: {
+                type: 'string',
+                description: `A detailed decision-making framework the assistant followed before selecting an action. You must think step about step about this, this process should include the following steps:
+1. **Contextual Analysis**: Assess the current information, including available permissions, current context, and task requirements. Identify any missing elements that are critical to task completion.
+2. **Decision Justification**: State the reasoning for the proposed action, considering whether planning, clarification, or action is required. If there's any ambiguity, prefer a confirmatory action (\`confirmCodeGeneration\`).
+3. **Minimal Action Selection**: Determine the minimal action that can make progress toward the task goal. Avoid requesting unnecessary permissions or context that isn't strictly needed.
+4. **Evaluation of Action Choice**: Double-check if the selected action aligns with task requirements and user-provided constraints.
+Output this in step-by-step format to ensure clarity in decision-making.`,
+            },
             actionType: {
                 type: 'string',
                 enum: [
-                    'requestAnswer',
+                    'sendMessage',
+                    'sendMessageWithImage',
                     'requestPermissions',
                     'requestFilesContent',
                     'removeFilesFromContext',
                     'confirmCodeGeneration',
-                    'startCodeGeneration',
                     'cancelCodeGeneration',
                     'contextOptimization',
                     ...Array.from(getRegisteredActionHandlers().keys()),
                 ],
                 description: getActionTypeDescription(),
             },
-            content: {
+            message: {
                 type: 'string',
-                description: 'The message you want to display to the user. It can be a question if you need more information, an analysis result, or a confirmation request before proceeding with code generation.',
+                description: 'The message to display to the user.',
             },
-            promptNecessity: {
-                type: 'number',
-                minimum: 0,
-                maximum: 100,
-                description: 'Indicates how strongly you need a response from the user. A higher value means a stronger need for user input.',
-            },
-            requestFilesContent: {
+        },
+        required: ['decisionMakingProcess', 'actionType', 'message'],
+    },
+});
+// requestFilesContent
+export const requestFilesContent = {
+    name: 'requestFilesContent',
+    description: 'Use this function to request the content of files that are missing from the context.',
+    parameters: {
+        type: 'object',
+        properties: {
+            filePaths: {
                 type: 'array',
-                description: 'An array of absolute file paths for which you need the content. Use this when specific files are missing and you need them to proceed with the task.',
                 items: {
                     type: 'string',
                 },
+                description: 'An array of absolute file paths for which you need the content.',
             },
-            removeFilesFromContext: {
+        },
+        required: ['filePaths'],
+    },
+};
+// sendMessageWithImage
+export const sendMessageWithImage = {
+    name: 'sendMessageWithImage',
+    description: 'Use this function to send message to the user with accompanying image that will be generated using provided prompt, and optionally using other image as a context.',
+    parameters: {
+        type: 'object',
+        properties: {
+            prompt: {
+                type: 'string',
+                description: 'The prompt used to generate the image.',
+            },
+            contextImage: {
+                type: 'string',
+                description: 'Optional path to an image file used as context for image generation.',
+            },
+        },
+        required: ['prompt'],
+    },
+};
+// removeFilesFromContext
+export const removeFilesFromContext = {
+    name: 'removeFilesFromContext',
+    description: 'Use this function to remove files from the context that are no longer needed for code generation.',
+    parameters: {
+        type: 'object',
+        properties: {
+            filePaths: {
                 type: 'array',
-                description: 'An array of absolute file paths that should be removed from the context. Use this to optimize token usage by removing unnecessary file contents.',
                 items: {
                     type: 'string',
                 },
+                description: 'An array of absolute file paths to remove from the context.',
             },
-            contextOptimization: {
+        },
+        required: ['filePaths'],
+    },
+};
+// contextOptimization
+export const contextOptimization = {
+    name: 'contextOptimization',
+    description: 'Use this function to optimize the context for code generation by specifying which parts are most relevant to keep.',
+    parameters: {
+        type: 'object',
+        properties: {
+            prompt: {
                 type: 'string',
-                description: 'A prompt generated to guide the LLM in optimizing context by specifying which parts of the context are most relevant to keep. This helps in efficient management during code generation tasks.',
+                description: 'The prompt used to guide the user in optimizing the context.',
             },
-            requestPermissions: {
-                type: 'object',
-                description: 'Use this to request additional permissions needed for code generation if they are not already granted.',
-                properties: {
-                    allowDirectoryCreate: {
-                        description: 'Set to true to request permission for creating directories.',
-                        type: 'boolean',
-                    },
-                    allowFileCreate: {
-                        description: 'Set to true to request permission for creating files.',
-                        type: 'boolean',
-                    },
-                    allowFileDelete: {
-                        description: 'Set to true to request permission for deleting files.',
-                        type: 'boolean',
-                    },
-                    allowFileMove: {
-                        description: 'Set to true to request permission for moving files.',
-                        type: 'boolean',
-                    },
-                    enableVision: {
-                        description: 'Set to true to request permission for vision capabilities, using images as context for code generation.',
-                        type: 'boolean',
-                    },
-                    enableImagen: {
-                        description: 'Set to true to request permission for generating images.',
-                        type: 'boolean',
-                    },
-                },
+        },
+        required: ['prompt'],
+    },
+};
+// requestPermissions
+export const requestPermissions = {
+    name: 'requestPermissions',
+    description: 'Use this function to request additional permissions needed for code generation if not already granted.',
+    parameters: {
+        type: 'object',
+        properties: {
+            allowDirectoryCreate: {
+                description: 'Request permission to create directories.',
+                type: 'boolean',
+            },
+            allowFileCreate: {
+                description: 'Request permission to create files.',
+                type: 'boolean',
+            },
+            allowFileDelete: {
+                description: 'Request permission to delete files.',
+                type: 'boolean',
+            },
+            allowFileMove: {
+                description: 'Request permission to move files.',
+                type: 'boolean',
+            },
+            enableVision: {
+                description: 'Request permission for vision capabilities, using images as context for code generation.',
+                type: 'boolean',
+            },
+            enableImagen: {
+                description: 'Request permission to generate images.',
+                type: 'boolean',
             },
         },
-        required: ['actionType', 'content', 'promptNecessity'],
+        required: [
+            'allowDirectoryCreate',
+            'allowFileCreate',
+            'allowFileDelete',
+            'allowFileMove',
+            'enableVision',
+            'enableImagen',
+        ],
     },
-});
+};
 //# sourceMappingURL=ask-question.js.map

package/dist/prompt/function-defs/ask-question.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"ask-question.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/ask-question.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sCAAsC,EAAE,2BAA2B,EAAE,MAAM,6BAA6B,CAAC;AAElH~~;;;;;GAKG;AACH~~,SAAS,wBAAwB;IAC/B,MAAM,kBAAkB,GAAG,KAAK,CAAC,IAAI,CAAC,sCAAsC,EAAE,CAAC,OAAO,EAAE,CAAC;SACtF,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,UAAU,KAAK,WAAW,EAAE,CAAC;SACtE,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO;;;;;;;;;;;EAWP,kBAAkB,EAAE,CAAC;AACvB,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAgB,EAAE,CAAC,CAAC;IACnD,IAAI,EAAE,aAAa;IACnB,WAAW,~~EACT,2OAA2O~~;~~IAC7O~~,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE;oBACJ,~~eAAe~~;~~oBACf~~,oBAAoB;oBACpB,qBAAqB;oBACrB,wBAAwB;oBACxB,uBAAuB;oBACvB,~~qBAAqB;oBACrB,~~sBAAsB;oBACtB,qBAAqB;oBACrB,GAAG,KAAK,CAAC,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,EAAE,CAAC;iBACpD;gBACD,WAAW,EAAE,wBAAwB,EAAE;aACxC;YACD,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,WAAW,~~EACT~~,~~uLAAuL~~;~~aAC1L~~;~~YACD~~,~~eAAe~~,EAAE;~~gBACf~~,IAAI,EAAE,QAAQ;~~gBACd~~,~~OAAO~~,EAAE,~~CAAC~~;~~gBACV~~,OAAO,EAAE,~~GAAG~~;~~gBACZ~~,WAAW,~~EACT~~,~~gHAAgH~~;~~aACnH~~;~~YACD~~,~~mBAAmB~~,EAAE;~~gBACnB~~,IAAI,EAAE,~~OAAO~~;~~gBACb~~,WAAW,EACT,~~sJAAsJ~~;~~gBACxJ~~,~~KAAK~~,EAAE;~~oBACL~~,IAAI,EAAE,QAAQ;~~iBACf~~;~~aACF~~;YACD,~~sBAAsB~~,EAAE;~~gBACtB~~,IAAI,EAAE,~~OAAO~~;~~gBACb~~,WAAW,~~EACT~~,~~kJAAkJ~~;~~gBACpJ~~,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;iBACf;~~aACF~~;~~YACD~~,~~mBAAmB~~,EAAE;~~gBACnB~~,IAAI,EAAE,~~QAAQ~~;~~gBACd~~,WAAW,EACT,~~gMAAgM~~;~~aACnM~~;~~YACD~~,~~kBAAkB~~,EAAE;~~gBAClB~~,IAAI,EAAE,QAAQ;gBACd,WAAW,~~EACT~~,wGAAwG;~~gBAC1G~~,UAAU,EAAE;~~oBACV~~,oBAAoB,EAAE;~~wBACpB~~,WAAW,EAAE,~~6DAA6D~~;~~wBAC1E~~,IAAI,EAAE,SAAS;~~qBAChB~~;~~oBACD~~,eAAe,EAAE;~~wBACf~~,WAAW,EAAE,~~uDAAuD~~;~~wBACpE~~,IAAI,EAAE,SAAS;~~qBAChB~~;~~oBACD~~,eAAe,EAAE;~~wBACf~~,WAAW,EAAE,~~uDAAuD~~;~~wBACpE~~,IAAI,EAAE,SAAS;~~qBAChB~~;~~oBACD~~,aAAa,EAAE;~~wBACb~~,WAAW,EAAE,~~qDAAqD~~;~~wBAClE~~,IAAI,EAAE,SAAS;~~qBAChB~~;~~oBACD~~,YAAY,EAAE;~~wBACZ~~,WAAW,~~EACT~~,~~yGAAyG~~;~~wBAC3G~~,IAAI,EAAE,SAAS;~~qBAChB~~;~~oBACD~~,YAAY,EAAE;~~wBACZ~~,WAAW,EAAE,~~0DAA0D~~;~~wBACvE~~,IAAI,EAAE,SAAS;~~qBAChB~~;~~iBACF;aACF;~~SACF;QACD,QAAQ,EAAE,~~CAAC~~,~~YAAY~~,~~EAAE~~,~~SAAS~~,~~EAAE~~,~~iBAAiB,CAAC~~;~~KACvD~~;CACF,CAAC~~,CAAC~~"}
1	+ {"version":3,"file":"ask-question.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/ask-question.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sCAAsC,EAAE,2BAA2B,EAAE,MAAM,6BAA6B,CAAC;AAElH,SAAS,wBAAwB;IAC/B,MAAM,kBAAkB,GAAG,KAAK,CAAC,IAAI,CAAC,sCAAsC,EAAE,CAAC,OAAO,EAAE,CAAC;SACtF,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,UAAU,KAAK,WAAW,EAAE,CAAC;SACtE,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO;;;;;;;;;;;EAWP,kBAAkB,EAAE,CAAC;AACvB,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAgB,EAAE,CAAC,CAAC;IACnD,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE;;;;;;;;;;;;;;GAcZ;IACD,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,qBAAqB,EAAE;gBACrB,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE;;;;;;;;;;yEAUoD;aAClE;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE;oBACJ,aAAa;oBACb,sBAAsB;oBACtB,oBAAoB;oBACpB,qBAAqB;oBACrB,wBAAwB;oBACxB,uBAAuB;oBACvB,sBAAsB;oBACtB,qBAAqB;oBACrB,GAAG,KAAK,CAAC,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,EAAE,CAAC;iBACpD;gBACD,WAAW,EAAE,wBAAwB,EAAE;aACxC;YACD,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,qCAAqC;aACnD;SACF;QACD,QAAQ,EAAE,CAAC,uBAAuB,EAAE,YAAY,EAAE,SAAS,CAAC;KAC7D;CACF,CAAC,CAAC;AAEH,sBAAsB;AACtB,MAAM,CAAC,MAAM,mBAAmB,GAAgB;IAC9C,IAAI,EAAE,qBAAqB;IAC3B,WAAW,EAAE,sFAAsF;IACnG,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;iBACf;gBACD,WAAW,EAAE,iEAAiE;aAC/E;SACF;QACD,QAAQ,EAAE,CAAC,WAAW,CAAC;KACxB;CACF,CAAC;AAEF,uBAAuB;AACvB,MAAM,CAAC,MAAM,oBAAoB,GAAgB;IAC/C,IAAI,EAAE,sBAAsB;IAC5B,WAAW,EACT,oKAAoK;IACtK,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,wCAAwC;aACtD;YACD,YAAY,EAAE;gBACZ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,sEAAsE;aACpF;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;KACrB;CACF,CAAC;AAEF,yBAAyB;AACzB,MAAM,CAAC,MAAM,sBAAsB,GAAgB;IACjD,IAAI,EAAE,wBAAwB;IAC9B,WAAW,EAAE,mGAAmG;IAChH,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;iBACf;gBACD,WAAW,EAAE,6DAA6D;aAC3E;SACF;QACD,QAAQ,EAAE,CAAC,WAAW,CAAC;KACxB;CACF,CAAC;AAEF,sBAAsB;AACtB,MAAM,CAAC,MAAM,mBAAmB,GAAgB;IAC9C,IAAI,EAAE,qBAAqB;IAC3B,WAAW,EACT,oHAAoH;IACtH,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,8DAA8D;aAC5E;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;KACrB;CACF,CAAC;AAEF,qBAAqB;AACrB,MAAM,CAAC,MAAM,kBAAkB,GAAgB;IAC7C,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EAAE,wGAAwG;IACrH,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,oBAAoB,EAAE;gBACpB,WAAW,EAAE,2CAA2C;gBACxD,IAAI,EAAE,SAAS;aAChB;YACD,eAAe,EAAE;gBACf,WAAW,EAAE,qCAAqC;gBAClD,IAAI,EAAE,SAAS;aAChB;YACD,eAAe,EAAE;gBACf,WAAW,EAAE,qCAAqC;gBAClD,IAAI,EAAE,SAAS;aAChB;YACD,aAAa,EAAE;gBACb,WAAW,EAAE,mCAAmC;gBAChD,IAAI,EAAE,SAAS;aAChB;YACD,YAAY,EAAE;gBACZ,WAAW,EAAE,0FAA0F;gBACvG,IAAI,EAAE,SAAS;aAChB;YACD,YAAY,EAAE;gBACZ,WAAW,EAAE,wCAAwC;gBACrD,IAAI,EAAE,SAAS;aAChB;SACF;QACD,QAAQ,EAAE;YACR,sBAAsB;YACtB,iBAAiB;YACjB,iBAAiB;YACjB,eAAe;YACf,cAAc;YACd,cAAc;SACf;KACF;CACF,CAAC"}

package/dist/prompt/function-defs/codegen-planning.d.ts ADDED Viewed

@@ -0,0 +1,8 @@
+import { FunctionDef } from '../../ai-service/common.js';
+/**
+ * Function definition for codegenPlanning
+ * This function analyzes the conversation and produces a detailed implementation plan
+ * before proceeding with actual code generation. This helps ensure all necessary files
+ * are identified and dependencies are properly handled.
+ */
+export declare const getCodegenPlanningDef: () => FunctionDef;

package/dist/prompt/function-defs/codegen-planning.js ADDED Viewed

@@ -0,0 +1,50 @@
+/**
+ * Function definition for codegenPlanning
+ * This function analyzes the conversation and produces a detailed implementation plan
+ * before proceeding with actual code generation. This helps ensure all necessary files
+ * are identified and dependencies are properly handled.
+ */
+export const getCodegenPlanningDef = () => ({
+    name: 'codegenPlanning',
+    description: 'Analyzes the conversation to produce an implementation plan, identifying all affected files and dependencies before code generation.',
+    parameters: {
+        type: 'object',
+        properties: {
+            problemAnalysis: {
+                type: 'string',
+                description: 'Detailed analysis of the problem.',
+            },
+            codeChanges: {
+                type: 'string',
+                description: 'A step by step plan of code changes neccessary to solve the problem.',
+            },
+            affectedFiles: {
+                type: 'array',
+                description: 'A list of all files that need to be modified. The reason must precede the file path.',
+                items: {
+                    type: 'object',
+                    properties: {
+                        reason: {
+                            type: 'string',
+                            description: 'Reason for modifying this file.',
+                        },
+                        filePath: {
+                            type: 'string',
+                            description: 'Absolute path of the file.',
+                        },
+                        dependencies: {
+                            type: 'array',
+                            description: 'Related files that depend on or are depended upon by this file.',
+                            items: {
+                                type: 'string',
+                            },
+                        },
+                    },
+                    required: ['reason', 'filePath'],
+                },
+            },
+        },
+        required: ['problemAnalysis', 'codeChanges', 'affectedFiles'],
+    },
+});
+//# sourceMappingURL=codegen-planning.js.map

package/dist/prompt/function-defs/codegen-planning.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"codegen-planning.js","sourceRoot":"","sources":["../../../src/prompt/function-defs/codegen-planning.ts"],"names":[],"mappings":"AAEA;;;;;GAKG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,GAAgB,EAAE,CAAC,CAAC;IACvD,IAAI,EAAE,iBAAiB;IACvB,WAAW,EACT,sIAAsI;IACxI,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,eAAe,EAAE;gBACf,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,mCAAmC;aACjD;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,sEAAsE;aACpF;YACD,aAAa,EAAE;gBACb,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,sFAAsF;gBACnG,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,MAAM,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,iCAAiC;yBAC/C;wBACD,QAAQ,EAAE;4BACR,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,4BAA4B;yBAC1C;wBACD,YAAY,EAAE;4BACZ,IAAI,EAAE,OAAO;4BACb,WAAW,EAAE,iEAAiE;4BAC9E,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;6BACf;yBACF;qBACF;oBACD,QAAQ,EAAE,CAAC,QAAQ,EAAE,UAAU,CAAC;iBACjC;aACF;SACF;QACD,QAAQ,EAAE,CAAC,iBAAiB,EAAE,aAAa,EAAE,eAAe,CAAC;KAC9D;CACF,CAAC,CAAC"}

package/dist/prompt/function-defs/codegen-summary.js CHANGED Viewed

@@ -6,9 +6,9 @@ export const getCodegenSummaryDef = () => ({
     name: 'codegenSummary',
     description: 'This function is called with a summary of proposed updates.\n' +
         '- `explanation`: A general explanation of the planned code generation updates or reasoning for no code changes.\n' +
-        '- `fileUpdates`: A list of proposed file updates that will be subject to subsequent code generation requests.\n' +
+        '- `fileUpdates`: A list of proposed file updates, including all affected files and dependencies.\n' +
         '- `contextPaths`: A list of file paths that make sense to use as context for code generation requests.\n' +
-        'It is critically important to adhere to the schema of parameters.',
+        'It is critically important to adhere to the schema of parameters and include all relevant files.',
     parameters: {
         type: 'object',
         properties: {

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

	@@ -1 +1 @@
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"}
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,oGAAoG;QACpG,0GAA0G;QAC1G,kGAAkG;IACpG,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/prompt/function-defs/get-source-code.js CHANGED Viewed

@@ -3,8 +3,38 @@
  */
 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.',
+    description: `This function returns source code of the application in the following format:
+\`\`\`
+{
+  [directoryPath: string]: {
+    [filePath: string]: {content: string | null} | {summary: string};
+  };
+}
+\`\`\`
+Some keys may not provide content. Some keys may provide a short summary of content.
+Here is an example of the returned object:
+\`\`\`
+{
+  '/path/to/directory': {
+    'file1.js': {content: 'console.log('Hello, World!');'},
+  },
+  '/path/to/directory/sub1/sub2': {
+    'file2.js': {summary: 'This file contains a simple log statement.'},
+  }
+}
+\`\`\`
+How to understand this object:
+- there are 2 files
+- the first file has content: 'console.log('Hello, World!');'
+- the second file has no content, but a summary: 'This file contains a simple log statement.'
+- path of first file: '/path/to/directory/file1.js'
+- path of second file: '/path/to/directory/sub1/sub2/file2.js'
+This function is used to provide context to the AI model.
+    `,
     parameters: {
         type: 'object',
         properties: {

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

	@@ -1 +1 @@
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"}
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,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA+BV;IACH,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/prompt/function-defs/optimize-context.js CHANGED Viewed

@@ -3,7 +3,39 @@
  */
 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.',
+    description: `This function helps prioritize files based on their importance to the user prompt.
+Example:
+1. User prompt: "I want to add a new test to util unit tests."
+2. Assistant makes a function call for \`getSourceCode\` to get the source code.
+3. User provides the source code in \`getSourceCode\` function response:
+\`\`\`
+{
+  '/path/to/directory': {
+    'something.js': ['console.log('Hello, World!');'],
+  },
+  '/path/to/directory/sub1/sub2': {
+    'util-tests.js': [null, 'This file contains unit tests.'],
+  }
+}
+\`\`\`
+4. Assistant calls the \`optimizeContext\` function to narrow the context to relevant files:
+\`\`\`
+{
+  userPrompt: "I want to add a new test to util unit tests.",
+  optimizedContext: [
+    {
+      filePath: '/path/to/directory/sub1/sub2/util-tests.js',
+      relevance: 0.8,
+    },
+    {
+      filePath: '/path/to/directory/something.js',
+      relevance: 0.2,
+    }
+  ]
+\`\`\`
+This means that the file \`something.js\` file is less relevant to the user prompt, and \`util-tests.js\` is more relevant.`,
     parameters: {
         type: 'object',
         properties: {
@@ -27,21 +59,12 @@ export const optimizeContext = {
                             minimum: 0,
                             maximum: 1,
                         },
-                        tokenCount: {
-                            type: 'integer',
-                            description: 'The estimated token count for the file content.',
-                            minimum: 0,
-                        },
                     },
-                    required: ['filePath', 'relevance', 'tokenCount'],
+                    required: ['filePath', 'relevance'],
                 },
             },
-            totalTokenCount: {
-                type: 'number',
-                description: 'The estimated token count for the entire context.',
-            },
         },
-        required: ['userPrompt', 'optimizedContext', 'totalTokenCount'],
+        required: ['userPrompt', 'optimizedContext'],
     },
 };
 //# sourceMappingURL=optimize-context.js.map

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

	@@ -1 +1 @@
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"}
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,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;4HAgC6G;IAC1H,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;qBACF;oBACD,QAAQ,EAAE,CAAC,UAAU,EAAE,WAAW,CAAC;iBACpC;aACF;SACF;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,kBAAkB,CAAC;KAC7C;CACF,CAAC"}

package/dist/prompt/limits.js CHANGED Viewed

@@ -1,7 +1,7 @@
 import assert from 'node:assert';
 import { estimateTokenCount } from './token-estimator.js';
-const SYSTEM_PROMPT_LIMIT = 2800;
-const CODEGEN_PROMPT_LIMIT = 850;
+const SYSTEM_PROMPT_LIMIT = 3500;
+const CODEGEN_PROMPT_LIMIT = 1000;
 const SOURCE_CODE_LIMIT = 100000;
 function verifyPromptLimit(promptType, prompt, limit) {
     const tokenCount = estimateTokenCount(prompt);

package/dist/prompt/limits.js.map CHANGED Viewed

	@@ -1 +1 @@
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"}
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,IAAI,CAAC;AAClC,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/prompt/limits.test.js CHANGED Viewed

@@ -7,8 +7,8 @@ describe('Prompt Limits', () => {
             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');
+            const invalidPrompt = 'A '.repeat(6000);
+            expect(() => verifySystemPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 3900 > 3500');
         });
     });
     describe('verifyCodegenPromptLimit', () => {
@@ -18,7 +18,7 @@ describe('Prompt Limits', () => {
         });
         it('should throw for a prompt exceeding the limit', () => {
             const invalidPrompt = 'B '.repeat(10000);
-            expect(() => verifyCodegenPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 6500 > 850');
+            expect(() => verifyCodegenPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 6500 > 1000');
         });
     });
     describe('verifySourceCodeLimit', () => {

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

	@@ -1 +1 @@
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"}
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,mCAAmC,CAAC,CAAC;QACrG,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"}