@dcrays/web-search 0.1.0 → 0.1.2

package/README.md

@@ -1,5 +1,5 @@
 # @dcrays/web-search
 
-书灵墨宝的 DSH 联网搜索插件。加载后注册 `web_search` 工具，把模型搜索请求映射到配置的 HTTP API。
+书灵墨宝的 DSH 联网搜索插件。加载后注册 `web_search` 工具，把模型搜索请求映射到搜索网关。
 
-本包对应 `mbh-chat` 环境，端点地址从 Cordis 插件配置读取。
+单一发布包同时适用于生产与测试桌面：运行时根据 `mobook.json` 中记录的环境选择搜索网关与 MBHChat 凭据通道；`endpointUrl` 可在 Cordis 插件配置中覆盖。

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@dcrays/web-search",
-  "version": "0.1.0",
-  "description": "DeepSeek Harness web search plugin for Mobook (production)",
+  "version": "0.1.2",
+  "description": "DeepSeek Harness web search plugin for Mobook",
   "type": "module",
   "main": "plugin/index.js",
   "types": "plugin/index.d.ts",
@@ -28,12 +28,12 @@
   "engines": {
     "node": ">=24.0.0"
   },
-  "dependencies": {
-    "@deepseek-ai/schemastery": "^3.18.1"
-  },
+  "dependencies": {},
   "peerDependencies": {
-    "@deepseek-ai/cordis": "^4.0.1",
-    "@deepseek-ai/dsh-tools": "^0.1.1-rc.2"
+    "@deepseek-ai/cordis": "^4.0.2",
+    "@deepseek-ai/dsh-home-paths": "0.1.2-rc.1",
+    "@deepseek-ai/dsh-tools": "^0.1.2-rc.1",
+    "@deepseek-ai/schemastery": "^3.18.1"
   },
   "publishConfig": {
     "access": "public"

package/plugin/index.d.ts

@@ -2,7 +2,6 @@ import type { Context } from '@deepseek-ai/cordis';
 import { Service } from '@deepseek-ai/cordis';
 import z from '@deepseek-ai/schemastery';
 import { type WebSearchRequest, type WebSearchResult } from '../src/index.js';
-export { DEFAULT_BUILD_ENV } from '../src/config/env-config.js';
 export declare const name = "mbh-web-search";
 export declare const inject: string[];
 export interface Config {
@@ -27,4 +26,5 @@ export declare class MbhWebSearchService extends Service {
     isEnabled(): boolean;
     search(request: WebSearchRequest, signal?: AbortSignal): Promise<WebSearchResult>;
 }
+export declare function resolveMetadataPath(configuredPath: string): string;
 export declare function apply(ctx: Context, config: Config): void;

package/plugin/index.js

@@ -1,14 +1,33 @@
 import { createRequire as __mobookCreateRequire } from 'node:module'; const require = __mobookCreateRequire(import.meta.url);
 
-// plugins/web-search/dist-npm/plugin/index.js
-import { homedir } from "node:os";
+// dist-npm/plugin/index.js
 import { resolve } from "node:path";
 import { Service } from "@deepseek-ai/cordis";
+import { resolveDshHome } from "@deepseek-ai/dsh-home-paths";
 import z from "@deepseek-ai/schemastery";
 
-// plugins/web-search/dist-npm/src/api-client.js
+// dist-npm/src/api-client.js
 import { createHmac, randomUUID } from "node:crypto";
 import { readFile } from "node:fs/promises";
+
+// dist-npm/src/config/env-config.js
+var ENV_CONFIG = {
+  production: {
+    channelId: "mbh-chat",
+    legacyChannelId: "mbhchat",
+    endpointUrl: "https://api-gateway.shuwenda.com/apikey-manage/api/search"
+  },
+  test: {
+    channelId: "mbh-chat-test",
+    legacyChannelId: "mbhchat-test",
+    endpointUrl: "https://api-gateway.shuwenda.icu/apikey-manage/api/search"
+  }
+};
+function resolveWebSearchEnvironment(value) {
+  return value === "production" ? "production" : "test";
+}
+
+// dist-npm/src/api-client.js
 var WebSearchApiError = class extends Error {
   code;
   constructor(code, message, options) {
@@ -17,6 +36,10 @@ var WebSearchApiError = class extends Error {
     this.name = "WebSearchApiError";
   }
 };
+var RESPONSE_TOO_LARGE_MESSAGE = "\u641C\u7D22\u670D\u52A1\u8FD4\u56DE\u7684\u5185\u5BB9\u8FC7\u5927\uFF0C\u6682\u65F6\u65E0\u6CD5\u5904\u7406\u3002\u8BF7\u7F29\u5C0F\u641C\u7D22\u8303\u56F4\u540E\u91CD\u8BD5\u3002";
+function searchServiceLabel(service) {
+  return service === "tavily_search" ? "Tavily \u641C\u7D22" : "\u535A\u67E5\u641C\u7D22";
+}
 function record(value) {
   return value && typeof value === "object" && !Array.isArray(value) ? value : void 0;
 }
@@ -35,7 +58,7 @@ function resolveSearchCredentials(metadata, channelId, legacyChannelId) {
     if (userId && botToken && deviceId)
       return { userId, botToken, deviceId };
   }
-  throw new WebSearchApiError("MISSING_CREDENTIALS", `mobook metadata must contain plugins.${channelId}.userId, botToken, and hardware.fingerprint`);
+  throw new WebSearchApiError("MISSING_CREDENTIALS", "\u641C\u7D22\u670D\u52A1\u5C1A\u672A\u51C6\u5907\u597D\uFF0C\u8BF7\u786E\u8BA4\u5DF2\u767B\u5F55\u58A8\u5B9D\u5E76\u5B8C\u6210\u5BA2\u6237\u7AEF\u521D\u59CB\u5316\u540E\u91CD\u8BD5\u3002");
 }
 function requestSignal(signal, timeoutMs) {
   const timeout = AbortSignal.timeout(timeoutMs);
@@ -45,7 +68,7 @@ async function responseText(response, maxBytes) {
   const declared = Number(response.headers.get("content-length"));
   if (Number.isFinite(declared) && declared > maxBytes) {
     await response.body?.cancel().catch(() => void 0);
-    throw new WebSearchApiError("RESPONSE_TOO_LARGE", `web search response exceeds ${maxBytes} bytes`);
+    throw new WebSearchApiError("RESPONSE_TOO_LARGE", RESPONSE_TOO_LARGE_MESSAGE);
   }
   if (!response.body)
     return "";
@@ -61,7 +84,7 @@ async function responseText(response, maxBytes) {
       size += item.value.byteLength;
       if (size > maxBytes) {
         await reader.cancel().catch(() => void 0);
-        throw new WebSearchApiError("RESPONSE_TOO_LARGE", `web search response exceeds ${maxBytes} bytes`);
+        throw new WebSearchApiError("RESPONSE_TOO_LARGE", RESPONSE_TOO_LARGE_MESSAGE);
       }
       text += decoder.decode(item.value, { stream: true });
     }
@@ -70,66 +93,92 @@ async function responseText(response, maxBytes) {
     reader.releaseLock();
   }
 }
-function parseResponse(text, fallbackQuery) {
+function searchItem(value, snippetKeys, publishedAtKeys) {
+  const item = record(value);
+  const url = nonEmptyString(item?.url);
+  if (!item || !url)
+    return [];
+  const result = {
+    title: nonEmptyString(item.title) ?? nonEmptyString(item.name) ?? url,
+    url,
+    snippet: snippetKeys.map((key) => nonEmptyString(item[key])).find(Boolean) ?? ""
+  };
+  const siteName = nonEmptyString(item.siteName);
+  const publishedAt = publishedAtKeys.map((key) => nonEmptyString(item[key])).find(Boolean);
+  if (siteName)
+    result.siteName = siteName;
+  if (publishedAt)
+    result.publishedAt = publishedAt;
+  return [result];
+}
+function parseResponse(text, fallbackQuery, service) {
   let payload;
   try {
     payload = JSON.parse(text);
   } catch (error) {
-    throw new WebSearchApiError("INVALID_RESPONSE", "web search API returned invalid JSON", { cause: error });
+    throw new WebSearchApiError("INVALID_RESPONSE", "\u641C\u7D22\u670D\u52A1\u8FD4\u56DE\u7684\u6570\u636E\u683C\u5F0F\u5F02\u5E38\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5\u3002", { cause: error });
   }
   const wrapper = record(payload);
   const upstream = record(wrapper?.data);
   if (wrapper?.success !== true || typeof upstream?.code === "number" && upstream.code !== 200) {
-    throw new WebSearchApiError("API_ERROR", "web search API reported an unsuccessful response");
+    throw new WebSearchApiError("API_ERROR", "\u641C\u7D22\u670D\u52A1\u6682\u65F6\u4E0D\u53EF\u7528\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5\u3002");
+  }
+  if (service === "tavily_search") {
+    const values = Array.isArray(upstream?.results) ? upstream.results : [];
+    return {
+      query: nonEmptyString(upstream?.query) ?? fallbackQuery,
+      provider: service,
+      results: values.flatMap((value) => searchItem(value, ["content", "snippet"], ["published_date", "publishedAt"]))
+    };
   }
   const searchData = record(upstream?.data);
   const pages = record(searchData?.webPages);
   const queryContext = record(searchData?.queryContext);
-  const values = Array.isArray(pages?.value) ? pages.value : [];
-  const results = values.flatMap((value) => {
-    const item = record(value);
-    const url = nonEmptyString(item?.url);
-    if (!item || !url)
-      return [];
-    const result2 = {
-      title: nonEmptyString(item.name) ?? url,
-      url,
-      snippet: nonEmptyString(item.snippet) ?? ""
-    };
-    const siteName = nonEmptyString(item.siteName);
-    const publishedAt = nonEmptyString(item.datePublished);
-    if (siteName)
-      result2.siteName = siteName;
-    if (publishedAt)
-      result2.publishedAt = publishedAt;
-    return [result2];
-  });
   const result = {
     query: nonEmptyString(queryContext?.originalQuery) ?? fallbackQuery,
-    results
+    provider: service,
+    results: (Array.isArray(pages?.value) ? pages.value : []).flatMap((value) => searchItem(value, ["snippet"], ["datePublished"]))
   };
   if (typeof pages?.totalEstimatedMatches === "number" && Number.isFinite(pages.totalEstimatedMatches)) {
     result.totalEstimatedMatches = pages.totalEstimatedMatches;
   }
   return result;
 }
+function toEndpoint(endpointUrl) {
+  let endpoint;
+  try {
+    endpoint = new URL(endpointUrl);
+  } catch (error) {
+    throw new WebSearchApiError("INVALID_CONFIG", "\u641C\u7D22\u670D\u52A1\u5730\u5740\u914D\u7F6E\u4E0D\u6B63\u786E\uFF0C\u8BF7\u8054\u7CFB\u7BA1\u7406\u5458\u68C0\u67E5\u63D2\u4EF6\u914D\u7F6E\u3002", { cause: error });
+  }
+  if (!["http:", "https:"].includes(endpoint.protocol)) {
+    throw new WebSearchApiError("INVALID_CONFIG", "\u641C\u7D22\u670D\u52A1\u5730\u5740\u4EC5\u652F\u6301 HTTP \u6216 HTTPS\uFF0C\u8BF7\u8054\u7CFB\u7BA1\u7406\u5458\u68C0\u67E5\u63D2\u4EF6\u914D\u7F6E\u3002");
+  }
+  return endpoint;
+}
 var WebSearchApiClient = class {
   config;
-  endpoint;
+  environments;
   fetchImpl;
   readFileImpl;
   nowSeconds;
   nonce;
   constructor(config, dependencies = {}) {
     this.config = config;
-    try {
-      this.endpoint = new URL(config.endpointUrl);
-    } catch (error) {
-      throw new WebSearchApiError("INVALID_CONFIG", "web search endpointUrl is invalid", { cause: error });
-    }
-    if (!["http:", "https:"].includes(this.endpoint.protocol)) {
-      throw new WebSearchApiError("INVALID_CONFIG", "web search endpointUrl must use http or https");
-    }
+    const environments = config.environments ?? ENV_CONFIG;
+    const override = config.endpointOverride?.trim();
+    this.environments = {
+      production: {
+        endpoint: toEndpoint(override || environments.production.endpointUrl),
+        channelId: environments.production.channelId,
+        legacyChannelId: environments.production.legacyChannelId
+      },
+      test: {
+        endpoint: toEndpoint(override || environments.test.endpointUrl),
+        channelId: environments.test.channelId,
+        legacyChannelId: environments.test.legacyChannelId
+      }
+    };
     for (const [name2, value] of [
       ["timeoutMs", config.timeoutMs],
       ["maxResponseBytes", config.maxResponseBytes],
@@ -137,11 +186,11 @@ var WebSearchApiClient = class {
       ["maxCount", config.maxCount]
     ]) {
       if (!Number.isSafeInteger(value) || value <= 0) {
-        throw new WebSearchApiError("INVALID_CONFIG", `${name2} must be a positive integer`);
+        throw new WebSearchApiError("INVALID_CONFIG", `\u641C\u7D22\u63D2\u4EF6\u914D\u7F6E\u9879 ${name2} \u5FC5\u987B\u662F\u5927\u4E8E 0 \u7684\u6574\u6570\uFF0C\u8BF7\u8054\u7CFB\u7BA1\u7406\u5458\u68C0\u67E5\u914D\u7F6E\u3002`);
       }
     }
     if (config.defaultCount > config.maxCount) {
-      throw new WebSearchApiError("INVALID_CONFIG", "defaultCount must not exceed maxCount");
+      throw new WebSearchApiError("INVALID_CONFIG", "\u9ED8\u8BA4\u641C\u7D22\u7ED3\u679C\u6570\u4E0D\u80FD\u5927\u4E8E\u6700\u5927\u7ED3\u679C\u6570\uFF0C\u8BF7\u8054\u7CFB\u7BA1\u7406\u5458\u68C0\u67E5\u63D2\u4EF6\u914D\u7F6E\u3002");
     }
     this.fetchImpl = dependencies.fetch ?? fetch;
     this.readFileImpl = dependencies.readFile ?? readFile;
@@ -151,73 +200,87 @@ var WebSearchApiClient = class {
   async search(request, signal) {
     const query = request.query.trim();
     if (!query)
-      throw new WebSearchApiError("INVALID_REQUEST", "web search query must not be empty");
+      throw new WebSearchApiError("INVALID_REQUEST", "\u8BF7\u8F93\u5165\u8981\u641C\u7D22\u7684\u5185\u5BB9\u540E\u91CD\u8BD5\u3002");
     const count = request.count ?? this.config.defaultCount;
     const offset = request.offset ?? 0;
     if (!Number.isSafeInteger(count) || count < 1 || count > this.config.maxCount) {
-      throw new WebSearchApiError("INVALID_REQUEST", `web search count must be between 1 and ${this.config.maxCount}`);
+      throw new WebSearchApiError("INVALID_REQUEST", `\u641C\u7D22\u7ED3\u679C\u6570\u91CF\u9700\u8BBE\u7F6E\u4E3A 1 \u5230 ${this.config.maxCount} \u4E4B\u95F4\u7684\u6574\u6570\u3002`);
     }
     if (!Number.isSafeInteger(offset) || offset < 0) {
-      throw new WebSearchApiError("INVALID_REQUEST", "web search offset must be a non-negative integer");
+      throw new WebSearchApiError("INVALID_REQUEST", "\u641C\u7D22\u7ED3\u679C\u504F\u79FB\u91CF\u5FC5\u987B\u662F\u5927\u4E8E\u6216\u7B49\u4E8E 0 \u7684\u6574\u6570\u3002");
     }
     let metadata;
     try {
       metadata = JSON.parse(await this.readFileImpl(this.config.metadataPath, "utf8"));
     } catch (error) {
-      throw new WebSearchApiError("MISSING_CREDENTIALS", "unable to read Mobook runtime metadata", { cause: error });
+      throw new WebSearchApiError("MISSING_CREDENTIALS", "\u6682\u65F6\u65E0\u6CD5\u8BFB\u53D6\u58A8\u5B9D\u8FD0\u884C\u914D\u7F6E\uFF0C\u8BF7\u786E\u8BA4\u5BA2\u6237\u7AEF\u5DF2\u6B63\u5E38\u521D\u59CB\u5316\u540E\u91CD\u8BD5\u3002", { cause: error });
     }
-    const credentials = resolveSearchCredentials(metadata, this.config.channelId, this.config.legacyChannelId);
-    const timestamp = String(this.nowSeconds());
-    const nonce = this.nonce();
-    const message = `${credentials.userId}
+    const environment = resolveWebSearchEnvironment(record(metadata)?.environment);
+    const resolved = this.environments[environment];
+    const credentials = resolveSearchCredentials(metadata, resolved.channelId, resolved.legacyChannelId);
+    const call = async (service) => {
+      const timestamp = String(this.nowSeconds());
+      const nonce = this.nonce();
+      const message = `${credentials.userId}
 ${credentials.deviceId}
 ${timestamp}
 ${nonce}`;
-    const signature = createHmac("sha256", credentials.botToken).update(message, "utf8").digest("hex");
-    const body = {
-      user_id: credentials.userId,
-      bot_token: credentials.botToken,
-      device_id: credentials.deviceId,
-      signature,
-      timestamp,
-      nonce,
-      service: "bocha_search",
-      query,
-      count,
-      offset,
-      language: request.language?.trim() || "zh-CN"
+      const signature = createHmac("sha256", credentials.botToken).update(message, "utf8").digest("hex");
+      const body = {
+        user_id: credentials.userId,
+        bot_token: credentials.botToken,
+        device_id: credentials.deviceId,
+        signature,
+        timestamp,
+        nonce,
+        service,
+        query,
+        count,
+        offset,
+        language: request.language?.trim() || "zh-CN"
+      };
+      if (request.includeDomains?.length)
+        body.include_domains = request.includeDomains;
+      if (request.excludeDomains?.length)
+        body.exclude_domains = request.excludeDomains;
+      const effectiveSignal = requestSignal(signal, this.config.timeoutMs);
+      let response;
+      try {
+        response = await this.fetchImpl(resolved.endpoint, {
+          method: "POST",
+          headers: { accept: "application/json", "content-type": "application/json" },
+          body: JSON.stringify(body),
+          signal: effectiveSignal
+        });
+      } catch (error) {
+        if (signal?.aborted)
+          throw error;
+        if (effectiveSignal.aborted) {
+          throw new WebSearchApiError("TIMEOUT", `${searchServiceLabel(service)}\u54CD\u5E94\u8D85\u65F6\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5\u3002`, {
+            cause: error
+          });
+        }
+        throw new WebSearchApiError("NETWORK_ERROR", `\u6682\u65F6\u65E0\u6CD5\u8FDE\u63A5${searchServiceLabel(service)}\uFF0C\u8BF7\u68C0\u67E5\u7F51\u7EDC\u540E\u91CD\u8BD5\u3002`, { cause: error });
+      }
+      const text = await responseText(response, this.config.maxResponseBytes);
+      if (!response.ok) {
+        throw new WebSearchApiError("HTTP_ERROR", `${searchServiceLabel(service)}\u6682\u65F6\u4E0D\u53EF\u7528\uFF08\u670D\u52A1\u72B6\u6001 ${response.status}\uFF09\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5\u3002`);
+      }
+      return parseResponse(text, query, service);
     };
-    if (request.includeDomains?.length)
-      body.include_domains = request.includeDomains;
-    if (request.excludeDomains?.length)
-      body.exclude_domains = request.excludeDomains;
-    const effectiveSignal = requestSignal(signal, this.config.timeoutMs);
-    let response;
     try {
-      response = await this.fetchImpl(this.endpoint, {
-        method: "POST",
-        headers: { accept: "application/json", "content-type": "application/json" },
-        body: JSON.stringify(body),
-        signal: effectiveSignal
-      });
+      const tavily = await call("tavily_search");
+      if (tavily.results.length > 0)
+        return tavily;
     } catch (error) {
       if (signal?.aborted)
         throw error;
-      if (effectiveSignal.aborted) {
-        throw new WebSearchApiError("TIMEOUT", `web search request timed out after ${this.config.timeoutMs}ms`, {
-          cause: error
-        });
-      }
-      throw new WebSearchApiError("NETWORK_ERROR", "web search request failed", { cause: error });
     }
-    const text = await responseText(response, this.config.maxResponseBytes);
-    if (!response.ok)
-      throw new WebSearchApiError("HTTP_ERROR", `web search request failed with HTTP ${response.status}`);
-    return parseResponse(text, query);
+    return call("bocha_search");
   }
 };
 
-// plugins/web-search/dist-npm/src/tools.js
+// dist-npm/src/tools.js
 import { defineTool } from "@deepseek-ai/dsh-tools";
 function createWebSearchTool(search) {
   return defineTool({
@@ -250,28 +313,10 @@ function createWebSearchTool(search) {
   });
 }
 
-// plugins/web-search/dist-npm/src/index.js
+// dist-npm/src/index.js
 var MBH_WEB_SEARCH_SERVICE_KEY = "mbhWebSearch";
 
-// plugins/web-search/dist-npm/src/config/env-config.js
-var ENV_CONFIG = {
-  production: {
-    packageName: "@dcrays/web-search",
-    channelId: "mbh-chat",
-    legacyChannelId: "mbhchat",
-    endpointUrl: "https://api-gateway.shuwenda.com/apikey-manage/api/search"
-  },
-  test: {
-    packageName: "@dcrays/web-search-test",
-    channelId: "mbh-chat-test",
-    legacyChannelId: "mbhchat-test",
-    endpointUrl: "https://api-gateway.shuwenda.icu/apikey-manage/api/search"
-  }
-};
-var DEFAULT_BUILD_ENV = "production";
-var ACTIVE_ENV_CONFIG = ENV_CONFIG[DEFAULT_BUILD_ENV];
-
-// plugins/web-search/dist-npm/plugin/index.js
+// dist-npm/plugin/index.js
 var name = "@dcrays/web-search";
 var inject = ["tools"];
 var Config = z.object({
@@ -291,12 +336,9 @@ var MbhWebSearchService = class extends Service {
     this.config = config;
     if (!config.enabled)
       return;
-    const dshHome = process.env.DSH_HOME?.trim() || resolve(homedir(), ".mobook-harness");
     this.operations = new WebSearchApiClient({
-      endpointUrl: config.endpointUrl.trim() || ACTIVE_ENV_CONFIG.endpointUrl,
-      metadataPath: config.metadataPath.trim() || resolve(dshHome, "mobook.json"),
-      channelId: ACTIVE_ENV_CONFIG.channelId,
-      legacyChannelId: ACTIVE_ENV_CONFIG.legacyChannelId,
+      metadataPath: resolveMetadataPath(config.metadataPath),
+      ...config.endpointUrl.trim() ? { endpointOverride: config.endpointUrl.trim() } : {},
       timeoutMs: config.timeoutMs,
       maxResponseBytes: config.maxResponseBytes,
       defaultCount: config.defaultCount,
@@ -307,26 +349,30 @@ var MbhWebSearchService = class extends Service {
     return this.config.enabled;
   }
   search(request, signal) {
-    if (!this.operations)
-      return Promise.reject(new Error("mbh-web-search is disabled"));
+    if (!this.operations) {
+      return Promise.reject(new Error("\u8054\u7F51\u641C\u7D22\u529F\u80FD\u5F53\u524D\u672A\u542F\u7528\uFF0C\u8BF7\u8054\u7CFB\u7BA1\u7406\u5458\u68C0\u67E5\u63D2\u4EF6\u914D\u7F6E\u3002"));
+    }
     return this.operations.search(request, signal);
   }
 };
+function resolveMetadataPath(configuredPath) {
+  return configuredPath.trim() || resolve(resolveDshHome(), "mobook.json");
+}
 function apply(ctx, config) {
   const service = new MbhWebSearchService(ctx, config);
   ctx.effect(() => {
     if (!service.isEnabled())
       return () => void 0;
     const unregister = ctx.tools.register(createWebSearchTool(service));
-    ctx.logger(name).info("%s plugin loaded with web_search tool for %s", name, DEFAULT_BUILD_ENV);
+    ctx.logger(name).info("%s plugin loaded with web_search tool", name);
     return unregister;
   }, `${name} lifecycle`);
 }
 export {
   Config,
-  DEFAULT_BUILD_ENV,
   MbhWebSearchService,
   apply,
   inject,
-  name
+  name,
+  resolveMetadataPath
 };

package/src/api-client.d.ts

@@ -1,3 +1,4 @@
+import { type EnvConfig, type WebSearchEnvironment } from './config/env-config.js';
 export interface WebSearchRequest {
     query: string;
     count?: number;
@@ -15,17 +16,18 @@ export interface WebSearchItem {
 }
 export interface WebSearchResult {
     query: string;
+    provider: SearchService;
     totalEstimatedMatches?: number;
     results: WebSearchItem[];
 }
+export type SearchService = 'tavily_search' | 'bocha_search';
 export interface WebSearchOperations {
     search(request: WebSearchRequest, signal?: AbortSignal): Promise<WebSearchResult>;
 }
 export interface WebSearchApiClientConfig {
-    endpointUrl: string;
     metadataPath: string;
-    channelId: string;
-    legacyChannelId: string;
+    endpointOverride?: string;
+    environments?: Record<WebSearchEnvironment, EnvConfig>;
     timeoutMs: number;
     maxResponseBytes: number;
     defaultCount: number;
@@ -51,7 +53,7 @@ interface SearchCredentials {
 export declare function resolveSearchCredentials(metadata: unknown, channelId: string, legacyChannelId: string): SearchCredentials;
 export declare class WebSearchApiClient implements WebSearchOperations {
     private readonly config;
-    private readonly endpoint;
+    private readonly environments;
     private readonly fetchImpl;
     private readonly readFileImpl;
     private readonly nowSeconds;

package/src/api-client.js

@@ -1,5 +1,6 @@
 import { createHmac, randomUUID } from 'node:crypto';
 import { readFile } from 'node:fs/promises';
+import { ENV_CONFIG, resolveWebSearchEnvironment } from './config/env-config.js';
 export class WebSearchApiError extends Error {
     code;
     constructor(code, message, options) {
@@ -8,6 +9,10 @@ export class WebSearchApiError extends Error {
         this.name = 'WebSearchApiError';
     }
 }
+const RESPONSE_TOO_LARGE_MESSAGE = '搜索服务返回的内容过大，暂时无法处理。请缩小搜索范围后重试。';
+function searchServiceLabel(service) {
+    return service === 'tavily_search' ? 'Tavily 搜索' : '博查搜索';
+}
 function record(value) {
     return value && typeof value === 'object' && !Array.isArray(value) ? value : undefined;
 }
@@ -26,7 +31,7 @@ export function resolveSearchCredentials(metadata, channelId, legacyChannelId) {
         if (userId && botToken && deviceId)
             return { userId, botToken, deviceId };
     }
-    throw new WebSearchApiError('MISSING_CREDENTIALS', `mobook metadata must contain plugins.${channelId}.userId, botToken, and hardware.fingerprint`);
+    throw new WebSearchApiError('MISSING_CREDENTIALS', '搜索服务尚未准备好，请确认已登录墨宝并完成客户端初始化后重试。');
 }
 function requestSignal(signal, timeoutMs) {
     const timeout = AbortSignal.timeout(timeoutMs);
@@ -36,7 +41,7 @@ async function responseText(response, maxBytes) {
     const declared = Number(response.headers.get('content-length'));
     if (Number.isFinite(declared) && declared > maxBytes) {
         await response.body?.cancel().catch(() => undefined);
-        throw new WebSearchApiError('RESPONSE_TOO_LARGE', `web search response exceeds ${maxBytes} bytes`);
+        throw new WebSearchApiError('RESPONSE_TOO_LARGE', RESPONSE_TOO_LARGE_MESSAGE);
     }
     if (!response.body)
         return '';
@@ -52,7 +57,7 @@ async function responseText(response, maxBytes) {
             size += item.value.byteLength;
             if (size > maxBytes) {
                 await reader.cancel().catch(() => undefined);
-                throw new WebSearchApiError('RESPONSE_TOO_LARGE', `web search response exceeds ${maxBytes} bytes`);
+                throw new WebSearchApiError('RESPONSE_TOO_LARGE', RESPONSE_TOO_LARGE_MESSAGE);
             }
             text += decoder.decode(item.value, { stream: true });
         }
@@ -62,68 +67,94 @@ async function responseText(response, maxBytes) {
         reader.releaseLock();
     }
 }
-function parseResponse(text, fallbackQuery) {
+function searchItem(value, snippetKeys, publishedAtKeys) {
+    const item = record(value);
+    const url = nonEmptyString(item?.url);
+    if (!item || !url)
+        return [];
+    const result = {
+        title: nonEmptyString(item.title) ?? nonEmptyString(item.name) ?? url,
+        url,
+        snippet: snippetKeys.map((key) => nonEmptyString(item[key])).find(Boolean) ?? ''
+    };
+    const siteName = nonEmptyString(item.siteName);
+    const publishedAt = publishedAtKeys.map((key) => nonEmptyString(item[key])).find(Boolean);
+    if (siteName)
+        result.siteName = siteName;
+    if (publishedAt)
+        result.publishedAt = publishedAt;
+    return [result];
+}
+function parseResponse(text, fallbackQuery, service) {
     let payload;
     try {
         payload = JSON.parse(text);
     }
     catch (error) {
-        throw new WebSearchApiError('INVALID_RESPONSE', 'web search API returned invalid JSON', { cause: error });
+        throw new WebSearchApiError('INVALID_RESPONSE', '搜索服务返回的数据格式异常，请稍后重试。', { cause: error });
     }
     const wrapper = record(payload);
     const upstream = record(wrapper?.data);
     if (wrapper?.success !== true || (typeof upstream?.code === 'number' && upstream.code !== 200)) {
-        throw new WebSearchApiError('API_ERROR', 'web search API reported an unsuccessful response');
+        throw new WebSearchApiError('API_ERROR', '搜索服务暂时不可用，请稍后重试。');
+    }
+    if (service === 'tavily_search') {
+        const values = Array.isArray(upstream?.results) ? upstream.results : [];
+        return {
+            query: nonEmptyString(upstream?.query) ?? fallbackQuery,
+            provider: service,
+            results: values.flatMap((value) => searchItem(value, ['content', 'snippet'], ['published_date', 'publishedAt']))
+        };
     }
     const searchData = record(upstream?.data);
     const pages = record(searchData?.webPages);
     const queryContext = record(searchData?.queryContext);
-    const values = Array.isArray(pages?.value) ? pages.value : [];
-    const results = values.flatMap((value) => {
-        const item = record(value);
-        const url = nonEmptyString(item?.url);
-        if (!item || !url)
-            return [];
-        const result = {
-            title: nonEmptyString(item.name) ?? url,
-            url,
-            snippet: nonEmptyString(item.snippet) ?? ''
-        };
-        const siteName = nonEmptyString(item.siteName);
-        const publishedAt = nonEmptyString(item.datePublished);
-        if (siteName)
-            result.siteName = siteName;
-        if (publishedAt)
-            result.publishedAt = publishedAt;
-        return [result];
-    });
     const result = {
         query: nonEmptyString(queryContext?.originalQuery) ?? fallbackQuery,
-        results
+        provider: service,
+        results: (Array.isArray(pages?.value) ? pages.value : []).flatMap((value) => searchItem(value, ['snippet'], ['datePublished']))
     };
     if (typeof pages?.totalEstimatedMatches === 'number' && Number.isFinite(pages.totalEstimatedMatches)) {
         result.totalEstimatedMatches = pages.totalEstimatedMatches;
     }
     return result;
 }
+function toEndpoint(endpointUrl) {
+    let endpoint;
+    try {
+        endpoint = new URL(endpointUrl);
+    }
+    catch (error) {
+        throw new WebSearchApiError('INVALID_CONFIG', '搜索服务地址配置不正确，请联系管理员检查插件配置。', { cause: error });
+    }
+    if (!['http:', 'https:'].includes(endpoint.protocol)) {
+        throw new WebSearchApiError('INVALID_CONFIG', '搜索服务地址仅支持 HTTP 或 HTTPS，请联系管理员检查插件配置。');
+    }
+    return endpoint;
+}
 export class WebSearchApiClient {
     config;
-    endpoint;
+    environments;
     fetchImpl;
     readFileImpl;
     nowSeconds;
     nonce;
     constructor(config, dependencies = {}) {
         this.config = config;
-        try {
-            this.endpoint = new URL(config.endpointUrl);
-        }
-        catch (error) {
-            throw new WebSearchApiError('INVALID_CONFIG', 'web search endpointUrl is invalid', { cause: error });
-        }
-        if (!['http:', 'https:'].includes(this.endpoint.protocol)) {
-            throw new WebSearchApiError('INVALID_CONFIG', 'web search endpointUrl must use http or https');
-        }
+        const environments = config.environments ?? ENV_CONFIG;
+        const override = config.endpointOverride?.trim();
+        this.environments = {
+            production: {
+                endpoint: toEndpoint(override || environments.production.endpointUrl),
+                channelId: environments.production.channelId,
+                legacyChannelId: environments.production.legacyChannelId
+            },
+            test: {
+                endpoint: toEndpoint(override || environments.test.endpointUrl),
+                channelId: environments.test.channelId,
+                legacyChannelId: environments.test.legacyChannelId
+            }
+        };
         for (const [name, value] of [
             ['timeoutMs', config.timeoutMs],
             ['maxResponseBytes', config.maxResponseBytes],
@@ -131,11 +162,11 @@ export class WebSearchApiClient {
             ['maxCount', config.maxCount]
         ]) {
             if (!Number.isSafeInteger(value) || value <= 0) {
-                throw new WebSearchApiError('INVALID_CONFIG', `${name} must be a positive integer`);
+                throw new WebSearchApiError('INVALID_CONFIG', `搜索插件配置项 ${name} 必须是大于 0 的整数，请联系管理员检查配置。`);
             }
         }
         if (config.defaultCount > config.maxCount) {
-            throw new WebSearchApiError('INVALID_CONFIG', 'defaultCount must not exceed maxCount');
+            throw new WebSearchApiError('INVALID_CONFIG', '默认搜索结果数不能大于最大结果数，请联系管理员检查插件配置。');
         }
         this.fetchImpl = dependencies.fetch ?? fetch;
         this.readFileImpl = dependencies.readFile ?? readFile;
@@ -145,67 +176,82 @@ export class WebSearchApiClient {
     async search(request, signal) {
         const query = request.query.trim();
         if (!query)
-            throw new WebSearchApiError('INVALID_REQUEST', 'web search query must not be empty');
+            throw new WebSearchApiError('INVALID_REQUEST', '请输入要搜索的内容后重试。');
         const count = request.count ?? this.config.defaultCount;
         const offset = request.offset ?? 0;
         if (!Number.isSafeInteger(count) || count < 1 || count > this.config.maxCount) {
-            throw new WebSearchApiError('INVALID_REQUEST', `web search count must be between 1 and ${this.config.maxCount}`);
+            throw new WebSearchApiError('INVALID_REQUEST', `搜索结果数量需设置为 1 到 ${this.config.maxCount} 之间的整数。`);
         }
         if (!Number.isSafeInteger(offset) || offset < 0) {
-            throw new WebSearchApiError('INVALID_REQUEST', 'web search offset must be a non-negative integer');
+            throw new WebSearchApiError('INVALID_REQUEST', '搜索结果偏移量必须是大于或等于 0 的整数。');
         }
         let metadata;
         try {
             metadata = JSON.parse(await this.readFileImpl(this.config.metadataPath, 'utf8'));
         }
         catch (error) {
-            throw new WebSearchApiError('MISSING_CREDENTIALS', 'unable to read Mobook runtime metadata', { cause: error });
+            throw new WebSearchApiError('MISSING_CREDENTIALS', '暂时无法读取墨宝运行配置，请确认客户端已正常初始化后重试。', { cause: error });
         }
-        const credentials = resolveSearchCredentials(metadata, this.config.channelId, this.config.legacyChannelId);
-        const timestamp = String(this.nowSeconds());
-        const nonce = this.nonce();
-        const message = `${credentials.userId}\n${credentials.deviceId}\n${timestamp}\n${nonce}`;
-        const signature = createHmac('sha256', credentials.botToken).update(message, 'utf8').digest('hex');
-        const body = {
-            user_id: credentials.userId,
-            bot_token: credentials.botToken,
-            device_id: credentials.deviceId,
-            signature,
-            timestamp,
-            nonce,
-            service: 'bocha_search',
-            query,
-            count,
-            offset,
-            language: request.language?.trim() || 'zh-CN'
+        const environment = resolveWebSearchEnvironment(record(metadata)?.environment);
+        const resolved = this.environments[environment];
+        const credentials = resolveSearchCredentials(metadata, resolved.channelId, resolved.legacyChannelId);
+        const call = async (service) => {
+            const timestamp = String(this.nowSeconds());
+            const nonce = this.nonce();
+            const message = `${credentials.userId}\n${credentials.deviceId}\n${timestamp}\n${nonce}`;
+            const signature = createHmac('sha256', credentials.botToken).update(message, 'utf8').digest('hex');
+            const body = {
+                user_id: credentials.userId,
+                bot_token: credentials.botToken,
+                device_id: credentials.deviceId,
+                signature,
+                timestamp,
+                nonce,
+                service,
+                query,
+                count,
+                offset,
+                language: request.language?.trim() || 'zh-CN'
+            };
+            if (request.includeDomains?.length)
+                body.include_domains = request.includeDomains;
+            if (request.excludeDomains?.length)
+                body.exclude_domains = request.excludeDomains;
+            const effectiveSignal = requestSignal(signal, this.config.timeoutMs);
+            let response;
+            try {
+                response = await this.fetchImpl(resolved.endpoint, {
+                    method: 'POST',
+                    headers: { accept: 'application/json', 'content-type': 'application/json' },
+                    body: JSON.stringify(body),
+                    signal: effectiveSignal
+                });
+            }
+            catch (error) {
+                if (signal?.aborted)
+                    throw error;
+                if (effectiveSignal.aborted) {
+                    throw new WebSearchApiError('TIMEOUT', `${searchServiceLabel(service)}响应超时，请稍后重试。`, {
+                        cause: error
+                    });
+                }
+                throw new WebSearchApiError('NETWORK_ERROR', `暂时无法连接${searchServiceLabel(service)}，请检查网络后重试。`, { cause: error });
+            }
+            const text = await responseText(response, this.config.maxResponseBytes);
+            if (!response.ok) {
+                throw new WebSearchApiError('HTTP_ERROR', `${searchServiceLabel(service)}暂时不可用（服务状态 ${response.status}），请稍后重试。`);
+            }
+            return parseResponse(text, query, service);
         };
-        if (request.includeDomains?.length)
-            body.include_domains = request.includeDomains;
-        if (request.excludeDomains?.length)
-            body.exclude_domains = request.excludeDomains;
-        const effectiveSignal = requestSignal(signal, this.config.timeoutMs);
-        let response;
         try {
-            response = await this.fetchImpl(this.endpoint, {
-                method: 'POST',
-                headers: { accept: 'application/json', 'content-type': 'application/json' },
-                body: JSON.stringify(body),
-                signal: effectiveSignal
-            });
+            const tavily = await call('tavily_search');
+            if (tavily.results.length > 0)
+                return tavily;
         }
         catch (error) {
             if (signal?.aborted)
                 throw error;
-            if (effectiveSignal.aborted) {
-                throw new WebSearchApiError('TIMEOUT', `web search request timed out after ${this.config.timeoutMs}ms`, {
-                    cause: error
-                });
-            }
-            throw new WebSearchApiError('NETWORK_ERROR', 'web search request failed', { cause: error });
         }
-        const text = await responseText(response, this.config.maxResponseBytes);
-        if (!response.ok)
-            throw new WebSearchApiError('HTTP_ERROR', `web search request failed with HTTP ${response.status}`);
-        return parseResponse(text, query);
+        return call('bocha_search');
     }
 }

package/src/config/env-config.d.ts

@@ -1,10 +1,15 @@
-export type BuildEnv = 'production' | 'test';
+export type WebSearchEnvironment = 'production' | 'test';
 export interface EnvConfig {
-    packageName: string;
     channelId: string;
     legacyChannelId: string;
     endpointUrl: string;
 }
-export declare const ENV_CONFIG: Record<BuildEnv, EnvConfig>;
-export declare const DEFAULT_BUILD_ENV: BuildEnv;
-export declare const ACTIVE_ENV_CONFIG: EnvConfig;
+export declare const PACKAGE_NAME = "@dcrays/web-search";
+export declare const ENV_CONFIG: Record<WebSearchEnvironment, EnvConfig>;
+/**
+ * Map the desktop runtime environment recorded in mobook.json to the search
+ * environment. Development desktops share the test gateway and MBHChat channel.
+ * Missing/unknown values fall back to test so a misconfigured desktop never
+ * silently reaches the production gateway with test credentials.
+ */
+export declare function resolveWebSearchEnvironment(value: unknown): WebSearchEnvironment;

package/src/config/env-config.js

@@ -1,16 +1,22 @@
+export const PACKAGE_NAME = '@dcrays/web-search';
 export const ENV_CONFIG = {
     production: {
-        packageName: '@dcrays/web-search',
         channelId: 'mbh-chat',
         legacyChannelId: 'mbhchat',
         endpointUrl: 'https://api-gateway.shuwenda.com/apikey-manage/api/search'
     },
     test: {
-        packageName: '@dcrays/web-search-test',
         channelId: 'mbh-chat-test',
         legacyChannelId: 'mbhchat-test',
         endpointUrl: 'https://api-gateway.shuwenda.icu/apikey-manage/api/search'
     }
 };
-export const DEFAULT_BUILD_ENV = 'production';
-export const ACTIVE_ENV_CONFIG = ENV_CONFIG[DEFAULT_BUILD_ENV];
+/**
+ * Map the desktop runtime environment recorded in mobook.json to the search
+ * environment. Development desktops share the test gateway and MBHChat channel.
+ * Missing/unknown values fall back to test so a misconfigured desktop never
+ * silently reaches the production gateway with test credentials.
+ */
+export function resolveWebSearchEnvironment(value) {
+    return value === 'production' ? 'production' : 'test';
+}