editcodewithai 2.3.0 → 2.4.0

package/dist/fileUtils.d.ts

@@ -5,9 +5,16 @@ import { VizFiles, VizFile, FileCollection } from "@vizhub/viz-types";
  */
 export declare function shouldDeleteFile(file?: VizFile): boolean;
 /**
- * Processes files for the prompt by truncating large files
+ * Checks if a filename is an image file based on extension
  */
-export declare function prepareFilesForPrompt(files: VizFiles): FileCollection;
+export declare function isImageFile(fileName: string): boolean;
+/**
+ * Processes files for the prompt by truncating large files and excluding images
+ */
+export declare function prepareFilesForPrompt(files: VizFiles): {
+    files: FileCollection;
+    imageFiles: string[];
+};
 /**
  * Merges original files with changes from the LLM
  */

package/dist/fileUtils.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.shouldDeleteFile = shouldDeleteFile;
+exports.isImageFile = isImageFile;
 exports.prepareFilesForPrompt = prepareFilesForPrompt;
 exports.mergeFileChanges = mergeFileChanges;
 exports.parseDiffs = parseDiffs;
@@ -19,11 +20,24 @@ function shouldDeleteFile(file) {
     return file.text.trim() === "";
 }
 /**
- * Processes files for the prompt by truncating large files
+ * Checks if a filename is an image file based on extension
+ */
+function isImageFile(fileName) {
+    const imageExtensions = /\.(png|jpg|jpeg|gif|bmp|svg|webp)$/i;
+    return imageExtensions.test(fileName);
+}
+/**
+ * Processes files for the prompt by truncating large files and excluding images
  */
 function prepareFilesForPrompt(files) {
     const result = {};
+    const imageFiles = [];
     Object.values(files).forEach((file) => {
+        // Check if it's an image file
+        if (isImageFile(file.name)) {
+            imageFiles.push(file.name);
+            return; // Skip processing image files
+        }
         // Example: truncate large files, etc.
         result[file.name] = file.text
             .split("\n")
@@ -31,7 +45,7 @@ function prepareFilesForPrompt(files) {
             .map((line) => line.slice(0, 200))
             .join("\n");
     });
-    return result;
+    return { files: result, imageFiles };
 }
 /**
  * Merges original files with changes from the LLM

package/dist/index.d.ts

@@ -1,9 +1,9 @@
 import type { PerformAiEditParams, PerformAiEditResult } from "./types";
 import { PROMPT_TEMPLATE_VERSION, assembleFullPrompt, FORMAT_INSTRUCTIONS } from "./prompt";
 import { getGenerationMetadata } from "./metadata";
-import { prepareFilesForPrompt, mergeFileChanges, parseDiffs, applyDiffs, parseDiffFenced, parseUdiffs, applyUdiffs } from "./fileUtils";
+import { prepareFilesForPrompt, mergeFileChanges, parseDiffs, applyDiffs, parseDiffFenced, parseUdiffs, applyUdiffs, isImageFile } from "./fileUtils";
 export type { LlmFunction, PerformAiEditParams, PerformAiEditResult, EditFormat, } from "./types";
-export { FORMAT_INSTRUCTIONS, mergeFileChanges, prepareFilesForPrompt, parseDiffs, applyDiffs, parseDiffFenced, parseUdiffs, applyUdiffs, assembleFullPrompt, getGenerationMetadata, PROMPT_TEMPLATE_VERSION, };
+export { FORMAT_INSTRUCTIONS, mergeFileChanges, prepareFilesForPrompt, isImageFile, parseDiffs, applyDiffs, parseDiffFenced, parseUdiffs, applyUdiffs, assembleFullPrompt, getGenerationMetadata, PROMPT_TEMPLATE_VERSION, };
 /**
  * Core AI logic for:
  * - Building the prompt (including context)

package/dist/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.PROMPT_TEMPLATE_VERSION = exports.getGenerationMetadata = exports.assembleFullPrompt = exports.applyUdiffs = exports.parseUdiffs = exports.parseDiffFenced = exports.applyDiffs = exports.parseDiffs = exports.prepareFilesForPrompt = exports.mergeFileChanges = exports.FORMAT_INSTRUCTIONS = void 0;
+exports.PROMPT_TEMPLATE_VERSION = exports.getGenerationMetadata = exports.assembleFullPrompt = exports.applyUdiffs = exports.parseUdiffs = exports.parseDiffFenced = exports.applyDiffs = exports.parseDiffs = exports.isImageFile = exports.prepareFilesForPrompt = exports.mergeFileChanges = exports.FORMAT_INSTRUCTIONS = void 0;
 exports.performAiEdit = performAiEdit;
 const llm_code_format_1 = require("llm-code-format");
 const prompt_1 = require("./prompt");
@@ -17,6 +17,7 @@ Object.defineProperty(exports, "applyDiffs", { enumerable: true, get: function (
 Object.defineProperty(exports, "parseDiffFenced", { enumerable: true, get: function () { return fileUtils_1.parseDiffFenced; } });
 Object.defineProperty(exports, "parseUdiffs", { enumerable: true, get: function () { return fileUtils_1.parseUdiffs; } });
 Object.defineProperty(exports, "applyUdiffs", { enumerable: true, get: function () { return fileUtils_1.applyUdiffs; } });
+Object.defineProperty(exports, "isImageFile", { enumerable: true, get: function () { return fileUtils_1.isImageFile; } });
 const debug = false;
 /**
  * Core AI logic for:
@@ -27,10 +28,15 @@ const debug = false;
  */
 async function performAiEdit({ prompt, files, llmFunction, apiKey, editFormat = "whole", }) {
     // 1. Format the existing files into the "markdown code block" format
-    const preparedFiles = (0, fileUtils_1.prepareFilesForPrompt)(files);
-    const filesContext = (0, llm_code_format_1.formatMarkdownFiles)(preparedFiles);
+    const preparedResult = (0, fileUtils_1.prepareFilesForPrompt)(files);
+    const filesContext = (0, llm_code_format_1.formatMarkdownFiles)(preparedResult.files);
     // 2. Assemble the final prompt
-    const fullPrompt = (0, prompt_1.assembleFullPrompt)({ filesContext, prompt, editFormat });
+    const fullPrompt = (0, prompt_1.assembleFullPrompt)({
+        filesContext,
+        prompt,
+        editFormat,
+        imageFiles: preparedResult.imageFiles,
+    });
     debug && console.log("[performAiEdit] fullPrompt:", fullPrompt);
     // 3. Invoke the model via the provided LLM function
     const result = await llmFunction(fullPrompt);

package/dist/prompt.d.ts

@@ -4,8 +4,9 @@ export declare const FORMAT_INSTRUCTIONS: Record<EditFormat, string>;
 /**
  * Assembles the full prompt by combining task, files context, and formatting instructions
  */
-export declare function assembleFullPrompt({ filesContext, prompt, editFormat, }: {
+export declare function assembleFullPrompt({ filesContext, prompt, editFormat, imageFiles, }: {
     filesContext: string;
     prompt: string;
     editFormat?: EditFormat;
+    imageFiles?: string[];
 }): string;

package/dist/prompt.js

@@ -20,11 +20,6 @@ exports.FORMAT_INSTRUCTIONS = {
         "Delete all unused files, but we need to keep `README.md`. ",
         "Files can be deleted by setting their content to empty, for example:\n\n",
         "**fileToDelete.js**\n\n```\n```\n\n",
-        "For D3 logic, make sure it remains idempotent (use data joins), ",
-        "and prefer function signatures like `someFunction(selection, options)` ",
-        "where `selection` is a D3 selection and `options` is an object. ",
-        'Only import from the top-level `"d3"` module, not from submodules ',
-        '(e.g. imports from `"d3-scale"` or `"d3-shape"` are not supported).\n\n',
     ].join(""),
     diff: [
         "## Formatting Instructions\n\n",
@@ -65,7 +60,14 @@ exports.FORMAT_INSTRUCTIONS = {
 /**
  * Assembles the full prompt by combining task, files context, and formatting instructions
  */
-function assembleFullPrompt({ filesContext, prompt, editFormat = "whole", }) {
+function assembleFullPrompt({ filesContext, prompt, editFormat = "whole", imageFiles = [], }) {
     const FORMAT = exports.FORMAT_INSTRUCTIONS[editFormat];
-    return [TASK(prompt), FILES(filesContext), FORMAT].join("\n\n");
+    // Add image files list if there are any
+    let imageFilesSection = "";
+    if (imageFiles.length > 0) {
+        imageFilesSection =
+            "\nImage files available:\n\n" +
+                imageFiles.map((fileName) => ` * \`${fileName}\``).join("\n");
+    }
+    return [TASK(prompt), FILES(filesContext), FORMAT + imageFilesSection].join("\n\n");
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "editcodewithai",
-  "version": "2.3.0",
+  "version": "2.4.0",
   "description": "Edit Code With AI",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -40,22 +40,22 @@
   },
   "homepage": "https://github.com/vizhub-core/editcodewithai#readme",
   "dependencies": {
-    "@vizhub/viz-types": "^0.3.0",
-    "@vizhub/viz-utils": "^1.3.0",
-    "dotenv": "^17.2.0",
+    "@vizhub/viz-types": "^0.5.0",
+    "@vizhub/viz-utils": "^1.5.0",
+    "dotenv": "^17.2.2",
     "llm-code-format": "^3.0.0"
   },
   "devDependencies": {
-    "@langchain/core": "^0.3.66",
-    "@langchain/openai": "^0.6.2",
-    "langchain": "^0.3.30",
+    "@langchain/core": "^0.3.73",
+    "@langchain/openai": "^0.6.11",
+    "langchain": "^0.3.32",
     "cors": "^2.8.5",
     "express": "^5.1.0",
-    "npm-check-updates": "^18.0.2",
+    "npm-check-updates": "^18.0.3",
     "prettier": "^3.6.2",
-    "puppeteer": "^24.15.0",
+    "puppeteer": "^24.19.0",
     "ts-node": "^10.9.2",
-    "typescript": "^5.8.3",
+    "typescript": "^5.9.2",
     "vitest": "^3.2.4"
   }
 }