dsh-account-pool 0.1.0

package/lib/trae-upstream.js

@@ -0,0 +1,485 @@
+/**
+ * Trae（TRAE SOLO）上游客户端：对话、模型目录、签到、积分。
+ *
+ * 与 WorkBuddy 的关键差异都集中在请求体改写里（见 prepareChatBody）：
+ *   - content 必须包成数组 [{type:'text',text:'...'}]
+ *   - 模型名要同时写进 config_name 与 model
+ *   - 必带 function: 'solo_work_lite'
+ *   - assistant 的 tool_calls 要转成上游的 function_call
+ */
+
+import { bridgeTraeSseToOpenAI } from './trae-bridge.js'
+import { TRAE } from './trae-accounts.js'
+
+const { AGENT_HOST, UG_HOST, SOLO_FUNCTION, DEFAULT_MODEL, soloHeaders, ugHeaders, postJson } = TRAE
+
+/** 对话与模型端点。 */
+const EP_CHAT = '/api/agent/v3/llm_utils_chat'
+const EP_MODELS = '/api/ide/v1/get_detail_param'
+
+/** 签到与积分端点。 */
+/**
+ * Trae 的业务码表（HTTP 200 也会出现的数字码）。
+ *
+ * 上游用「HTTP 200 + code != 0」表达业务失败，靠英文措辞判断不可靠——
+ * 换个近义说法就漏掉。数字码是稳定契约，用它分类。
+ *
+ * 取值来自实测（2026-09-21，真实账号）：
+ *   4008  对话额度用尽      → 该换号（与本插件 hard_credit 同类）
+ *   1005  额度/鉴权问题     → 该换号
+ *   9074  参与用户过多      → 稍后重试（软失败）
+ *   9095  该设备今日已签到  → 不是错误，是"已完成"，中性提示
+ *   9004  请求参数不正确    → 客户端问题（如缺 x-device-id）
+ *   4023  需要自带上游 key  → BYOK 模板未配置
+ */
+const TRAE_BIZ_CODES = {
+  4008: { kind: 'hard_credit', label: '对话额度已用尽' },
+  1005: { kind: 'hard_credit', label: '额度或鉴权异常' },
+  9074: { kind: 'soft_rate', label: '参与用户太多，请稍后再试' },
+  9095: { kind: 'checkin_done', label: '该设备今日已签到' },
+  9004: { kind: 'client', label: '请求参数不正确' },
+  4023: { kind: 'client', label: '该模型需要自带上游 API key' },
+}
+
+const EP_CHECKIN_STATUS = '/trae/api/v2/ug/checkin_credits/status'
+const EP_CHECKIN_CLAIM = '/trae/api/v2/ug/checkin_credits/claim'
+const EP_ENT_USAGE = '/trae/api/v2/pay/ide_user_ent_usage'
+
+// ---------------------------------------------------------------------------
+// 请求体改写
+// ---------------------------------------------------------------------------
+
+/** tool_choice 压平成字符串（上游只认字符串形式）。 */
+function normalizeToolChoice(obj) {
+  const choice = obj.tool_choice
+  if (choice === undefined || choice === null || typeof choice === 'string') return
+  if (typeof choice === 'object' && typeof choice.type === 'string') {
+    obj.tool_choice = choice.type
+    return
+  }
+  delete obj.tool_choice
+}
+
+/**
+ * 把 DSH 发来的 OpenAI 请求体改写成 Trae 认的形态。
+ *
+ * 解析失败时原样返回，让上游自己报错，不在本地吞掉。
+ */
+export function prepareChatBody(source) {
+  let body
+  try {
+    body = JSON.parse(source)
+  } catch {
+    return source
+  }
+  if (typeof body !== 'object' || body === null || Array.isArray(body)) return source
+
+  // 上游只支持流式
+  body.stream = true
+  body.function = SOLO_FUNCTION
+
+  if (Array.isArray(body.messages)) {
+    for (const message of body.messages) {
+      if (typeof message !== 'object' || message === null || Array.isArray(message)) continue
+
+      // assistant 带 tool_calls 时：OpenAI 的 function{name,arguments}
+      // 要转成上游的 function_call{name,arguments}；没有 name 的丢弃
+      // （上游要求 Name 必填，留着会让整个请求 400）。
+      if (message.role === 'assistant' && Array.isArray(message.tool_calls)) {
+        const kept = []
+        for (const call of message.tool_calls) {
+          if (typeof call !== 'object' || call === null) continue
+          if (typeof call.function === 'object' && call.function !== null) {
+            call.function_call = call.function
+            delete call.function
+          }
+          const name = call.function_call?.name
+          if (typeof name !== 'string' || name.trim() === '') continue
+          kept.push(call)
+        }
+        if (kept.length === 0) delete message.tool_calls
+        else message.tool_calls = kept
+      }
+
+      // content 字符串要包成数组形式；已经是数组就透传（多模态）
+      if (message.content === undefined || message.content === null) continue
+      if (typeof message.content === 'string') {
+        message.content = [{ type: 'text', text: message.content }]
+      }
+    }
+  }
+
+  // tools[].function.parameters 必须序列化成 JSON 字符串。
+  // 上游要的是字符串，传对象会被判无效（参照实现明确标注过这一点）。
+  if (Array.isArray(body.tools)) {
+    for (const tool of body.tools) {
+      const fn = tool?.function
+      if (typeof fn !== 'object' || fn === null) continue
+      if (typeof fn.parameters === 'object' && fn.parameters !== null) {
+        fn.parameters = JSON.stringify(fn.parameters)
+      }
+    }
+  }
+
+  // 模型名：空则用默认，且要同时写 config_name
+  let model = typeof body.model === 'string' ? body.model.trim() : ''
+  if (model === '') model = DEFAULT_MODEL
+  body.config_name = model
+  body.model = model
+
+  normalizeToolChoice(body)
+  return JSON.stringify(body)
+}
+
+// ---------------------------------------------------------------------------
+// 对话
+// ---------------------------------------------------------------------------
+
+/**
+ * 发一次对话请求，返回上游原始 Response（SSE 流）。
+ * 不做账号选择——由调用方决定用哪个凭证。
+ */
+export async function chatStream(credential, bodyJson, signal) {
+  let response
+  try {
+    response = await fetch(`${AGENT_HOST}${EP_CHAT}`, {
+      method: 'POST',
+      headers: soloHeaders(credential, true),
+      body: bodyJson,
+      signal,
+    })
+  } catch (error) {
+    return { ok: false, status: 0, kind: 'server', message: `网络错误：${error?.message ?? error}` }
+  }
+
+  if (response.ok) {
+    // shim 对 pi-ai 承诺 OpenAI 格式的流；Trae 上游是自己的命名事件格式，
+    // 必须在这里转换（详见 trae-bridge.js）。没有这层，pi-ai 等不到
+    // delta 会一直等到流空闲超时——表现为「很慢、几乎都失败」。
+    const model = (() => {
+      try { return JSON.parse(bodyJson)?.model ?? 'unknown' } catch { return 'unknown' }
+    })()
+    return { ok: true, response: bridgeTraeSseToOpenAI(response, model) }
+  }
+
+  const text = (await response.text()).slice(0, 4096)
+  return {
+    ok: false,
+    status: response.status,
+    kind: classifyError(response.status, text),
+    message: text,
+  }
+}
+
+/**
+ * 失败分类。
+ *
+ * Trae 的限流/欠费信号与 WorkBuddy 不同（业务码走 body 里的 code），
+ * 这里按状态码 + 关键词组合判断。
+ */
+/** 从响应体里取出业务码（HTTP 200 也可能带 code）。 */
+function businessCodeOf(text) {
+  const match = /"code"\s*:\s*(\d+)/.exec(text)
+  return match === null ? undefined : Number(match[1])
+}
+
+export function classifyError(status, body) {
+  const text = String(body ?? '')
+
+  // 先看业务码：Trae 用「HTTP 200 + code != 0」表达失败，数字码比
+  // 英文措辞稳定。不先看它就只能靠关键词碰运气（曾靠 "quota" 侥幸命中）。
+  const code = businessCodeOf(text)
+  if (code !== undefined && code !== 0) {
+    const known = TRAE_BIZ_CODES[code]
+    if (known !== undefined) return known.kind
+    // 未知业务码：按区间粗分。4xxx 多为额度/鉴权类，9xxx 多为限流类
+    if (code >= 4000 && code < 5000) return 'hard_credit'
+    if (code >= 9000) return 'soft_rate'
+    return 'client'
+  }
+
+  if (status === 401 || /unauthorized|token.*(invalid|expired)/i.test(text)) return 'session_dead'
+  if (status === 403 || /forbidden|permission/i.test(text)) return 'hard_credit'
+  if (status === 402 || /balance|quota|credit.*(insufficient|exhaust)/i.test(text)) return 'hard_credit'
+  if (status === 429 || /rate.?limit|too many/i.test(text)) return 'soft_rate'
+  if (status === 404) return 'not_found'
+  if (status >= 500) return 'server'
+  return 'client'
+}
+
+/** 哪些失败值得换个账号重试。 */
+export function isRetryableWithAnotherAccount(kind) {
+  return kind === 'hard_credit' || kind === 'soft_rate'
+    || kind === 'session_dead' || kind === 'server' || kind === 'not_found'
+}
+
+// ---------------------------------------------------------------------------
+// 模型目录
+// ---------------------------------------------------------------------------
+
+/**
+ * 拉取可用模型。
+ *
+ * 上游返回的是 config 列表，每项带 config_name 与 display_name；
+ * 超出列表的字段一律不编造。
+ */
+/**
+ * 目录里有、但不该展示的模型。
+ *
+ * 目录（get_detail_param）与调用通道（llm_utils_chat）并不一致，
+ * 有些条目走 SOLO 轻量通道会失败。不剔除的话，用户选中后每次对话
+ * 都失败，而不是"看不到"。名单来自受控最小对话实测（2026-09-20），
+ * 对照组 glm-5.2 / DeepSeek-V4-Flash 同时间窗正常出字。
+ */
+export const UNCALLABLE_MODELS = new Set([
+  // —— 通道不支持（4001 param is invalid）——
+  'auto',
+  'glm-5.3',
+  'muse-spark-1.2-contributor-free',
+  'n-mimo-v2.5-free',
+  'nvidia/deepseek-v4-flash-0731',
+  // —— Trae 内部的辅助 agent（不是给用户直接对话用的）——
+  'computer_use_subagent',
+  'browser_use_subagent',
+  'file_search_agent',
+  'explore_sub_agent_v2',
+  'summary',
+  // —— 未公开命名的内部模型（display_name 是占位符 "-"）——
+  'sagitta',
+  'aquila',
+])
+
+/**
+ * BYOK 模板：需要用户自带上游 API key 才有意义。
+ *
+ * 不配 key 时实测回 4023（gemini/claude/kimi/deepseek 系等 12 个）。
+ * gpt-5 / gpt-6 虽然实测有默认映射能调通，但那是上游的灰度行为，
+ * 随时可能收回——按「只保留正式模型」的口径统一剔除。
+ */
+export const BYOK_MODEL_PREFIX = 'custom_model_'
+
+/** 内部辅助 agent 的命名特征（防新增变体漏网）。 */
+const INTERNAL_AGENT_PATTERN = /(^|_)(sub_?agent|agent|summary)(_|$)/i
+
+/** 该模型是否属于不展示的类别。 */
+export function isHiddenModel(id) {
+  const text = String(id)
+  if (UNCALLABLE_MODELS.has(text)) return true
+  if (text.startsWith(BYOK_MODEL_PREFIX)) return true
+  return INTERNAL_AGENT_PATTERN.test(text)
+}
+
+export async function fetchModels(credential) {
+  const envelope = await postJson(`${AGENT_HOST}${EP_MODELS}`, {
+    headers: soloHeaders(credential, false),
+    body: {
+      function: SOLO_FUNCTION,
+      config_names: null,
+      need_prompt: false,
+      current_config_info: null,
+      poly_prompt: true,
+      mode_type: null,
+      agent_type: null,
+    },
+  })
+
+  // 实测（2026-09-20，真实凭证）：顶层就是 config_info_list（小写下划线），
+  // 没有 Result 包装、也不是大写驼峰。之前找错字段名导致
+  // 「HTTP 200 却解析出 0 个模型」。
+  const list = envelope.config_info_list
+    ?? envelope.Result?.ConfigInfoList
+    ?? envelope.ConfigInfoList
+    ?? []
+  if (!Array.isArray(list)) return []
+
+  const models = []
+  // DSH 的模型校验要求 id 唯一（重复直接抛 INVALID_CATALOG，
+  // 整个 provider 的模型都加载失败）。上游确实会给出重复的
+  // config_name（实测 DeepSeek-V4-Flash 出现两次），所以这里
+  // 必须按 id 去重：同 id 后出现的条目丢弃。
+  const seen = new Set()
+  for (const item of list) {
+    const id = String(item?.config_name ?? '').trim()
+    if (id === '') continue
+    if (isHiddenModel(id)) continue
+    if (seen.has(id)) continue
+    seen.add(id)
+    // display_name 为 "-" 或空是上游的占位符（未公开命名的内部模型，
+    // 实测 sagitta / aquila）。显示成 "- " 会让人以为数据坏了，
+    // 回退到 config_name。
+    const rawDisplay = String(item?.display_config?.display_name ?? '').trim()
+    const display = rawDisplay === '' || rawDisplay === '-' ? id : rawDisplay
+
+    // 字段名对照 dsh-connect-trae 的实测记录（2026-08-30）：
+    //   上下文 = model_detail_list[].prompt_max_tokens，或 context_window_tokens.dev
+    //   最大输出 = model_detail_list[].max_tokens
+    //   （不存在 max_input_tokens / max_output_tokens，读它们只会得到空值）
+    const detail = Array.isArray(item?.model_detail_list) && item.model_detail_list.length > 0
+      ? item.model_detail_list[0]
+      : {}
+    const ctxTokens = typeof item?.context_window_tokens === 'object' && item?.context_window_tokens !== null
+      ? item.context_window_tokens
+      : {}
+    const contextWindow = Number(detail.prompt_max_tokens ?? ctxTokens.dev ?? 0) || 128_000
+    const maxTokens = Number(detail.max_tokens ?? 0) || 16_000
+
+    models.push({
+      id,
+      // 展示名取 display_name，拿不到就用 id
+      name: display === '' ? id : display,
+      contextWindow,
+      maxTokens,
+      multimodal: false,
+      reasoning: undefined,
+      supportsToolCall: true,
+    })
+  }
+  return models
+}
+
+// ---------------------------------------------------------------------------
+// 签到与积分
+// ---------------------------------------------------------------------------
+
+/**
+ * 查签到状态（**只读**）。
+ * 比 WorkBuddy 好——那边没有只读接口，只能靠调幂等写接口判断。
+ *
+ * @returns {{checkedIn:boolean, credits:number, enabled:boolean}}
+ */
+export async function checkinStatus(credential) {
+  const envelope = await postJson(`${UG_HOST}${EP_CHECKIN_STATUS}`, {
+    headers: ugHeaders(credential),
+    body: {},
+  })
+  const data = envelope.Result ?? envelope
+  return {
+    checkedIn: data.checked_in === true,
+    // did_checked_in 表示「这台设备今天已经签过了」——本账号没签，
+    // 但同设备上的另一个账号签了。Trae 的签到按设备计数，所以这个
+    // 状态必须与 checkedIn 区分开，否则界面会显示"未签到"让人反复点。
+    deviceCheckedIn: data.did_checked_in === true,
+    credits: Number(data.credits ?? 0),
+    enabled: data.enable !== false,
+  }
+}
+
+/** 领取签到奖励。 */
+export async function checkinClaim(credential) {
+  await postJson(`${UG_HOST}${EP_CHECKIN_CLAIM}`, {
+    headers: ugHeaders(credential),
+    body: {},
+  })
+}
+
+/**
+ * 一键签到：先查状态，未签到才领（幂等，重复调用不会重复领）。
+ *
+ * @returns {{status:'done'|'already'|'failed'|'disabled', message:string}}
+ */
+export async function checkin(credential) {
+  try {
+    const status = await checkinStatus(credential)
+    if (!status.enabled) return { status: 'disabled', message: '该账号未开启签到' }
+    if (status.checkedIn) return { status: 'already', message: '今天已签到' }
+
+    await checkinClaim(credential)
+    return { status: 'done', message: '签到成功' }
+  } catch (error) {
+    // 9095「该设备今日已签到」不是失败：Trae 的签到按**设备**计数，
+    // 同一台机器上的第二个账号领取时必然遇到。显示成红色"失败"会让人
+    // 以为账号坏了，其实它只是"今天已经签过了"。
+    if (error?.code === 9095) {
+      return { status: 'already', message: '该设备今日已签到（同设备的账号共享签到）' }
+    }
+    return { status: 'failed', message: error?.message ?? String(error) }
+  }
+}
+
+/** 查剩余积分（汇总所有权益包）。 */
+export async function fetchCredits(credential) {
+  const envelope = await postJson(`${UG_HOST}${EP_ENT_USAGE}`, {
+    headers: ugHeaders(credential),
+    body: { require_usage: true },
+  })
+  // 实测：顶层直接是 usage_summary / user_entitlement_pack_list（无 Result 包装）
+  const data = envelope.Result ?? envelope
+
+  /**
+   * 口径必须与 Trae 桌面端的「用量管理」一致：**剩余 = 总量 − 已消耗**。
+   *
+   * 实测对照（2026-09-20）：
+   *   usage_summary = { total_amount: 2000, consumed_amount: 378.27 }
+   *   2000 − 378.27 = 1621.73 —— 与桌面显示完全一致。
+   *
+   * 之前按「各积分包的额度上限求和」算——但实测 quota 里放的是
+   * enable_solo_* 功能开关，根本没有数字字段，求和恒为 0；
+   * 恰好又撞上别处返回的 2000，掩盖了错误。
+   */
+  const summary = data?.usage_summary
+  const totalAmount = Number(summary?.total_amount)
+  const consumed = Number(summary?.consumed_amount)
+  if (Number.isFinite(totalAmount) && Number.isFinite(consumed)) {
+    // 保留两位小数：桌面显示也是两位（1621.73）。
+    // size = 总量，供界面悬停展示「总量 / 已用 / 剩余」。
+    return {
+      total: Math.round((totalAmount - consumed) * 100) / 100,
+      size: totalAmount,
+    }
+  }
+  return { total: 0 }
+}
+
+// ---------------------------------------------------------------------------
+// 每日任务编排
+// ---------------------------------------------------------------------------
+
+/**
+ * 对一个账号执行全部每日任务。
+ *
+ * Trae 的任务比 WorkBuddy 少：只有签到与积分查询，
+ * 没有旅行 / 连登兑换 / 抽奖那套活动体系。
+ *
+ * 顺序：先签到（可能有奖励），再查积分（拿到签到后的最新值）。
+ *
+ * @param {object} credential
+ * @returns {Promise<object>} 各任务结果
+ */
+export async function runDailyTasks(credential) {
+  const out = {
+    accountId: credential.id,
+    nickname: credential.nickname ?? credential.uid,
+  }
+
+  out.checkin = await checkin(credential)
+
+  try {
+    const credits = await fetchCredits(credential)
+    out.credits = credits.total
+  } catch (error) {
+    out.creditsError = error?.message ?? String(error)
+  }
+
+  return out
+}
+
+/**
+ * 只读：账号当前状态（签到情况 + 积分）。
+ * 不改动任何东西，供界面展示。
+ */
+export async function readStatus(credential) {
+  const out = { accountId: credential.id, nickname: credential.nickname ?? credential.uid }
+  try {
+    out.checkin = await checkinStatus(credential)
+  } catch (error) {
+    out.checkinError = error?.message ?? String(error)
+  }
+  try {
+    const credits = await fetchCredits(credential)
+    out.credits = credits.total
+    out.creditsSize = credits.size
+  } catch (error) {
+    out.creditsError = error?.message ?? String(error)
+  }
+  return out
+}