dsh-agentone 0.5.0

package/lib/model.js

@@ -0,0 +1,200 @@
+// 平台模型自动配置：登录后把 AgentOne 平台模型写入本机 dsh。
+//
+// 与平台托管实例（app/orchestrator/dsh_orchestrator.py build_dsh_settings_yaml）
+// 同一条命名空间语义，只覆盖两处平台段，用户其它配置原样保留：
+//   settings.yaml      llm-pi-ai.providers.platform + agent-default-model
+//   .credentials.yaml  refs.AGENTONE_API_KEY（dsh 凭据库，owner-only，watch 热加载）
+//
+// 密钥形态是 30 天插件令牌（M1 exchange 签发），指向平台 /api/plugin/v1 代理；
+// 真实 Token Hub 密钥不出平台。
+import { readFile, rename, writeFile } from 'node:fs/promises';
+import { join } from 'node:path';
+import { parse, stringify } from 'yaml';
+
+import { platformError, platformFetch } from './http.js';
+import { loadCredentials, resolveDshHome } from './store.js';
+
+export const PROVIDER_KEY = 'platform';
+export const CREDENTIAL_REF = 'AGENTONE_API_KEY';
+
+const REASONING_EFFORTS = { off: null, low: 'low', medium: 'medium', high: 'high', max: 'max' };
+
+/**
+ * 把平台 /models 条目转成 llm-pi-ai models 配置。
+ * input（视觉）与 reasoning（思考）是平台能力表下发的声明，插件只翻译：
+ * 未声明的旧平台版本回退到全档位思考 + 纯文本。
+ */
+function platformModelEntry(item) {
+  const id = typeof item === 'string' ? item : item?.id;
+  if (!id) return null;
+  const entry = { id, name: friendlyModelName(id) };
+  const input = Array.isArray(item?.input) ? item.input : ['text'];
+  if (input.includes('image')) entry.input = ['text', 'image'];
+  if (item?.reasoning === false) entry.reasoningEfforts = false;
+  else entry.reasoningEfforts = { ...REASONING_EFFORTS };
+  return entry;
+}
+
+/** 已知厂家的展示名（小写 id 前缀 → 名称）。 */
+const VENDOR_NAMES = {
+  deepseek: 'DeepSeek',
+  glm: 'GLM',
+  gpt: 'GPT',
+  minimax: 'MiniMax',
+  kimi: 'Kimi',
+  doubao: '豆包',
+  qwen: 'Qwen',
+  claude: 'Claude',
+  grok: 'Grok',
+  gemini: 'Gemini',
+  llava: 'LLaVA',
+};
+
+/** 模型 id → 选择器里的友好显示名：deepseek-v4.1-flash → DeepSeek V4.1 Flash。 */
+export function friendlyModelName(id) {
+  const parts = String(id || '').split('-').filter(Boolean);
+  if (!parts.length) return String(id || '');
+  const [rawVendor, ...rest] = parts;
+  const vendor = VENDOR_NAMES[rawVendor.toLowerCase()] ?? rawVendor.toUpperCase();
+  const tail = rest
+    .map((part) => (/^[a-z]/.test(part) ? part.charAt(0).toUpperCase() + part.slice(1) : part))
+    .join(' ');
+  return tail ? `${vendor} ${tail}` : vendor;
+}
+
+function parseYamlObject(text) {
+  if (!text || !text.trim()) return {};
+  try {
+    const loaded = parse(text);
+    if (loaded && typeof loaded === 'object' && !Array.isArray(loaded)) return loaded;
+  } catch {
+    // 损坏的 YAML 视为空配置，避免因为一段坏配置让用户彻底不可用
+  }
+  return {};
+}
+
+/** 合并出 settings.yaml 文本：覆盖 platform provider 与 agent-default-model。 */
+export function buildSettingsYaml(existingText, { baseUrl, models, defaultModel }) {
+  const settings = parseYamlObject(existingText);
+  const existing = settings['llm-pi-ai'];
+  const llmPiAi =
+    existing && typeof existing === 'object' && existing.providers && typeof existing.providers === 'object'
+      ? existing
+      : { providers: {} };
+  llmPiAi.providers[PROVIDER_KEY] = {
+    displayName: '平台模型',
+    apiKeyEnv: CREDENTIAL_REF,
+    api: 'openai-completions',
+    baseURL: baseUrl,
+    reasoning: 'high',
+    models: models
+      .map((item) => platformModelEntry(item))
+      .filter((entry) => entry !== null),
+  };
+  settings['llm-pi-ai'] = llmPiAi;
+  settings['agent-default-model'] = {
+    provider: PROVIDER_KEY,
+    model: defaultModel,
+    reasoningEffort: 'high',
+  };
+  return stringify(settings);
+}
+
+/** 合并出 .credentials.yaml 文本：写入 refs.AGENTONE_API_KEY，其余 refs/records 保留。 */
+export function buildCredentialsYaml(existingText, token) {
+  const document = parseYamlObject(existingText);
+  if (!document.refs || typeof document.refs !== 'object') document.refs = {};
+  document.refs[CREDENTIAL_REF] = token;
+  return stringify(document);
+}
+
+/** 登出清理：删除凭据引用；settings 中的 platform provider 与默认模型一并移除。 */
+export function buildLogoutYaml(existingText) {
+  const document = parseYamlObject(existingText);
+  if (document.refs && typeof document.refs === 'object') delete document.refs[CREDENTIAL_REF];
+  return stringify(document);
+}
+
+export function buildLogoutSettingsYaml(existingText) {
+  const settings = parseYamlObject(existingText);
+  const providers = settings['llm-pi-ai']?.providers;
+  if (providers && typeof providers === 'object') delete providers[PROVIDER_KEY];
+  if (settings['agent-default-model']?.provider === PROVIDER_KEY) {
+    delete settings['agent-default-model'];
+  }
+  return stringify(settings);
+}
+
+export async function fetchPlatformModels(platformUrl, accessToken) {
+  const response = await platformFetch(`${platformUrl.replace(/\/+$/, '')}/api/plugin/v1/models`, {
+    headers: { Authorization: `Bearer ${accessToken}` },
+  });
+  if (!response.ok) {
+    // 平台 403 带 error.code=plan_required：错误码透传给前端做确定性判断，
+    // 不再依赖中文文案正则（文案一改判定就失效）。
+    throw await platformError(response, '获取平台模型列表失败');
+  }
+  const data = await response.json().catch(() => ({}));
+  const models = Array.isArray(data.data)
+    ? data.data
+        .map((item) => (typeof item === 'string' ? item : item?.id ? { id: item.id, input: item.input, reasoning: item.reasoning } : null))
+        .filter(Boolean)
+    : [];
+  if (models.length === 0) throw new Error('平台返回的可用模型列表为空');
+  return { models, defaultModel: data.default_model || (models[0]?.id ?? models[0]) };
+}
+
+async function writeFileAtomic(path, text, mode) {
+  const tmp = tmpName(path);
+  await writeFile(tmp, text, mode ? { mode } : undefined);
+  await rename(tmp, path);
+}
+
+function tmpName(path) {
+  return `${path}.agentone-${process.pid}-${Date.now()}.tmp`;
+}
+
+/**
+ * 拉取平台模型并写入本机 dsh 配置。返回 { models, defaultModel }。
+ * 失败抛 PlatformError（中文 message + 平台错误码），由调用方记录/展示
+ * （不影响已生效的旧配置）。
+ */
+export async function applyPlatformModel(config) {
+  const credentials = await loadCredentials(config);
+  if (!credentials) throw new Error('尚未登录平台');
+  const baseUrl = `${String(credentials.platform_url || config.platformUrl).replace(/\/+$/, '')}/api/plugin/v1`;
+  const { models, defaultModel } = await fetchPlatformModels(
+    credentials.platform_url || config.platformUrl,
+    credentials.access_token,
+  );
+  const home = resolveDshHome(config);
+  const settingsPath = join(home, 'settings.yaml');
+  const credentialsPath = join(home, '.credentials.yaml');
+  const [existingSettings, existingCredentials] = await Promise.all([
+    readFile(settingsPath, 'utf8').catch(() => ''),
+    readFile(credentialsPath, 'utf8').catch(() => ''),
+  ]);
+  await writeFileAtomic(credentialsPath, buildCredentialsYaml(existingCredentials, credentials.access_token), 0o600);
+  await writeFileAtomic(
+    settingsPath,
+    buildSettingsYaml(existingSettings, { baseUrl, models, defaultModel }),
+  );
+  return { models, defaultModel };
+}
+
+/** 登出时清理本机配置：删除凭据引用与 platform provider。 */
+export async function removePlatformModel(config) {
+  const home = resolveDshHome(config);
+  const settingsPath = join(home, 'settings.yaml');
+  const credentialsPath = join(home, '.credentials.yaml');
+  const [existingSettings, existingCredentials] = await Promise.all([
+    readFile(settingsPath, 'utf8').catch(() => ''),
+    readFile(credentialsPath, 'utf8').catch(() => ''),
+  ]);
+  if (existingCredentials) {
+    await writeFileAtomic(credentialsPath, buildLogoutYaml(existingCredentials), 0o600);
+  }
+  if (existingSettings) {
+    await writeFileAtomic(settingsPath, buildLogoutSettingsYaml(existingSettings));
+  }
+}