mslxdff 0.1.203 → 0.1.207

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (330) hide show
  1. package/README.md +3 -3
  2. package/bin/mslxdff.js +1 -1
  3. package/docs/cli_help_mini.md +146 -0
  4. package/package.json +38 -38
  5. package/plugins/README.md +14 -14
  6. package/plugins/prefer-model.mjs.example +14 -14
  7. package/src/anthropic/canonical-json.js +55 -0
  8. package/src/anthropic/count-tokens.js +61 -61
  9. package/src/anthropic/request-body.js +183 -126
  10. package/src/anthropic/request-shape.js +60 -0
  11. package/src/anthropic/session-id.js +58 -0
  12. package/src/anthropic/shared.js +18 -18
  13. package/src/anthropic/sse-aggregate.js +56 -56
  14. package/src/anthropic/tool-media.js +75 -0
  15. package/src/anthropic/translate.js +296 -287
  16. package/src/anthropic/web-search.js +361 -361
  17. package/src/api-kind.js +31 -31
  18. package/src/autostart.js +288 -288
  19. package/src/bench/probe.js +1 -1
  20. package/src/bench/runner.js +1 -1
  21. package/src/bench/via-probe.js +1 -1
  22. package/src/bench/via.js +144 -144
  23. package/src/bench/workbuddy-bench.js +1 -1
  24. package/src/chat/config.js +11 -11
  25. package/src/chat/cooling.js +81 -81
  26. package/src/chat/direct.js +58 -58
  27. package/src/chat/index.js +6 -6
  28. package/src/chat/spinner.js +38 -38
  29. package/src/chat/sse.js +69 -69
  30. package/src/chat/stats.js +247 -247
  31. package/src/chat/store.js +48 -48
  32. package/src/chat/tool-handlers.js +67 -67
  33. package/src/chat/tools.js +1 -1
  34. package/src/chat/upstream.js +59 -59
  35. package/src/chat-pipeline/auto-race.js +158 -158
  36. package/src/chat-pipeline/empty-turn.js +147 -147
  37. package/src/chat-pipeline/engine.js +17 -17
  38. package/src/chat-pipeline/index.js +8 -40
  39. package/src/chat-pipeline/serial-trial.js +279 -283
  40. package/src/chat-pipeline/terminal-verdict.js +132 -147
  41. package/src/chooser.js +54 -54
  42. package/src/claude-desktop/apply.js +101 -101
  43. package/src/claude-desktop/context1m.js +38 -38
  44. package/src/claude-desktop/doctor.js +146 -146
  45. package/src/claude-desktop/slots.js +152 -152
  46. package/src/cli/bootstrap.js +1 -1
  47. package/src/cli/commands/claude-desktop.js +284 -284
  48. package/src/cli/commands/daemon.js +231 -231
  49. package/src/cli/commands/model/list-live.js +22 -22
  50. package/src/cli/commands/model/list-providers.js +76 -76
  51. package/src/cli/commands/model/list-render.js +77 -77
  52. package/src/cli/commands/model/list-sort.js +52 -52
  53. package/src/cli/commands/model/list-tty.js +76 -76
  54. package/src/cli/commands/model/list.js +166 -166
  55. package/src/cli/commands/model/live-models.js +24 -24
  56. package/src/cli/commands/model/stats.js +43 -43
  57. package/src/cli/commands/model/status.js +47 -47
  58. package/src/cli/commands/model.js +51 -51
  59. package/src/cli/commands/provider/add.js +108 -108
  60. package/src/cli/commands/provider/allowlist.js +99 -99
  61. package/src/cli/commands/provider/cline-free.js +99 -99
  62. package/src/cli/commands/provider/cline-login.js +218 -218
  63. package/src/cli/commands/provider/cline-quota.js +65 -65
  64. package/src/cli/commands/provider/codearts-login.js +45 -45
  65. package/src/cli/commands/provider/config.js +119 -119
  66. package/src/cli/commands/provider/globalqwenwork-login.js +126 -126
  67. package/src/cli/commands/provider/keys-guard.js +50 -50
  68. package/src/cli/commands/provider/keys.js +161 -161
  69. package/src/cli/commands/provider/qoder-checkin.js +81 -81
  70. package/src/cli/commands/provider/qoder-login.js +102 -102
  71. package/src/cli/commands/provider/raccoon-login.js +180 -180
  72. package/src/cli/commands/provider/raccoon-password.js +162 -162
  73. package/src/cli/commands/provider/raccoon-quota.js +109 -109
  74. package/src/cli/commands/provider/traework-checkin.js +144 -144
  75. package/src/cli/commands/provider/traework-login.js +164 -164
  76. package/src/cli/commands/provider/workbuddy-login.js +153 -153
  77. package/src/cli/commands/provider/zcode-login.js +77 -77
  78. package/src/cli/commands/provider/zcode-quota.js +55 -55
  79. package/src/cli/commands/provider.js +1 -1
  80. package/src/cli/commands/stats.js +158 -149
  81. package/src/cli/commands/system.js +261 -261
  82. package/src/cli/commands/timezone.js +47 -47
  83. package/src/cli/commands/use-group.js +60 -60
  84. package/src/cli/commands/workbuddy.js +189 -189
  85. package/src/cli/format.js +141 -141
  86. package/src/cli/group-helpers.js +70 -70
  87. package/src/cli/help.js +80 -80
  88. package/src/cli/index.js +70 -70
  89. package/src/cli/interactive.js +121 -121
  90. package/src/cli/join-core.js +99 -99
  91. package/src/cli/join-wizard.js +175 -175
  92. package/src/cli/policy.js +136 -136
  93. package/src/cli/provider-row.js +126 -126
  94. package/src/cli/util.js +24 -24
  95. package/src/compat.js +53 -53
  96. package/src/events.js +38 -38
  97. package/src/free-lane.js +131 -131
  98. package/src/free-watcher.js +63 -63
  99. package/src/groups.js +262 -262
  100. package/src/logs.js +195 -195
  101. package/src/metrics.js +71 -62
  102. package/src/model-capabilities/context-lookup.js +65 -65
  103. package/src/model-capabilities/merge.js +142 -142
  104. package/src/model-trace.js +187 -187
  105. package/src/models.js +1 -1
  106. package/src/opencode-identity.js +72 -72
  107. package/src/peers.js +254 -254
  108. package/src/plugins.js +107 -107
  109. package/src/providers/AGENTS.md +49 -0
  110. package/src/providers/base.js +164 -164
  111. package/src/providers/classify.js +22 -22
  112. package/src/providers/cline/allowlist-sync.js +93 -93
  113. package/src/providers/cline/free-catalog.js +71 -71
  114. package/src/providers/cline/headers.js +37 -37
  115. package/src/providers/cline/index.js +130 -130
  116. package/src/providers/cline/models.js +150 -150
  117. package/src/providers/cline/usage.js +274 -274
  118. package/src/providers/codearts/auth-pool.js +148 -148
  119. package/src/providers/codearts/concurrency.js +34 -34
  120. package/src/providers/codearts/const.js +65 -65
  121. package/src/providers/codearts/dpop.js +95 -95
  122. package/src/providers/codearts/index.js +175 -175
  123. package/src/providers/codearts/login.js +214 -214
  124. package/src/providers/codearts/models.js +182 -182
  125. package/src/providers/codearts/sign.js +96 -96
  126. package/src/providers/codearts/sse.js +204 -204
  127. package/src/providers/codearts/stream.js +209 -209
  128. package/src/providers/codearts/sts.js +124 -124
  129. package/src/providers/codearts.js +1 -1
  130. package/src/providers/dispatcher.js +146 -146
  131. package/src/providers/generic.js +156 -156
  132. package/src/providers/globalqwenwork/account-store.js +141 -141
  133. package/src/providers/globalqwenwork/constants.js +73 -73
  134. package/src/providers/globalqwenwork/crypto.js +2 -2
  135. package/src/providers/globalqwenwork/http.js +2 -2
  136. package/src/providers/globalqwenwork/index.js +331 -331
  137. package/src/providers/globalqwenwork/payload.js +147 -147
  138. package/src/providers/globalqwenwork/rsa.js +2 -2
  139. package/src/providers/globalqwenwork/sse.js +2 -2
  140. package/src/providers/globalqwenwork/stream.js +2 -2
  141. package/src/providers/globalqwenwork.js +1 -1
  142. package/src/providers/globalworkbuddy/account-store.js +198 -198
  143. package/src/providers/globalworkbuddy/auth.js +97 -97
  144. package/src/providers/globalworkbuddy/catalog.js +138 -138
  145. package/src/providers/globalworkbuddy/constants.js +73 -73
  146. package/src/providers/globalworkbuddy/device-auth.js +120 -120
  147. package/src/providers/globalworkbuddy/errors.js +98 -98
  148. package/src/providers/globalworkbuddy/headers.js +89 -89
  149. package/src/providers/globalworkbuddy/models.js +96 -96
  150. package/src/providers/globalworkbuddy/payload.js +80 -80
  151. package/src/providers/index.js +5 -5
  152. package/src/providers/model-id.js +172 -172
  153. package/src/providers/opencode.js +19 -19
  154. package/src/providers/openrouter.js +31 -31
  155. package/src/providers/qoder/account-store.js +106 -106
  156. package/src/providers/qoder/aggregate.js +60 -60
  157. package/src/providers/qoder/baseprompt.json +591 -591
  158. package/src/providers/qoder/checkin.js +160 -160
  159. package/src/providers/qoder/constants.js +44 -44
  160. package/src/providers/qoder/encode.js +41 -41
  161. package/src/providers/qoder/fingerprint.js +38 -38
  162. package/src/providers/qoder/models.js +77 -77
  163. package/src/providers/qoder/oauth.js +84 -84
  164. package/src/providers/qoder/payload.js +162 -162
  165. package/src/providers/qoder/request.js +27 -27
  166. package/src/providers/qoder/session.js +134 -134
  167. package/src/providers/qoder/sticky.js +38 -38
  168. package/src/providers/qoder/stream.js +116 -116
  169. package/src/providers/qoder.js +1 -1
  170. package/src/providers/raccoon/account-store.js +216 -216
  171. package/src/providers/raccoon/auth.js +67 -67
  172. package/src/providers/raccoon/chat.js +228 -228
  173. package/src/providers/raccoon/const.js +180 -180
  174. package/src/providers/raccoon/credits.js +185 -185
  175. package/src/providers/raccoon/crypto.js +43 -43
  176. package/src/providers/raccoon/envelope.js +31 -31
  177. package/src/providers/raccoon/headers.js +45 -45
  178. package/src/providers/raccoon/index.js +216 -216
  179. package/src/providers/raccoon/login.js +279 -279
  180. package/src/providers/raccoon/models.js +160 -160
  181. package/src/providers/raccoon/password-login.js +102 -102
  182. package/src/providers/raccoon/qr-codewords.js +115 -115
  183. package/src/providers/raccoon/qr.js +236 -236
  184. package/src/providers/raccoon/sse.js +135 -135
  185. package/src/providers/responses-channel.js +120 -120
  186. package/src/providers/share-keys.js +53 -53
  187. package/src/providers/traework/account-store.js +234 -234
  188. package/src/providers/traework/chat.js +217 -217
  189. package/src/providers/traework/checkin.js +77 -77
  190. package/src/providers/traework/constants.js +41 -41
  191. package/src/providers/traework/errors.js +52 -52
  192. package/src/providers/traework/headers.js +79 -79
  193. package/src/providers/traework/index.js +132 -132
  194. package/src/providers/traework/market-identity.js +116 -116
  195. package/src/providers/traework/models.js +98 -98
  196. package/src/providers/traework/payload.js +86 -86
  197. package/src/providers/traework/sse.js +184 -184
  198. package/src/providers/traework/token.js +52 -52
  199. package/src/providers/traework.js +2 -2
  200. package/src/providers/workbuddy/account-store.js +186 -186
  201. package/src/providers/workbuddy/auth.js +158 -158
  202. package/src/providers/workbuddy/balance.js +86 -86
  203. package/src/providers/workbuddy/cat-travel.js +125 -125
  204. package/src/providers/workbuddy/checkin.js +75 -75
  205. package/src/providers/workbuddy/desktop-info.js +78 -78
  206. package/src/providers/workbuddy/growth-api.js +67 -67
  207. package/src/providers/workbuddy/growth-plans.js +63 -63
  208. package/src/providers/workbuddy/growth.js +181 -181
  209. package/src/providers/workbuddy/index.js +217 -217
  210. package/src/providers/workbuddy/models.js +126 -126
  211. package/src/providers/workbuddy/payload.js +141 -141
  212. package/src/providers/workbuddy/rotation-log.js +83 -83
  213. package/src/providers/workbuddy/sanitize-tools.js +61 -61
  214. package/src/providers/workbuddy/sdk-chat.js +26 -26
  215. package/src/providers/workbuddy-balance.js +78 -78
  216. package/src/providers/zcode/account-store.js +129 -129
  217. package/src/providers/zcode/auth.js +28 -28
  218. package/src/providers/zcode/chat.js +179 -179
  219. package/src/providers/zcode/const.js +54 -54
  220. package/src/providers/zcode/context-shape.js +162 -162
  221. package/src/providers/zcode/headers.js +93 -93
  222. package/src/providers/zcode/index.js +124 -124
  223. package/src/providers/zcode/models.js +65 -65
  224. package/src/providers/zcode/oauth.js +120 -120
  225. package/src/providers/zcode/quota.js +176 -176
  226. package/src/providers/zcode/sse.js +196 -196
  227. package/src/readline-compat.js +25 -25
  228. package/src/responses/translate.js +311 -311
  229. package/src/routes/AGENTS.md +37 -0
  230. package/src/routes/chat/broadband-handler.js +72 -72
  231. package/src/routes/chat/empty-turn-report.js +31 -31
  232. package/src/routes/chat/exhausted-handler.js +87 -87
  233. package/src/routes/chat/hedge-handler.js +130 -130
  234. package/src/routes/chat/index.js +1 -1
  235. package/src/routes/chat/local-handler.js +63 -63
  236. package/src/routes/chat/peer-handler.js +81 -81
  237. package/src/routes/chat.js +3 -3
  238. package/src/routes/fallback.js +88 -88
  239. package/src/routes/groups-relay.js +214 -214
  240. package/src/routes/groups.js +122 -122
  241. package/src/routes/helpers.js +93 -83
  242. package/src/routes/index.js +147 -147
  243. package/src/routes/messages-route.js +375 -365
  244. package/src/routes/peers.js +10 -2
  245. package/src/routes/relay-queue.js +181 -181
  246. package/src/routes/relay.js +60 -60
  247. package/src/routes/responses-route.js +213 -213
  248. package/src/routes/stream-body.js +84 -84
  249. package/src/routes/stream-hold.js +337 -337
  250. package/src/routes/stream-scan.js +366 -366
  251. package/src/routes.js +3 -3
  252. package/src/runtime/auto-update.js +113 -113
  253. package/src/runtime/broadband-stream.js +79 -79
  254. package/src/runtime/broadband.js +103 -103
  255. package/src/runtime/group-sync.js +81 -81
  256. package/src/runtime/keep-warm.js +120 -120
  257. package/src/runtime/lifecycle-forensics.js +116 -116
  258. package/src/runtime/lifecycle-log.js +87 -87
  259. package/src/runtime/log-stamp.js +73 -73
  260. package/src/runtime/proc-info.js +137 -137
  261. package/src/runtime/qoder-checkin.js +93 -93
  262. package/src/runtime/raccoon-checkin.js +126 -126
  263. package/src/runtime/server-lifecycle.js +189 -189
  264. package/src/runtime/traework-checkin.js +118 -118
  265. package/src/runtime/workbuddy-checkin.js +120 -120
  266. package/src/runtime/workbuddy-growth.js +133 -133
  267. package/src/state/facade.js +58 -58
  268. package/src/state/memory.js +162 -162
  269. package/src/state/merge.js +25 -25
  270. package/src/state/migrations/cline-unify.js +94 -94
  271. package/src/state/migrations.js +33 -33
  272. package/src/state/persist.js +42 -42
  273. package/src/state/provider-config.js +142 -142
  274. package/src/state/schemas/group.js +31 -31
  275. package/src/state/schemas/model.js +94 -94
  276. package/src/state/schemas/peer.js +31 -31
  277. package/src/state/schemas/port.js +10 -10
  278. package/src/state/schemas/provider.js +202 -202
  279. package/src/state/schemas/timezone.js +70 -70
  280. package/src/state/schemas/token.js +48 -48
  281. package/src/state/schemas/use-group.js +102 -102
  282. package/src/state/store.js +176 -176
  283. package/src/state.js +1 -1
  284. package/src/sync-claude-desktop.js +234 -234
  285. package/src/sync-claude.js +152 -152
  286. package/src/sync-codex.js +116 -116
  287. package/src/sync-workbuddy.js +158 -158
  288. package/src/talk-full-redact.js +129 -129
  289. package/src/talk-full.js +131 -131
  290. package/src/talk-log.js +257 -257
  291. package/src/text-tokens.js +52 -52
  292. package/src/thinking-effort.js +67 -67
  293. package/src/time.js +106 -106
  294. package/src/timeline.js +26 -26
  295. package/src/transport/index.js +1 -1
  296. package/src/upstream-engine/index.js +74 -74
  297. package/src/upstream-engine/mode.js +15 -15
  298. package/src/upstream-engine/sdk/attempt.js +221 -221
  299. package/src/upstream-engine/sdk/chat.js +28 -28
  300. package/src/upstream-engine/sdk/convert.js +192 -192
  301. package/src/upstream-engine/sdk/diagnose.js +62 -62
  302. package/src/upstream-engine/sdk/dispatch.js +55 -55
  303. package/src/upstream-engine/sdk/responses.js +176 -176
  304. package/src/upstream-engine/sdk/sse.js +166 -166
  305. package/src/upstream-responses.js +305 -305
  306. package/src/usage/cache-fields.js +58 -0
  307. package/src/usage/record.js +109 -106
  308. package/src/usage/report.js +209 -195
  309. package/src/v6pool/claim.js +157 -157
  310. package/src/v6pool/config.js +65 -65
  311. package/src/v6pool/detect.js +130 -130
  312. package/src/v6pool/index.js +152 -152
  313. package/src/v6pool/prefix.js +140 -140
  314. package/src/v6pool/probe.js +40 -40
  315. package/src/v6pool/report.js +104 -104
  316. package/src/v6pool/rotate.js +263 -263
  317. package/src/warp-pool/config.js +74 -74
  318. package/src/warp-pool/health.js +106 -106
  319. package/src/warp-pool/identity.js +141 -141
  320. package/src/warp-pool/index.js +358 -358
  321. package/src/warp-pool/keys.js +29 -29
  322. package/src/warp-pool/nodes.js +95 -95
  323. package/src/warp-pool/ports.js +160 -160
  324. package/src/warp-pool/register.js +113 -113
  325. package/src/warp-pool/report.js +232 -232
  326. package/src/warp-pool/revive.js +168 -168
  327. package/src/warp-pool/rotate.js +214 -214
  328. package/src/warp-pool/strategy.js +43 -43
  329. package/src/warp-pool/verdict.js +69 -69
  330. package/src/warp-pool/wireproxy.js +162 -162
@@ -1,41 +1,41 @@
1
- // traework SOLO 上游技术常量(照抄 traework2api internal/upstream/constants.go,禁止改值)。
2
- export const AGENT_HOST = "https://trae-api-cn.mchost.guru";
3
- export const UG_HOST = "https://api.trae.cn";
4
- export const OAUTH_HOST = "https://api.trae.com.cn";
5
- export const CONSOLE_HOST = "https://www.trae.cn";
6
- export const CLIENT_ID = "en1oxy7wnw8j9n";
7
- export const APP_ID = "6eefa01c-1036-4c7e-9ca5-d891f63bfcd8";
8
- export const IDE_VERSION = "0.1.43";
9
- export const IDE_VERSION_CODE = "20260716";
10
- export const DEVICE_BRAND = "83DG";
11
- export const OS_VERSION = "Windows 11 Pro";
12
- export const FUNCTION = "solo_work_lite";
13
- export const TRAE_UA = "Trae/0.1.43";
14
- export const DEFAULT_MODEL = "glm-5.2";
15
- export const EP_CHAT = "/api/agent/v3/llm_utils_chat";
16
- export const EP_MODELS = "/api/ide/v1/get_detail_param";
17
- export const EP_EXCHANGE = "/cloudide/api/v3/trae/oauth/ExchangeToken";
18
- export const EP_USER_INFO = "/cloudide/api/v3/trae/GetUserInfo";
19
- export const EP_CHECKIN_STATUS = "/trae/api/v2/ug/checkin_credits/status";
20
- export const EP_CHECKIN_CLAIM = "/trae/api/v2/ug/checkin_credits/claim";
21
- export const EP_ENT_USAGE = "/trae/api/v2/pay/ide_user_ent_usage";
22
-
23
- // ── 市场/UG 侧线上契约(status/claim/ent_usage 共用)────────────────────────
24
- // 出处:抓包复刻自 NextAgentX/trae-workbuddy-switch `modules/trae/credits.rs:49-99`,
25
- // 不是从客户端源码抄的 —— 该仓自述「把 `x-market-client-id`/`x-market-user-id`/`x-lscbd-aid`
26
- // 与整串 `VSCode 1.107.1 (TRAE SOLO CN)` 在 Trae 安装目录 5141 个文件(含 .exe/.asar)里全扫一遍,
27
- // **一个字面量都搜不到**」,即无法从客户端侧核对,只能以「线上实测能用」为准。
28
- // ⚠ 因此这些值是**线上契约**,不要"顺手改成看起来更对的";要改必须先拿到抓包证据。
29
- // 上面那段纪律原文见该仓 credits.rs:53-76。
30
- export const MARKET_UA_PRODUCT = "TRAE SOLO CN"; // UA 括号里的产品名
31
- export const MARKET_VSCODE_VERSION = "1.107.1"; // 真机 resources/app/package.json 的 version
32
- export const MARKET_UA = `VSCode ${MARKET_VSCODE_VERSION} (${MARKET_UA_PRODUCT})`;
33
- export const MARKET_AID = "787976"; // x-lscbd-aid(活动/埋点侧应用标识)
34
- export const MARKET_SDK_TYPE = "3"; // x-lgw-req-sdk-type
35
- export const MARKET_PACKAGE_TYPE = "stable_cn"; // package-type,CN 稳定版通道
36
- export const MARKET_USER_REGION = "CN"; // x-user-region,产品级 CN 通道,不随账号 Region 变
37
- export const MARKET_APP_VERSION = "0.1.45"; // app-version(≠ 上面的 IDE_VERSION 0.1.43:两条通道各自取值,别合)
38
-
39
- // 市场伪设备身份的派生算法版本(抄 device.rs:35 DEVICE_GEN)。
40
- // auth 文档里的 `market.gen` 缺失或小于此值 → 视为旧算法产物,就地重建(自愈)。
41
- export const MARKET_DEVICE_GEN = 2;
1
+ // traework SOLO 上游技术常量(照抄 traework2api internal/upstream/constants.go,禁止改值)。
2
+ export const AGENT_HOST = "https://trae-api-cn.mchost.guru";
3
+ export const UG_HOST = "https://api.trae.cn";
4
+ export const OAUTH_HOST = "https://api.trae.com.cn";
5
+ export const CONSOLE_HOST = "https://www.trae.cn";
6
+ export const CLIENT_ID = "en1oxy7wnw8j9n";
7
+ export const APP_ID = "6eefa01c-1036-4c7e-9ca5-d891f63bfcd8";
8
+ export const IDE_VERSION = "0.1.43";
9
+ export const IDE_VERSION_CODE = "20260716";
10
+ export const DEVICE_BRAND = "83DG";
11
+ export const OS_VERSION = "Windows 11 Pro";
12
+ export const FUNCTION = "solo_work_lite";
13
+ export const TRAE_UA = "Trae/0.1.43";
14
+ export const DEFAULT_MODEL = "glm-5.2";
15
+ export const EP_CHAT = "/api/agent/v3/llm_utils_chat";
16
+ export const EP_MODELS = "/api/ide/v1/get_detail_param";
17
+ export const EP_EXCHANGE = "/cloudide/api/v3/trae/oauth/ExchangeToken";
18
+ export const EP_USER_INFO = "/cloudide/api/v3/trae/GetUserInfo";
19
+ export const EP_CHECKIN_STATUS = "/trae/api/v2/ug/checkin_credits/status";
20
+ export const EP_CHECKIN_CLAIM = "/trae/api/v2/ug/checkin_credits/claim";
21
+ export const EP_ENT_USAGE = "/trae/api/v2/pay/ide_user_ent_usage";
22
+
23
+ // ── 市场/UG 侧线上契约(status/claim/ent_usage 共用)────────────────────────
24
+ // 出处:抓包复刻自 NextAgentX/trae-workbuddy-switch `modules/trae/credits.rs:49-99`,
25
+ // 不是从客户端源码抄的 —— 该仓自述「把 `x-market-client-id`/`x-market-user-id`/`x-lscbd-aid`
26
+ // 与整串 `VSCode 1.107.1 (TRAE SOLO CN)` 在 Trae 安装目录 5141 个文件(含 .exe/.asar)里全扫一遍,
27
+ // **一个字面量都搜不到**」,即无法从客户端侧核对,只能以「线上实测能用」为准。
28
+ // ⚠ 因此这些值是**线上契约**,不要"顺手改成看起来更对的";要改必须先拿到抓包证据。
29
+ // 上面那段纪律原文见该仓 credits.rs:53-76。
30
+ export const MARKET_UA_PRODUCT = "TRAE SOLO CN"; // UA 括号里的产品名
31
+ export const MARKET_VSCODE_VERSION = "1.107.1"; // 真机 resources/app/package.json 的 version
32
+ export const MARKET_UA = `VSCode ${MARKET_VSCODE_VERSION} (${MARKET_UA_PRODUCT})`;
33
+ export const MARKET_AID = "787976"; // x-lscbd-aid(活动/埋点侧应用标识)
34
+ export const MARKET_SDK_TYPE = "3"; // x-lgw-req-sdk-type
35
+ export const MARKET_PACKAGE_TYPE = "stable_cn"; // package-type,CN 稳定版通道
36
+ export const MARKET_USER_REGION = "CN"; // x-user-region,产品级 CN 通道,不随账号 Region 变
37
+ export const MARKET_APP_VERSION = "0.1.45"; // app-version(≠ 上面的 IDE_VERSION 0.1.43:两条通道各自取值,别合)
38
+
39
+ // 市场伪设备身份的派生算法版本(抄 device.rs:35 DEVICE_GEN)。
40
+ // auth 文档里的 `market.gen` 缺失或小于此值 → 视为旧算法产物,就地重建(自愈)。
41
+ export const MARKET_DEVICE_GEN = 2;
@@ -1,52 +1,52 @@
1
- // SOLO 上游错误分类(照抄 traework2api internal/upstream/client.go Classify)。
2
- export const ErrKind = {
3
- NONE: "none",
4
- PLAN_LIMIT: "plan_limit",
5
- SOFT_RATE: "soft_rate",
6
- SESSION_DEAD: "session_dead",
7
- NOT_FOUND: "not_found",
8
- SERVER: "server",
9
- CLIENT: "client",
10
- };
11
-
12
- // 长冷却时长:plan_limit → 12h。
13
- export const PLAN_COOLDOWN_MS = 12 * 60 * 60 * 1000;
14
- export const SOFT_COOLDOWN_MS = 60 * 1000;
15
-
16
- const SESSION_DEAD_MARKERS = ["login", "token 失效", "token invalid", "session", "unauthorized", "401"];
17
-
18
- export class UpstreamError extends Error {
19
- constructor(kind, status, msg) {
20
- super(`upstream ${kind} (http ${status}): ${msg}`);
21
- this.kind = kind;
22
- this.status = status;
23
- }
24
- }
25
-
26
- // 流内业务错误(event:error)。
27
- export class SOLOStreamError extends Error {
28
- constructor(code, msg) {
29
- super(`solo error code=${code} msg=${msg}`);
30
- this.code = code;
31
- this.msg = msg;
32
- }
33
- kind() { return this.code === 1005 ? ErrKind.PLAN_LIMIT : ErrKind.CLIENT; }
34
- }
35
-
36
- export function classify(status, body) {
37
- const txt = String(body || "");
38
- const lower = txt.toLowerCase();
39
- if (txt.includes('"code":1005') || (txt.includes("1005") && lower.includes("plan"))) return ErrKind.PLAN_LIMIT;
40
- if (status === 401) return ErrKind.SESSION_DEAD;
41
- if (status === 429) return ErrKind.SOFT_RATE;
42
- if (status === 404) return ErrKind.NOT_FOUND;
43
- if (status >= 500) return ErrKind.SERVER;
44
- if (status >= 400) return ErrKind.CLIENT;
45
- return ErrKind.NONE;
46
- }
47
-
48
- export function isSessionDead(status, body) {
49
- if (status === 401) return true;
50
- const lower = String(body || "").toLowerCase();
51
- return SESSION_DEAD_MARKERS.some((m) => lower.includes(m.toLowerCase()));
52
- }
1
+ // SOLO 上游错误分类(照抄 traework2api internal/upstream/client.go Classify)。
2
+ export const ErrKind = {
3
+ NONE: "none",
4
+ PLAN_LIMIT: "plan_limit",
5
+ SOFT_RATE: "soft_rate",
6
+ SESSION_DEAD: "session_dead",
7
+ NOT_FOUND: "not_found",
8
+ SERVER: "server",
9
+ CLIENT: "client",
10
+ };
11
+
12
+ // 长冷却时长:plan_limit → 12h。
13
+ export const PLAN_COOLDOWN_MS = 12 * 60 * 60 * 1000;
14
+ export const SOFT_COOLDOWN_MS = 60 * 1000;
15
+
16
+ const SESSION_DEAD_MARKERS = ["login", "token 失效", "token invalid", "session", "unauthorized", "401"];
17
+
18
+ export class UpstreamError extends Error {
19
+ constructor(kind, status, msg) {
20
+ super(`upstream ${kind} (http ${status}): ${msg}`);
21
+ this.kind = kind;
22
+ this.status = status;
23
+ }
24
+ }
25
+
26
+ // 流内业务错误(event:error)。
27
+ export class SOLOStreamError extends Error {
28
+ constructor(code, msg) {
29
+ super(`solo error code=${code} msg=${msg}`);
30
+ this.code = code;
31
+ this.msg = msg;
32
+ }
33
+ kind() { return this.code === 1005 ? ErrKind.PLAN_LIMIT : ErrKind.CLIENT; }
34
+ }
35
+
36
+ export function classify(status, body) {
37
+ const txt = String(body || "");
38
+ const lower = txt.toLowerCase();
39
+ if (txt.includes('"code":1005') || (txt.includes("1005") && lower.includes("plan"))) return ErrKind.PLAN_LIMIT;
40
+ if (status === 401) return ErrKind.SESSION_DEAD;
41
+ if (status === 429) return ErrKind.SOFT_RATE;
42
+ if (status === 404) return ErrKind.NOT_FOUND;
43
+ if (status >= 500) return ErrKind.SERVER;
44
+ if (status >= 400) return ErrKind.CLIENT;
45
+ return ErrKind.NONE;
46
+ }
47
+
48
+ export function isSessionDead(status, body) {
49
+ if (status === 401) return true;
50
+ const lower = String(body || "").toLowerCase();
51
+ return SESSION_DEAD_MARKERS.some((m) => lower.includes(m.toLowerCase()));
52
+ }
@@ -1,79 +1,79 @@
1
- // SOLO 三类请求头(照抄 traework2api internal/upstream/headers.go)。纯函数,可测。
2
- import { APP_ID, DEVICE_BRAND, IDE_VERSION, IDE_VERSION_CODE, OS_VERSION, TRAE_UA, MARKET_UA, MARKET_VSCODE_VERSION, MARKET_USER_REGION, MARKET_SDK_TYPE, MARKET_PACKAGE_TYPE, MARKET_AID, MARKET_APP_VERSION } from "./constants.js";
3
- import { uuid } from "../../compat.js";
4
-
5
- // 对话/模型表(Cloud-IDE-JWT 鉴权 + IDE 指纹)。
6
- export function soloHeaders(cred = {}, stream = true) {
7
- const at = cred?.accessToken || "";
8
- const h = {
9
- "Content-Type": "application/json",
10
- Accept: stream ? "text/event-stream" : "application/json",
11
- "User-Agent": TRAE_UA,
12
- Authorization: `Cloud-IDE-JWT ${at}`,
13
- "X-Cloudide-Token": at,
14
- "X-Ide-Token": at,
15
- "X-App-Id": APP_ID,
16
- "X-App-Version": "default",
17
- "X-Ide-Version": IDE_VERSION,
18
- "X-Ide-Version-Code": IDE_VERSION_CODE,
19
- "X-App-Version-Code": IDE_VERSION_CODE,
20
- "X-Ide-Version-Type": "stable",
21
- "X-Device-Type": "windows",
22
- "X-OS-Version": OS_VERSION,
23
- "X-Device-Brand": DEVICE_BRAND,
24
- "Request-Traffic-Type": "prod",
25
- };
26
- if (cred?.uid) h["X-Uid"] = cred.uid;
27
- if (cred?.machineId) h["X-Machine-Id"] = cred.machineId;
28
- if (cred?.deviceId) h["X-Device-Id"] = cred.deviceId;
29
- return h;
30
- }
31
-
32
- // 签到/积分/额度(api.trae.cn 的 UG 与 pay 系接口)。
33
- // ⚠ 这套头是**线上契约**:三件设备身份(x-device-id / x-market-user-id / vscode-sessionid)缺一或形态不对,
34
- // `checkin_credits/claim` 就回 `code=9004 下单参数不正确`(2026-10-15 现网取证)。
35
- // 逐字对齐 NextAgentX/trae-workbuddy-switch `modules/trae/credits.rs:116-147`,取值出处与该仓纪律见 constants.js 注释。
36
- export function ugHeaders(cred = {}) {
37
- const m = cred?.market || {};
38
- const at = String(cred?.accessToken || "");
39
- return {
40
- accept: "*/*",
41
- "accept-language": "zh-CN",
42
- "content-type": "application/json",
43
- "user-agent": MARKET_UA,
44
- // 防双前缀:传进来的可能已带 `Cloud-IDE-JWT `(口径同参考实现 jwt.rs:53-61)
45
- authorization: at.startsWith("Cloud-IDE-JWT ") ? at : `Cloud-IDE-JWT ${at}`,
46
- "x-market-client-id": `VSCode ${MARKET_VSCODE_VERSION}`,
47
- "x-market-user-id": String(m.marketUserId || ""),
48
- "x-user-region": MARKET_USER_REGION,
49
- "x-device-id": String(m.deviceId || ""),
50
- "x-lgw-req-sdk-type": MARKET_SDK_TYPE,
51
- "package-type": MARKET_PACKAGE_TYPE,
52
- "x-request-id": uuid(),
53
- "x-lscbd-aid": MARKET_AID,
54
- "x-lscbd-platform": ugPlatformTag(),
55
- "app-version": MARKET_APP_VERSION,
56
- "x-tt-trace-id": ugTraceId(),
57
- "vscode-sessionid": String(m.sessionId || ""),
58
- "sec-fetch-dest": "empty",
59
- "sec-fetch-mode": "no-cors",
60
- "sec-fetch-site": "none",
61
- };
62
- }
63
-
64
- /** `x-lscbd-platform`:按真实平台报,不自称 Windows(参考实现 credits.rs:150-160 的理由:减少跨平台特征矛盾)。 */
65
- export function ugPlatformTag(p = process.platform) {
66
- if (p === "win32") return "windows";
67
- if (p === "darwin") return "macos";
68
- return "linux";
69
- }
70
-
71
- /** `x-tt-trace-id` = `00-<uuid 去连字符前 16 位>-01`(参考实现 credits.rs:162-166)。 */
72
- export function ugTraceId(rnd = uuid) {
73
- return `00-${String(rnd()).replace(/-/g, "").slice(0, 16)}-01`;
74
- }
75
-
76
- // ExchangeToken / GetUserInfo(无签名,仅 UA)。
77
- export function oauthHeaders() {
78
- return { "Content-Type": "application/json", Accept: "application/json", "User-Agent": TRAE_UA };
79
- }
1
+ // SOLO 三类请求头(照抄 traework2api internal/upstream/headers.go)。纯函数,可测。
2
+ import { APP_ID, DEVICE_BRAND, IDE_VERSION, IDE_VERSION_CODE, OS_VERSION, TRAE_UA, MARKET_UA, MARKET_VSCODE_VERSION, MARKET_USER_REGION, MARKET_SDK_TYPE, MARKET_PACKAGE_TYPE, MARKET_AID, MARKET_APP_VERSION } from "./constants.js";
3
+ import { uuid } from "../../compat.js";
4
+
5
+ // 对话/模型表(Cloud-IDE-JWT 鉴权 + IDE 指纹)。
6
+ export function soloHeaders(cred = {}, stream = true) {
7
+ const at = cred?.accessToken || "";
8
+ const h = {
9
+ "Content-Type": "application/json",
10
+ Accept: stream ? "text/event-stream" : "application/json",
11
+ "User-Agent": TRAE_UA,
12
+ Authorization: `Cloud-IDE-JWT ${at}`,
13
+ "X-Cloudide-Token": at,
14
+ "X-Ide-Token": at,
15
+ "X-App-Id": APP_ID,
16
+ "X-App-Version": "default",
17
+ "X-Ide-Version": IDE_VERSION,
18
+ "X-Ide-Version-Code": IDE_VERSION_CODE,
19
+ "X-App-Version-Code": IDE_VERSION_CODE,
20
+ "X-Ide-Version-Type": "stable",
21
+ "X-Device-Type": "windows",
22
+ "X-OS-Version": OS_VERSION,
23
+ "X-Device-Brand": DEVICE_BRAND,
24
+ "Request-Traffic-Type": "prod",
25
+ };
26
+ if (cred?.uid) h["X-Uid"] = cred.uid;
27
+ if (cred?.machineId) h["X-Machine-Id"] = cred.machineId;
28
+ if (cred?.deviceId) h["X-Device-Id"] = cred.deviceId;
29
+ return h;
30
+ }
31
+
32
+ // 签到/积分/额度(api.trae.cn 的 UG 与 pay 系接口)。
33
+ // ⚠ 这套头是**线上契约**:三件设备身份(x-device-id / x-market-user-id / vscode-sessionid)缺一或形态不对,
34
+ // `checkin_credits/claim` 就回 `code=9004 下单参数不正确`(2026-10-15 现网取证)。
35
+ // 逐字对齐 NextAgentX/trae-workbuddy-switch `modules/trae/credits.rs:116-147`,取值出处与该仓纪律见 constants.js 注释。
36
+ export function ugHeaders(cred = {}) {
37
+ const m = cred?.market || {};
38
+ const at = String(cred?.accessToken || "");
39
+ return {
40
+ accept: "*/*",
41
+ "accept-language": "zh-CN",
42
+ "content-type": "application/json",
43
+ "user-agent": MARKET_UA,
44
+ // 防双前缀:传进来的可能已带 `Cloud-IDE-JWT `(口径同参考实现 jwt.rs:53-61)
45
+ authorization: at.startsWith("Cloud-IDE-JWT ") ? at : `Cloud-IDE-JWT ${at}`,
46
+ "x-market-client-id": `VSCode ${MARKET_VSCODE_VERSION}`,
47
+ "x-market-user-id": String(m.marketUserId || ""),
48
+ "x-user-region": MARKET_USER_REGION,
49
+ "x-device-id": String(m.deviceId || ""),
50
+ "x-lgw-req-sdk-type": MARKET_SDK_TYPE,
51
+ "package-type": MARKET_PACKAGE_TYPE,
52
+ "x-request-id": uuid(),
53
+ "x-lscbd-aid": MARKET_AID,
54
+ "x-lscbd-platform": ugPlatformTag(),
55
+ "app-version": MARKET_APP_VERSION,
56
+ "x-tt-trace-id": ugTraceId(),
57
+ "vscode-sessionid": String(m.sessionId || ""),
58
+ "sec-fetch-dest": "empty",
59
+ "sec-fetch-mode": "no-cors",
60
+ "sec-fetch-site": "none",
61
+ };
62
+ }
63
+
64
+ /** `x-lscbd-platform`:按真实平台报,不自称 Windows(参考实现 credits.rs:150-160 的理由:减少跨平台特征矛盾)。 */
65
+ export function ugPlatformTag(p = process.platform) {
66
+ if (p === "win32") return "windows";
67
+ if (p === "darwin") return "macos";
68
+ return "linux";
69
+ }
70
+
71
+ /** `x-tt-trace-id` = `00-<uuid 去连字符前 16 位>-01`(参考实现 credits.rs:162-166)。 */
72
+ export function ugTraceId(rnd = uuid) {
73
+ return `00-${String(rnd()).replace(/-/g, "").slice(0, 16)}-01`;
74
+ }
75
+
76
+ // ExchangeToken / GetUserInfo(无签名,仅 UA)。
77
+ export function oauthHeaders() {
78
+ return { "Content-Type": "application/json", Accept: "application/json", "User-Agent": TRAE_UA };
79
+ }
@@ -1,132 +1,132 @@
1
- // traework provider 门面:薄聚合(读 state keys/auths,无则扫 auth 目录)。
2
- import { createKeyRing } from "../keyring.js";
3
- import { loadProviderKeys, loadProviderAuths, loadProviderBaseUrl } from "../../state.js";
4
- import { compatFetch, getUndici } from "../../compat.js";
5
- import { envInt, createAgent } from "../base.js";
6
- import { AGENT_HOST } from "./constants.js";
7
- import { listAccountDocs, withTraeworkIdentity } from "./account-store.js";
8
- import { createChatService } from "./chat.js";
9
- import { createModelsService } from "./models.js";
10
-
11
- const { UndiciFetch } = getUndici();
12
-
13
- function resolveBaseUrl(baseUrl) {
14
- if (baseUrl) return String(baseUrl).trim().replace(/\/+$/, "");
15
- try {
16
- const env = loadProviderBaseUrl("traework");
17
- if (env) return env;
18
- } catch {}
19
- return AGENT_HOST;
20
- }
21
-
22
- export function createTraeworkProvider({
23
- id = "traework",
24
- baseUrl,
25
- apiKeys,
26
- apiKey,
27
- auths,
28
- connectTimeoutMs = Number(process.env.MSLXDFF_TRAEWORK_TIMEOUT_MS) || 30_000,
29
- cooldownMs = envInt("MSLXDFF_TRAEWORK_COOLDOWN_MS", 30_000),
30
- fetchImpl,
31
- file,
32
- clock = Date.now,
33
- } = {}) {
34
- if (id !== "traework") throw new Error(`traework provider id mismatch: ${id}`);
35
- const resolvedBase = resolveBaseUrl(baseUrl);
36
- if (!fetchImpl) fetchImpl = UndiciFetch || compatFetch;
37
-
38
- const keysFromState = (() => { try { return loadProviderKeys(id, file ? { file } : {}); } catch { return []; } })();
39
- const authsFromState = (() => { try { return loadProviderAuths(id, file ? { file } : {}); } catch { return []; } })();
40
- const keys = (() => {
41
- const list = [
42
- ...(Array.isArray(apiKeys) ? apiKeys : [apiKeys].filter(Boolean)),
43
- apiKey,
44
- ...(apiKeys === undefined && apiKey === undefined ? keysFromState : []),
45
- ].filter((k) => typeof k === "string" && k.trim().length);
46
- return [...new Set(list.map((k) => k.trim()))];
47
- })();
48
- let authList = Array.isArray(auths) && auths.length ? [...auths] : [...authsFromState];
49
-
50
- if (!authList.length && !keys.length) {
51
- try {
52
- for (const { uid, doc } of listAccountDocs()) {
53
- const at = doc?.auth?.accessToken || doc?.accessToken;
54
- if (at && !keys.includes(at)) keys.push(at);
55
- authList.push({
56
- uid,
57
- domain: doc?.auth?.domain || "trae.cn",
58
- apiHost: doc?.auth?.apiHost || "",
59
- machineId: doc?.auth?.machineId || "",
60
- deviceId: doc?.auth?.deviceId || "",
61
- enterpriseId: doc?.account?.enterpriseId || "",
62
- refreshToken: doc?.auth?.refreshToken || "",
63
- expiresAt: doc?.auth?.expiresAt || 0,
64
- nickname: doc?.account?.nickname || "",
65
- });
66
- }
67
- } catch {}
68
- }
69
- // 无论 authList 来自 state 还是刚扫的目录,都回 auth 文档补一次设备身份:
70
- // state 的 auths 行被 `normalizeAuths` 四字段白名单剥掉了 machineId/deviceId(见 account-store.js 同段注释),
71
- // 不补则 SOLO 对话永远不发 X-Machine-Id / X-Device-Id。
72
- authList = withTraeworkIdentity(authList);
73
- // auths 行自带 refreshToken 但无 accessToken(login 后 keys 与 auths 平行);
74
- // 补齐触发条件不能只看长度相等:2026-10-15 实测 `-provider traework <未接线子命令>` 掉进 keys 兜底分支,
75
- // 把 keys[0] 覆盖成 5 字符垃圾串,长度仍与 authList 相等 → 旧判据放行,真拿垃圾串当 accessToken 鉴权。
76
- // 故加形态判据:不像凭据(短、含空白)的槽位一律回 auth 目录重取。
77
- // (判空白逐字符 trim 比对,不用 \s 正则——本仓 Edit 通道会把单反斜杠吃成两个,留可读写法防再踩)
78
- const hasBlank = (k) => [...k].some((c) => c.trim() === "");
79
- const looksLikeCredential = (k) => typeof k === "string" && k.trim().length >= 20 && !hasBlank(k);
80
- const needRefill = authList.length && (keys.length !== authList.length || keys.some((k, i) => authList[i]?.uid && !looksLikeCredential(k)));
81
- if (needRefill) {
82
- try {
83
- const docs = new Map(listAccountDocs().map((d) => [String(d.uid), d.doc]));
84
- authList.forEach((a, i) => {
85
- const at = docs.get(String(a.uid))?.auth?.accessToken;
86
- if (at) keys[i] = at; // 按位序覆盖:keys 与 authList 是平行数组(account-store 同语义),去重判据会留垃圾槽位
87
- });
88
- } catch {}
89
- }
90
-
91
- const ring = createKeyRing(keys, { cooldownMs, now: clock });
92
- let agent = null; let dispatcher = null;
93
- try {
94
- const a = createAgent({
95
- keepAliveTimeout: envInt("MSLXDFF_TRAEWORK_KEEPALIVE_TIMEOUT", 30_000),
96
- keepAliveMaxTimeout: envInt("MSLXDFF_TRAEWORK_KEEPALIVE_MAX_TIMEOUT", 60_000),
97
- connections: envInt("MSLXDFF_TRAEWORK_KEEPALIVE_CONNECTIONS", 20),
98
- });
99
- agent = a.agent; dispatcher = a.dispatcher;
100
- } catch {}
101
-
102
- function getKey() { return ring.next() || keys[0] || ""; }
103
- function getAuth(k) {
104
- const idx = keys.indexOf(k);
105
- if (idx >= 0 && authList[idx]) return authList[idx];
106
- if (authList.length) return authList[0];
107
- return { uid: "", accessToken: k || "", refreshToken: "", expiresAt: 0, domain: "trae.cn", apiHost: "", machineId: "", deviceId: "" };
108
- }
109
-
110
- const chatSvc = createChatService({ id, baseUrl: resolvedBase, keys, authList, ring, fetchImpl, dispatcher, connectTimeoutMs, file, clock, cooldownMs });
111
- const modelsSvc = createModelsService({ id, baseUrl: resolvedBase, fetchImpl, getKey, getAuth, clock });
112
-
113
- async function chat(body, opts = {}) { return chatSvc.runChat(body, ring, opts); }
114
-
115
- async function chatWithKeys(body, keysOverride) {
116
- const tmpKeys = [...keysOverride].filter((k) => typeof k === "string" && k.trim().length).map((k) => k.trim());
117
- const tmpAuth = authList[0] || getAuth("");
118
- const tmpAuthList = tmpKeys.map(() => tmpAuth);
119
- const tmpRing = createKeyRing(tmpKeys, { cooldownMs, now: clock });
120
- const tmpSvc = createChatService({ id, baseUrl: resolvedBase, keys: tmpKeys, authList: tmpAuthList, ring: tmpRing, fetchImpl, dispatcher: null, connectTimeoutMs, file, clock, cooldownMs });
121
- return tmpSvc.runChat(body, tmpRing);
122
- }
123
-
124
- async function close() { if (agent && typeof agent.close === "function") { try { await agent.close(); } catch {} } }
125
-
126
- return { id, chat, chatWithKeys, listModels: modelsSvc.listModels, preheat: modelsSvc.preheat, close, agent, keyRing: ring, baseUrl: resolvedBase };
127
- }
128
-
129
- export { mapModel, normalizeModelName, staticModels } from "./models.js";
130
- export { soloHeaders, ugHeaders, oauthHeaders } from "./headers.js";
131
- export { prepareBody } from "./payload.js";
132
- export { classify, ErrKind } from "./errors.js";
1
+ // traework provider 门面:薄聚合(读 state keys/auths,无则扫 auth 目录)。
2
+ import { createKeyRing } from "../keyring.js";
3
+ import { loadProviderKeys, loadProviderAuths, loadProviderBaseUrl } from "../../state.js";
4
+ import { compatFetch, getUndici } from "../../compat.js";
5
+ import { envInt, createAgent } from "../base.js";
6
+ import { AGENT_HOST } from "./constants.js";
7
+ import { listAccountDocs, withTraeworkIdentity } from "./account-store.js";
8
+ import { createChatService } from "./chat.js";
9
+ import { createModelsService } from "./models.js";
10
+
11
+ const { UndiciFetch } = getUndici();
12
+
13
+ function resolveBaseUrl(baseUrl) {
14
+ if (baseUrl) return String(baseUrl).trim().replace(/\/+$/, "");
15
+ try {
16
+ const env = loadProviderBaseUrl("traework");
17
+ if (env) return env;
18
+ } catch {}
19
+ return AGENT_HOST;
20
+ }
21
+
22
+ export function createTraeworkProvider({
23
+ id = "traework",
24
+ baseUrl,
25
+ apiKeys,
26
+ apiKey,
27
+ auths,
28
+ connectTimeoutMs = Number(process.env.MSLXDFF_TRAEWORK_TIMEOUT_MS) || 30_000,
29
+ cooldownMs = envInt("MSLXDFF_TRAEWORK_COOLDOWN_MS", 30_000),
30
+ fetchImpl,
31
+ file,
32
+ clock = Date.now,
33
+ } = {}) {
34
+ if (id !== "traework") throw new Error(`traework provider id mismatch: ${id}`);
35
+ const resolvedBase = resolveBaseUrl(baseUrl);
36
+ if (!fetchImpl) fetchImpl = UndiciFetch || compatFetch;
37
+
38
+ const keysFromState = (() => { try { return loadProviderKeys(id, file ? { file } : {}); } catch { return []; } })();
39
+ const authsFromState = (() => { try { return loadProviderAuths(id, file ? { file } : {}); } catch { return []; } })();
40
+ const keys = (() => {
41
+ const list = [
42
+ ...(Array.isArray(apiKeys) ? apiKeys : [apiKeys].filter(Boolean)),
43
+ apiKey,
44
+ ...(apiKeys === undefined && apiKey === undefined ? keysFromState : []),
45
+ ].filter((k) => typeof k === "string" && k.trim().length);
46
+ return [...new Set(list.map((k) => k.trim()))];
47
+ })();
48
+ let authList = Array.isArray(auths) && auths.length ? [...auths] : [...authsFromState];
49
+
50
+ if (!authList.length && !keys.length) {
51
+ try {
52
+ for (const { uid, doc } of listAccountDocs()) {
53
+ const at = doc?.auth?.accessToken || doc?.accessToken;
54
+ if (at && !keys.includes(at)) keys.push(at);
55
+ authList.push({
56
+ uid,
57
+ domain: doc?.auth?.domain || "trae.cn",
58
+ apiHost: doc?.auth?.apiHost || "",
59
+ machineId: doc?.auth?.machineId || "",
60
+ deviceId: doc?.auth?.deviceId || "",
61
+ enterpriseId: doc?.account?.enterpriseId || "",
62
+ refreshToken: doc?.auth?.refreshToken || "",
63
+ expiresAt: doc?.auth?.expiresAt || 0,
64
+ nickname: doc?.account?.nickname || "",
65
+ });
66
+ }
67
+ } catch {}
68
+ }
69
+ // 无论 authList 来自 state 还是刚扫的目录,都回 auth 文档补一次设备身份:
70
+ // state 的 auths 行被 `normalizeAuths` 四字段白名单剥掉了 machineId/deviceId(见 account-store.js 同段注释),
71
+ // 不补则 SOLO 对话永远不发 X-Machine-Id / X-Device-Id。
72
+ authList = withTraeworkIdentity(authList);
73
+ // auths 行自带 refreshToken 但无 accessToken(login 后 keys 与 auths 平行);
74
+ // 补齐触发条件不能只看长度相等:2026-10-15 实测 `-provider traework <未接线子命令>` 掉进 keys 兜底分支,
75
+ // 把 keys[0] 覆盖成 5 字符垃圾串,长度仍与 authList 相等 → 旧判据放行,真拿垃圾串当 accessToken 鉴权。
76
+ // 故加形态判据:不像凭据(短、含空白)的槽位一律回 auth 目录重取。
77
+ // (判空白逐字符 trim 比对,不用 \s 正则——本仓 Edit 通道会把单反斜杠吃成两个,留可读写法防再踩)
78
+ const hasBlank = (k) => [...k].some((c) => c.trim() === "");
79
+ const looksLikeCredential = (k) => typeof k === "string" && k.trim().length >= 20 && !hasBlank(k);
80
+ const needRefill = authList.length && (keys.length !== authList.length || keys.some((k, i) => authList[i]?.uid && !looksLikeCredential(k)));
81
+ if (needRefill) {
82
+ try {
83
+ const docs = new Map(listAccountDocs().map((d) => [String(d.uid), d.doc]));
84
+ authList.forEach((a, i) => {
85
+ const at = docs.get(String(a.uid))?.auth?.accessToken;
86
+ if (at) keys[i] = at; // 按位序覆盖:keys 与 authList 是平行数组(account-store 同语义),去重判据会留垃圾槽位
87
+ });
88
+ } catch {}
89
+ }
90
+
91
+ const ring = createKeyRing(keys, { cooldownMs, now: clock });
92
+ let agent = null; let dispatcher = null;
93
+ try {
94
+ const a = createAgent({
95
+ keepAliveTimeout: envInt("MSLXDFF_TRAEWORK_KEEPALIVE_TIMEOUT", 30_000),
96
+ keepAliveMaxTimeout: envInt("MSLXDFF_TRAEWORK_KEEPALIVE_MAX_TIMEOUT", 60_000),
97
+ connections: envInt("MSLXDFF_TRAEWORK_KEEPALIVE_CONNECTIONS", 20),
98
+ });
99
+ agent = a.agent; dispatcher = a.dispatcher;
100
+ } catch {}
101
+
102
+ function getKey() { return ring.next() || keys[0] || ""; }
103
+ function getAuth(k) {
104
+ const idx = keys.indexOf(k);
105
+ if (idx >= 0 && authList[idx]) return authList[idx];
106
+ if (authList.length) return authList[0];
107
+ return { uid: "", accessToken: k || "", refreshToken: "", expiresAt: 0, domain: "trae.cn", apiHost: "", machineId: "", deviceId: "" };
108
+ }
109
+
110
+ const chatSvc = createChatService({ id, baseUrl: resolvedBase, keys, authList, ring, fetchImpl, dispatcher, connectTimeoutMs, file, clock, cooldownMs });
111
+ const modelsSvc = createModelsService({ id, baseUrl: resolvedBase, fetchImpl, getKey, getAuth, clock });
112
+
113
+ async function chat(body, opts = {}) { return chatSvc.runChat(body, ring, opts); }
114
+
115
+ async function chatWithKeys(body, keysOverride) {
116
+ const tmpKeys = [...keysOverride].filter((k) => typeof k === "string" && k.trim().length).map((k) => k.trim());
117
+ const tmpAuth = authList[0] || getAuth("");
118
+ const tmpAuthList = tmpKeys.map(() => tmpAuth);
119
+ const tmpRing = createKeyRing(tmpKeys, { cooldownMs, now: clock });
120
+ const tmpSvc = createChatService({ id, baseUrl: resolvedBase, keys: tmpKeys, authList: tmpAuthList, ring: tmpRing, fetchImpl, dispatcher: null, connectTimeoutMs, file, clock, cooldownMs });
121
+ return tmpSvc.runChat(body, tmpRing);
122
+ }
123
+
124
+ async function close() { if (agent && typeof agent.close === "function") { try { await agent.close(); } catch {} } }
125
+
126
+ return { id, chat, chatWithKeys, listModels: modelsSvc.listModels, preheat: modelsSvc.preheat, close, agent, keyRing: ring, baseUrl: resolvedBase };
127
+ }
128
+
129
+ export { mapModel, normalizeModelName, staticModels } from "./models.js";
130
+ export { soloHeaders, ugHeaders, oauthHeaders } from "./headers.js";
131
+ export { prepareBody } from "./payload.js";
132
+ export { classify, ErrKind } from "./errors.js";