genaicode 0.0.37 → 0.1.1

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;
-}

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

@@ -1,116 +0,0 @@
-/*eslint-disable no-import-assign*/
-
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { getCodeGenPrompt, getLintFixPrompt } from './prompt-codegen.js';
-import * as findFiles from '../files/find-files.js';
-import * as cliParams from '../cli/cli-params.js';
-import * as limits from './limits.js';
-import '../files/read-files.js';
-
-vi.mock('../files/find-files.js', () => ({
-  rcConfig: {},
-  getSourceFiles: vi.fn(),
-}));
-vi.mock('../files/read-files.js', () => ({
-  getSourceCode: () => ({}),
-}));
-vi.mock('../cli/cli-params.js', () => ({
-  requireExplanations: false,
-  considerAllFiles: false,
-  dependencyTree: false,
-  explicitPrompt: false,
-  allowFileCreate: false,
-  allowFileDelete: false,
-  allowDirectoryCreate: false,
-  allowFileMove: false,
-  verbosePrompt: false,
-  imagen: false,
-  vision: false,
-}));
-vi.mock('./limits.js');
-
-describe('getCodeGenPrompt', () => {
-  beforeEach(() => {
-    vi.resetAllMocks();
-  });
-
-  it('should generate prompt for all files when considerAllFiles is true', () => {
-    vi.spyOn(findFiles, 'getSourceFiles').mockReturnValue(['file1.js', 'file2.js']);
-    cliParams.considerAllFiles = true;
-    cliParams.explicitPrompt = null;
-    cliParams.allowFileCreate = false;
-    cliParams.allowFileDelete = false;
-    cliParams.allowDirectoryCreate = false;
-    cliParams.allowFileMove = false;
-    vi.spyOn(limits, 'verifyCodegenPromptLimit').mockImplementation(() => {});
-
-    const prompt = getCodeGenPrompt();
-
-    expect(prompt).toContain('You are allowed to modify all files in the application');
-    expect(prompt).toContain('Do not create new files.');
-    expect(prompt).toContain('Do not delete files.');
-    expect(prompt).toContain('Do not create new directories.');
-    expect(prompt).toContain('Do not move files.');
-  });
-
-  // Add more test cases for getCodeGenPrompt as needed
-});
-
-describe('getLintFixPrompt', () => {
-  beforeEach(() => {
-    cliParams.verbosePrompt = false;
-    vi.spyOn(limits, 'verifyCodegenPromptLimit').mockImplementation(() => {});
-  });
-
-  it('should generate a lint fix prompt with provided command, stdout, and stderr', () => {
-    const command = 'eslint --fix';
-    const stdout = 'Fixed 2 errors';
-    const stderr = '';
-
-    const prompt = getLintFixPrompt(command, stdout, stderr);
-
-    expect(prompt).toContain('The following lint errors were encountered after the initial code generation:');
-    expect(prompt).toContain(`Lint command: ${command}`);
-    expect(prompt).toContain('Lint command stdout:');
-    expect(prompt).toContain(stdout);
-    expect(prompt).toContain('Lint command stderr:');
-    expect(prompt).toContain('Please suggest changes to fix these lint errors.');
-  });
-
-  it('should include stderr in the prompt when provided', () => {
-    const command = 'eslint --fix';
-    const stdout = '';
-    const stderr = 'Error: Unable to resolve path';
-
-    const prompt = getLintFixPrompt(command, stdout, stderr);
-
-    expect(prompt).toContain('Lint command stderr:');
-    expect(prompt).toContain(stderr);
-  });
-
-  it('should call verifyCodegenPromptLimit with the generated prompt', () => {
-    const command = 'eslint --fix';
-    const stdout = 'Fixed 1 error';
-    const stderr = '';
-
-    getLintFixPrompt(command, stdout, stderr);
-
-    expect(limits.verifyCodegenPromptLimit).toHaveBeenCalledWith(expect.any(String));
-  });
-
-  it('should log the prompt when verbosePrompt is true', () => {
-    cliParams.verbosePrompt = true;
-    const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => {});
-
-    const command = 'eslint --fix';
-    const stdout = 'Fixed 1 error';
-    const stderr = '';
-
-    getLintFixPrompt(command, stdout, stderr);
-
-    expect(consoleSpy).toHaveBeenCalledWith('Lint fix prompt:');
-    expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining('The following lint errors were encountered'));
-
-    consoleSpy.mockRestore();
-  });
-});