YeonGyu-Kim
45 lines
1.2 KiB
TypeScript
45 lines
1.2 KiB
TypeScript
import type { WebsearchConfig } from "../config/schema"
|
|
import { log } from "../shared/logger"
|
|
|
|
type RemoteMcpConfig = {
|
|
type: "remote"
|
|
url: string
|
|
enabled: boolean
|
|
headers?: Record<string, string>
|
|
oauth?: false
|
|
}
|
|
|
|
export function createWebsearchConfig(config?: WebsearchConfig): RemoteMcpConfig | undefined {
|
|
const provider = config?.provider || "exa"
|
|
|
|
if (provider === "tavily") {
|
|
const tavilyKey = process.env.TAVILY_API_KEY
|
|
if (!tavilyKey) {
|
|
log("[websearch] Tavily API key not found, skipping websearch MCP")
|
|
return undefined
|
|
}
|
|
|
|
return {
|
|
type: "remote" as const,
|
|
url: "https://mcp.tavily.com/mcp/",
|
|
enabled: true,
|
|
headers: {
|
|
Authorization: `Bearer ${tavilyKey}`,
|
|
},
|
|
oauth: false as const,
|
|
}
|
|
}
|
|
|
|
return {
|
|
type: "remote" as const,
|
|
url: process.env.EXA_API_KEY
|
|
? `https://mcp.exa.ai/mcp?tools=web_search_exa&exaApiKey=${encodeURIComponent(process.env.EXA_API_KEY)}`
|
|
: "https://mcp.exa.ai/mcp?tools=web_search_exa",
|
|
enabled: true,
|
|
...(process.env.EXA_API_KEY ? { headers: { "x-api-key": process.env.EXA_API_KEY } } : {}),
|
|
oauth: false as const,
|
|
}
|
|
}
|
|
|
|
export const websearch = createWebsearchConfig()
|