voyageai-cli 1.12.1 → 1.15.0

package/test/commands/ping.test.js

@@ -1,172 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach, mock } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerPing } = require('../../src/commands/ping');
-
-describe('ping command', () => {
-  let originalLog;
-  let originalError;
-  let originalExit;
-  let originalKey;
-  let originalMongoUri;
-  let output;
-  let errorOutput;
-
-  // Strip ANSI escape codes for reliable string assertions in CI
-  // (GitHub Actions sets FORCE_COLOR which adds ANSI codes via picocolors)
-  const stripAnsi = (s) => s.replace(/\x1b\[[0-9;]*m/g, '');
-
-  beforeEach(() => {
-    originalLog = console.log;
-    originalError = console.error;
-    originalExit = process.exit;
-    originalKey = process.env.VOYAGE_API_KEY;
-    originalMongoUri = process.env.MONGODB_URI;
-    output = [];
-    errorOutput = [];
-    console.log = (...args) => output.push(args.join(' '));
-    console.error = (...args) => errorOutput.push(args.join(' '));
-    // Remove MONGODB_URI by default so we don't accidentally test mongo
-    delete process.env.MONGODB_URI;
-  });
-
-  afterEach(() => {
-    console.log = originalLog;
-    console.error = originalError;
-    process.exit = originalExit;
-    if (originalKey !== undefined) {
-      process.env.VOYAGE_API_KEY = originalKey;
-    } else {
-      delete process.env.VOYAGE_API_KEY;
-    }
-    if (originalMongoUri !== undefined) {
-      process.env.MONGODB_URI = originalMongoUri;
-    } else {
-      delete process.env.MONGODB_URI;
-    }
-    mock.restoreAll();
-  });
-
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerPing(program);
-    const pingCmd = program.commands.find(c => c.name() === 'ping');
-    assert.ok(pingCmd, 'ping command should be registered');
-  });
-
-  it('prints success on valid API response', async () => {
-    process.env.VOYAGE_API_KEY = 'test-key';
-
-    mock.method(global, 'fetch', async () => ({
-      ok: true,
-      status: 200,
-      json: async () => ({
-        data: [{ embedding: new Array(1024).fill(0) }],
-        usage: { total_tokens: 1 },
-      }),
-    }));
-
-    const program = new Command();
-    program.exitOverride();
-    registerPing(program);
-
-    await program.parseAsync(['node', 'test', 'ping']);
-
-    const combined = stripAnsi(output.join('\n'));
-    assert.ok(combined.includes('✓ Connected to Voyage AI API'), 'Should show success message');
-    assert.ok(combined.includes('voyage-4-lite'), 'Should show model name');
-    assert.ok(combined.includes('1024'), 'Should show dimensions');
-  });
-
-  it('exits with error on auth failure', async () => {
-    process.env.VOYAGE_API_KEY = 'bad-key';
-
-    mock.method(global, 'fetch', async () => ({
-      ok: false,
-      status: 401,
-      text: async () => 'Unauthorized',
-    }));
-
-    let exitCode = null;
-    process.exit = (code) => {
-      exitCode = code;
-      throw new Error('process.exit called');
-    };
-
-    const program = new Command();
-    program.exitOverride();
-    registerPing(program);
-
-    await assert.rejects(
-      () => program.parseAsync(['node', 'test', 'ping']),
-      /process\.exit called/
-    );
-
-    assert.equal(exitCode, 1);
-    const combined = stripAnsi(errorOutput.join('\n'));
-    assert.ok(combined.includes('Authentication failed'), 'Should show auth error');
-  });
-
-  it('exits when VOYAGE_API_KEY is not set and no config', async () => {
-    delete process.env.VOYAGE_API_KEY;
-    // Mock config to return nothing so the key isn't found in ~/.vai/config.json
-    delete require.cache[require.resolve('../../src/lib/config')];
-    delete require.cache[require.resolve('../../src/lib/api')];
-    delete require.cache[require.resolve('../../src/commands/ping')];
-    const config = require('../../src/lib/config');
-    const origGetConfigValue = config.getConfigValue;
-    config.getConfigValue = () => undefined;
-
-    const { registerPing: registerPingFresh } = require('../../src/commands/ping');
-
-    let exitCode = null;
-    process.exit = (code) => {
-      exitCode = code;
-      throw new Error('process.exit called');
-    };
-
-    const program = new Command();
-    program.exitOverride();
-    registerPingFresh(program);
-
-    try {
-      await assert.rejects(
-        () => program.parseAsync(['node', 'test', 'ping']),
-        /process\.exit called/
-      );
-
-      assert.equal(exitCode, 1);
-      const combined = stripAnsi(errorOutput.join('\n'));
-      assert.ok(combined.includes('VOYAGE_API_KEY'), 'Should mention missing key');
-    } finally {
-      config.getConfigValue = origGetConfigValue;
-    }
-  });
-
-  it('outputs JSON when --json flag is used', async () => {
-    process.env.VOYAGE_API_KEY = 'test-key';
-
-    mock.method(global, 'fetch', async () => ({
-      ok: true,
-      status: 200,
-      json: async () => ({
-        data: [{ embedding: new Array(1024).fill(0) }],
-        usage: { total_tokens: 1 },
-      }),
-    }));
-
-    const program = new Command();
-    program.exitOverride();
-    registerPing(program);
-
-    await program.parseAsync(['node', 'test', 'ping', '--json']);
-
-    const combined = output.join('\n');
-    const parsed = JSON.parse(combined);
-    assert.equal(parsed.ok, true);
-    assert.ok(parsed.voyage);
-    assert.equal(parsed.voyage.ok, true);
-  });
-});

package/test/commands/playground.test.js

@@ -1,137 +0,0 @@
-'use strict';
-
-const { describe, it, after } = require('node:test');
-const assert = require('node:assert/strict');
-const http = require('http');
-const { Command } = require('commander');
-const { registerPlayground, createPlaygroundServer } = require('../../src/commands/playground');
-
-describe('playground command', () => {
-  let server;
-  let port;
-
-  // Start server once for all tests
-  const serverReady = new Promise((resolve) => {
-    server = createPlaygroundServer();
-    server.listen(0, () => {
-      port = server.address().port;
-      resolve();
-    });
-  });
-
-  after(() => {
-    return new Promise((resolve) => {
-      if (server) server.close(resolve);
-      else resolve();
-    });
-  });
-
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerPlayground(program);
-    const cmd = program.commands.find(c => c.name() === 'playground');
-    assert.ok(cmd, 'playground command should be registered');
-    assert.ok(cmd.description().includes('playground') || cmd.description().includes('Playground') || cmd.description().includes('web'),
-      'should have a relevant description');
-  });
-
-  it('serves HTML on GET /', async () => {
-    await serverReady;
-    const body = await httpGet(`http://localhost:${port}/`);
-    assert.ok(body.includes('<!DOCTYPE html>'), 'should return HTML');
-    assert.ok(body.includes('Voyage AI Playground'), 'should include playground title');
-  });
-
-  it('returns JSON from GET /api/models', async () => {
-    await serverReady;
-    const body = await httpGet(`http://localhost:${port}/api/models`);
-    const data = JSON.parse(body);
-    assert.ok(Array.isArray(data.models), 'models should be an array');
-    assert.ok(data.models.length > 0, 'should have at least one model');
-    assert.ok(data.models.every(m => !m.legacy), 'should not include legacy models');
-    // Check a known model exists
-    const names = data.models.map(m => m.name);
-    assert.ok(names.includes('voyage-4-large'), 'should include voyage-4-large');
-  });
-
-  it('returns config with hasKey boolean from GET /api/config', async () => {
-    await serverReady;
-    const body = await httpGet(`http://localhost:${port}/api/config`);
-    const data = JSON.parse(body);
-    assert.ok(typeof data.hasKey === 'boolean', 'hasKey should be a boolean');
-    assert.ok(typeof data.baseUrl === 'string', 'baseUrl should be a string');
-    // Should never expose the actual key
-    assert.ok(!data.apiKey, 'should not expose apiKey');
-    assert.ok(!data.key, 'should not expose key');
-  });
-
-  it('returns 404 for unknown routes', async () => {
-    await serverReady;
-    const { statusCode } = await httpGetFull(`http://localhost:${port}/nonexistent`);
-    assert.equal(statusCode, 404);
-  });
-
-  it('returns 400 for POST /api/embed with invalid body', async () => {
-    await serverReady;
-    const { statusCode, body } = await httpPostFull(`http://localhost:${port}/api/embed`, { texts: 'not-an-array' });
-    assert.equal(statusCode, 400);
-    const data = JSON.parse(body);
-    assert.ok(data.error, 'should return an error message');
-  });
-});
-
-/**
- * Simple HTTP GET that returns the body string.
- */
-function httpGet(url) {
-  return new Promise((resolve, reject) => {
-    http.get(url, (res) => {
-      const chunks = [];
-      res.on('data', c => chunks.push(c));
-      res.on('end', () => resolve(Buffer.concat(chunks).toString()));
-      res.on('error', reject);
-    }).on('error', reject);
-  });
-}
-
-/**
- * HTTP GET returning { statusCode, body }.
- */
-function httpGetFull(url) {
-  return new Promise((resolve, reject) => {
-    http.get(url, (res) => {
-      const chunks = [];
-      res.on('data', c => chunks.push(c));
-      res.on('end', () => resolve({ statusCode: res.statusCode, body: Buffer.concat(chunks).toString() }));
-      res.on('error', reject);
-    }).on('error', reject);
-  });
-}
-
-/**
- * HTTP POST returning { statusCode, body }.
- */
-function httpPostFull(url, data) {
-  return new Promise((resolve, reject) => {
-    const payload = JSON.stringify(data);
-    const parsed = new URL(url);
-    const req = http.request({
-      hostname: parsed.hostname,
-      port: parsed.port,
-      path: parsed.pathname,
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Content-Length': Buffer.byteLength(payload),
-      },
-    }, (res) => {
-      const chunks = [];
-      res.on('data', c => chunks.push(c));
-      res.on('end', () => resolve({ statusCode: res.statusCode, body: Buffer.concat(chunks).toString() }));
-      res.on('error', reject);
-    });
-    req.on('error', reject);
-    req.write(payload);
-    req.end();
-  });
-}

package/test/commands/rerank.test.js

@@ -1,32 +0,0 @@
-'use strict';
-
-const { describe, it } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerRerank } = require('../../src/commands/rerank');
-
-describe('rerank command', () => {
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerRerank(program);
-    const rerankCmd = program.commands.find(c => c.name() === 'rerank');
-    assert.ok(rerankCmd, 'rerank command should be registered');
-  });
-
-  it('has --truncation flag', () => {
-    const program = new Command();
-    registerRerank(program);
-    const rerankCmd = program.commands.find(c => c.name() === 'rerank');
-    const optionNames = rerankCmd.options.map(o => o.long);
-    assert.ok(optionNames.includes('--truncation'), 'should have --truncation option');
-    assert.ok(optionNames.includes('--no-truncation'), 'should have --no-truncation option');
-  });
-
-  it('has --return-documents flag', () => {
-    const program = new Command();
-    registerRerank(program);
-    const rerankCmd = program.commands.find(c => c.name() === 'rerank');
-    const optionNames = rerankCmd.options.map(o => o.long);
-    assert.ok(optionNames.includes('--return-documents'), 'should have --return-documents option');
-  });
-});

package/test/commands/similarity.test.js

@@ -1,79 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach, mock } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerSimilarity } = require('../../src/commands/similarity');
-
-describe('similarity command', () => {
-  let originalLog, originalError, originalExit;
-  let output, errors;
-
-  beforeEach(() => {
-    originalLog = console.log;
-    originalError = console.error;
-    originalExit = process.exit;
-    output = [];
-    errors = [];
-    console.log = (...args) => output.push(args.join(' '));
-    console.error = (...args) => errors.push(args.join(' '));
-  });
-
-  afterEach(() => {
-    console.log = originalLog;
-    console.error = originalError;
-    process.exit = originalExit;
-    mock.restoreAll();
-  });
-
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerSimilarity(program);
-    const cmd = program.commands.find(c => c.name() === 'similarity');
-    assert.ok(cmd, 'similarity command should be registered');
-    assert.ok(cmd.description().includes('similarity'), 'should have a description about similarity');
-  });
-
-  it('requires at least two inputs', async () => {
-    let exitCode;
-    process.exit = (code) => { exitCode = code; throw new Error('exit'); };
-
-    const program = new Command();
-    program.exitOverride();
-    registerSimilarity(program);
-
-    try {
-      await program.parseAsync(['node', 'test', 'similarity', 'only-one-text']);
-    } catch {
-      // expected
-    }
-
-    assert.equal(exitCode, 1, 'should exit with code 1');
-    const combined = errors.join('\n');
-    assert.ok(combined.includes('two texts') || combined.includes('second text'), 'should mention needing two texts');
-  });
-
-  it('accepts --against flag with multiple values', () => {
-    const program = new Command();
-    registerSimilarity(program);
-    const cmd = program.commands.find(c => c.name() === 'similarity');
-    const againstOpt = cmd.options.find(o => o.long === '--against');
-    assert.ok(againstOpt, '--against option should exist');
-  });
-
-  it('accepts --json flag', () => {
-    const program = new Command();
-    registerSimilarity(program);
-    const cmd = program.commands.find(c => c.name() === 'similarity');
-    const jsonOpt = cmd.options.find(o => o.long === '--json');
-    assert.ok(jsonOpt, '--json option should exist');
-  });
-
-  it('cosineSimilarity integration — identical texts would score 1.0', () => {
-    // Direct test of math module integration
-    const { cosineSimilarity } = require('../../src/lib/math');
-    const vec = [0.1, 0.2, 0.3, 0.4];
-    const result = cosineSimilarity(vec, vec);
-    assert.ok(Math.abs(result - 1.0) < 1e-10);
-  });
-});

package/test/commands/store.test.js

@@ -1,26 +0,0 @@
-'use strict';
-
-const { describe, it } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerStore } = require('../../src/commands/store');
-
-describe('store command', () => {
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerStore(program);
-    const storeCmd = program.commands.find(c => c.name() === 'store');
-    assert.ok(storeCmd, 'store command should be registered');
-  });
-
-  it('has --input-type flag defaulting to document', () => {
-    const program = new Command();
-    registerStore(program);
-    const storeCmd = program.commands.find(c => c.name() === 'store');
-    const optionNames = storeCmd.options.map(o => o.long);
-    assert.ok(optionNames.includes('--input-type'), 'should have --input-type option');
-    // Check the default value
-    const inputTypeOpt = storeCmd.options.find(o => o.long === '--input-type');
-    assert.equal(inputTypeOpt.defaultValue, 'document', '--input-type should default to document');
-  });
-});

package/test/fixtures/sample.csv

@@ -1,6 +0,0 @@
-title,content,category
-"MongoDB Overview","MongoDB is a document database designed for ease of development and scaling.",databases
-"Vector Search","Atlas Vector Search enables semantic search over your data using embeddings.",search
-"Voyage AI","Voyage AI provides state-of-the-art embedding models for retrieval.",ai
-"RAG Explained","RAG combines retrieval with generation for more accurate AI responses.",ai
-"Node.js Intro","Node.js is a JavaScript runtime built on Chrome's V8 engine.",runtime

package/test/fixtures/sample.json

@@ -1,7 +0,0 @@
-[
-  {"text": "MongoDB is a document database designed for ease of development and scaling.", "source": "docs"},
-  {"text": "Atlas Vector Search enables semantic search over your data using embeddings.", "source": "atlas"},
-  {"text": "Voyage AI provides state-of-the-art embedding models for retrieval.", "source": "voyage"},
-  {"text": "RAG combines retrieval with generation for more accurate AI responses.", "source": "ai"},
-  {"text": "Node.js is a JavaScript runtime built on Chrome's V8 engine.", "source": "nodejs"}
-]

package/test/fixtures/sample.jsonl

@@ -1,5 +0,0 @@
-{"text": "MongoDB is a document database designed for ease of development and scaling.", "source": "docs"}
-{"text": "Atlas Vector Search enables semantic search over your data using embeddings.", "source": "atlas"}
-{"text": "Voyage AI provides state-of-the-art embedding models for retrieval.", "source": "voyage"}
-{"text": "RAG combines retrieval with generation for more accurate AI responses.", "source": "ai"}
-{"text": "Node.js is a JavaScript runtime built on Chrome's V8 engine.", "source": "nodejs"}

package/test/fixtures/sample.txt

@@ -1,5 +0,0 @@
-MongoDB is a document database designed for ease of development and scaling.
-Atlas Vector Search enables semantic search over your data using embeddings.
-Voyage AI provides state-of-the-art embedding models for retrieval.
-RAG combines retrieval with generation for more accurate AI responses.
-Node.js is a JavaScript runtime built on Chrome's V8 engine.

package/test/lib/api.test.js

@@ -1,133 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach, mock } = require('node:test');
-const assert = require('node:assert/strict');
-
-describe('api', () => {
-  let originalKey;
-  let originalExit;
-
-  beforeEach(() => {
-    originalKey = process.env.VOYAGE_API_KEY;
-    originalExit = process.exit;
-  });
-
-  afterEach(() => {
-    if (originalKey !== undefined) {
-      process.env.VOYAGE_API_KEY = originalKey;
-    } else {
-      delete process.env.VOYAGE_API_KEY;
-    }
-    process.exit = originalExit;
-    mock.restoreAll();
-  });
-
-  describe('requireApiKey', () => {
-    it('throws/exits when VOYAGE_API_KEY is not set', () => {
-      delete process.env.VOYAGE_API_KEY;
-      // Re-require to get fresh modules (clear config cache too)
-      delete require.cache[require.resolve('../../src/lib/api')];
-      delete require.cache[require.resolve('../../src/lib/config')];
-      const config = require('../../src/lib/config');
-      const originalGetConfigValue = config.getConfigValue;
-      config.getConfigValue = () => undefined;
-
-      const { requireApiKey } = require('../../src/lib/api');
-
-      let exitCode = null;
-      process.exit = (code) => {
-        exitCode = code;
-        throw new Error('process.exit called');
-      };
-
-      try {
-        assert.throws(() => requireApiKey(), /process\.exit called/);
-        assert.equal(exitCode, 1);
-      } finally {
-        config.getConfigValue = originalGetConfigValue;
-      }
-    });
-
-    it('returns key when VOYAGE_API_KEY is set', () => {
-      process.env.VOYAGE_API_KEY = 'test-key-123';
-      delete require.cache[require.resolve('../../src/lib/api')];
-      const { requireApiKey } = require('../../src/lib/api');
-
-      const key = requireApiKey();
-      assert.equal(key, 'test-key-123');
-    });
-  });
-
-  describe('apiRequest', () => {
-    it('returns parsed JSON on success', async () => {
-      process.env.VOYAGE_API_KEY = 'test-key';
-      delete require.cache[require.resolve('../../src/lib/api')];
-      const { apiRequest } = require('../../src/lib/api');
-
-      const mockResponse = {
-        ok: true,
-        status: 200,
-        json: async () => ({ data: [{ embedding: [1, 2, 3] }] }),
-        headers: new Map(),
-      };
-      mock.method(global, 'fetch', async () => mockResponse);
-
-      const result = await apiRequest('/embeddings', { input: ['test'], model: 'voyage-4-lite' });
-      assert.deepEqual(result, { data: [{ embedding: [1, 2, 3] }] });
-    });
-
-    it('exits on non-200 response', async () => {
-      process.env.VOYAGE_API_KEY = 'test-key';
-      delete require.cache[require.resolve('../../src/lib/api')];
-      const { apiRequest } = require('../../src/lib/api');
-
-      let exitCode = null;
-      process.exit = (code) => {
-        exitCode = code;
-        throw new Error('process.exit called');
-      };
-
-      const mockResponse = {
-        ok: false,
-        status: 400,
-        json: async () => ({ detail: 'Bad request' }),
-        headers: new Map(),
-      };
-      mock.method(global, 'fetch', async () => mockResponse);
-
-      await assert.rejects(
-        () => apiRequest('/embeddings', { input: ['test'], model: 'voyage-4-lite' }),
-        /API Error \(400\)/
-      );
-    });
-
-    it('retries on 429', async () => {
-      process.env.VOYAGE_API_KEY = 'test-key';
-      delete require.cache[require.resolve('../../src/lib/api')];
-      const { apiRequest } = require('../../src/lib/api');
-
-      let callCount = 0;
-      mock.method(global, 'fetch', async () => {
-        callCount++;
-        if (callCount === 1) {
-          return {
-            ok: false,
-            status: 429,
-            headers: { get: () => '0' },
-            json: async () => ({ detail: 'Rate limited' }),
-          };
-        }
-        return {
-          ok: true,
-          status: 200,
-          json: async () => ({ data: 'success' }),
-          headers: new Map(),
-        };
-      });
-
-      const result = await apiRequest('/embeddings', { input: ['test'], model: 'voyage-4-lite' });
-      assert.deepEqual(result, { data: 'success' });
-      assert.equal(callCount, 2, 'Should have retried once');
-    });
-  });
-});

package/test/lib/banner.test.js

@@ -1,44 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach } = require('node:test');
-const assert = require('node:assert/strict');
-const { showBanner, showQuickStart, getVersion } = require('../../src/lib/banner');
-
-describe('banner', () => {
-  let originalLog;
-  let output;
-
-  beforeEach(() => {
-    originalLog = console.log;
-    output = [];
-    console.log = (...args) => output.push(args.join(' '));
-  });
-
-  afterEach(() => {
-    console.log = originalLog;
-  });
-
-  it('getVersion returns a semver string', () => {
-    const version = getVersion();
-    assert.ok(/^\d+\.\d+\.\d+/.test(version), `Expected semver, got: ${version}`);
-  });
-
-  it('showBanner prints box with vai and Voyage AI CLI', () => {
-    showBanner();
-    const combined = output.join('\n');
-    assert.ok(combined.includes('vai'), 'Should include "vai"');
-    assert.ok(combined.includes('Voyage AI CLI'), 'Should include "Voyage AI CLI"');
-    assert.ok(combined.includes('╭'), 'Should include top border');
-    assert.ok(combined.includes('╰'), 'Should include bottom border');
-  });
-
-  it('showQuickStart prints quick start commands', () => {
-    showQuickStart();
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Quick start'), 'Should include header');
-    assert.ok(combined.includes('vai ping'), 'Should include ping');
-    assert.ok(combined.includes('vai embed'), 'Should include embed');
-    assert.ok(combined.includes('vai models'), 'Should include models');
-    assert.ok(combined.includes('vai demo'), 'Should include demo');
-  });
-});