duojie-helper 0.2.21 → 0.2.23

package/src/tools/openclaw.js

@@ -1,292 +1,292 @@
-import fs from 'fs-extra';
-import path from 'path';
-import os from 'os';
-import { getModels, getApiBaseForProtocol } from '../index.js';
-
-const DUOJIE_PROVIDER_PREFIX = 'duojie-';
-
-/**
- * 获取 OpenClaw 配置路径
- * 支持 macOS / Linux / Windows（os.homedir() 跨平台处理）
- */
-function getOpenClawConfigPaths() {
-  const home = os.homedir();
-  return {
-    configDir: path.join(home, '.openclaw'),
-    configFile: path.join(home, '.openclaw', 'openclaw.json'),
-  };
-}
-
-/**
- * 读取并解析 openclaw.json（支持 JSON5 注释/尾逗号）
- */
-async function readConfig(filePath) {
-  try {
-    const content = await fs.readFile(filePath, 'utf-8');
-    const clean = content
-      .replace(/\/\/.*$/gm, '')
-      .replace(/\/\*[\s\S]*?\*\//g, '')
-      .replace(/,(\s*[}\]])/g, '$1');
-    return JSON.parse(clean);
-  } catch {
-    return {};
-  }
-}
-
-/**
- * 将模型对象转换为 OpenClaw models.providers 格式
- */
-function buildModelEntry(m, contextWindow, maxTokens) {
-  return {
-    id: m.id,
-    name: m.name || m.id,
-    reasoning: false,
-    input: ['text', 'image'],
-    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-    contextWindow,
-    maxTokens,
-  };
-}
-
-function canonicalizeApi(api) {
-  const normalized = `${api || ''}`.toLowerCase().trim();
-  if (!normalized) return 'openai-completions';
-  if (normalized === 'openai-response') return 'openai-responses';
-  if (normalized === 'openai-completion') return 'openai-completions';
-  if (normalized === 'openai') return 'openai-completions';
-  return normalized;
-}
-
-function providerNameForApi(api) {
-  const normalized = canonicalizeApi(api);
-  if (normalized === 'anthropic-messages') return 'duojie-claude';
-  if (normalized === 'openai-completions') return 'duojie-openai';
-  if (normalized === 'openai-responses') return 'duojie-openai-responses';
-  return `duojie-${normalized.replace(/[^a-z0-9]+/g, '-')}`;
-}
-
-function stripDuojieKeysFromObject(obj) {
-  if (!obj || typeof obj !== 'object') return {};
-  const next = { ...obj };
-  for (const key of Object.keys(next)) {
-    if (`${key}`.startsWith(DUOJIE_PROVIDER_PREFIX)) {
-      delete next[key];
-    }
-  }
-  return next;
-}
-
-/**
- * 配置 OpenClaw
- *
- * 使用官方 models.providers 自定义 provider 格式：
- *   models.providers.<name>.baseUrl / apiKey / api / models[]
- *   agents.defaults.model.primary → "<provider>/<model-id>"
- *
- * 参考文档：https://docs.openclaw.ai/gateway/configuration-reference#custom-providers-and-base-urls
- */
-export async function configureOpenClaw(apiKey) {
-  const paths = getOpenClawConfigPaths();
-
-  try {
-    await fs.ensureDir(paths.configDir);
-
-    const existingConfig = await fs.pathExists(paths.configFile)
-      ? await readConfig(paths.configFile)
-      : {};
-
-    // OpenClaw 配置会做 schema 校验：根级未知字段可能导致校验失败
-    // 旧版本写入过 `_duojie`，这里显式剔除
-    const { _duojie: _ignoredDuojie, ...baseConfig } = existingConfig || {};
-    void _ignoredDuojie;
-
-    const models = getModels();
-
-    // 构建各类模型列表（保留 api 协议，后续按 api 拆分 providers）
-    const claudeModels = (models.claude || []).map(m => ({
-      ...m,
-      api: canonicalizeApi(m.api || 'anthropic-messages'),
-      _entry: buildModelEntry(m, 200000, 64000),
-    }));
-    const gptModels = (models.gpt || []).map(m => ({
-      ...m,
-      api: canonicalizeApi(m.api || 'openai-completions'),
-      _entry: buildModelEntry(m, 128000, 16384),
-    }));
-    const geminiModels = (models.gemini || []).map(m => ({
-      ...m,
-      api: canonicalizeApi(m.api || 'openai-completions'),
-      _entry: buildModelEntry(m, 1000000, 65536),
-    }));
-    const otherModels = (models.other || []).map(m => ({
-      ...m,
-      api: canonicalizeApi(m.api || 'openai-completions'),
-      _entry: buildModelEntry(m, 128000, 16384),
-    }));
-
-    const allModels = [...claudeModels, ...gptModels, ...geminiModels, ...otherModels];
-
-    // 构建 providers（按 api 协议分组，避免在同一个 provider 混用不同协议）
-    const providers = {};
-    const modelIdToProvider = new Map();
-
-    for (const m of allModels) {
-      const api = canonicalizeApi(m.api);
-      const providerName = providerNameForApi(api);
-
-      if (!providers[providerName]) {
-        providers[providerName] = {
-          baseUrl: getApiBaseForProtocol(api),
-          apiKey,
-          api,
-          models: [],
-        };
-      }
-
-      providers[providerName].models.push(m._entry);
-      modelIdToProvider.set(m.id, providerName);
-    }
-
-    // 默认模型：优先 gpt-5.2（如存在）
-    const gpt52 = gptModels.find(m => m.id === 'gpt-5.2')?.id || null;
-    let defaultProvider = 'duojie-claude';
-    let defaultModelId = claudeModels[0]?.id || allModels[0]?.id || '';
-
-    if (gpt52) {
-      defaultProvider = modelIdToProvider.get(gpt52) || providerNameForApi('openai-completions');
-      defaultModelId = gpt52;
-    } else if (claudeModels.length > 0) {
-      defaultModelId = (
-        claudeModels.find(m => m.id.includes('opus-4-6')) ||
-        claudeModels.find(m => m.id.includes('opus')) ||
-        claudeModels.find(m => m.id.includes('sonnet')) ||
-        claudeModels[0]
-      ).id;
-      defaultProvider = modelIdToProvider.get(defaultModelId) || 'duojie-claude';
-    } else if (gptModels.length > 0) {
-      defaultModelId = gptModels[0].id;
-      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(gptModels[0].api);
-    } else if (geminiModels.length > 0) {
-      defaultModelId = geminiModels[0].id;
-      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(geminiModels[0].api);
-    } else if (otherModels.length > 0) {
-      defaultModelId = otherModels[0].id;
-      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(otherModels[0].api);
-    } else if (allModels.length > 0) {
-      defaultModelId = allModels[0].id;
-      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(allModels[0].api);
-    }
-
-    // 构建模型 allowlist（用于 /model 切换）
-    const modelAllowList = {};
-    for (const m of allModels) {
-      const providerName = modelIdToProvider.get(m.id) || providerNameForApi(m.api);
-      modelAllowList[`${providerName}/${m.id}`] = { alias: m.name || m.id };
-    }
-
-    const totalModels = allModels.length;
-
-    // 合并配置，保留用户已有的非 Duojie 设置
-    const existingProviders = stripDuojieKeysFromObject(baseConfig.models?.providers);
-    const existingAllowList = stripDuojieKeysFromObject(baseConfig.agents?.defaults?.models);
-
-    const newConfig = {
-      ...baseConfig,
-      models: {
-        ...baseConfig.models,
-        mode: 'merge',
-        providers: {
-          ...existingProviders,
-          ...providers,
-        },
-      },
-      agents: {
-        ...baseConfig.agents,
-        defaults: {
-          ...baseConfig.agents?.defaults,
-          model: {
-            ...baseConfig.agents?.defaults?.model,
-            primary: `${defaultProvider}/${defaultModelId}`,
-          },
-          models: {
-            ...existingAllowList,
-            ...modelAllowList,
-          },
-        },
-      },
-    };
-
-    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
-
-    return {
-      success: true,
-      message: `已配置 ${totalModels} 个模型 → ${paths.configFile}`,
-      configPath: paths.configFile,
-      hint: '运行 openclaw gateway restart 使配置生效',
-    };
-  } catch (error) {
-    return {
-      success: false,
-      message: `配置失败: ${error.message}`,
-    };
-  }
-}
-
-/**
- * 检查 OpenClaw 配置状态
- */
-configureOpenClaw.checkStatus = async function () {
-  const paths = getOpenClawConfigPaths();
-
-  if (await fs.pathExists(paths.configFile)) {
-    const config = await readConfig(paths.configFile);
-    const providerKeys = Object.keys(config.models?.providers || {});
-    if (providerKeys.some(k => `${k}`.startsWith(DUOJIE_PROVIDER_PREFIX))) {
-      return { configured: true, message: '已配置 Duojie API' };
-    }
-    if (providerKeys.length > 0) {
-      return { configured: true, message: '已配置（非 Duojie）' };
-    }
-  }
-
-  return { configured: false };
-};
-
-/**
- * 撤销 OpenClaw 配置
- */
-configureOpenClaw.revoke = async function () {
-  const paths = getOpenClawConfigPaths();
-
-  if (await fs.pathExists(paths.configFile)) {
-    const config = await readConfig(paths.configFile);
-
-    if (config.models?.providers) {
-      for (const key of Object.keys(config.models.providers)) {
-        if (`${key}`.startsWith(DUOJIE_PROVIDER_PREFIX)) {
-          delete config.models.providers[key];
-        }
-      }
-    }
-
-    if (config.agents?.defaults?.models) {
-      for (const key of Object.keys(config.agents.defaults.models)) {
-        if (`${key}`.startsWith(DUOJIE_PROVIDER_PREFIX)) {
-          delete config.agents.defaults.models[key];
-        }
-      }
-    }
-
-    if (config.agents?.defaults?.model?.primary?.startsWith?.(DUOJIE_PROVIDER_PREFIX)) {
-      delete config.agents.defaults.model.primary;
-    }
-
-    if (config._duojie) {
-      delete config._duojie;
-    }
-
-    await fs.writeJson(paths.configFile, config, { spaces: 2 });
-  }
-};
-
-export default configureOpenClaw;
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+import { getModels, getApiBaseForProtocol } from '../index.js';
+
+const DUOJIE_PROVIDER_PREFIX = 'duojie-';
+
+/**
+ * 获取 OpenClaw 配置路径
+ * 支持 macOS / Linux / Windows（os.homedir() 跨平台处理）
+ */
+function getOpenClawConfigPaths() {
+  const home = os.homedir();
+  return {
+    configDir: path.join(home, '.openclaw'),
+    configFile: path.join(home, '.openclaw', 'openclaw.json'),
+  };
+}
+
+/**
+ * 读取并解析 openclaw.json（支持 JSON5 注释/尾逗号）
+ */
+async function readConfig(filePath) {
+  try {
+    const content = await fs.readFile(filePath, 'utf-8');
+    const clean = content
+      .replace(/\/\/.*$/gm, '')
+      .replace(/\/\*[\s\S]*?\*\//g, '')
+      .replace(/,(\s*[}\]])/g, '$1');
+    return JSON.parse(clean);
+  } catch {
+    return {};
+  }
+}
+
+/**
+ * 将模型对象转换为 OpenClaw models.providers 格式
+ */
+function buildModelEntry(m, contextWindow, maxTokens) {
+  return {
+    id: m.id,
+    name: m.name || m.id,
+    reasoning: false,
+    input: ['text', 'image'],
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    contextWindow,
+    maxTokens,
+  };
+}
+
+function canonicalizeApi(api) {
+  const normalized = `${api || ''}`.toLowerCase().trim();
+  if (!normalized) return 'openai-completions';
+  if (normalized === 'openai-response') return 'openai-responses';
+  if (normalized === 'openai-completion') return 'openai-completions';
+  if (normalized === 'openai') return 'openai-completions';
+  return normalized;
+}
+
+function providerNameForApi(api) {
+  const normalized = canonicalizeApi(api);
+  if (normalized === 'anthropic-messages') return 'duojie-claude';
+  if (normalized === 'openai-completions') return 'duojie-openai';
+  if (normalized === 'openai-responses') return 'duojie-openai-responses';
+  return `duojie-${normalized.replace(/[^a-z0-9]+/g, '-')}`;
+}
+
+function stripDuojieKeysFromObject(obj) {
+  if (!obj || typeof obj !== 'object') return {};
+  const next = { ...obj };
+  for (const key of Object.keys(next)) {
+    if (`${key}`.startsWith(DUOJIE_PROVIDER_PREFIX)) {
+      delete next[key];
+    }
+  }
+  return next;
+}
+
+/**
+ * 配置 OpenClaw
+ *
+ * 使用官方 models.providers 自定义 provider 格式：
+ *   models.providers.<name>.baseUrl / apiKey / api / models[]
+ *   agents.defaults.model.primary → "<provider>/<model-id>"
+ *
+ * 参考文档：https://docs.openclaw.ai/gateway/configuration-reference#custom-providers-and-base-urls
+ */
+export async function configureOpenClaw(apiKey) {
+  const paths = getOpenClawConfigPaths();
+
+  try {
+    await fs.ensureDir(paths.configDir);
+
+    const existingConfig = await fs.pathExists(paths.configFile)
+      ? await readConfig(paths.configFile)
+      : {};
+
+    // OpenClaw 配置会做 schema 校验：根级未知字段可能导致校验失败
+    // 旧版本写入过 `_duojie`，这里显式剔除
+    const { _duojie: _ignoredDuojie, ...baseConfig } = existingConfig || {};
+    void _ignoredDuojie;
+
+    const models = getModels();
+
+    // 构建各类模型列表（保留 api 协议，后续按 api 拆分 providers）
+    const claudeModels = (models.claude || []).map(m => ({
+      ...m,
+      api: canonicalizeApi(m.api || 'anthropic-messages'),
+      _entry: buildModelEntry(m, 200000, 64000),
+    }));
+    const gptModels = (models.gpt || []).map(m => ({
+      ...m,
+      api: canonicalizeApi(m.api || 'openai-completions'),
+      _entry: buildModelEntry(m, 128000, 16384),
+    }));
+    const geminiModels = (models.gemini || []).map(m => ({
+      ...m,
+      api: canonicalizeApi(m.api || 'openai-completions'),
+      _entry: buildModelEntry(m, 1000000, 65536),
+    }));
+    const otherModels = (models.other || []).map(m => ({
+      ...m,
+      api: canonicalizeApi(m.api || 'openai-completions'),
+      _entry: buildModelEntry(m, 128000, 16384),
+    }));
+
+    const allModels = [...claudeModels, ...gptModels, ...geminiModels, ...otherModels];
+
+    // 构建 providers（按 api 协议分组，避免在同一个 provider 混用不同协议）
+    const providers = {};
+    const modelIdToProvider = new Map();
+
+    for (const m of allModels) {
+      const api = canonicalizeApi(m.api);
+      const providerName = providerNameForApi(api);
+
+      if (!providers[providerName]) {
+        providers[providerName] = {
+          baseUrl: getApiBaseForProtocol(api),
+          apiKey,
+          api,
+          models: [],
+        };
+      }
+
+      providers[providerName].models.push(m._entry);
+      modelIdToProvider.set(m.id, providerName);
+    }
+
+    // 默认模型：优先 gpt-5.2（如存在）
+    const gpt52 = gptModels.find(m => m.id === 'gpt-5.2')?.id || null;
+    let defaultProvider = 'duojie-claude';
+    let defaultModelId = claudeModels[0]?.id || allModels[0]?.id || '';
+
+    if (gpt52) {
+      defaultProvider = modelIdToProvider.get(gpt52) || providerNameForApi('openai-completions');
+      defaultModelId = gpt52;
+    } else if (claudeModels.length > 0) {
+      defaultModelId = (
+        claudeModels.find(m => m.id.includes('opus-4-6')) ||
+        claudeModels.find(m => m.id.includes('opus')) ||
+        claudeModels.find(m => m.id.includes('sonnet')) ||
+        claudeModels[0]
+      ).id;
+      defaultProvider = modelIdToProvider.get(defaultModelId) || 'duojie-claude';
+    } else if (gptModels.length > 0) {
+      defaultModelId = gptModels[0].id;
+      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(gptModels[0].api);
+    } else if (geminiModels.length > 0) {
+      defaultModelId = geminiModels[0].id;
+      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(geminiModels[0].api);
+    } else if (otherModels.length > 0) {
+      defaultModelId = otherModels[0].id;
+      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(otherModels[0].api);
+    } else if (allModels.length > 0) {
+      defaultModelId = allModels[0].id;
+      defaultProvider = modelIdToProvider.get(defaultModelId) || providerNameForApi(allModels[0].api);
+    }
+
+    // 构建模型 allowlist（用于 /model 切换）
+    const modelAllowList = {};
+    for (const m of allModels) {
+      const providerName = modelIdToProvider.get(m.id) || providerNameForApi(m.api);
+      modelAllowList[`${providerName}/${m.id}`] = { alias: m.name || m.id };
+    }
+
+    const totalModels = allModels.length;
+
+    // 合并配置，保留用户已有的非 Duojie 设置
+    const existingProviders = stripDuojieKeysFromObject(baseConfig.models?.providers);
+    const existingAllowList = stripDuojieKeysFromObject(baseConfig.agents?.defaults?.models);
+
+    const newConfig = {
+      ...baseConfig,
+      models: {
+        ...baseConfig.models,
+        mode: 'merge',
+        providers: {
+          ...existingProviders,
+          ...providers,
+        },
+      },
+      agents: {
+        ...baseConfig.agents,
+        defaults: {
+          ...baseConfig.agents?.defaults,
+          model: {
+            ...baseConfig.agents?.defaults?.model,
+            primary: `${defaultProvider}/${defaultModelId}`,
+          },
+          models: {
+            ...existingAllowList,
+            ...modelAllowList,
+          },
+        },
+      },
+    };
+
+    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
+
+    return {
+      success: true,
+      message: `已配置 ${totalModels} 个模型 → ${paths.configFile}`,
+      configPath: paths.configFile,
+      hint: '运行 openclaw gateway restart 使配置生效',
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `配置失败: ${error.message}`,
+    };
+  }
+}
+
+/**
+ * 检查 OpenClaw 配置状态
+ */
+configureOpenClaw.checkStatus = async function () {
+  const paths = getOpenClawConfigPaths();
+
+  if (await fs.pathExists(paths.configFile)) {
+    const config = await readConfig(paths.configFile);
+    const providerKeys = Object.keys(config.models?.providers || {});
+    if (providerKeys.some(k => `${k}`.startsWith(DUOJIE_PROVIDER_PREFIX))) {
+      return { configured: true, message: '已配置 Duojie API' };
+    }
+    if (providerKeys.length > 0) {
+      return { configured: true, message: '已配置（非 Duojie）' };
+    }
+  }
+
+  return { configured: false };
+};
+
+/**
+ * 撤销 OpenClaw 配置
+ */
+configureOpenClaw.revoke = async function () {
+  const paths = getOpenClawConfigPaths();
+
+  if (await fs.pathExists(paths.configFile)) {
+    const config = await readConfig(paths.configFile);
+
+    if (config.models?.providers) {
+      for (const key of Object.keys(config.models.providers)) {
+        if (`${key}`.startsWith(DUOJIE_PROVIDER_PREFIX)) {
+          delete config.models.providers[key];
+        }
+      }
+    }
+
+    if (config.agents?.defaults?.models) {
+      for (const key of Object.keys(config.agents.defaults.models)) {
+        if (`${key}`.startsWith(DUOJIE_PROVIDER_PREFIX)) {
+          delete config.agents.defaults.models[key];
+        }
+      }
+    }
+
+    if (config.agents?.defaults?.model?.primary?.startsWith?.(DUOJIE_PROVIDER_PREFIX)) {
+      delete config.agents.defaults.model.primary;
+    }
+
+    if (config._duojie) {
+      delete config._duojie;
+    }
+
+    await fs.writeJson(paths.configFile, config, { spaces: 2 });
+  }
+};
+
+export default configureOpenClaw;