badgr-cli 1.1.0 → 1.1.2

package/tests/workload-templates.test.js

@@ -1,865 +0,0 @@
-/**
- * Workload template tests: comfyui, train, transcribe, embed
- *
- * Covers arg parsing, input validation, API dispatch, and receipt recording.
- */
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { parseComfyuiArgs, comfyuiCommand } from '../src/commands/comfyui.js';
-import { parseTrainArgs, detectFramework, findLocalDatasetPaths, trainCommand } from '../src/commands/train.js';
-import { parseTranscribeArgs, resolveAudioInput, transcribeCommand } from '../src/commands/transcribe.js';
-import { parseEmbedArgs, resolveEmbedInput, embedCommand } from '../src/commands/embed.js';
-
-// ── Module mocks ──────────────────────────────────────────────────────────────
-
-vi.mock('../src/api.js', () => ({
-  callApi:             vi.fn(),
-  terminateDeployment: vi.fn().mockResolvedValue({}),
-  listDeployments:     vi.fn().mockResolvedValue({ deployments: [], count: 0 }),
-}));
-
-vi.mock('../src/store.js', () => ({
-  addDeployment:        vi.fn(),
-  addReceipt:           vi.fn(),
-  updateReceipt:        vi.fn(),
-  generateReceiptId:    vi.fn(() => 'rcpt-test-001'),
-  generateDeploymentId: vi.fn(() => 'dep-test-001'),
-  listDeployments:      vi.fn(() => []),
-  listReceipts:         vi.fn(() => []),
-  findDeployment:       vi.fn(() => null),
-  updateDeployment:     vi.fn(),
-  removeDeployment:     vi.fn(),
-}));
-
-// Mock callWithFallback directly so tests don't depend on API call ordering.
-vi.mock('../src/fallback.js', async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    callWithFallback: vi.fn(),
-  };
-});
-
-vi.mock('fs', async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    readFileSync: vi.fn(),
-    existsSync:   vi.fn(),
-    statSync:     vi.fn(),
-  };
-});
-
-import * as api     from '../src/api.js';
-import * as store   from '../src/store.js';
-import * as fallback from '../src/fallback.js';
-import * as fs      from 'fs';
-
-// ── Helpers ───────────────────────────────────────────────────────────────────
-
-const config = { apiKey: 'sk-test', baseUrl: 'https://api.test/v1' };
-
-const chalk = {
-  bold:   s => s, dim:    s => s, red:    s => s,
-  yellow: s => s, green:  s => s, cyan:   s => s,
-};
-
-const ENDPOINT_URL = 'https://dep-test-001.aibadgr.com/v1';
-
-function makeServeDep(overrides = {}) {
-  return {
-    deployment_id: 'dep-test-001',
-    status:        'running',
-    gpu_type:      'RTX_4090',
-    gpu_count:     1,
-    cost_per_hour: 0.72,
-    provider:      'test-provider',
-    receipt_id:    'rcpt-test-001',
-    tier:          '1',
-    endpoint_url:  ENDPOINT_URL,
-    ...overrides,
-  };
-}
-
-function makeRunDep(overrides = {}) {
-  return {
-    deployment_id: 'dep-test-001',
-    status:        'queued',
-    gpu_type:      'RTX_4090',
-    gpu_count:     1,
-    cost_per_hour: 0.72,
-    provider:      'test-provider',
-    receipt_id:    'rcpt-test-001',
-    tier:          '1',
-    ...overrides,
-  };
-}
-
-// ── Setup / teardown ──────────────────────────────────────────────────────────
-
-beforeEach(() => {
-  vi.useFakeTimers();
-  process.exitCode = undefined;
-  vi.spyOn(console, 'log').mockImplementation(() => {});
-  vi.spyOn(console, 'error').mockImplementation(() => {});
-  vi.spyOn(process.stdout, 'write').mockImplementation(() => true);
-  vi.clearAllMocks();
-  store.generateReceiptId.mockReturnValue('rcpt-test-001');
-  api.terminateDeployment.mockResolvedValue({});
-  api.listDeployments.mockResolvedValue({ deployments: [], count: 0 });
-});
-
-afterEach(() => {
-  vi.useRealTimers();
-  vi.restoreAllMocks();
-  process.exitCode = undefined;
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// parseComfyuiArgs
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('parseComfyuiArgs', () => {
-  it('parses workflow positional after "run" subcommand', () => {
-    const { workflow, flags } = parseComfyuiArgs(['run', 'workflow.json']);
-    expect(workflow).toBe('workflow.json');
-    expect(flags.gpu).toBeUndefined();
-  });
-
-  it('parses workflow without "run" subcommand', () => {
-    const { workflow } = parseComfyuiArgs(['workflow.json']);
-    expect(workflow).toBe('workflow.json');
-  });
-
-  it('parses --gpu flag', () => {
-    const { flags } = parseComfyuiArgs(['run', 'wf.json', '--gpu', 'L40S']);
-    expect(flags.gpu).toBe('L40S');
-  });
-
-  it('parses --max-cost flag', () => {
-    const { flags } = parseComfyuiArgs(['wf.json', '--max-cost', '5']);
-    expect(flags.maxCost).toBe(5);
-  });
-
-  it('parses --no-wait flag', () => {
-    const { flags } = parseComfyuiArgs(['wf.json', '--no-wait']);
-    expect(flags.noWait).toBe(true);
-  });
-
-  it('parses --check-nodes flag', () => {
-    const { flags } = parseComfyuiArgs(['wf.json', '--check-nodes', 'CLIPTextEncode,KSampler']);
-    expect(flags.checkNodes).toBe('CLIPTextEncode,KSampler');
-  });
-
-  it('parses --persistent flag', () => {
-    const { flags } = parseComfyuiArgs(['wf.json', '--persistent']);
-    expect(flags.persistent).toBe(true);
-  });
-
-  it('returns null workflow when no positional arg', () => {
-    const { workflow } = parseComfyuiArgs(['--gpu', 'RTX_4090']);
-    expect(workflow).toBeNull();
-  });
-
-  it('parses --env flag', () => {
-    const { flags } = parseComfyuiArgs(['wf.json', '--env', 'FOO=bar', '--max-cost', '5']);
-    expect(flags.env).toEqual(['FOO=bar']);
-  });
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// comfyuiCommand
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('comfyuiCommand', () => {
-  it('errors when no workflow provided', async () => {
-    await comfyuiCommand(config, [], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('errors when --max-cost and --persistent are both absent', async () => {
-    await comfyuiCommand(config, ['run', 'wf.json'], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('errors when workflow file does not exist', async () => {
-    fs.existsSync.mockReturnValue(false);
-    await comfyuiCommand(config, ['run', 'wf.json', '--max-cost', '5'], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('errors when workflow JSON is invalid', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue('not valid json {{{');
-    await comfyuiCommand(config, ['wf.json', '--max-cost', '5'], chalk);
-    expect(process.exitCode).toBe(1);
-  });
-
-  it('launches ComfyUI and records deployment', async () => {
-    const wfContent = JSON.stringify({ '1': { class_type: 'KSampler' } });
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(wfContent);
-    fallback.callWithFallback.mockResolvedValue(makeServeDep());
-
-    global.fetch = vi.fn().mockResolvedValue({ ok: true, status: 200 });
-
-    const p = comfyuiCommand(config, ['wf.json', '--no-wait', '--max-cost', '5'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    expect(store.addDeployment).toHaveBeenCalledWith(expect.objectContaining({
-      id: 'dep-test-001',
-    }));
-    expect(store.updateReceipt).toHaveBeenCalledWith('rcpt-test-001', expect.objectContaining({
-      status: 'starting',
-    }));
-    expect(process.exitCode).toBeFalsy();
-  });
-
-  it('passes COMFYUI_WORKFLOW_B64 env var to callWithFallback', async () => {
-    const wfContent = JSON.stringify({ '1': { class_type: 'CLIPTextEncode' } });
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(wfContent);
-    fallback.callWithFallback.mockResolvedValue(makeServeDep());
-
-    const p = comfyuiCommand(config, ['wf.json', '--no-wait', '--max-cost', '5'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    expect(fallback.callWithFallback).toHaveBeenCalledWith(
-      '/serve',
-      expect.anything(),
-      expect.any(Function),
-      expect.anything(),
-      expect.anything(),
-      expect.anything(),
-    );
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.env.COMFYUI_WORKFLOW_B64).toBe(Buffer.from(wfContent).toString('base64'));
-    expect(body.image).toBe('yanwk/comfyui-boot:cu126-megapak');
-  });
-
-  it('skips health check when --no-wait', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(JSON.stringify({ '1': {} }));
-    fallback.callWithFallback.mockResolvedValue(makeServeDep());
-    global.fetch = vi.fn();
-
-    const p = comfyuiCommand(config, ['wf.json', '--no-wait', '--max-cost', '5'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    expect(global.fetch).not.toHaveBeenCalled();
-    expect(store.updateReceipt).toHaveBeenCalledWith('rcpt-test-001', expect.objectContaining({
-      status: 'starting',
-    }));
-    expect(process.exitCode).toBeFalsy();
-  });
-
-  it('health-checks /system_stats and records ready state', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(JSON.stringify({ '1': {} }));
-    fallback.callWithFallback.mockResolvedValue(makeServeDep());
-
-    api.callApi.mockResolvedValue({ status: 'running' }); // dep status polls
-
-    const fetchedUrls = [];
-    global.fetch = vi.fn().mockImplementation((url) => {
-      fetchedUrls.push(url);
-      return Promise.resolve({ ok: true, status: 200 });
-    });
-
-    const p = comfyuiCommand(config, ['wf.json', '--max-cost', '5'], chalk);
-    await vi.advanceTimersByTimeAsync(5000);
-    await p;
-
-    expect(fetchedUrls.some(u => u.includes('/system_stats'))).toBe(true);
-    expect(store.updateReceipt).toHaveBeenCalledWith('rcpt-test-001', expect.objectContaining({
-      status: 'ready',
-    }));
-    expect(process.exitCode).toBeFalsy();
-  });
-
-  it('records failed receipt when deployment crashes mid-health-check', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(JSON.stringify({ '1': {} }));
-    fallback.callWithFallback.mockResolvedValue(makeServeDep());
-
-    api.callApi.mockResolvedValue({ status: 'failed', error: 'OOM' });
-    global.fetch = vi.fn().mockRejectedValue(new Error('Connection refused'));
-
-    const p = comfyuiCommand(config, ['wf.json', '--max-cost', '5'], chalk);
-    await vi.advanceTimersByTimeAsync(10000);
-    await p;
-
-    expect(store.updateReceipt).toHaveBeenCalledWith('rcpt-test-001', expect.objectContaining({
-      status: 'failed',
-    }));
-    expect(process.exitCode).toBe(1);
-  });
-
-  it('prints failure_class/next_action from the backend using the shared vocabulary', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(JSON.stringify({ '1': {} }));
-    fallback.callWithFallback.mockResolvedValue(makeServeDep());
-
-    api.callApi.mockResolvedValue({
-      status: 'failed',
-      error: 'OOM',
-      failure_class: 'container_start_failed',
-      next_action: 'Check the image and command, then retry.',
-    });
-    global.fetch = vi.fn().mockRejectedValue(new Error('Connection refused'));
-
-    const p = comfyuiCommand(config, ['wf.json', '--max-cost', '5'], chalk);
-    await vi.advanceTimersByTimeAsync(10000);
-    await p;
-
-    const errLines = console.error.mock.calls.flat().join('\n');
-    expect(errLines).toContain('Class: container_start_failed');
-    expect(errLines).toContain('Next: Check the image and command, then retry.');
-  });
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// parseTrainArgs + helpers
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('parseTrainArgs', () => {
-  it('parses config file positional', () => {
-    const { configFile } = parseTrainArgs(['config.yaml']);
-    expect(configFile).toBe('config.yaml');
-  });
-
-  it('parses --gpu flag', () => {
-    const { flags } = parseTrainArgs(['config.yaml', '--gpu', 'A100']);
-    expect(flags.gpu).toBe('A100');
-  });
-
-  it('parses --max-runtime flag', () => {
-    const { flags } = parseTrainArgs(['config.yaml', '--max-runtime', '90']);
-    expect(flags.maxRuntime).toBe(90);
-  });
-
-  it('parses --max-cost flag', () => {
-    const { flags } = parseTrainArgs(['config.yaml', '--max-cost', '20']);
-    expect(flags.maxCost).toBe(20);
-  });
-
-  it('parses --framework flag', () => {
-    const { flags } = parseTrainArgs(['config.yaml', '--framework', 'unsloth']);
-    expect(flags.framework).toBe('unsloth');
-  });
-
-  it('parses --detach flag', () => {
-    const { flags } = parseTrainArgs(['config.yaml', '--detach']);
-    expect(flags.detach).toBe(true);
-  });
-
-  it('returns null configFile when no positional arg', () => {
-    const { configFile } = parseTrainArgs(['--gpu', 'A100']);
-    expect(configFile).toBeNull();
-  });
-});
-
-describe('detectFramework', () => {
-  it('detects axolotl from base_model field', () => {
-    expect(detectFramework('base_model: meta-llama/Llama-2-7b-hf\nsequence_len: 2048')).toBe('axolotl');
-  });
-
-  it('detects unsloth from unsloth keyword', () => {
-    expect(detectFramework('# uses unsloth for fast training\nmodel: llama')).toBe('unsloth');
-  });
-
-  it('detects trl from SFTTrainer', () => {
-    expect(detectFramework('trainer: SFTTrainer\nmodel: mistral')).toBe('trl');
-  });
-
-  it('detects trl from DPOTrainer', () => {
-    expect(detectFramework('trainer: DPOTrainer')).toBe('trl');
-  });
-
-  it('falls back to generic for unknown config', () => {
-    expect(detectFramework('some_key: some_value\nother: 123')).toBe('generic');
-  });
-});
-
-describe('findLocalDatasetPaths', () => {
-  it('detects local relative paths', () => {
-    const cfg = 'datasets:\n  - path: ./my_data\n    type: completion';
-    expect(findLocalDatasetPaths(cfg)).toContain('./my_data');
-  });
-
-  it('detects absolute paths', () => {
-    const cfg = 'datasets:\n  - path: /mnt/data/train.jsonl';
-    expect(findLocalDatasetPaths(cfg)).toContain('/mnt/data/train.jsonl');
-  });
-
-  it('ignores HuggingFace dataset IDs', () => {
-    const cfg = 'datasets:\n  - path: tatsu-lab/alpaca\n    type: alpaca';
-    expect(findLocalDatasetPaths(cfg)).toEqual([]);
-  });
-
-  it('ignores s3:// URIs', () => {
-    const cfg = 'datasets:\n  - path: s3://my-bucket/data.jsonl';
-    expect(findLocalDatasetPaths(cfg)).toEqual([]);
-  });
-
-  it('ignores https:// URLs', () => {
-    const cfg = 'datasets:\n  - path: https://example.com/data.jsonl';
-    expect(findLocalDatasetPaths(cfg)).toEqual([]);
-  });
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// trainCommand
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('trainCommand', () => {
-  it('errors when no config file provided', async () => {
-    await trainCommand(config, [], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('errors when config file does not exist', async () => {
-    fs.existsSync.mockReturnValue(false);
-    await trainCommand(config, ['config.yaml'], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('launches training job and records receipt', async () => {
-    const axolotlConfig = 'base_model: meta-llama/Llama-2-7b-hf\nsequence_len: 2048';
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(axolotlConfig);
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = trainCommand(config, ['config.yaml', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    expect(store.addReceipt).toHaveBeenCalledWith(expect.objectContaining({
-      action: 'badgr train',
-      deploymentId: 'dep-test-001',
-    }));
-    expect(process.exitCode).toBeFalsy();
-  });
-
-  it('uses axolotl image for axolotl config', async () => {
-    const axolotlConfig = 'base_model: meta-llama/Llama-2-7b-hf\nsequence_len: 2048';
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(axolotlConfig);
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = trainCommand(config, ['config.yaml', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.image).toBe('winglian/axolotl:main-latest');
-  });
-
-  it('encodes config as TRAIN_CONFIG_B64', async () => {
-    const configContent = 'base_model: test-model\n';
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue(configContent);
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = trainCommand(config, ['config.yaml', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.env.TRAIN_CONFIG_B64).toBe(Buffer.from(configContent).toString('base64'));
-  });
-
-  it('applies default 120min max-runtime', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue('base_model: test\n');
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = trainCommand(config, ['config.yaml', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.max_runtime_seconds).toBe(120 * 60);
-  });
-
-  it('honours explicit --max-runtime override', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue('base_model: test\n');
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = trainCommand(config, ['config.yaml', '--max-runtime', '60', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.max_runtime_seconds).toBe(60 * 60);
-  });
-
-  it('blocks unsloth config instead of running a mismatched command', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue('# unsloth training config\nmodel: llama\n');
-
-    await trainCommand(config, ['config.yaml', '--detach'], chalk);
-
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('blocks generic (unrecognized) config instead of guessing a command', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue('some_key: some_value\nother: 123\n');
-
-    await trainCommand(config, ['config.yaml', '--detach'], chalk);
-
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('uses the trl CLI command for a trl-detected config', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue('trainer: SFTTrainer\nmodel: mistral\n');
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = trainCommand(config, ['config.yaml', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.image).toBe('huggingface/trl-source:latest');
-    expect(body.command[2]).toContain('trl sft --config /tmp/config.yaml');
-  });
-
-  it('uses the axolotl CLI command for an axolotl-detected config', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.readFileSync.mockReturnValue('base_model: meta-llama/Llama-2-7b-hf\nsequence_len: 2048\n');
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = trainCommand(config, ['config.yaml', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.command[2]).toContain('axolotl train /tmp/config.yaml');
-  });
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// parseTranscribeArgs + resolveAudioInput
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('parseTranscribeArgs', () => {
-  it('parses input positional', () => {
-    const { input } = parseTranscribeArgs(['meeting.mp3']);
-    expect(input).toBe('meeting.mp3');
-  });
-
-  it('parses --model flag', () => {
-    const { flags } = parseTranscribeArgs(['audio.mp3', '--model', 'medium']);
-    expect(flags.model).toBe('medium');
-  });
-
-  it('parses --language flag', () => {
-    const { flags } = parseTranscribeArgs(['audio.mp3', '--language', 'fr']);
-    expect(flags.language).toBe('fr');
-  });
-
-  it('parses --max-cost flag', () => {
-    const { flags } = parseTranscribeArgs(['audio.mp3', '--max-cost', '2']);
-    expect(flags.maxCost).toBe(2);
-  });
-
-  it('returns null input for empty args', () => {
-    const { input } = parseTranscribeArgs([]);
-    expect(input).toBeNull();
-  });
-});
-
-describe('resolveAudioInput', () => {
-  it('accepts https:// URL', () => {
-    const r = resolveAudioInput('https://example.com/audio.mp3');
-    expect(r.audioUrl).toBe('https://example.com/audio.mp3');
-    expect(r.audioB64).toBeNull();
-  });
-
-  it('accepts s3:// URI', () => {
-    const r = resolveAudioInput('s3://my-bucket/audio.mp3');
-    expect(r.audioUrl).toBe('s3://my-bucket/audio.mp3');
-  });
-
-  it('returns error for missing local file', () => {
-    fs.existsSync.mockReturnValue(false);
-    const r = resolveAudioInput('missing.mp3');
-    expect(r.error).toMatch(/not found/i);
-  });
-
-  it('returns error for oversized local file', () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.statSync.mockReturnValue({ size: 60 * 1024 * 1024 });
-    const r = resolveAudioInput('big.mp3');
-    expect(r.error).toMatch(/too large/i);
-  });
-
-  it('base64-encodes small local file', () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.statSync.mockReturnValue({ size: 1024 });
-    fs.readFileSync.mockReturnValue(Buffer.from('fake-audio-data'));
-    const r = resolveAudioInput('small.mp3');
-    expect(r.audioB64).toBeTruthy();
-    expect(r.audioUrl).toBeNull();
-  });
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// transcribeCommand
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('transcribeCommand', () => {
-  it('errors when no input provided', async () => {
-    await transcribeCommand(config, [], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('errors for missing local file', async () => {
-    fs.existsSync.mockReturnValue(false);
-    await transcribeCommand(config, ['missing.mp3'], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('launches transcription job for URL input', async () => {
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = transcribeCommand(
-      config,
-      ['https://example.com/audio.mp3', '--detach'],
-      chalk,
-    );
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    expect(store.addReceipt).toHaveBeenCalledWith(expect.objectContaining({
-      action: 'badgr transcribe',
-      deploymentId: 'dep-test-001',
-    }));
-    expect(process.exitCode).toBeFalsy();
-  });
-
-  it('passes AUDIO_URL env var and uses Whisper image', async () => {
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = transcribeCommand(
-      config,
-      ['s3://bucket/audio.mp3', '--detach'],
-      chalk,
-    );
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.env.AUDIO_URL).toBe('s3://bucket/audio.mp3');
-    expect(body.image).toBe('fedirz/faster-whisper-server:latest-cuda');
-  });
-
-  it('passes AUDIO_B64 env var for small local file', async () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.statSync.mockReturnValue({ size: 1024 });
-    fs.readFileSync.mockReturnValue(Buffer.from('audio-data'));
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = transcribeCommand(config, ['recording.mp3', '--detach'], chalk);
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.env.AUDIO_B64).toBeTruthy();
-    expect(body.env.AUDIO_URL).toBeUndefined();
-  });
-
-  it('defaults to large-v3 Whisper model', async () => {
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = transcribeCommand(
-      config,
-      ['https://example.com/audio.mp3', '--detach'],
-      chalk,
-    );
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.env.WHISPER_MODEL).toBe('large-v3');
-  });
-
-  it('honours --model override', async () => {
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = transcribeCommand(
-      config,
-      ['https://example.com/audio.mp3', '--model', 'medium', '--detach'],
-      chalk,
-    );
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.env.WHISPER_MODEL).toBe('medium');
-  });
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// parseEmbedArgs + resolveEmbedInput
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('parseEmbedArgs', () => {
-  it('parses model + input positional args', () => {
-    const { model, input } = parseEmbedArgs(['BAAI/bge-large-en-v1.5', 'docs.txt']);
-    expect(model).toBe('BAAI/bge-large-en-v1.5');
-    expect(input).toBe('docs.txt');
-  });
-
-  it('uses default model when only input is given', () => {
-    const { model, input } = parseEmbedArgs(['docs.txt']);
-    expect(model).toBe('BAAI/bge-large-en-v1.5');
-    expect(input).toBe('docs.txt');
-  });
-
-  it('--model flag takes precedence when only one positional', () => {
-    const { model, input } = parseEmbedArgs(['--model', 'my-model', 'docs.txt']);
-    expect(model).toBe('my-model');
-    expect(input).toBe('docs.txt');
-  });
-
-  it('parses --max-cost flag', () => {
-    const { flags } = parseEmbedArgs(['BAAI/bge-large-en-v1.5', 'docs.txt', '--max-cost', '3']);
-    expect(flags.maxCost).toBe(3);
-  });
-
-  it('returns null input for empty args', () => {
-    const { input } = parseEmbedArgs([]);
-    expect(input).toBeNull();
-  });
-});
-
-describe('resolveEmbedInput', () => {
-  it('accepts https:// URL', () => {
-    const r = resolveEmbedInput('https://example.com/docs.txt');
-    expect(r.inputUrl).toBe('https://example.com/docs.txt');
-    expect(r.inputB64).toBeNull();
-  });
-
-  it('accepts s3:// URI', () => {
-    const r = resolveEmbedInput('s3://bucket/corpus.jsonl');
-    expect(r.inputUrl).toBe('s3://bucket/corpus.jsonl');
-  });
-
-  it('returns error for missing local file', () => {
-    fs.existsSync.mockReturnValue(false);
-    const r = resolveEmbedInput('missing.txt');
-    expect(r.error).toMatch(/not found/i);
-  });
-
-  it('returns error for oversized file (> 10 MB)', () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.statSync.mockReturnValue({ size: 15 * 1024 * 1024 });
-    const r = resolveEmbedInput('large.txt');
-    expect(r.error).toMatch(/too large/i);
-  });
-
-  it('base64-encodes small local file', () => {
-    fs.existsSync.mockReturnValue(true);
-    fs.statSync.mockReturnValue({ size: 512 });
-    fs.readFileSync.mockReturnValue(Buffer.from('line1\nline2\n'));
-    const r = resolveEmbedInput('docs.txt');
-    expect(r.inputB64).toBeTruthy();
-    expect(r.inputUrl).toBeNull();
-  });
-});
-
-// ─────────────────────────────────────────────────────────────────────────────
-// embedCommand
-// ─────────────────────────────────────────────────────────────────────────────
-
-describe('embedCommand', () => {
-  it('errors when no input provided', async () => {
-    await embedCommand(config, [], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('errors for missing local file', async () => {
-    fs.existsSync.mockReturnValue(false);
-    await embedCommand(config, ['BAAI/bge-large-en-v1.5', 'missing.txt'], chalk);
-    expect(process.exitCode).toBe(1);
-    expect(fallback.callWithFallback).not.toHaveBeenCalled();
-  });
-
-  it('launches embeddings job for URL input', async () => {
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = embedCommand(
-      config,
-      ['BAAI/bge-large-en-v1.5', 'https://example.com/docs.txt', '--detach'],
-      chalk,
-    );
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    expect(store.addReceipt).toHaveBeenCalledWith(expect.objectContaining({
-      action: 'badgr embed',
-      deploymentId: 'dep-test-001',
-    }));
-    expect(process.exitCode).toBeFalsy();
-  });
-
-  it('uses vLLM image with VLLM_TASK=embed', async () => {
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = embedCommand(
-      config,
-      ['BAAI/bge-large-en-v1.5', 's3://bucket/docs.jsonl', '--detach'],
-      chalk,
-    );
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.image).toBe('vllm/vllm-openai:latest');
-    expect(body.env.VLLM_TASK).toBe('embed');
-  });
-
-  it('passes model name in env', async () => {
-    fallback.callWithFallback.mockResolvedValue(makeRunDep());
-
-    const p = embedCommand(
-      config,
-      ['my-org/my-embed-model', 's3://bucket/docs.jsonl', '--detach'],
-      chalk,
-    );
-    await vi.advanceTimersByTimeAsync(100);
-    await p;
-
-    const bodyBuilder = fallback.callWithFallback.mock.calls[0][2];
-    const body = bodyBuilder();
-    expect(body.env.EMBED_MODEL).toBe('my-org/my-embed-model');
-  });
-});