genaicode 0.0.37 → 0.2.0

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

@@ -1,87 +0,0 @@
-import { temperature, cheap, vision } from '../../cli/cli-params.js';
-
-/**
- * Function definition for codegenSummary
- */
-export const codegenSummary = {
-  name: 'codegenSummary',
-  description:
-    'This function is called with a summary of proposed updates.' +
-    'Summary is a data structure(object) which contains:\n' +
-    '- `fileUpdates`: list of proposed file updates that will be subject to subsequent code generation request\n' +
-    '- `contextPaths`: list of file paths that make sense to use as a context for code generation requests.' +
-    '- `explanation`: general explanation of planned code generation updates\n' +
-    'It is crticially important to adhere to the schema of parameters',
-  parameters: {
-    type: 'object',
-    properties: {
-      contextPaths: {
-        type: 'array',
-        description:
-          'An array of absolute paths of files that should be used to provide context for the following updates. Context files could be for example the dependencies, or files that depend on one of the files that we want to update in the next step.',
-        items: {
-          type: 'string',
-        },
-      },
-      fileUpdates: {
-        type: 'array',
-        description: 'An array of proposed file updates, each update is an object with several properties.',
-        items: {
-          type: 'object',
-          description:
-            'Proposed update of a file. The update is an object which contains properties like file path, update tool name, and few other important properties.',
-          properties: {
-            path: { type: 'string', description: 'An absolute path of the project file that will be updated' },
-            updateToolName: {
-              type: 'string',
-              enum: [
-                'createFile',
-                'updateFile',
-                'patchFile',
-                'deleteFile',
-                'createDirectory',
-                'moveFile',
-                'generateImage',
-                'downloadFile',
-                'splitImage',
-                'resizeImage',
-                'imglyRemoveBackground',
-              ],
-              description: 'A name of the tool that will be used to perform the update.',
-            },
-            temperature: {
-              type: 'number',
-              description:
-                'Temperature parameter that will be used for LLM request. The value is adjusted to the characteristic of the update. If there is a need for a more creative solution, the value should be lower, but stil within [0.0, 2.0] range. The default value is: ' +
-                temperature,
-            },
-            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. The default value is: ' +
-                !!cheap,
-            },
-            prompt: {
-              type: 'string',
-              description:
-                'Prompt that will be passed to the model request together with the tool request. It summarizes the planned changes for this particular file, so it should be detailed enough for the model to generate necessary changes.',
-            },
-            contextImageAssets: {
-              type: 'array',
-              description:
-                'A list of of absolute image asset paths that should be included to the context of LLM request.' +
-                'This parameter must be used if there is a need to analyze an image.',
-              items: { type: 'string' },
-            },
-          },
-          required: ['path', 'updateToolName', ...(vision ? ['contextImageAssets'] : [])],
-        },
-      },
-      explanation: {
-        type: 'string',
-        description: 'Explanation of planned changes or explanation of reasoning for no code changes',
-      },
-    },
-    required: ['fileUpdates', 'contextPaths'],
-  },
-};

package/src/prompt/function-defs/create-directory.js

@@ -1,21 +0,0 @@
-/**
- * Function definition for createDirectory
- */
-export const createDirectory = {
-  name: 'createDirectory',
-  description: 'Create a new directory',
-  parameters: {
-    type: 'object',
-    properties: {
-      filePath: {
-        type: 'string',
-        description: 'The directory path to create.',
-      },
-      explanation: {
-        type: 'string',
-        description: 'The explanation of the reasoning behind creating this directory',
-      },
-    },
-    required: ['filePath'],
-  },
-};

package/src/prompt/function-defs/create-file.js

@@ -1,26 +0,0 @@
-/**
- * Function definition for createFile
- */
-export const createFile = {
-  name: 'createFile',
-  description:
-    'Create a new file with specified content. The file will be created inside of project folder structure. This tool should not be used of creation if image files.',
-  parameters: {
-    type: 'object',
-    properties: {
-      filePath: {
-        type: 'string',
-        description: 'Path of the file that will be created, it must not be empty.',
-      },
-      newContent: {
-        type: 'string',
-        description: 'Content of the file that will be created, it must no be empty.',
-      },
-      explanation: {
-        type: 'string',
-        description: 'The explanation of the reasoning behind creating this file',
-      },
-    },
-    required: ['filePath', 'newContent'],
-  },
-};

package/src/prompt/function-defs/delete-file.js

@@ -1,21 +0,0 @@
-/**
- * Function definition for deleteFile
- */
-export const deleteFile = {
-  name: 'deleteFile',
-  description: 'Delete a specified file from the application source code.',
-  parameters: {
-    type: 'object',
-    properties: {
-      filePath: {
-        type: 'string',
-        description: 'The file path to delete.',
-      },
-      explanation: {
-        type: 'string',
-        description: 'The explanation of the reasoning behind deleting this file',
-      },
-    },
-    required: ['filePath'],
-  },
-};

package/src/prompt/function-defs/download-file.js

@@ -1,25 +0,0 @@
-/**
- * Function definition for downloadFile
- */
-export const downloadFile = {
-  name: 'downloadFile',
-  description: 'Download file from url, and save to file',
-  parameters: {
-    type: 'object',
-    properties: {
-      filePath: {
-        type: 'string',
-        description: 'The file path to save the downloaded file.',
-      },
-      downloadUrl: {
-        type: 'string',
-        description: 'The url of the file that will be used for downloading.',
-      },
-      explanation: {
-        type: 'string',
-        description: 'The reasoning behind downloading this image.',
-      },
-    },
-    required: ['filePath', 'downloadUrl'],
-  },
-};

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

@@ -1,17 +0,0 @@
-/**
- * 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'],
-  },
-};

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

@@ -1,51 +0,0 @@
-/**
- * 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: {
-      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.',
-      },
-      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.',
-      },
-      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.',
-      },
-      explanation: {
-        type: 'string',
-        description: 'The explanation of the reasoning behind generating this image',
-      },
-    },
-    required: ['prompt', 'filePath', 'size', 'cheap'],
-  },
-};

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

@@ -1,21 +0,0 @@
-/**
- * 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: [],
-  },
-};

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

@@ -1,21 +0,0 @@
-/**
- * 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. 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: [],
-  },
-};

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

@@ -1,25 +0,0 @@
-/**
- * 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'],
-  },
-};

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

@@ -1,25 +0,0 @@
-/**
- * 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'],
-  },
-};

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

@@ -1,39 +0,0 @@
-/**
- * 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'],
-  },
-};

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

@@ -1,36 +0,0 @@
-/**
- * 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'],
-  },
-};

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

@@ -1,45 +0,0 @@
-/**
- * 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'],
-  },
-};

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

@@ -1,26 +0,0 @@
-/**
- * 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'],
-  },
-};

package/src/prompt/limits.js

@@ -1,23 +0,0 @@
-import assert from 'node:assert';
-
-const SYSTEM_PROMPT_LIMIT = 200;
-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);
-}

package/src/prompt/limits.test.js

@@ -1,40 +0,0 @@
-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(201);
-      expect(() => verifySystemPromptLimit(invalidPrompt)).toThrow('Token limit exceeded: 202 > 200');
-    });
-  });
-
-  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');
-    });
-  });
-});

package/src/prompt/prompt-codegen.js

@@ -1,106 +0,0 @@
-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.`
-        : `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;
-}