feat(websearch): add real provider chain

2026-10-14 22:17:20 +00:00 · 2026-03-23 15:26:05 -04:00
1 parent 84fd2b001c
commit 6c7d215ecc
@@ -0,0 +1,127 @@
+import { afterEach, beforeEach, describe, expect, it } from 'bun:test';
+import * as fs from 'fs';
+import * as os from 'os';
+import * as path from 'path';
+import { getWebSearchReadiness } from '../../../../src/utils/websearch/status';
+
+function writeWebSearchConfig(tempRoot: string, lines: string[]): void {
+  fs.writeFileSync(path.join(tempRoot, '.ccs', 'config.yaml'), lines.join('\n'), 'utf8');
+}
+
+describe('websearch readiness', () => {
+  const originalCcsHome = process.env.CCS_HOME;
+  const originalBraveKey = process.env.BRAVE_API_KEY;
+  const originalExaKey = process.env.EXA_API_KEY;
+  const originalTavilyKey = process.env.TAVILY_API_KEY;
+  let tempRoot = '';
+
+  beforeEach(() => {
+    tempRoot = fs.mkdtempSync(path.join(os.tmpdir(), 'ccs-websearch-status-'));
+    process.env.CCS_HOME = tempRoot;
+    delete process.env.BRAVE_API_KEY;
+    delete process.env.EXA_API_KEY;
+    delete process.env.TAVILY_API_KEY;
+    fs.mkdirSync(path.join(tempRoot, '.ccs'), { recursive: true });
+  });
+
+  afterEach(() => {
+    if (originalCcsHome !== undefined) process.env.CCS_HOME = originalCcsHome;
+    else delete process.env.CCS_HOME;
+
+    if (originalBraveKey !== undefined) process.env.BRAVE_API_KEY = originalBraveKey;
+    else delete process.env.BRAVE_API_KEY;
+
+    if (originalExaKey !== undefined) process.env.EXA_API_KEY = originalExaKey;
+    else delete process.env.EXA_API_KEY;
+
+    if (originalTavilyKey !== undefined) process.env.TAVILY_API_KEY = originalTavilyKey;
+    else delete process.env.TAVILY_API_KEY;
+
+    fs.rmSync(tempRoot, { recursive: true, force: true });
+  });
+
+  it('is ready by default because DuckDuckGo is enabled', () => {
+    const readiness = getWebSearchReadiness();
+
+    expect(readiness.readiness).toBe('ready');
+    expect(readiness.message).toContain('DuckDuckGo');
+  });
+
+  it('reports setup required when only Tavily is enabled without an API key', () => {
+    writeWebSearchConfig(tempRoot, [
+      'version: 10',
+      'websearch:',
+      '  enabled: true',
+      '  providers:',
+      '    exa:',
+      '      enabled: false',
+      '      max_results: 5',
+      '    tavily:',
+      '      enabled: true',
+      '      max_results: 5',
+      '    duckduckgo:',
+      '      enabled: false',
+      '      max_results: 5',
+      '    brave:',
+      '      enabled: false',
+      '      max_results: 5',
+      '    gemini:',
+      '      enabled: false',
+      '      model: "gemini-2.5-flash"',
+      '      timeout: 55',
+      '    opencode:',
+      '      enabled: false',
+      '      model: "opencode/grok-code"',
+      '      timeout: 90',
+      '    grok:',
+      '      enabled: false',
+      '      timeout: 55',
+      '',
+    ]);
+
+    const readiness = getWebSearchReadiness();
+
+    expect(readiness.readiness).toBe('needs_setup');
+    expect(readiness.message).toContain('Tavily');
+    expect(readiness.message).toContain('TAVILY_API_KEY');
+  });
+
+  it('prefers API-backed readiness when Exa is enabled and configured', () => {
+    process.env.EXA_API_KEY = 'exa-test-key';
+    writeWebSearchConfig(tempRoot, [
+      'version: 10',
+      'websearch:',
+      '  enabled: true',
+      '  providers:',
+      '    exa:',
+      '      enabled: true',
+      '      max_results: 5',
+      '    tavily:',
+      '      enabled: false',
+      '      max_results: 5',
+      '    duckduckgo:',
+      '      enabled: false',
+      '      max_results: 5',
+      '    brave:',
+      '      enabled: false',
+      '      max_results: 5',
+      '    gemini:',
+      '      enabled: false',
+      '      model: "gemini-2.5-flash"',
+      '      timeout: 55',
+      '    opencode:',
+      '      enabled: false',
+      '      model: "opencode/grok-code"',
+      '      timeout: 90',
+      '    grok:',
+      '      enabled: false',
+      '      timeout: 55',
+      '',
+    ]);
+
+    const readiness = getWebSearchReadiness();
+
+    expect(readiness.readiness).toBe('ready');
+    expect(readiness.message).toContain('Exa');
+  });
+});