editcodewithai

A lightweight, flexible library for AI-powered code editing.

See also vizhub-benchmarks.

Overview

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.

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.

Installation

npm install editcodewithai

Usage

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: "anthropic/claude-3.5-sonnet",
        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",
  files: files,
  llmFunction: myLlmFunction,
  apiKey: "your-openrouter-api-key",
});

console.log(result.changedFiles);

API Reference

performAiEdit(params)

The main function that processes files with an AI model and returns edited code.

Parameters

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 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:

File Operations

The library handles several file operations automatically:

• 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/

• Bolt.new / Bolt.diy: A platform that allows users to prompt, run, edit, and deploy full-stack web and mobile applications. https://bolt.new/

• Cline: An AI-powered code assistant designed to help developers write and debug code more efficiently. https://cline.ai/

• Cerebras Coder: A code generation tool developed by Cerebras Systems, leveraging advanced AI models to assist in coding tasks. https://www.cerebras.net/

• Pear AI: An open-source AI code editor that accelerates the development process by integrating features like AI chat, code generation, and debugging assistance. https://trypear.ai/

• Void: An open-source alternative to proprietary AI code editors, offering AI-assisted coding features while prioritizing user privacy and control. 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

Contributing

To contribute to this project:

# 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:

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 file for details.