rol-websocket-channel 1.4.2 → 1.4.8

package/dist/src/admin/methods/models-extended.js

@@ -1,190 +1,190 @@
-import path from 'node:path';
-import { readJsonFile, writeJsonFile } from '../lib/fs.js';
-import { JsonRpcException, JSON_RPC_ERRORS } from '../jsonrpc.js';
-// 允许修改的 provider 配置字段白名单
-const ALLOWED_PROVIDER_FIELDS = {
-    'apiKey': true,
-    'api': true,
-    'baseUrl': true,
-    'models': true,
-    'name': true,
-    'label': true,
-    'model': true,
-    'temperature': true,
-    'maxTokens': true,
-    'topP': true,
-    'frequencyPenalty': true,
-    'presencePenalty': true,
-};
-/**
- * 更新模型配置
- * 支持修改 primary model 和 provider 配置
- */
-export const updateModels = async (params, context) => {
-    const objectParams = isObject(params) ? params : {};
-    const primaryModel = pickString(objectParams.primaryModel);
-    const modelProvider = pickString(objectParams.modelProvider);
-    const provider = pickString(objectParams.provider);
-    const providerConfig = isObject(objectParams.providerConfig) ? objectParams.providerConfig : null;
-    const configPath = path.join(context.openclawRoot, 'openclaw.json');
-    const config = await readJsonFile(configPath);
-    let updated = false;
-    const changes = [];
-    // 更新 primary model
-    if (primaryModel) {
-        if (!config.agents)
-            config.agents = {};
-        if (!config.agents.defaults)
-            config.agents.defaults = {};
-        if (!config.agents.defaults.model)
-            config.agents.defaults.model = {};
-        config.agents.defaults.model.primary = primaryModel;
-        const inferredProvider = inferProviderFromPrimaryModel(primaryModel);
-        if (modelProvider && modelProvider !== inferredProvider) {
-            throwModelError('MODEL_PROVIDER_MISMATCH', 'modelProvider does not match primaryModel');
-        }
-        updated = true;
-        changes.push(`Updated primary model to: ${primaryModel}`);
-    }
-    if (modelProvider && !primaryModel) {
-        throwModelError('MODEL_PROVIDER_DERIVED', 'modelProvider cannot be updated directly; set primaryModel in provider/model format');
-    }
-    // 更新 provider 配置
-    if (provider && providerConfig) {
-        if (!config.models)
-            config.models = {};
-        if (!config.models.providers)
-            config.models.providers = {};
-        // 验证字段白名单
-        for (const field of Object.keys(providerConfig)) {
-            if (!ALLOWED_PROVIDER_FIELDS[field]) {
-                throw new JsonRpcException(JSON_RPC_ERRORS.invalidParams, `Field not allowed for provider config: ${field}. Allowed fields: ${Object.keys(ALLOWED_PROVIDER_FIELDS).join(', ')}`);
-            }
-        }
-        // 合并配置
-        if (!config.models.providers[provider]) {
-            config.models.providers[provider] = {};
-        }
-        config.models.providers[provider] = {
-            ...config.models.providers[provider],
-            ...providerConfig
-        };
-        updated = true;
-        changes.push(`Updated provider config for: ${provider}`);
-    }
-    if (!updated) {
-        throw new JsonRpcException(JSON_RPC_ERRORS.invalidParams, 'No valid update parameters provided. Use primaryModel, modelProvider, or provider+providerConfig.');
-    }
-    // 写回配置文件
-    await writeJsonFile(configPath, config);
-    return {
-        success: true,
-        configFile: configPath,
-        changes,
-        updatedConfig: {
-            primaryModel: config.agents?.defaults?.model?.primary ?? null,
-            modelProvider: config.agents?.defaults?.model?.primary
-                ? inferProviderFromPrimaryModel(config.agents.defaults.model.primary)
-                : null,
-            defaultModel: config.agents?.defaults?.model ?? {},
-            providers: redactSecrets(config.models?.providers ?? {})
-        }
-    };
-};
-export const setModel = async (params, context) => {
-    const objectParams = isObject(params) ? params : {};
-    const primary = pickString(objectParams.primary);
-    const provider = pickString(objectParams.provider);
-    if (!primary) {
-        throwModelError('MODEL_PRIMARY_REQUIRED', 'primary is required');
-    }
-    const inferredProvider = inferProviderFromPrimaryModel(primary);
-    if (!inferredProvider) {
-        throwModelError('MODEL_PRIMARY_INVALID', 'primary must be in provider/model format');
-    }
-    if (provider && provider !== inferredProvider) {
-        throwModelError('MODEL_PROVIDER_MISMATCH', 'provider does not match primary');
-    }
-    const configPath = path.join(context.openclawRoot, 'openclaw.json');
-    const config = await readJsonFile(configPath);
-    if (!isAllowedModel(config, primary)) {
-        throwModelError('MODEL_NOT_ALLOWED', 'model is not in allowed model options');
-    }
-    if (!config.agents)
-        config.agents = {};
-    if (!config.agents.defaults)
-        config.agents.defaults = {};
-    if (!config.agents.defaults.model)
-        config.agents.defaults.model = {};
-    config.agents.defaults.model.primary = primary;
-    await writeJsonFile(configPath, config);
-    return {
-        ok: true,
-        model: {
-            primary,
-            provider: inferredProvider
-        },
-        defaults: {
-            model: config.agents.defaults.model
-        }
-    };
-};
-function inferProviderFromPrimaryModel(primaryModel) {
-    const slashIndex = primaryModel.indexOf('/');
-    if (slashIndex <= 0) {
-        return null;
-    }
-    return primaryModel.slice(0, slashIndex);
-}
-function isAllowedModel(config, primary) {
-    const catalog = config.agents?.defaults?.models;
-    if (catalog && typeof catalog === 'object' && !Array.isArray(catalog)) {
-        return Object.prototype.hasOwnProperty.call(catalog, primary);
-    }
-    const currentPrimary = pickString(config.agents?.defaults?.model?.primary);
-    if (currentPrimary === primary) {
-        return true;
-    }
-    const agents = Array.isArray(config.agents?.list) ? config.agents.list : [];
-    return agents.some((agent) => pickString(agent?.model?.primary) === primary);
-}
-function throwModelError(errorCode, message) {
-    throw new JsonRpcException(JSON_RPC_ERRORS.invalidParams, message, { code: errorCode });
-}
-function pickString(value) {
-    if (typeof value !== 'string') {
-        return null;
-    }
-    const trimmed = value.trim();
-    return trimmed.length > 0 ? trimmed : null;
-}
-function isObject(value) {
-    return Boolean(value) && typeof value === 'object' && !Array.isArray(value);
-}
-function redactSecrets(value) {
-    if (Array.isArray(value)) {
-        return value.map(redactSecrets);
-    }
-    if (!value || typeof value !== 'object') {
-        return value;
-    }
-    const result = {};
-    for (const [key, nestedValue] of Object.entries(value)) {
-        if (isSecretKey(key) && typeof nestedValue === 'string') {
-            result[key] = redactString(nestedValue);
-            continue;
-        }
-        result[key] = redactSecrets(nestedValue);
-    }
-    return result;
-}
-function isSecretKey(key) {
-    const normalized = key.toLowerCase();
-    return normalized.includes('apikey') || normalized.includes('token') || normalized.includes('secret');
-}
-function redactString(value) {
-    if (value.length <= 8) {
-        return '********';
-    }
-    return `${value.slice(0, 4)}***${value.slice(-4)}`;
-}
+import path from 'node:path';
+import { readJsonFile, writeJsonFile } from '../lib/fs.js';
+import { JsonRpcException, JSON_RPC_ERRORS } from '../jsonrpc.js';
+// 允许修改的 provider 配置字段白名单
+const ALLOWED_PROVIDER_FIELDS = {
+    'apiKey': true,
+    'api': true,
+    'baseUrl': true,
+    'models': true,
+    'name': true,
+    'label': true,
+    'model': true,
+    'temperature': true,
+    'maxTokens': true,
+    'topP': true,
+    'frequencyPenalty': true,
+    'presencePenalty': true,
+};
+/**
+ * 更新模型配置
+ * 支持修改 primary model 和 provider 配置
+ */
+export const updateModels = async (params, context) => {
+    const objectParams = isObject(params) ? params : {};
+    const primaryModel = pickString(objectParams.primaryModel);
+    const modelProvider = pickString(objectParams.modelProvider);
+    const provider = pickString(objectParams.provider);
+    const providerConfig = isObject(objectParams.providerConfig) ? objectParams.providerConfig : null;
+    const configPath = path.join(context.openclawRoot, 'openclaw.json');
+    const config = await readJsonFile(configPath);
+    let updated = false;
+    const changes = [];
+    // 更新 primary model
+    if (primaryModel) {
+        if (!config.agents)
+            config.agents = {};
+        if (!config.agents.defaults)
+            config.agents.defaults = {};
+        if (!config.agents.defaults.model)
+            config.agents.defaults.model = {};
+        config.agents.defaults.model.primary = primaryModel;
+        const inferredProvider = inferProviderFromPrimaryModel(primaryModel);
+        if (modelProvider && modelProvider !== inferredProvider) {
+            throwModelError('MODEL_PROVIDER_MISMATCH', 'modelProvider does not match primaryModel');
+        }
+        updated = true;
+        changes.push(`Updated primary model to: ${primaryModel}`);
+    }
+    if (modelProvider && !primaryModel) {
+        throwModelError('MODEL_PROVIDER_DERIVED', 'modelProvider cannot be updated directly; set primaryModel in provider/model format');
+    }
+    // 更新 provider 配置
+    if (provider && providerConfig) {
+        if (!config.models)
+            config.models = {};
+        if (!config.models.providers)
+            config.models.providers = {};
+        // 验证字段白名单
+        for (const field of Object.keys(providerConfig)) {
+            if (!ALLOWED_PROVIDER_FIELDS[field]) {
+                throw new JsonRpcException(JSON_RPC_ERRORS.invalidParams, `Field not allowed for provider config: ${field}. Allowed fields: ${Object.keys(ALLOWED_PROVIDER_FIELDS).join(', ')}`);
+            }
+        }
+        // 合并配置
+        if (!config.models.providers[provider]) {
+            config.models.providers[provider] = {};
+        }
+        config.models.providers[provider] = {
+            ...config.models.providers[provider],
+            ...providerConfig
+        };
+        updated = true;
+        changes.push(`Updated provider config for: ${provider}`);
+    }
+    if (!updated) {
+        throw new JsonRpcException(JSON_RPC_ERRORS.invalidParams, 'No valid update parameters provided. Use primaryModel, modelProvider, or provider+providerConfig.');
+    }
+    // 写回配置文件
+    await writeJsonFile(configPath, config);
+    return {
+        success: true,
+        configFile: configPath,
+        changes,
+        updatedConfig: {
+            primaryModel: config.agents?.defaults?.model?.primary ?? null,
+            modelProvider: config.agents?.defaults?.model?.primary
+                ? inferProviderFromPrimaryModel(config.agents.defaults.model.primary)
+                : null,
+            defaultModel: config.agents?.defaults?.model ?? {},
+            providers: redactSecrets(config.models?.providers ?? {})
+        }
+    };
+};
+export const setModel = async (params, context) => {
+    const objectParams = isObject(params) ? params : {};
+    const primary = pickString(objectParams.primary);
+    const provider = pickString(objectParams.provider);
+    if (!primary) {
+        throwModelError('MODEL_PRIMARY_REQUIRED', 'primary is required');
+    }
+    const inferredProvider = inferProviderFromPrimaryModel(primary);
+    if (!inferredProvider) {
+        throwModelError('MODEL_PRIMARY_INVALID', 'primary must be in provider/model format');
+    }
+    if (provider && provider !== inferredProvider) {
+        throwModelError('MODEL_PROVIDER_MISMATCH', 'provider does not match primary');
+    }
+    const configPath = path.join(context.openclawRoot, 'openclaw.json');
+    const config = await readJsonFile(configPath);
+    if (!isAllowedModel(config, primary)) {
+        throwModelError('MODEL_NOT_ALLOWED', 'model is not in allowed model options');
+    }
+    if (!config.agents)
+        config.agents = {};
+    if (!config.agents.defaults)
+        config.agents.defaults = {};
+    if (!config.agents.defaults.model)
+        config.agents.defaults.model = {};
+    config.agents.defaults.model.primary = primary;
+    await writeJsonFile(configPath, config);
+    return {
+        ok: true,
+        model: {
+            primary,
+            provider: inferredProvider
+        },
+        defaults: {
+            model: config.agents.defaults.model
+        }
+    };
+};
+function inferProviderFromPrimaryModel(primaryModel) {
+    const slashIndex = primaryModel.indexOf('/');
+    if (slashIndex <= 0) {
+        return null;
+    }
+    return primaryModel.slice(0, slashIndex);
+}
+function isAllowedModel(config, primary) {
+    const catalog = config.agents?.defaults?.models;
+    if (catalog && typeof catalog === 'object' && !Array.isArray(catalog)) {
+        return Object.prototype.hasOwnProperty.call(catalog, primary);
+    }
+    const currentPrimary = pickString(config.agents?.defaults?.model?.primary);
+    if (currentPrimary === primary) {
+        return true;
+    }
+    const agents = Array.isArray(config.agents?.list) ? config.agents.list : [];
+    return agents.some((agent) => pickString(agent?.model?.primary) === primary);
+}
+function throwModelError(errorCode, message) {
+    throw new JsonRpcException(JSON_RPC_ERRORS.invalidParams, message, { code: errorCode });
+}
+function pickString(value) {
+    if (typeof value !== 'string') {
+        return null;
+    }
+    const trimmed = value.trim();
+    return trimmed.length > 0 ? trimmed : null;
+}
+function isObject(value) {
+    return Boolean(value) && typeof value === 'object' && !Array.isArray(value);
+}
+function redactSecrets(value) {
+    if (Array.isArray(value)) {
+        return value.map(redactSecrets);
+    }
+    if (!value || typeof value !== 'object') {
+        return value;
+    }
+    const result = {};
+    for (const [key, nestedValue] of Object.entries(value)) {
+        if (isSecretKey(key) && typeof nestedValue === 'string') {
+            result[key] = redactString(nestedValue);
+            continue;
+        }
+        result[key] = redactSecrets(nestedValue);
+    }
+    return result;
+}
+function isSecretKey(key) {
+    const normalized = key.toLowerCase();
+    return normalized.includes('apikey') || normalized.includes('token') || normalized.includes('secret');
+}
+function redactString(value) {
+    if (value.length <= 8) {
+        return '********';
+    }
+    return `${value.slice(0, 4)}***${value.slice(-4)}`;
+}