editcodewithai 0.1.0 → 1.1.0

package/dist/benchmarks/challenges.js

@@ -1,158 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.challenges = void 0;
-// Challenges definition. Each has multiple files (some with TODO).
-// "index.mjs" includes the test code that calls process.exit(1) if it fails.
-exports.challenges = [
-    {
-        name: "add",
-        prompt: "Implement the 'add' function to correctly add two numbers (a+b) and pass the test in index.mjs.",
-        files: {
-            file1: {
-                name: "index.mjs",
-                text: `
-import { add } from "./functions.mjs";
-
-// A simple test:
-const result = add(3, 4);
-if (result !== 7) {
-  console.error("Test failed: expected 7, got", result);
-  process.exit(1);
-}
-console.log("Add test passed");
-process.exit(0);
-        `,
-            },
-            file2: {
-                name: "functions.mjs",
-                text: `
-// TODO: Implement the add function
-export function add(a, b) {
-  // TODO
-}
-        `,
-            },
-        },
-    },
-    {
-        name: "multiply",
-        prompt: "Implement the 'multiply' function to correctly multiply two numbers and pass the unit test in index.mjs.",
-        files: {
-            file1: {
-                name: "index.mjs",
-                text: `
-import { multiply } from "./functions.mjs";
-
-const result = multiply(6, 7);
-if (result !== 42) {
-  console.error("Test failed: expected 42, got", result);
-  process.exit(1);
-}
-console.log("Multiply test passed");
-process.exit(0);
-      `,
-            },
-            file2: {
-                name: "functions.mjs",
-                text: `
-// TODO: Implement the multiply function
-export function multiply(a, b) {
-  // TODO
-}
-      `,
-            },
-        },
-    },
-    {
-        name: "square",
-        prompt: "Implement the 'square' function that returns x*x.",
-        files: {
-            file1: {
-                name: "index.mjs",
-                text: `
-import { square } from "./functions.mjs";
-
-const input = 5;
-const result = square(input);
-if (result !== 25) {
-  console.error("Test failed: expected 25, got", result);
-  process.exit(1);
-}
-console.log("Square test passed");
-process.exit(0);
-      `,
-            },
-            file2: {
-                name: "functions.mjs",
-                text: `
-// TODO: Implement the square function
-export function square(x) {
-  // TODO
-}
-      `,
-            },
-        },
-    },
-    {
-        name: "toUpperCase",
-        prompt: "Implement the toUpperCase function that returns the given string in uppercase.",
-        files: {
-            file1: {
-                name: "index.mjs",
-                text: `
-import { toUpperCase } from "./functions.mjs";
-
-const input = "hello";
-const result = toUpperCase(input);
-if (result !== "HELLO") {
-  console.error("Test failed: expected 'HELLO', got", result);
-  process.exit(1);
-}
-console.log("toUpperCase test passed");
-process.exit(0);
-      `,
-            },
-            file2: {
-                name: "functions.mjs",
-                text: `
-// TODO: Implement the toUpperCase function
-export function toUpperCase(str) {
-  // TODO
-}
-      `,
-            },
-        },
-    },
-    {
-        name: "reverseString",
-        prompt: "Implement the reverseString function that reverses the given string.",
-        files: {
-            file1: {
-                name: "index.mjs",
-                text: `
-import { reverseString } from "./functions.mjs";
-
-const input = "OpenAI";
-const expected = "IAnepO";
-
-const result = reverseString(input);
-if (result !== expected) {
-  console.error("Test failed: expected", expected, "but got", result);
-  process.exit(1);
-}
-console.log("reverseString test passed");
-process.exit(0);
-      `,
-            },
-            file2: {
-                name: "functions.mjs",
-                text: `
-// TODO: Implement the reverseString function
-export function reverseString(str) {
-  // TODO
-}
-      `,
-            },
-        },
-    },
-];

package/dist/benchmarks/fileUtils.d.ts

@@ -1,15 +0,0 @@
-import { VizFiles } from "@vizhub/viz-types";
-import { ChallengeResult } from "./types";
-/**
- * Writes the files for a given challenge into a folder named after challenge.name + model
- * e.g. "benchmarks/challenges/add/gpt4" to keep them separate per model.
- */
-export declare function writeChallengeFiles(challengeName: string, model: string, changedFiles: VizFiles, llmResponse?: string): string;
-/**
- * Writes results to a CSV file
- */
-export declare function writeResultsToCsv(results: ChallengeResult[], filePath?: string): string;
-/**
- * Ensures a cache directory exists
- */
-export declare function ensureCacheDir(cacheDirBase: string): void;

package/dist/benchmarks/fileUtils.js

@@ -1,55 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.writeChallengeFiles = writeChallengeFiles;
-exports.writeResultsToCsv = writeResultsToCsv;
-exports.ensureCacheDir = ensureCacheDir;
-const fs_1 = __importDefault(require("fs"));
-const path_1 = __importDefault(require("path"));
-/**
- * Writes the files for a given challenge into a folder named after challenge.name + model
- * e.g. "benchmarks/challenges/add/gpt4" to keep them separate per model.
- */
-function writeChallengeFiles(challengeName, model, changedFiles, llmResponse) {
-    const challengeDir = path_1.default.join("benchmarks", "challenges", challengeName, model);
-    if (!fs_1.default.existsSync(challengeDir)) {
-        fs_1.default.mkdirSync(challengeDir, { recursive: true });
-    }
-    Object.keys(changedFiles).forEach((fileKey) => {
-        const { name, text } = changedFiles[fileKey];
-        const filePath = path_1.default.join(challengeDir, name);
-        fs_1.default.writeFileSync(filePath, text, "utf-8");
-    });
-    // Write the LLM response to a file if provided
-    if (llmResponse) {
-        const responsePath = path_1.default.join(challengeDir, "llmResponse.md");
-        fs_1.default.writeFileSync(responsePath, llmResponse, "utf-8");
-    }
-    return challengeDir;
-}
-/**
- * Writes results to a CSV file
- */
-function writeResultsToCsv(results, filePath = "benchmarks/results.csv") {
-    let csv = "challenge,model,passFail\n";
-    for (const r of results) {
-        csv += `${r.challenge},${r.model},${r.passFail}\n`;
-    }
-    // Ensure benchmarks directory exists
-    const dir = path_1.default.dirname(filePath);
-    if (!fs_1.default.existsSync(dir)) {
-        fs_1.default.mkdirSync(dir, { recursive: true });
-    }
-    fs_1.default.writeFileSync(filePath, csv, "utf-8");
-    return csv;
-}
-/**
- * Ensures a cache directory exists
- */
-function ensureCacheDir(cacheDirBase) {
-    if (!fs_1.default.existsSync(cacheDirBase)) {
-        fs_1.default.mkdirSync(cacheDirBase, { recursive: true });
-    }
-}

package/dist/benchmarks/index.d.ts

@@ -1,5 +0,0 @@
-export { runAllChallenges } from "./benchmarkRunner";
-export { challenges } from "./challenges";
-export { createOpenRouterLlmFunction } from "./llm";
-export { runNodeTest } from "./runner";
-export { writeChallengeFiles, writeResultsToCsv } from "./fileUtils";

package/dist/benchmarks/index.js

@@ -1,44 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.writeResultsToCsv = exports.writeChallengeFiles = exports.runNodeTest = exports.createOpenRouterLlmFunction = exports.challenges = exports.runAllChallenges = void 0;
-/**********************************************************************
- * benchmark.ts
- * --------------------------------------------------------------------
- * Requires environment variables to be set in a .env file
- * --------------------------------------------------------------------
- * Defines 5 challenges, each with index.js (ES module) that performs
- * a simple unit test. The code calls process.exit(0) on success or
- * process.exit(1) on failure.
- *
- * Runs all challenges on 5 different models:
- *    ["gpt3", "gpt4", "gpt5", "deepseek", "qwen32"]
- *
- * Writes final results to "results.csv" with columns:
- *    challenge, model, passFail, exitCode
- **********************************************************************/
-const dotenv_1 = __importDefault(require("dotenv"));
-const benchmarkRunner_1 = require("./benchmarkRunner");
-// Load environment variables from .env file
-dotenv_1.default.config();
-// Run if invoked directly
-if (require.main === module) {
-    (0, benchmarkRunner_1.runAllChallenges)().catch((error) => {
-        console.error("Error running challenges:", error);
-        process.exit(1);
-    });
-}
-// Re-export for programmatic usage
-var benchmarkRunner_2 = require("./benchmarkRunner");
-Object.defineProperty(exports, "runAllChallenges", { enumerable: true, get: function () { return benchmarkRunner_2.runAllChallenges; } });
-var challenges_1 = require("./challenges");
-Object.defineProperty(exports, "challenges", { enumerable: true, get: function () { return challenges_1.challenges; } });
-var llm_1 = require("./llm");
-Object.defineProperty(exports, "createOpenRouterLlmFunction", { enumerable: true, get: function () { return llm_1.createOpenRouterLlmFunction; } });
-var runner_1 = require("./runner");
-Object.defineProperty(exports, "runNodeTest", { enumerable: true, get: function () { return runner_1.runNodeTest; } });
-var fileUtils_1 = require("./fileUtils");
-Object.defineProperty(exports, "writeChallengeFiles", { enumerable: true, get: function () { return fileUtils_1.writeChallengeFiles; } });
-Object.defineProperty(exports, "writeResultsToCsv", { enumerable: true, get: function () { return fileUtils_1.writeResultsToCsv; } });

package/dist/benchmarks/llm.d.ts

@@ -1,10 +0,0 @@
-import { LlmFunction } from "../types";
-import { LocalFileCache } from "langchain/cache/file_system";
-/**
- * Creates an LLM function that connects to OpenRouter
- * @param model The model identifier to use
- * @param apiKey OpenRouter API key
- * @param cache Optional local file cache to avoid duplicate requests
- * @returns A function that can be used with performAiEdit
- */
-export declare function createOpenRouterLlmFunction(model: string, apiKey: string, cache?: LocalFileCache): LlmFunction;

package/dist/benchmarks/llm.js

@@ -1,29 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createOpenRouterLlmFunction = createOpenRouterLlmFunction;
-const openai_1 = require("@langchain/openai");
-const output_parsers_1 = require("@langchain/core/output_parsers");
-/**
- * Creates an LLM function that connects to OpenRouter
- * @param model The model identifier to use
- * @param apiKey OpenRouter API key
- * @param cache Optional local file cache to avoid duplicate requests
- * @returns A function that can be used with performAiEdit
- */
-function createOpenRouterLlmFunction(model, apiKey, cache) {
-    return async (prompt) => {
-        // Create OpenAI chat model with OpenRouter configuration
-        const chatModel = new openai_1.ChatOpenAI({
-            modelName: model,
-            configuration: { apiKey, baseURL: "https://openrouter.ai/api/v1" },
-            streaming: false,
-            cache,
-        });
-        // Invoke the model
-        const result = await chatModel.invoke(prompt);
-        // Parse to string
-        const parser = new output_parsers_1.StringOutputParser();
-        const resultString = await parser.invoke(result);
-        return { content: resultString, generationId: result.lc_kwargs.id };
-    };
-}

package/dist/benchmarks/models.d.ts

@@ -1 +0,0 @@
-export declare const models: string[];

package/dist/benchmarks/models.js

@@ -1,21 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.models = void 0;
-exports.models = [
-    "anthropic/claude-3.7-sonnet",
-    "anthropic/claude-3.7-sonnet:thinking",
-    "google/gemini-2.0-flash-001",
-    "anthropic/claude-3.5-sonnet",
-    "deepseek/deepseek-r1",
-    "openai/o3-mini-high",
-    "openai/o3-mini",
-    "deepseek/deepseek-chat",
-    "qwen/qwen-2.5-72b-instruct",
-    "qwen/qwq-32b",
-    "anthropic/claude-3.5-haiku",
-    "qwen/qwen-2.5-7b-instruct",
-    "amazon/nova-micro-v1",
-    "meta-llama/llama-3.1-8b-instruct",
-    "meta-llama/llama-3.2-3b-instruct",
-    "meta-llama/llama-3.2-1b-instruct",
-];

package/dist/benchmarks/runner.d.ts

@@ -1,4 +0,0 @@
-/**
- * Runs "index.mjs" for a given challenge directory with Node and returns exit code.
- */
-export declare function runNodeTest(challengeDir: string): number;

package/dist/benchmarks/runner.js

@@ -1,35 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.runNodeTest = runNodeTest;
-const child_process_1 = require("child_process");
-/**
- * Runs "index.mjs" for a given challenge directory with Node and returns exit code.
- */
-function runNodeTest(challengeDir) {
-    const result = runNodeProcess(challengeDir);
-    logTestResult(challengeDir, result);
-    return result.status ?? 1; // If status is null, consider that a failure
-}
-/**
- * Executes the Node.js process for a test
- */
-function runNodeProcess(challengeDir) {
-    const child = (0, child_process_1.spawnSync)("node", ["index.mjs"], {
-        cwd: challengeDir,
-        encoding: "utf-8",
-    });
-    return {
-        status: child.status,
-        stdout: child.stdout,
-        stderr: child.stderr
-    };
-}
-/**
- * Logs the test result to the console
- */
-function logTestResult(challengeDir, result) {
-    console.log(`--- Running test in ${challengeDir} ---`);
-    console.log("stdout:\n", result.stdout.trim());
-    console.log("stderr:\n", result.stderr.trim());
-    console.log("exit code:", result.status, "\n");
-}

package/dist/benchmarks/types.d.ts

@@ -1,16 +0,0 @@
-import { VizFiles } from "@vizhub/viz-types";
-export interface Challenge {
-    name: string;
-    prompt: string;
-    files: VizFiles;
-}
-export interface ChallengeResult {
-    challenge: string;
-    model: string;
-    passFail: "pass" | "fail" | "error";
-}
-export interface TestRunResult {
-    status: number | null;
-    stdout: string;
-    stderr: string;
-}

package/dist/benchmarks/types.js

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

package/dist/cli.d.ts

@@ -1,2 +0,0 @@
-#!/usr/bin/env node
-export {};

package/dist/cli.js

@@ -1,157 +0,0 @@
-#!/usr/bin/env node
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const commander_1 = require("commander");
-const path = __importStar(require("path"));
-const readline = __importStar(require("readline"));
-const index_1 = require("./index");
-const llm_1 = require("./benchmarks/llm");
-const computeInitialDocument_1 = require("./computeInitialDocument");
-const dotenv_1 = __importDefault(require("dotenv"));
-const fs = __importStar(require("fs/promises"));
-// Load environment variables
-dotenv_1.default.config();
-const program = new commander_1.Command();
-program
-    .name("editcode")
-    .description("Edit code files using AI")
-    .option("-p, --prompt <prompt>", "The instruction for editing the code")
-    .option("-d, --dir <directory>", "Directory to process", process.cwd())
-    .option("--dry-run", "Show changes without writing them", false)
-    .option("--model <model>", "OpenRouter model to use", "anthropic/claude-3.7-sonnet")
-    .action(async (options) => {
-    try {
-        // Get prompt from user if not provided as argument
-        if (!options.prompt) {
-            const rl = readline.createInterface({
-                input: process.stdin,
-                output: process.stdout
-            });
-            options.prompt = await new Promise((resolve) => {
-                rl.question("> ", (answer) => {
-                    rl.close();
-                    resolve(answer);
-                });
-            });
-        }
-        if (!options.prompt.trim()) {
-            console.error("Error: Prompt cannot be empty");
-            process.exit(1);
-        }
-        // Debug: Log environment
-        console.log('Environment variables:', {
-            OPENROUTER_API_KEY: process.env.OPENROUTER_API_KEY ? '[PRESENT]' : '[MISSING]',
-            NODE_ENV: process.env.NODE_ENV,
-            PATH: process.env.PATH?.substring(0, 50) + '...'
-        });
-        // Validate API key
-        const apiKey = process.env.OPENROUTER_API_KEY;
-        if (!apiKey) {
-            console.error("Error: OPENROUTER_API_KEY environment variable is required");
-            process.exit(1);
-        }
-        // Get absolute path
-        const fullPath = path.resolve(options.dir);
-        console.log(`Processing directory: ${fullPath}`);
-        // Use computeInitialDocument to get files (respects .ignore files)
-        const initialDocument = (0, computeInitialDocument_1.computeInitialDocument)({ fullPath });
-        // Convert to VizFiles format
-        const vizFiles = {};
-        Object.entries(initialDocument.files).forEach(([id, file]) => {
-            if (file.text !== null) {
-                // Skip directories
-                vizFiles[id] = {
-                    name: path.relative(process.cwd(), path.join(fullPath, file.name)),
-                    text: file.text,
-                };
-            }
-        });
-        const fileCount = Object.keys(vizFiles).length;
-        if (fileCount === 0) {
-            console.error("No files found to process");
-            process.exit(1);
-        }
-        console.log(`Found ${fileCount} files to process`);
-        // Create LLM function
-        const llmFunction = (0, llm_1.createOpenRouterLlmFunction)(options.model, apiKey);
-        // Perform the edit
-        console.log("\nRequesting changes from AI...");
-        const result = await (0, index_1.performAiEdit)({
-            prompt: options.prompt,
-            files: vizFiles,
-            llmFunction,
-            apiKey,
-        });
-        // Show changes
-        console.log("\nProposed changes:");
-        for (const [id, file] of Object.entries(result.changedFiles)) {
-            const original = vizFiles[id];
-            if (!original) {
-                console.log(`\nNew file: ${file.name}`);
-                console.log(file.text);
-            }
-            else if (original.text !== file.text) {
-                console.log(`\nModified: ${file.name}`);
-                console.log(file.text);
-            }
-        }
-        // Write changes if not dry run
-        if (!options.dryRun) {
-            console.log("\nWriting changes...");
-            for (const [_, file] of Object.entries(result.changedFiles)) {
-                const fullPath = path.resolve(file.name);
-                await fs.mkdir(path.dirname(fullPath), { recursive: true });
-                await fs.writeFile(fullPath, file.text);
-                console.log(`Updated ${fullPath}`);
-            }
-        }
-        // Show metadata
-        console.log("\nMetadata:");
-        console.log(`Model: ${options.model}`);
-        console.log(`Provider: ${result.provider}`);
-        console.log(`Cost: ${result.upstreamCostCents / 100} USD`);
-        console.log(`Input tokens: ${result.inputTokens}`);
-        console.log(`Output tokens: ${result.outputTokens}`);
-    }
-    catch (error) {
-        console.error("Error:", error);
-        process.exit(1);
-    }
-});
-program.parse();

package/dist/computeInitialDocument.d.ts

@@ -1,13 +0,0 @@
-interface InitialDocument {
-    files: {
-        [key: string]: {
-            text: string | null;
-            name: string;
-        };
-    };
-}
-export declare const isDirectory: (path: string) => boolean;
-export declare const computeInitialDocument: ({ fullPath }: {
-    fullPath: string;
-}) => InitialDocument;
-export {};