editcodewithai 0.0.1 → 1.0.0

package/README.md

@@ -1,52 +1,118 @@
 # editcodewithai
-Edit Code With AI 
 
-## A Simple Open Source AI Code Editor
+A lightweight, flexible library for AI-powered code editing.
 
-The premise of this project is to build and iterate an open source AI code editing system. The core idea is that you can feed this system a set of source code files and high level instructions, and it will return the edited code after some time. It aims to be compatible with many LLMs, and flexible enough to adapt to future models as they come out.
+## Overview
 
-## Installation
+`editcodewithai` is a JavaScript/TypeScript library that enables AI-powered code editing in your applications. It provides a simple interface to send code files and instructions to an LLM (Large Language Model) and receive edited code in return.
 
-```bash
-# Clone the repository
-git clone https://github.com/yourusername/editcodewithai.git
+The library is designed to be model-agnostic, allowing you to use any LLM provider while handling the prompt engineering, file parsing, and response processing for you.
 
-# Navigate to project directory
-cd editcodewithai
+## Installation
 
-# Install dependencies
-npm install
+```bash
+npm install editcodewithai
 ```
 
 ## Usage
 
 ```typescript
-import { performAiEdit } from 'editcodewithai';
+import { performAiEdit } from "editcodewithai";
+import { VizFiles } from "@vizhub/viz-types";
+
+// Define your LLM function that will process the prompt
+const myLlmFunction = async (prompt: string) => {
+  // Call your preferred LLM API here
+  // This example assumes using OpenRouter
+  const response = await fetch(
+    "https://openrouter.ai/api/v1/chat/completions",
+    {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${apiKey}`,
+      },
+      body: JSON.stringify({
+        model: "openai/gpt-4",
+        messages: [{ role: "user", content: prompt }],
+      }),
+    },
+  );
+
+  const data = await response.json();
+  return {
+    content: data.choices[0].message.content,
+    generationId: data.id,
+  };
+};
+
+// Your files
+const files: VizFiles = {
+  file1: {
+    name: "index.js",
+    text: "console.log('Hello world');",
+  },
+};
 
+// Perform the AI edit
 const result = await performAiEdit({
   prompt: "Update the code to use async/await",
-  modelName: "openai/gpt-4",
-  files: {
-    // Your files object
-  },
+  files: files,
+  llmFunction: myLlmFunction,
   apiKey: "your-openrouter-api-key",
-  baseURL: "https://openrouter.ai/api/v1"
 });
+
+console.log(result.changedFiles);
 ```
 
-The `files` object should be a map of file IDs to file objects containing:
-- `name`: The filename (e.g. "index.js")
+## API Reference
+
+### performAiEdit(params)
+
+The main function that processes files with an AI model and returns edited code.
+
+#### Parameters
+
+| Parameter     | Type          | Description                                                       |
+| ------------- | ------------- | ----------------------------------------------------------------- |
+| `prompt`      | `string`      | Instructions for the AI on how to modify the code                 |
+| `files`       | `VizFiles`    | Object containing file information (see below)                    |
+| `llmFunction` | `LlmFunction` | Function that sends the prompt to an LLM and returns the response |
+| `apiKey`      | `string`      | OpenRouter API key for retrieving cost metadata                   |
+
+The `VizFiles` type is a map of file IDs to file objects, where each file object has:
+
+- `name`: The filename (e.g., "index.js")
 - `text`: The file contents as a string
 
-The function returns:
-- `changedFiles`: Updated files with modifications
-- `openRouterGenerationId`: ID of the generation
-- `upstreamCostCents`: Cost in cents
-- `provider`: The AI provider used
-- `inputTokens`: Number of input tokens
-- `outputTokens`: Number of output tokens
+The `LlmFunction` type is a function that takes a prompt string and returns a Promise with:
+
+- `content`: The LLM's response text
+- `generationId`: A unique ID for the generation (used for cost tracking)
+
+#### Return Value
+
+The function returns an object with:
+
+| Property                 | Type       | Description                                |
+| ------------------------ | ---------- | ------------------------------------------ |
+| `changedFiles`           | `VizFiles` | Updated files with AI modifications        |
+| `openRouterGenerationId` | `string`   | ID of the generation from the LLM provider |
+| `upstreamCostCents`      | `number`   | Cost of the API call in cents              |
+| `provider`               | `string`   | The AI provider used (e.g., "openai")      |
+| `inputTokens`            | `number`   | Number of input tokens used                |
+| `outputTokens`           | `number`   | Number of output tokens generated          |
+| `promptTemplateVersion`  | `number`   | Version of the prompt template used        |
+
+### File Operations
+
+The library handles several file operations automatically:
 
-## Similar open source projects
+- **Updating existing files**: When the AI modifies a file's content
+- **Creating new files**: When the AI suggests new files to add
+- **Deleting files**: When the AI returns empty content for a file
+
+## Similar Projects
 
 - **Aider**: An AI pair programming tool that integrates with your terminal to assist in code editing within your local git repository. [https://aider.chat/](https://aider.chat/)
 
@@ -61,3 +127,39 @@ The function returns:
 - **Void**: An open-source alternative to proprietary AI code editors, offering AI-assisted coding features while prioritizing user privacy and control. [https://void.dev/](https://void.dev/)
 
 - **Cody**: An advanced AI coding assistant developed by Sourcegraph, integrating seamlessly with popular IDEs to provide features like AI-driven chat, code autocompletion, and inline editing. [https://github.com/sourcegraph/cody](https://github.com/sourcegraph/cody)
+
+## Contributing
+
+To contribute to this project:
+
+```bash
+# Clone the repository
+git clone https://github.com/yourusername/editcodewithai.git
+
+# Navigate to project directory
+cd editcodewithai
+
+# Install dependencies
+npm install
+```
+
+Run tests to ensure everything is working correctly:
+
+```bash
+npm test
+```
+
+Please submit pull requests with clear descriptions of changes and ensure all tests pass. Protocol for wrapping up a PR:
+
+```
+npm test
+npm run typecheck
+npm run prettier
+# Verify the README is up to date
+```
+
+Please create an issue first before creating a PR to discuss the changes you want to make. This helps ensure that your contributions align with the project's goals and vision.
+
+## License
+
+This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for details.

package/dist/fileUtils.d.ts

@@ -0,0 +1,14 @@
+import { VizFiles, VizFile, FileCollection } from "@vizhub/viz-types";
+/**
+ * If the LLM outputs empty text for a file, we interpret this
+ * as a request to delete the file.
+ */
+export declare function shouldDeleteFile(file?: VizFile): boolean;
+/**
+ * Processes files for the prompt by truncating large files
+ */
+export declare function prepareFilesForPrompt(files: VizFiles): FileCollection;
+/**
+ * Merges original files with changes from the LLM
+ */
+export declare function mergeFileChanges(originalFiles: VizFiles, parsedFiles: FileCollection): VizFiles;

package/dist/fileUtils.js

@@ -0,0 +1,66 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.shouldDeleteFile = shouldDeleteFile;
+exports.prepareFilesForPrompt = prepareFilesForPrompt;
+exports.mergeFileChanges = mergeFileChanges;
+const viz_utils_1 = require("@vizhub/viz-utils");
+/**
+ * If the LLM outputs empty text for a file, we interpret this
+ * as a request to delete the file.
+ */
+function shouldDeleteFile(file) {
+    if (!file)
+        return false;
+    return file.text.trim() === "";
+}
+/**
+ * Processes files for the prompt by truncating large files
+ */
+function prepareFilesForPrompt(files) {
+    const result = {};
+    Object.values(files).forEach((file) => {
+        // Example: truncate large files, etc.
+        result[file.name] = file.text
+            .split("\n")
+            .slice(0, file.name.endsWith(".csv") || file.name.endsWith(".json") ? 50 : 500)
+            .map((line) => line.slice(0, 200))
+            .join("\n");
+    });
+    return result;
+}
+/**
+ * Merges original files with changes from the LLM
+ */
+function mergeFileChanges(originalFiles, parsedFiles) {
+    // Start with existing files
+    let changedFiles = Object.keys(originalFiles).reduce((acc, fileId) => {
+        const original = originalFiles[fileId];
+        const changedText = parsedFiles[original.name];
+        const changedFile = changedText !== undefined
+            ? { name: original.name, text: changedText }
+            : undefined;
+        if (shouldDeleteFile(changedFile)) {
+            // Exclude from new set
+            return acc;
+        }
+        // If changedFile is present, use it; otherwise use original
+        acc[fileId] = {
+            ...original,
+            text: changedFile ? changedFile.text : original.text,
+        };
+        return acc;
+    }, {});
+    // Handle newly-created files
+    Object.entries(parsedFiles).forEach(([fileName, fileText]) => {
+        const existingFile = Object.values(changedFiles).find((file) => file.name === fileName);
+        // If no existing file and not empty => it's a new file
+        if (!existingFile && fileText.trim() !== "") {
+            const newFileId = (0, viz_utils_1.generateVizFileId)();
+            changedFiles[newFileId] = {
+                name: fileName,
+                text: fileText,
+            };
+        }
+    });
+    return changedFiles;
+}

package/dist/index.d.ts

@@ -1,25 +1,9 @@
-import { VizFiles } from "@vizhub/viz-types";
-export interface PerformAiEditParams {
-    prompt: string;
-    modelName: string;
-    files: VizFiles;
-    apiKey: string;
-    baseURL: string | undefined;
-}
-export interface PerformAiEditResult {
-    changedFiles: VizFiles;
-    openRouterGenerationId: string;
-    upstreamCostCents: number;
-    provider: string;
-    inputTokens: number;
-    outputTokens: number;
-    promptTemplateVersion: number;
-}
+import { PerformAiEditParams, PerformAiEditResult } from "./types";
 /**
  * Core AI logic for:
  * - Building the prompt (including context)
- * - Calling the LLM (ChatOpenAI)
+ * - Calling the provided LLM function
  * - Parsing and merging file changes
  * - Retrieving cost metadata
  */
-export declare function performAiEdit({ prompt, modelName, files, apiKey, baseURL, }: PerformAiEditParams): Promise<PerformAiEditResult>;
+export declare function performAiEdit({ prompt, files, llmFunction, apiKey, }: PerformAiEditParams): Promise<PerformAiEditResult>;

package/dist/index.js

@@ -1,161 +1,40 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.performAiEdit = performAiEdit;
-const openai_1 = require("@langchain/openai");
-const output_parsers_1 = require("@langchain/core/output_parsers");
 const llm_code_format_1 = require("llm-code-format");
-const viz_types_1 = require("@vizhub/viz-types");
+const prompt_1 = require("./prompt");
+const metadata_1 = require("./metadata");
+const fileUtils_1 = require("./fileUtils");
 const debug = false;
-// Versions of the prompt template.
-const PROMPT_TEMPLATE_VERSION = 1;
-// Template pieces
-const TASK = (prompt) => `## Your Task\n\n${prompt}`;
-const FILES = (filesContext) => `## Original Files\n\n${filesContext}`;
-const FORMAT = [
-    "## Formatting Instructions\n\n",
-    "Suggest changes to the original files using this exact format:\n\n",
-    "**fileA.js**\n\n```js\n// Entire updated code for fileA\n```\n\n",
-    "**fileB.js**\n\n```js\n// Entire updated code for fileB\n```\n\n",
-    "Only include the files that need to be updated or created.\n\n",
-    "To suggest changes you MUST include the ENTIRE content of the updated file.\n\n",
-    'NEVER leave out sections as in "... rest of the code remain the same ...".\n\n',
-    "Refactor large files into smaller files in the same directory.\n\n",
-    "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.\n\n",
-].join("");
-function assembleFullPrompt({ filesContext, prompt, }) {
-    return [TASK(prompt), FILES(filesContext), FORMAT].join("\n\n");
-}
-/**
- * If the LLM outputs empty text for a file, we interpret this
- * as a request to delete the file.
- */
-function shouldDeleteFile(file) {
-    return file && file.text.trim() === "";
-}
-/**
- * Get cost and metadata for a generation from OpenRouter.
- */
-async function getGenerationMetadata({ apiKey, generationId, }) {
-    const url = `https://openrouter.ai/api/v1/generation?id=${generationId}`;
-    const headers = {
-        Authorization: `Bearer ${apiKey}`,
-    };
-    // Retry logic for potential 404 from OpenRouter
-    const maxRetries = 10;
-    const retryDelay = 1000; // 1 second
-    for (let attempt = 1; attempt <= maxRetries; attempt++) {
-        const response = await fetch(url, {
-            method: "GET",
-            headers,
-        });
-        if (response.ok) {
-            const data = await response.json();
-            debug && console.log("[getCost] data:", JSON.stringify(data, null, 2));
-            const upstreamCostInDollars = data.data.total_cost;
-            const upstreamCostCents = upstreamCostInDollars * 100;
-            const provider = data.data.provider_name;
-            const inputTokens = data.data.tokens_prompt;
-            const outputTokens = data.data.tokens_completion;
-            return {
-                upstreamCostCents,
-                provider,
-                inputTokens,
-                outputTokens,
-            };
-        }
-        else {
-            const text = await response.text();
-            debug &&
-                console.error(`Attempt ${attempt} failed. Status: ${response.status}, response text: ${text}`);
-            if (attempt < maxRetries) {
-                debug && console.log(`Retrying in ${retryDelay / 1000} seconds...`);
-                await new Promise((resolve) => setTimeout(resolve, retryDelay));
-            }
-            else {
-                throw new Error(`HTTP error! Status: ${response.status} after ${maxRetries} attempts.`);
-            }
-        }
-    }
-    // You theoretically never reach here because of the throw above
-    throw new Error("Failed to get generation metadata.");
-}
 /**
  * Core AI logic for:
  * - Building the prompt (including context)
- * - Calling the LLM (ChatOpenAI)
+ * - Calling the provided LLM function
  * - Parsing and merging file changes
  * - Retrieving cost metadata
  */
-async function performAiEdit({ prompt, modelName, files, apiKey, baseURL, }) {
-    // 1. Serialize the existing files into the “markdown code block” format
-    const filesContext = (0, llm_code_format_1.serializeMarkdownFiles)(Object.values(files).map((file) => ({
-        name: file.name,
-        // Example: truncate large files, etc.
-        text: file.text
-            .split("\n")
-            .slice(0, file.name.endsWith(".csv") || file.name.endsWith(".json") ? 50 : 500)
-            .map((line) => line.slice(0, 200))
-            .join("\n"),
-    })));
+async function performAiEdit({ prompt, files, llmFunction, apiKey, }) {
+    // 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);
     // 2. Assemble the final prompt
-    const fullPrompt = assembleFullPrompt({ filesContext, prompt });
+    const fullPrompt = (0, prompt_1.assembleFullPrompt)({ filesContext, prompt });
     debug && console.log("[performAiEdit] fullPrompt:", fullPrompt);
-    // 3. Invoke the model via LangChain’s ChatOpenAI
-    const options = {
-        modelName,
-        configuration: {
-            apiKey,
-            baseURL,
-        },
-        streaming: false,
-    };
-    const chatModel = new openai_1.ChatOpenAI(options);
-    const result = await chatModel.invoke(fullPrompt);
+    // 3. Invoke the model via the provided LLM function
+    const result = await llmFunction(fullPrompt);
     // 4. We parse the output to figure out which files changed
-    const parser = new output_parsers_1.StringOutputParser();
-    const resultString = await parser.invoke(result);
+    const resultString = result.content;
     const parsed = (0, llm_code_format_1.parseMarkdownFiles)(resultString, "bold");
     // 5. Merge the changes into a new `Files` object
-    let changedFiles = Object.keys(files).reduce((acc, fileId) => {
-        const original = files[fileId];
-        const changedFile = parsed.files.find((f) => f.name === original.name);
-        if (shouldDeleteFile(changedFile)) {
-            // Exclude from new set
-            return acc;
-        }
-        // If changedFile is present, use it; otherwise use original
-        acc[fileId] = {
-            ...original,
-            text: changedFile ? changedFile.text : original.text,
-        };
-        return acc;
-    }, {});
-    // Handle newly-created files
-    parsed.files.forEach((changedFile) => {
-        const existingFile = Object.values(changedFiles).find((file) => file.name === changedFile.name);
-        // If no existing file and not empty => it's a new file
-        if (!existingFile && !shouldDeleteFile(changedFile)) {
-            const newFileId = (0, viz_types_1.generateVizFileId)();
-            changedFiles[newFileId] = {
-                name: changedFile.name,
-                text: changedFile.text,
-            };
-        }
-    });
+    const changedFiles = (0, fileUtils_1.mergeFileChanges)(files, parsed.files);
     // 6. Retrieve cost metadata for charging the user
-    //    The openRouterGenerationId is stored in `result.lc_kwargs.id` by LangChain
-    const openRouterGenerationId = result.lc_kwargs?.id;
+    const openRouterGenerationId = result.generationId || "";
     let upstreamCostCents = 0;
     let provider = "";
     let inputTokens = 0;
     let outputTokens = 0;
-    if (openRouterGenerationId) {
-        const costData = await getGenerationMetadata({
+    if (openRouterGenerationId && apiKey) {
+        const costData = await (0, metadata_1.getGenerationMetadata)({
             apiKey,
             generationId: openRouterGenerationId,
         });
@@ -166,11 +45,12 @@ async function performAiEdit({ prompt, modelName, files, apiKey, baseURL, }) {
     }
     return {
         changedFiles,
-        openRouterGenerationId: openRouterGenerationId ?? "",
+        openRouterGenerationId,
         upstreamCostCents,
         provider,
         inputTokens,
         outputTokens,
-        promptTemplateVersion: PROMPT_TEMPLATE_VERSION,
+        promptTemplateVersion: prompt_1.PROMPT_TEMPLATE_VERSION,
+        rawResponse: resultString, // Include the raw response
     };
 }

package/dist/metadata.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Get cost and metadata for a generation from OpenRouter.
+ */
+export declare function getGenerationMetadata({ apiKey, generationId, }: {
+    apiKey: string;
+    generationId: string;
+}): Promise<{
+    upstreamCostCents: number;
+    provider: string;
+    inputTokens: number;
+    outputTokens: number;
+}>;

package/dist/metadata.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getGenerationMetadata = getGenerationMetadata;
+const debug = false;
+/**
+ * Get cost and metadata for a generation from OpenRouter.
+ */
+async function getGenerationMetadata({ apiKey, generationId, }) {
+    const url = `https://openrouter.ai/api/v1/generation?id=${generationId}`;
+    const headers = {
+        Authorization: `Bearer ${apiKey}`,
+    };
+    // Retry logic for potential 404 from OpenRouter
+    const maxRetries = 10;
+    const retryDelay = 1000; // 1 second
+    for (let attempt = 1; attempt <= maxRetries; attempt++) {
+        const response = await fetch(url, {
+            method: "GET",
+            headers,
+        });
+        if (response.ok) {
+            const data = await response.json();
+            debug && console.log("[getCost] data:", JSON.stringify(data, null, 2));
+            const upstreamCostInDollars = data.data.total_cost;
+            const upstreamCostCents = upstreamCostInDollars * 100;
+            const provider = data.data.provider_name;
+            const inputTokens = data.data.tokens_prompt;
+            const outputTokens = data.data.tokens_completion;
+            return {
+                upstreamCostCents,
+                provider,
+                inputTokens,
+                outputTokens,
+            };
+        }
+        else {
+            const text = await response.text();
+            debug &&
+                console.error(`Attempt ${attempt} failed. Status: ${response.status}, response text: ${text}`);
+            if (attempt < maxRetries) {
+                debug && console.log(`Retrying in ${retryDelay / 1000} seconds...`);
+                await new Promise((resolve) => setTimeout(resolve, retryDelay));
+            }
+            else {
+                throw new Error(`HTTP error! Status: ${response.status} after ${maxRetries} attempts.`);
+            }
+        }
+    }
+    // You theoretically never reach here because of the throw above
+    throw new Error("Failed to get generation metadata.");
+}

package/dist/prompt.d.ts

@@ -0,0 +1,8 @@
+export declare const PROMPT_TEMPLATE_VERSION = 1;
+/**
+ * Assembles the full prompt by combining task, files context, and formatting instructions
+ */
+export declare function assembleFullPrompt({ filesContext, prompt, }: {
+    filesContext: string;
+    prompt: string;
+}): string;

package/dist/prompt.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PROMPT_TEMPLATE_VERSION = void 0;
+exports.assembleFullPrompt = assembleFullPrompt;
+// Versions of the prompt template.
+exports.PROMPT_TEMPLATE_VERSION = 1;
+// Template pieces
+const TASK = (prompt) => `## Your Task\n\n${prompt}`;
+const FILES = (filesContext) => `## Original Files\n\n${filesContext}`;
+const FORMAT = [
+    "## Formatting Instructions\n\n",
+    "Suggest changes to the original files using this exact format:\n\n",
+    "**fileA.js**\n\n```js\n// Entire updated code for fileA\n```\n\n",
+    "**fileB.js**\n\n```js\n// Entire updated code for fileB\n```\n\n",
+    "Only include the files that need to be updated or created.\n\n",
+    "To suggest changes you MUST include the ENTIRE content of the updated file.\n\n",
+    'NEVER leave out sections as in "... rest of the code remain the same ...".\n\n',
+    "Refactor large files into smaller files in the same directory.\n\n",
+    "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.\n\n",
+].join("");
+/**
+ * Assembles the full prompt by combining task, files context, and formatting instructions
+ */
+function assembleFullPrompt({ filesContext, prompt, }) {
+    return [TASK(prompt), FILES(filesContext), FORMAT].join("\n\n");
+}

package/dist/types.d.ts

@@ -0,0 +1,22 @@
+import { VizFiles } from "@vizhub/viz-types";
+export type LlmFunction = (prompt: string) => Promise<{
+    content: string;
+    generationId?: string;
+}>;
+export interface PerformAiEditParams {
+    prompt: string;
+    files: VizFiles;
+    llmFunction: LlmFunction;
+    apiKey?: string;
+    baseURL?: string;
+}
+export interface PerformAiEditResult {
+    changedFiles: VizFiles;
+    openRouterGenerationId?: string;
+    upstreamCostCents?: number;
+    provider?: string;
+    inputTokens?: number;
+    outputTokens?: number;
+    promptTemplateVersion?: number;
+    rawResponse?: string;
+}

package/dist/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "editcodewithai",
-  "version": "0.0.1",
+  "version": "1.0.0",
   "description": "Edit Code With AI",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -12,7 +12,12 @@
     "test": "vitest run",
     "build": "tsc",
     "prepublishOnly": "npm run build",
-    "type-check": "tsc --noEmit"
+    "typecheck": "tsc --noEmit",
+    "benchmark": "ts-node src/benchmarks/cli.ts run; cp -r ./benchmarks grader-app/public",
+    "grade": "ts-node src/benchmarks/cli.ts grade",
+    "benchmark:help": "ts-node src/benchmarks/cli.ts help",
+    "upgrade": "ncu -u",
+    "prettier": "prettier {*.*,**/*.*} --write"
   },
   "repository": {
     "type": "git",
@@ -35,13 +40,24 @@
   },
   "homepage": "https://github.com/vizhub-core/editcodewithai#readme",
   "dependencies": {
-    "@langchain/core": "^0.3.42",
-    "@langchain/openai": "^0.4.6",
-    "@vizhub/viz-types": "^0.0.1",
-    "llm-code-format": "^0.1.2"
+    "@langchain/core": "^0.3.43",
+    "@langchain/openai": "^0.5.4",
+    "@types/d3": "^7.4.3",
+    "@vizhub/viz-types": "^0.1.0",
+    "@vizhub/viz-utils": "^0.1.0",
+    "d3": "^7.9.0",
+    "dotenv": "^16.4.7",
+    "langchain": "^0.3.20",
+    "llm-code-format": "^2.0.1"
   },
   "devDependencies": {
-    "typescript": "^5.3.3",
-    "vitest": "^3.0.9"
+    "cors": "^2.8.5",
+    "express": "^5.1.0",
+    "npm-check-updates": "^17.1.16",
+    "prettier": "^3.5.3",
+    "puppeteer": "^24.6.0",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.8.3",
+    "vitest": "^3.1.1"
   }
 }