voyageai-cli 1.12.1 → 1.13.0

package/test/commands/completions.test.js

@@ -1,166 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerCompletions, generateBashCompletions, generateZshCompletions } = require('../../src/commands/completions');
-
-describe('completions command', () => {
-  let originalLog;
-  let originalWrite;
-  let originalError;
-  let originalExit;
-  let output;
-  let stdoutOutput;
-  let stderrOutput;
-
-  beforeEach(() => {
-    originalLog = console.log;
-    originalWrite = process.stdout.write;
-    originalError = console.error;
-    originalExit = process.exit;
-    output = [];
-    stdoutOutput = [];
-    stderrOutput = [];
-    console.log = (...args) => output.push(args.join(' '));
-    process.stdout.write = (data) => { stdoutOutput.push(data); return true; };
-    console.error = (...args) => stderrOutput.push(args.join(' '));
-    process.exit = (code) => { throw new Error(`EXIT_${code}`); };
-  });
-
-  afterEach(() => {
-    console.log = originalLog;
-    process.stdout.write = originalWrite;
-    console.error = originalError;
-    process.exit = originalExit;
-  });
-
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerCompletions(program);
-    const cmd = program.commands.find(c => c.name() === 'completions');
-    assert.ok(cmd, 'completions command should be registered');
-    assert.ok(cmd.description().includes('completion'), 'should have a description about completions');
-  });
-
-  it('shows usage when called without shell argument', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerCompletions(program);
-
-    await program.parseAsync(['node', 'test', 'completions']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('bash'), 'should mention bash');
-    assert.ok(combined.includes('zsh'), 'should mention zsh');
-  });
-
-  it('outputs bash completion script', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerCompletions(program);
-
-    await program.parseAsync(['node', 'test', 'completions', 'bash']);
-
-    const combined = stdoutOutput.join('');
-    assert.ok(combined.includes('_vai_completions'), 'should contain bash completion function');
-    assert.ok(combined.includes('complete -F _vai_completions vai'), 'should register completion');
-  });
-
-  it('outputs zsh completion script', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerCompletions(program);
-
-    await program.parseAsync(['node', 'test', 'completions', 'zsh']);
-
-    const combined = stdoutOutput.join('');
-    assert.ok(combined.includes('#compdef vai'), 'should contain zsh compdef header');
-    assert.ok(combined.includes('_vai'), 'should contain zsh completion function');
-  });
-
-  it('rejects unknown shell', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerCompletions(program);
-
-    await assert.rejects(
-      () => program.parseAsync(['node', 'test', 'completions', 'fish']),
-      /EXIT_1/,
-      'should exit with code 1 for unsupported shell'
-    );
-    const combined = stderrOutput.join('\n');
-    assert.ok(combined.includes('fish'), 'should mention the unknown shell name');
-  });
-});
-
-describe('generateBashCompletions', () => {
-  it('includes all 14 commands (including completions)', () => {
-    const script = generateBashCompletions();
-    const commands = ['embed', 'rerank', 'store', 'search', 'index', 'models', 'ping', 'config', 'demo', 'explain', 'similarity', 'ingest', 'completions', 'help'];
-    for (const cmd of commands) {
-      assert.ok(script.includes(cmd), `should include command: ${cmd}`);
-    }
-  });
-
-  it('includes model completions', () => {
-    const script = generateBashCompletions();
-    assert.ok(script.includes('voyage-4-large'), 'should include voyage-4-large model');
-    assert.ok(script.includes('rerank-2.5'), 'should include rerank-2.5 model');
-  });
-
-  it('includes flag completions for embed', () => {
-    const script = generateBashCompletions();
-    assert.ok(script.includes('--model'), 'should include --model flag');
-    assert.ok(script.includes('--dimensions'), 'should include --dimensions flag');
-    assert.ok(script.includes('--input-type'), 'should include --input-type flag');
-  });
-
-  it('includes index subcommands', () => {
-    const script = generateBashCompletions();
-    assert.ok(script.includes('create list delete'), 'should include index subcommands');
-  });
-
-  it('includes config subcommands', () => {
-    const script = generateBashCompletions();
-    assert.ok(script.includes('set get delete path reset'), 'should include config subcommands');
-  });
-
-  it('includes input-type values', () => {
-    const script = generateBashCompletions();
-    assert.ok(script.includes('query document'), 'should include input-type values');
-  });
-});
-
-describe('generateZshCompletions', () => {
-  it('includes compdef header', () => {
-    const script = generateZshCompletions();
-    assert.ok(script.startsWith('#compdef vai'), 'should start with #compdef vai');
-  });
-
-  it('includes all commands with descriptions', () => {
-    const script = generateZshCompletions();
-    const commands = ['embed', 'rerank', 'store', 'search', 'index', 'models', 'ping', 'config', 'demo', 'explain', 'similarity', 'ingest', 'completions'];
-    for (const cmd of commands) {
-      assert.ok(script.includes(`'${cmd}:`), `should include command with description: ${cmd}`);
-    }
-  });
-
-  it('includes model names', () => {
-    const script = generateZshCompletions();
-    assert.ok(script.includes('voyage-4-large'), 'should include voyage-4-large model');
-    assert.ok(script.includes('voyage-code-3'), 'should include voyage-code-3 model');
-  });
-
-  it('includes explain topics', () => {
-    const script = generateZshCompletions();
-    assert.ok(script.includes('embeddings'), 'should include embeddings topic');
-    assert.ok(script.includes('cosine-similarity'), 'should include cosine-similarity topic');
-    assert.ok(script.includes('batch-processing'), 'should include batch-processing topic');
-  });
-
-  it('includes file completion for --file flags', () => {
-    const script = generateZshCompletions();
-    assert.ok(script.includes('_files'), 'should use _files completion');
-  });
-});

package/test/commands/config.test.js

@@ -1,35 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerConfig } = require('../../src/commands/config');
-
-describe('config command', () => {
-  it('registers config command with subcommands', () => {
-    const program = new Command();
-    program.exitOverride();
-    registerConfig(program);
-
-    const configCmd = program.commands.find(c => c.name() === 'config');
-    assert.ok(configCmd, 'config command should be registered');
-
-    const subNames = configCmd.commands.map(c => c.name());
-    assert.ok(subNames.includes('set'), 'should have set subcommand');
-    assert.ok(subNames.includes('get'), 'should have get subcommand');
-    assert.ok(subNames.includes('delete'), 'should have delete subcommand');
-    assert.ok(subNames.includes('path'), 'should have path subcommand');
-    assert.ok(subNames.includes('reset'), 'should have reset subcommand');
-  });
-
-  it('config set/get/delete round-trip via lib functions', () => {
-    // This test exercises the underlying lib (already tested in config.test.js)
-    // but validates the key mapping used by the command
-    const { KEY_MAP } = require('../../src/lib/config');
-
-    assert.equal(KEY_MAP['api-key'], 'apiKey');
-    assert.equal(KEY_MAP['mongodb-uri'], 'mongodbUri');
-    assert.equal(KEY_MAP['default-model'], 'defaultModel');
-    assert.equal(KEY_MAP['default-dimensions'], 'defaultDimensions');
-  });
-});

package/test/commands/demo.test.js

@@ -1,46 +0,0 @@
-'use strict';
-
-const { describe, it } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerDemo } = require('../../src/commands/demo');
-
-describe('demo command', () => {
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerDemo(program);
-    const demoCmd = program.commands.find(c => c.name() === 'demo');
-    assert.ok(demoCmd, 'demo command should be registered');
-  });
-
-  it('has --no-pause option', () => {
-    const program = new Command();
-    registerDemo(program);
-    const demoCmd = program.commands.find(c => c.name() === 'demo');
-    const opts = demoCmd.options.map(o => o.long);
-    assert.ok(opts.includes('--no-pause'), 'Should have --no-pause option');
-  });
-
-  it('has --skip-pipeline option', () => {
-    const program = new Command();
-    registerDemo(program);
-    const demoCmd = program.commands.find(c => c.name() === 'demo');
-    const opts = demoCmd.options.map(o => o.long);
-    assert.ok(opts.includes('--skip-pipeline'), 'Should have --skip-pipeline option');
-  });
-
-  it('has --keep option', () => {
-    const program = new Command();
-    registerDemo(program);
-    const demoCmd = program.commands.find(c => c.name() === 'demo');
-    const opts = demoCmd.options.map(o => o.long);
-    assert.ok(opts.includes('--keep'), 'Should have --keep option');
-  });
-
-  it('has correct description', () => {
-    const program = new Command();
-    registerDemo(program);
-    const demoCmd = program.commands.find(c => c.name() === 'demo');
-    assert.ok(demoCmd.description().includes('walkthrough'), 'Should mention walkthrough');
-  });
-});

package/test/commands/embed.test.js

@@ -1,42 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach, mock } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerEmbed } = require('../../src/commands/embed');
-
-describe('embed command', () => {
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerEmbed(program);
-    const embedCmd = program.commands.find(c => c.name() === 'embed');
-    assert.ok(embedCmd, 'embed command should be registered');
-  });
-
-  it('has --truncation flag', () => {
-    const program = new Command();
-    registerEmbed(program);
-    const embedCmd = program.commands.find(c => c.name() === 'embed');
-    const optionNames = embedCmd.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 --input-type flag', () => {
-    const program = new Command();
-    registerEmbed(program);
-    const embedCmd = program.commands.find(c => c.name() === 'embed');
-    const optionNames = embedCmd.options.map(o => o.long);
-    assert.ok(optionNames.includes('--input-type'), 'should have --input-type option');
-  });
-
-  it('has --output-dtype flag with float default', () => {
-    const program = new Command();
-    registerEmbed(program);
-    const embedCmd = program.commands.find(c => c.name() === 'embed');
-    const optionNames = embedCmd.options.map(o => o.long);
-    assert.ok(optionNames.includes('--output-dtype'), 'should have --output-dtype option');
-    const opt = embedCmd.options.find(o => o.long === '--output-dtype');
-    assert.equal(opt.defaultValue, 'float');
-  });
-});

package/test/commands/explain.test.js

@@ -1,207 +0,0 @@
-'use strict';
-
-const { describe, it, beforeEach, afterEach } = require('node:test');
-const assert = require('node:assert/strict');
-const { Command } = require('commander');
-const { registerExplain } = require('../../src/commands/explain');
-
-describe('explain command', () => {
-  let originalLog;
-  let originalError;
-  let originalExit;
-  let output;
-  let errorOutput;
-
-  beforeEach(() => {
-    originalLog = console.log;
-    originalError = console.error;
-    originalExit = process.exit;
-    output = [];
-    errorOutput = [];
-    console.log = (...args) => output.push(args.join(' '));
-    console.error = (...args) => errorOutput.push(args.join(' '));
-  });
-
-  afterEach(() => {
-    console.log = originalLog;
-    console.error = originalError;
-    process.exit = originalExit;
-  });
-
-  it('registers correctly on a program', () => {
-    const program = new Command();
-    registerExplain(program);
-    const cmd = program.commands.find(c => c.name() === 'explain');
-    assert.ok(cmd, 'explain command should be registered');
-  });
-
-  it('lists topics when no argument given', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Available topics'), 'Should show available topics header');
-    assert.ok(combined.includes('embeddings'), 'Should list embeddings');
-    assert.ok(combined.includes('reranking'), 'Should list reranking');
-    assert.ok(combined.includes('rag'), 'Should list rag');
-    assert.ok(combined.includes('vector-search'), 'Should list vector-search');
-    assert.ok(combined.includes('vai explain <topic>'), 'Should show usage hint');
-  });
-
-  it('shows content for a known concept', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'embeddings']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Embeddings'), 'Should show title');
-    assert.ok(combined.includes('vector'), 'Should contain explanation about vectors');
-    assert.ok(combined.includes('Try it'), 'Should show Try it section');
-    assert.ok(combined.includes('Learn more'), 'Should show Learn more section');
-    assert.ok(combined.includes('not affiliated'), 'Should show disclaimer');
-  });
-
-  it('resolves alias "embed" to embeddings', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'embed']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Embeddings'), 'Should resolve alias and show Embeddings');
-  });
-
-  it('resolves alias "rerank" to reranking', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'rerank']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Reranking'), 'Should resolve alias and show Reranking');
-  });
-
-  it('resolves alias "vectors" to vector-search', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'vectors']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Vector Search'), 'Should resolve alias and show Vector Search');
-  });
-
-  it('resolves alias "similarity" to cosine-similarity', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'similarity']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Cosine Similarity'), 'Should resolve alias');
-  });
-
-  it('resolves alias "two-stage" to two-stage-retrieval', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'two-stage']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Two-Stage Retrieval'), 'Should resolve alias');
-  });
-
-  it('resolves alias "keys" to api-keys', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'keys']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('API Keys'), 'Should resolve alias');
-  });
-
-  it('resolves alias "batch" to batch-processing', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'batch']);
-
-    const combined = output.join('\n');
-    assert.ok(combined.includes('Batch Processing'), 'Should resolve alias');
-  });
-
-  it('shows error and suggestions for unknown concept', async () => {
-    let exitCode = null;
-    process.exit = (code) => {
-      exitCode = code;
-      throw new Error('process.exit called');
-    };
-
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await assert.rejects(
-      () => program.parseAsync(['node', 'test', 'explain', 'nonexistent']),
-      /process\.exit called/
-    );
-
-    assert.equal(exitCode, 1);
-    const combined = errorOutput.join('\n');
-    assert.ok(combined.includes('Unknown topic'), 'Should show unknown topic error');
-    assert.ok(combined.includes('vai explain'), 'Should suggest running vai explain');
-  });
-
-  it('outputs JSON for topic list with --json flag', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', '--json']);
-
-    const combined = output.join('\n');
-    const parsed = JSON.parse(combined);
-    assert.ok(Array.isArray(parsed.topics), 'Should have topics array');
-    assert.ok(parsed.topics.length >= 10, 'Should have at least 10 topics');
-    assert.ok(parsed.topics[0].key, 'Each topic should have a key');
-    assert.ok(parsed.topics[0].title, 'Each topic should have a title');
-    assert.ok(parsed.topics[0].summary, 'Each topic should have a summary');
-  });
-
-  it('outputs JSON for a specific concept with --json flag', async () => {
-    const program = new Command();
-    program.exitOverride();
-    registerExplain(program);
-
-    await program.parseAsync(['node', 'test', 'explain', 'rag', '--json']);
-
-    const combined = output.join('\n');
-    const parsed = JSON.parse(combined);
-    assert.equal(parsed.concept, 'rag');
-    assert.equal(parsed.title, 'RAG (Retrieval-Augmented Generation)');
-    assert.ok(parsed.content, 'Should have content');
-    assert.ok(Array.isArray(parsed.links), 'Should have links');
-    assert.ok(Array.isArray(parsed.tryIt), 'Should have tryIt');
-  });
-
-  it('has --json option', () => {
-    const program = new Command();
-    registerExplain(program);
-    const cmd = program.commands.find(c => c.name() === 'explain');
-    const opts = cmd.options.map(o => o.long);
-    assert.ok(opts.includes('--json'), 'Should have --json option');
-  });
-});