wyxrouter 0.4.71

package/open-sse/handlers/search/chatSearch.js

@@ -0,0 +1,409 @@
+/**
+ * Wrap chat-completions endpoints (with built-in web search) into the unified
+ * /v1/search response format. Supports gemini, openai, xai, kimi, minimax, perplexity.
+ */
+
+const REQUEST_TIMEOUT_MS = 15000;
+const DEFAULT_MAX_RESULTS = 10;
+
+/**
+ * Normalize a citation entry into the unified result shape.
+ * @param {{url:string, title?:string, snippet?:string}} c
+ * @param {number} index
+ * @param {string} provider
+ * @param {string} retrievedAt
+ */
+function toResult(c, index, provider, retrievedAt) {
+  return {
+    title: c.title || "",
+    url: c.url,
+    snippet: c.snippet || "",
+    position: index + 1,
+    score: null,
+    published_at: null,
+    favicon_url: null,
+    content: null,
+    metadata: {},
+    citation: { provider, retrieved_at: retrievedAt, rank: index + 1 },
+    provider_raw: null
+  };
+}
+
+/** Coerce a citation that might be a raw URL string or an object. */
+function normalizeCitation(c) {
+  if (!c) return null;
+  if (typeof c === "string") return { url: c };
+  if (typeof c === "object" && c.url) return c;
+  return null;
+}
+
+/**
+ * Provider-specific configuration map. All providers must implement:
+ * { endpoint, defaultModel, buildBody, buildHeaders, extractAnswer }
+ */
+const CHAT_SEARCH_CONFIG = {
+  gemini: {
+    endpoint: (model) =>
+      `https://generativelanguage.googleapis.com/v1beta/models/${model}:generateContent`,
+    defaultModel: "gemini-2.5-flash",
+    buildBody: (query) => ({
+      contents: [{ role: "user", parts: [{ text: query }] }],
+      tools: [{ google_search: {} }]
+    }),
+    buildHeaders: (token) => ({
+      "Content-Type": "application/json",
+      "x-goog-api-key": token
+    }),
+    extractAnswer: (data) => {
+      const candidate = data?.candidates?.[0];
+      const parts = candidate?.content?.parts || [];
+      const text = parts.map((p) => p?.text || "").filter(Boolean).join("");
+      const chunks = candidate?.groundingMetadata?.groundingChunks || [];
+      const citations = chunks
+        .map((ch) => ch?.web)
+        .filter(Boolean)
+        .map((w) => ({ url: w.uri || w.url, title: w.title || "" }))
+        .filter((c) => c.url);
+      const tokens = data?.usageMetadata?.totalTokenCount || 0;
+      return { text, citations, tokens };
+    }
+  },
+
+  openai: {
+    endpoint: () => "https://api.openai.com/v1/chat/completions",
+    defaultModel: "gpt-4o-mini",
+    buildBody: (query, model) => {
+      const body = {
+        model,
+        messages: [{ role: "user", content: query }]
+      };
+      // Non-search-preview models need explicit web_search tool
+      if (!/search/i.test(model)) {
+        body.tools = [{ type: "web_search" }];
+      }
+      return body;
+    },
+    buildHeaders: (token) => ({
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${token}`
+    }),
+    extractAnswer: (data) => {
+      const msg = data?.choices?.[0]?.message || {};
+      const text = msg.content || "";
+      const annotations = Array.isArray(msg.annotations) ? msg.annotations : [];
+      const fromAnn = annotations
+        .map((a) => a?.url_citation)
+        .filter(Boolean)
+        .map((u) => ({ url: u.url, title: u.title || "" }));
+      const fromTop = Array.isArray(data?.citations)
+        ? data.citations.map(normalizeCitation).filter(Boolean)
+        : [];
+      const citations = fromAnn.length ? fromAnn : fromTop;
+      const tokens = data?.usage?.total_tokens || 0;
+      return { text, citations, tokens };
+    }
+  },
+
+  xai: {
+    endpoint: () => "https://api.x.ai/v1/responses",
+    defaultModel: "grok-4.20-reasoning",
+    buildBody: (query, model) => ({
+      model,
+      input: [{ role: "user", content: query }],
+      tools: [{ type: "web_search" }]
+    }),
+    buildHeaders: (token) => ({
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${token}`
+    }),
+    extractAnswer: (data) => {
+      // /v1/responses returns output[] array of message/tool blocks
+      const output = Array.isArray(data?.output) ? data.output : [];
+      let text = "";
+      const citations = [];
+      for (const item of output) {
+        const parts = Array.isArray(item?.content) ? item.content : [];
+        for (const p of parts) {
+          if (typeof p?.text === "string") text += p.text;
+          const anns = Array.isArray(p?.annotations) ? p.annotations : [];
+          for (const a of anns) {
+            const c = normalizeCitation(a?.url ? a : a?.url_citation);
+            if (c) citations.push(c);
+          }
+        }
+      }
+      // Fallback: top-level citations array (some response variants)
+      if (!citations.length && Array.isArray(data?.citations)) {
+        for (const c of data.citations) {
+          const n = normalizeCitation(c);
+          if (n) citations.push(n);
+        }
+      }
+      const tokens = data?.usage?.total_tokens || 0;
+      return { text, citations, tokens };
+    }
+  },
+
+  kimi: {
+    endpoint: () => "https://api.moonshot.cn/v1/chat/completions",
+    defaultModel: "kimi-k2.5",
+    buildBody: (query, model) => ({
+      model,
+      messages: [{ role: "user", content: query }],
+      tools: [
+        { type: "builtin_function", function: { name: "$web_search" } }
+      ]
+    }),
+    buildHeaders: (token) => ({
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${token}`
+    }),
+    extractAnswer: (data) => {
+      const msg = data?.choices?.[0]?.message || {};
+      const text = msg.content || "";
+      const calls = Array.isArray(msg.tool_calls) ? msg.tool_calls : [];
+      const citations = [];
+      for (const call of calls) {
+        const argStr = call?.function?.arguments;
+        if (!argStr) continue;
+        let parsed;
+        try {
+          parsed = typeof argStr === "string" ? JSON.parse(argStr) : argStr;
+        } catch {
+          continue;
+        }
+        const items =
+          parsed?.search_results ||
+          parsed?.results ||
+          parsed?.references ||
+          [];
+        if (Array.isArray(items)) {
+          for (const it of items) {
+            const url = it?.url || it?.link;
+            if (!url) continue;
+            citations.push({
+              url,
+              title: it.title || "",
+              snippet: it.snippet || it.summary || ""
+            });
+          }
+        }
+      }
+      const tokens = data?.usage?.total_tokens || 0;
+      return { text, citations, tokens };
+    }
+  },
+
+  minimax: {
+    endpoint: () => "https://api.minimaxi.com/v1/text/chatcompletion_v2",
+    defaultModel: "MiniMax-M2.7",
+    buildBody: (query, model) => ({
+      model,
+      messages: [{ role: "user", content: query }],
+      tools: [{ type: "web_search" }]
+    }),
+    buildHeaders: (token) => ({
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${token}`
+    }),
+    extractAnswer: (data) => {
+      const msg = data?.choices?.[0]?.message || {};
+      const text = msg.content || "";
+      const citations = [];
+      const direct = Array.isArray(data?.web_search_results)
+        ? data.web_search_results
+        : [];
+      for (const it of direct) {
+        const url = it?.url || it?.link;
+        if (url) {
+          citations.push({
+            url,
+            title: it.title || "",
+            snippet: it.snippet || it.summary || ""
+          });
+        }
+      }
+      if (!citations.length) {
+        const calls = Array.isArray(msg.tool_calls) ? msg.tool_calls : [];
+        for (const call of calls) {
+          const argStr = call?.function?.arguments;
+          if (!argStr) continue;
+          let parsed;
+          try {
+            parsed = typeof argStr === "string" ? JSON.parse(argStr) : argStr;
+          } catch {
+            continue;
+          }
+          const items = parsed?.results || parsed?.search_results || [];
+          if (Array.isArray(items)) {
+            for (const it of items) {
+              const url = it?.url || it?.link;
+              if (!url) continue;
+              citations.push({
+                url,
+                title: it.title || "",
+                snippet: it.snippet || ""
+              });
+            }
+          }
+        }
+      }
+      const tokens = data?.usage?.total_tokens || 0;
+      return { text, citations, tokens };
+    }
+  },
+
+  perplexity: {
+    endpoint: () => "https://api.perplexity.ai/chat/completions",
+    defaultModel: "sonar",
+    buildBody: (query, model) => ({
+      model,
+      messages: [{ role: "user", content: query }]
+    }),
+    buildHeaders: (token) => ({
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${token}`
+    }),
+    extractAnswer: (data) => {
+      const msg = data?.choices?.[0]?.message || {};
+      const text = msg.content || "";
+      const raw = data?.citations || [];
+      const citations = Array.isArray(raw)
+        ? raw.map(normalizeCitation).filter(Boolean)
+        : [];
+      const tokens = data?.usage?.total_tokens || 0;
+      return { text, citations, tokens };
+    }
+  }
+};
+
+/**
+ * Execute a chat-search request against the chosen provider.
+ * @param {object} params
+ * @param {string} params.provider
+ * @param {string} params.query
+ * @param {number} [params.maxResults]
+ * @param {string} [params.model]
+ * @param {{apiKey?:string, accessToken?:string}} params.credentials
+ * @param {{info?:Function, warn?:Function, error?:Function}} [params.log]
+ * @returns {Promise<{success:boolean, status?:number, error?:string, data?:object}>}
+ */
+export async function handleChatSearch({
+  provider,
+  query,
+  maxResults,
+  model,
+  credentials,
+  log
+}) {
+  const startTime = Date.now();
+  const cfg = CHAT_SEARCH_CONFIG[provider];
+
+  if (!cfg) {
+    return {
+      success: false,
+      status: 400,
+      error: `Unsupported chat-search provider: ${provider}`
+    };
+  }
+
+  if (!query || typeof query !== "string") {
+    return { success: false, status: 400, error: "Missing query" };
+  }
+
+  const token = credentials?.apiKey || credentials?.accessToken;
+  if (!token) {
+    return {
+      success: false,
+      status: 401,
+      error: "Missing credentials (apiKey or accessToken)"
+    };
+  }
+
+  const limit =
+    Number.isFinite(maxResults) && maxResults > 0
+      ? Math.floor(maxResults)
+      : DEFAULT_MAX_RESULTS;
+  const useModel = model || cfg.defaultModel;
+  const url = cfg.endpoint(useModel);
+  const body = cfg.buildBody(query, useModel);
+  const headers = cfg.buildHeaders(token);
+
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS);
+
+  let upstreamStart = Date.now();
+  let resp;
+  try {
+    resp = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(body),
+      signal: controller.signal
+    });
+  } catch (err) {
+    clearTimeout(timer);
+    if (err?.name === "AbortError") {
+      log?.warn?.(`[chatSearch] timeout provider=${provider}`);
+      return { success: false, status: 504, error: "Upstream timeout" };
+    }
+    log?.error?.(`[chatSearch] network error provider=${provider}: ${err?.message}`);
+    return {
+      success: false,
+      status: 502,
+      error: `Network error: ${err?.message || "unknown"}`
+    };
+  }
+  clearTimeout(timer);
+  const upstreamLatency = Date.now() - upstreamStart;
+
+  let data;
+  try {
+    data = await resp.json();
+  } catch {
+    return {
+      success: false,
+      status: 502,
+      error: `Invalid upstream response (status ${resp.status})`
+    };
+  }
+
+  if (!resp.ok) {
+    const errMsg =
+      data?.error?.message ||
+      data?.error ||
+      data?.message ||
+      `Upstream HTTP ${resp.status}`;
+    log?.warn?.(`[chatSearch] upstream error provider=${provider} status=${resp.status}`);
+    return {
+      success: false,
+      status: resp.status,
+      error: typeof errMsg === "string" ? errMsg : JSON.stringify(errMsg)
+    };
+  }
+
+  const { text, citations, tokens } = cfg.extractAnswer(data);
+  const retrievedAt = new Date().toISOString();
+  const limited = (citations || []).slice(0, limit);
+  const results = limited.map((c, i) => toResult(c, i, provider, retrievedAt));
+
+  return {
+    success: true,
+    status: 200,
+    data: {
+      provider,
+      query,
+      results,
+      answer: { source: provider, text: text || "", model: useModel },
+      usage: { queries_used: 1, search_cost_usd: 0, llm_tokens: tokens || 0 },
+      metrics: {
+        response_time_ms: Date.now() - startTime,
+        upstream_latency_ms: upstreamLatency,
+        total_results_available: null
+      },
+      errors: []
+    }
+  };
+}
+
+export { CHAT_SEARCH_CONFIG };

package/open-sse/handlers/search/index.js

@@ -0,0 +1,201 @@
+/**
+ * Search Dispatcher — routes /v1/search requests to dedicated search APIs
+ * or chat-based LLM search wrappers, with retry-friendly error envelope.
+ *
+ * Dependency map:
+ *   provider.searchConfig    → dedicated search API (callers + normalizers)
+ *   provider.searchViaChat   → wrap chat-completions (chatSearch.js)
+ */
+
+import { buildSearchRequest } from "./callers.js";
+import { normalizeSearchResponse } from "./normalizers.js";
+import { handleChatSearch } from "./chatSearch.js";
+
+const GLOBAL_TIMEOUT_MS = 15000;
+const NON_RETRIABLE = new Set([400, 401, 403, 404]);
+
+const CONTROL_CHAR_RE = /[\x00-\x08\x0B\x0C\x0E-\x1F\x7F]/;
+
+/** Normalize and validate query string. */
+function sanitizeQuery(query) {
+  if (CONTROL_CHAR_RE.test(query)) return { error: "Query contains invalid control characters" };
+  const clean = query.normalize("NFKC").trim().replace(/\s+/g, " ");
+  if (!clean) return { error: "Query is empty after normalization" };
+  return { clean };
+}
+
+// Strip non-ASCII chars from header values (HTTP headers must be ByteString).
+function sanitizeHeaders(headers) {
+  if (!headers) return headers;
+  const out = {};
+  for (const [k, v] of Object.entries(headers)) {
+    out[k] = typeof v === "string" ? v.replace(/[^\x00-\xFF]/g, "").trim() : v;
+  }
+  return out;
+}
+
+/** Build a JSON Response wrapper used by the auth layer. */
+function jsonResponse(payload, status = 200) {
+  return new Response(JSON.stringify(payload), {
+    status,
+    headers: { "Content-Type": "application/json", "Access-Control-Allow-Origin": "*" }
+  });
+}
+
+/** Wrap an error result with a Response object so the auth wrapper can return it directly. */
+function errorResult(status, error) {
+  return {
+    success: false,
+    status,
+    error,
+    response: jsonResponse({ error: { message: error, code: status } }, status)
+  };
+}
+
+/** Wrap a success payload. */
+function successResult(data) {
+  return { success: true, data, response: jsonResponse(data, 200) };
+}
+
+/**
+ * Run a single dedicated search provider attempt.
+ * @returns {Promise<{success:boolean, status?:number, error?:string, data?:object}>}
+ */
+async function tryDedicatedProvider({ provider, providerConfig, body, credentials, log, globalStartTime }) {
+  const startTime = Date.now();
+  const token = credentials?.apiKey || credentials?.accessToken || undefined;
+
+  if (providerConfig.authType !== "none" && !token) {
+    return { success: false, status: 401, error: `No credentials for provider: ${provider.id}` };
+  }
+
+  const params = {
+    query: body.query,
+    searchType: body.search_type || (providerConfig.searchTypes?.[0] || "web"),
+    maxResults: Math.min(body.max_results || providerConfig.defaultMaxResults || 5, providerConfig.maxMaxResults || 100),
+    token,
+    country: body.country,
+    language: body.language,
+    timeRange: body.time_range,
+    offset: body.offset,
+    domainFilter: body.domain_filter,
+    contentOptions: body.content_options,
+    providerOptions: body.provider_options,
+    providerSpecificData: credentials?.providerSpecificData
+  };
+
+  let url, init;
+  try {
+    ({ url, init } = buildSearchRequest({ id: provider.id, ...providerConfig }, params));
+  } catch (err) {
+    return { success: false, status: 400, error: err?.message || `Invalid request for ${provider.id}` };
+  }
+
+  // Timeout = min(provider timeout, remaining global)
+  const remaining = GLOBAL_TIMEOUT_MS - (Date.now() - globalStartTime);
+  const timeout = Math.min(providerConfig.timeoutMs || 10000, Math.max(remaining, 1000));
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), timeout);
+
+  log?.info?.("SEARCH", `${provider.id} | "${params.query.slice(0, 80)}" | type=${params.searchType}`);
+
+  try {
+    const resp = await fetch(url, { ...init, headers: sanitizeHeaders(init.headers), signal: controller.signal });
+    clearTimeout(timer);
+    if (!resp.ok) {
+      const errText = await resp.text().catch(() => "");
+      log?.error?.("SEARCH", `${provider.id} ${resp.status}: ${errText.slice(0, 200)}`);
+      return { success: false, status: resp.status, error: `${provider.id} returned ${resp.status}: ${errText.slice(0, 200)}` };
+    }
+    const data = await resp.json();
+    const normalized = normalizeSearchResponse(provider.id, data, params.query, params.searchType);
+    const results = normalized.results.slice(0, params.maxResults);
+    const duration = Date.now() - startTime;
+
+    return {
+      success: true,
+      data: {
+        provider: provider.id,
+        query: params.query,
+        results,
+        answer: null,
+        usage: { queries_used: 1, search_cost_usd: providerConfig.costPerQuery || 0 },
+        metrics: { response_time_ms: duration, upstream_latency_ms: duration, total_results_available: normalized.totalResults },
+        errors: []
+      }
+    };
+  } catch (err) {
+    clearTimeout(timer);
+    const isTimeout = err.name === "AbortError";
+    const status = isTimeout ? 504 : 502;
+    log?.error?.("SEARCH", `${provider.id} ${isTimeout ? "timeout" : "error"}: ${err.message}`);
+    return { success: false, status, error: `${provider.id} ${isTimeout ? "timeout" : "error"}: ${err.message}` };
+  }
+}
+
+/**
+ * Core search handler. Dispatches to dedicated API or chat-based LLM.
+ * Same calling convention as handleEmbeddingsCore: returns `{success, response, status?, error?}`.
+ *
+ * @param {object}   options
+ * @param {object}   options.body            Sanitized body from auth wrapper
+ * @param {object}   options.provider        Provider entry from AI_PROVIDERS
+ * @param {object}   [options.providerConfig] Provider's searchConfig (if dedicated)
+ * @param {object|null} options.credentials  Provider credentials
+ * @param {object}   [options.log]           Logger
+ */
+export async function handleSearchCore({ body, provider, providerConfig, credentials, log }) {
+  const globalStartTime = Date.now();
+
+  // 1. Sanitize query
+  const { clean, error: sanitizeError } = sanitizeQuery(body.query || "");
+  if (sanitizeError) return errorResult(400, sanitizeError);
+  const normalizedBody = { ...body, query: clean };
+
+  // 2. Route: dedicated search API takes priority over chat-based
+  let result;
+  if (providerConfig) {
+    result = await tryDedicatedProvider({
+      provider,
+      providerConfig,
+      body: normalizedBody,
+      credentials,
+      log,
+      globalStartTime
+    });
+  } else if (provider.searchViaChat) {
+    result = await handleChatSearch({
+      provider: provider.id,
+      query: clean,
+      maxResults: normalizedBody.max_results,
+      model: provider.searchViaChat.defaultModel,
+      credentials,
+      log
+    });
+  } else {
+    return errorResult(400, `Provider ${provider.id} does not support web search`);
+  }
+
+  if (result.success) return successResult(result.data);
+
+  // 3. Failover within global timeout for retriable errors
+  if (
+    !NON_RETRIABLE.has(result.status || 0) &&
+    Date.now() - globalStartTime < GLOBAL_TIMEOUT_MS &&
+    provider.searchViaChat &&
+    providerConfig
+  ) {
+    log?.warn?.("SEARCH", `${provider.id} dedicated failed (${result.status}), falling back to chat-based search`);
+    const fallback = await handleChatSearch({
+      provider: provider.id,
+      query: clean,
+      maxResults: normalizedBody.max_results,
+      model: provider.searchViaChat.defaultModel,
+      credentials,
+      log
+    });
+    if (fallback.success) return successResult(fallback.data);
+  }
+
+  return errorResult(result.status || 502, result.error || "Search failed");
+}