mcp-omnisearch/dist/providers/unified/web_search.js

MCP server for integrating Omnisearch with LLMs

import { ErrorType, ProviderError, } from '../../common/types.js';
import { BraveSearchProvider } from '../search/brave/index.js';
import { ExaSearchProvider } from '../search/exa/index.js';
import { KagiSearchProvider } from '../search/kagi/index.js';
import { TavilySearchProvider } from '../search/tavily/index.js';
export class UnifiedWebSearchProvider {
    constructor() {
        this.name = 'web_search';
        this.description = 'Search the web. Providers: tavily (factual/citations), brave (privacy/operators), kagi (quality/operators), exa (AI-semantic). Brave/Kagi support query operators like site:, filetype:, lang:, etc.';
        this.providers = new Map();
        this.providers.set('tavily', new TavilySearchProvider());
        this.providers.set('brave', new BraveSearchProvider());
        this.providers.set('kagi', new KagiSearchProvider());
        this.providers.set('exa', new ExaSearchProvider());
    }
    async search(params) {
        const { provider, ...searchParams } = params;
        if (!provider) {
            throw new ProviderError(ErrorType.INVALID_INPUT, 'Provider parameter is required', this.name);
        }
        const selectedProvider = this.providers.get(provider);
        if (!selectedProvider) {
            throw new ProviderError(ErrorType.INVALID_INPUT, `Invalid provider: ${provider}. Valid options: ${Array.from(this.providers.keys()).join(', ')}`, this.name);
        }
        return selectedProvider.search(searchParams);
    }
}