mslxdff 0.1.203 → 0.1.208

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 (332) hide show
  1. package/README.md +3 -3
  2. package/bin/mslxdff.js +1 -1
  3. package/docs/cli_help_mini.md +147 -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/v6-pool.js +123 -0
  85. package/src/cli/commands/workbuddy.js +189 -189
  86. package/src/cli/format.js +141 -141
  87. package/src/cli/group-helpers.js +70 -70
  88. package/src/cli/help.js +80 -80
  89. package/src/cli/index.js +73 -70
  90. package/src/cli/interactive.js +121 -121
  91. package/src/cli/join-core.js +99 -99
  92. package/src/cli/join-wizard.js +175 -175
  93. package/src/cli/policy.js +136 -136
  94. package/src/cli/provider-row.js +126 -126
  95. package/src/cli/util.js +24 -24
  96. package/src/compat.js +53 -53
  97. package/src/events.js +38 -38
  98. package/src/free-lane.js +131 -131
  99. package/src/free-watcher.js +63 -63
  100. package/src/groups.js +262 -262
  101. package/src/logs.js +195 -195
  102. package/src/metrics.js +71 -62
  103. package/src/model-capabilities/context-lookup.js +65 -65
  104. package/src/model-capabilities/merge.js +142 -142
  105. package/src/model-trace.js +187 -187
  106. package/src/models.js +1 -1
  107. package/src/opencode-identity.js +72 -72
  108. package/src/peers.js +254 -254
  109. package/src/plugins.js +107 -107
  110. package/src/providers/AGENTS.md +49 -0
  111. package/src/providers/base.js +164 -164
  112. package/src/providers/classify.js +22 -22
  113. package/src/providers/cline/allowlist-sync.js +93 -93
  114. package/src/providers/cline/free-catalog.js +71 -71
  115. package/src/providers/cline/headers.js +37 -37
  116. package/src/providers/cline/index.js +130 -130
  117. package/src/providers/cline/models.js +150 -150
  118. package/src/providers/cline/usage.js +274 -274
  119. package/src/providers/codearts/auth-pool.js +148 -148
  120. package/src/providers/codearts/concurrency.js +34 -34
  121. package/src/providers/codearts/const.js +65 -65
  122. package/src/providers/codearts/dpop.js +95 -95
  123. package/src/providers/codearts/index.js +175 -175
  124. package/src/providers/codearts/login.js +214 -214
  125. package/src/providers/codearts/models.js +182 -182
  126. package/src/providers/codearts/sign.js +96 -96
  127. package/src/providers/codearts/sse.js +204 -204
  128. package/src/providers/codearts/stream.js +209 -209
  129. package/src/providers/codearts/sts.js +124 -124
  130. package/src/providers/codearts.js +1 -1
  131. package/src/providers/dispatcher.js +146 -146
  132. package/src/providers/generic.js +156 -156
  133. package/src/providers/globalqwenwork/account-store.js +141 -141
  134. package/src/providers/globalqwenwork/constants.js +73 -73
  135. package/src/providers/globalqwenwork/crypto.js +2 -2
  136. package/src/providers/globalqwenwork/http.js +2 -2
  137. package/src/providers/globalqwenwork/index.js +331 -331
  138. package/src/providers/globalqwenwork/payload.js +147 -147
  139. package/src/providers/globalqwenwork/rsa.js +2 -2
  140. package/src/providers/globalqwenwork/sse.js +2 -2
  141. package/src/providers/globalqwenwork/stream.js +2 -2
  142. package/src/providers/globalqwenwork.js +1 -1
  143. package/src/providers/globalworkbuddy/account-store.js +198 -198
  144. package/src/providers/globalworkbuddy/auth.js +97 -97
  145. package/src/providers/globalworkbuddy/catalog.js +138 -138
  146. package/src/providers/globalworkbuddy/constants.js +73 -73
  147. package/src/providers/globalworkbuddy/device-auth.js +120 -120
  148. package/src/providers/globalworkbuddy/errors.js +98 -98
  149. package/src/providers/globalworkbuddy/headers.js +89 -89
  150. package/src/providers/globalworkbuddy/models.js +96 -96
  151. package/src/providers/globalworkbuddy/payload.js +80 -80
  152. package/src/providers/index.js +5 -5
  153. package/src/providers/model-id.js +172 -172
  154. package/src/providers/opencode.js +19 -19
  155. package/src/providers/openrouter.js +31 -31
  156. package/src/providers/qoder/account-store.js +106 -106
  157. package/src/providers/qoder/aggregate.js +60 -60
  158. package/src/providers/qoder/baseprompt.json +591 -591
  159. package/src/providers/qoder/checkin.js +160 -160
  160. package/src/providers/qoder/constants.js +44 -44
  161. package/src/providers/qoder/encode.js +41 -41
  162. package/src/providers/qoder/fingerprint.js +38 -38
  163. package/src/providers/qoder/models.js +77 -77
  164. package/src/providers/qoder/oauth.js +84 -84
  165. package/src/providers/qoder/payload.js +162 -162
  166. package/src/providers/qoder/request.js +27 -27
  167. package/src/providers/qoder/session.js +134 -134
  168. package/src/providers/qoder/sticky.js +38 -38
  169. package/src/providers/qoder/stream.js +116 -116
  170. package/src/providers/qoder.js +1 -1
  171. package/src/providers/raccoon/account-store.js +216 -216
  172. package/src/providers/raccoon/auth.js +67 -67
  173. package/src/providers/raccoon/chat.js +228 -228
  174. package/src/providers/raccoon/const.js +180 -180
  175. package/src/providers/raccoon/credits.js +185 -185
  176. package/src/providers/raccoon/crypto.js +43 -43
  177. package/src/providers/raccoon/envelope.js +31 -31
  178. package/src/providers/raccoon/headers.js +45 -45
  179. package/src/providers/raccoon/index.js +216 -216
  180. package/src/providers/raccoon/login.js +279 -279
  181. package/src/providers/raccoon/models.js +160 -160
  182. package/src/providers/raccoon/password-login.js +102 -102
  183. package/src/providers/raccoon/qr-codewords.js +115 -115
  184. package/src/providers/raccoon/qr.js +236 -236
  185. package/src/providers/raccoon/sse.js +135 -135
  186. package/src/providers/responses-channel.js +120 -120
  187. package/src/providers/share-keys.js +53 -53
  188. package/src/providers/traework/account-store.js +234 -234
  189. package/src/providers/traework/chat.js +217 -217
  190. package/src/providers/traework/checkin.js +77 -77
  191. package/src/providers/traework/constants.js +41 -41
  192. package/src/providers/traework/errors.js +52 -52
  193. package/src/providers/traework/headers.js +79 -79
  194. package/src/providers/traework/index.js +132 -132
  195. package/src/providers/traework/market-identity.js +116 -116
  196. package/src/providers/traework/models.js +98 -98
  197. package/src/providers/traework/payload.js +86 -86
  198. package/src/providers/traework/sse.js +184 -184
  199. package/src/providers/traework/token.js +52 -52
  200. package/src/providers/traework.js +2 -2
  201. package/src/providers/workbuddy/account-store.js +186 -186
  202. package/src/providers/workbuddy/auth.js +158 -158
  203. package/src/providers/workbuddy/balance.js +86 -86
  204. package/src/providers/workbuddy/cat-travel.js +125 -125
  205. package/src/providers/workbuddy/checkin.js +75 -75
  206. package/src/providers/workbuddy/desktop-info.js +78 -78
  207. package/src/providers/workbuddy/growth-api.js +67 -67
  208. package/src/providers/workbuddy/growth-plans.js +63 -63
  209. package/src/providers/workbuddy/growth.js +181 -181
  210. package/src/providers/workbuddy/index.js +217 -217
  211. package/src/providers/workbuddy/models.js +126 -126
  212. package/src/providers/workbuddy/payload.js +141 -141
  213. package/src/providers/workbuddy/rotation-log.js +83 -83
  214. package/src/providers/workbuddy/sanitize-tools.js +61 -61
  215. package/src/providers/workbuddy/sdk-chat.js +26 -26
  216. package/src/providers/workbuddy-balance.js +78 -78
  217. package/src/providers/zcode/account-store.js +129 -129
  218. package/src/providers/zcode/auth.js +28 -28
  219. package/src/providers/zcode/chat.js +179 -179
  220. package/src/providers/zcode/const.js +54 -54
  221. package/src/providers/zcode/context-shape.js +162 -162
  222. package/src/providers/zcode/headers.js +93 -93
  223. package/src/providers/zcode/index.js +124 -124
  224. package/src/providers/zcode/models.js +65 -65
  225. package/src/providers/zcode/oauth.js +120 -120
  226. package/src/providers/zcode/quota.js +176 -176
  227. package/src/providers/zcode/sse.js +196 -196
  228. package/src/readline-compat.js +25 -25
  229. package/src/responses/translate.js +311 -311
  230. package/src/routes/AGENTS.md +37 -0
  231. package/src/routes/chat/broadband-handler.js +72 -72
  232. package/src/routes/chat/empty-turn-report.js +31 -31
  233. package/src/routes/chat/exhausted-handler.js +87 -87
  234. package/src/routes/chat/hedge-handler.js +130 -130
  235. package/src/routes/chat/index.js +1 -1
  236. package/src/routes/chat/local-handler.js +63 -63
  237. package/src/routes/chat/peer-handler.js +81 -81
  238. package/src/routes/chat.js +3 -3
  239. package/src/routes/fallback.js +88 -88
  240. package/src/routes/groups-relay.js +214 -214
  241. package/src/routes/groups.js +122 -122
  242. package/src/routes/helpers.js +93 -83
  243. package/src/routes/index.js +147 -147
  244. package/src/routes/messages-route.js +375 -365
  245. package/src/routes/peers.js +10 -2
  246. package/src/routes/relay-queue.js +181 -181
  247. package/src/routes/relay.js +60 -60
  248. package/src/routes/responses-route.js +213 -213
  249. package/src/routes/stream-body.js +84 -84
  250. package/src/routes/stream-hold.js +337 -337
  251. package/src/routes/stream-scan.js +366 -366
  252. package/src/routes.js +3 -3
  253. package/src/runtime/auto-update.js +113 -113
  254. package/src/runtime/broadband-stream.js +79 -79
  255. package/src/runtime/broadband.js +103 -103
  256. package/src/runtime/group-sync.js +81 -81
  257. package/src/runtime/keep-warm.js +120 -120
  258. package/src/runtime/lifecycle-forensics.js +116 -116
  259. package/src/runtime/lifecycle-log.js +87 -87
  260. package/src/runtime/log-stamp.js +73 -73
  261. package/src/runtime/proc-info.js +137 -137
  262. package/src/runtime/qoder-checkin.js +93 -93
  263. package/src/runtime/raccoon-checkin.js +126 -126
  264. package/src/runtime/server-lifecycle.js +189 -189
  265. package/src/runtime/traework-checkin.js +118 -118
  266. package/src/runtime/workbuddy-checkin.js +120 -120
  267. package/src/runtime/workbuddy-growth.js +133 -133
  268. package/src/state/facade.js +58 -58
  269. package/src/state/memory.js +162 -162
  270. package/src/state/merge.js +25 -25
  271. package/src/state/migrations/cline-unify.js +94 -94
  272. package/src/state/migrations.js +33 -33
  273. package/src/state/persist.js +42 -42
  274. package/src/state/provider-config.js +142 -142
  275. package/src/state/schemas/group.js +31 -31
  276. package/src/state/schemas/model.js +94 -94
  277. package/src/state/schemas/peer.js +31 -31
  278. package/src/state/schemas/port.js +10 -10
  279. package/src/state/schemas/provider.js +202 -202
  280. package/src/state/schemas/timezone.js +70 -70
  281. package/src/state/schemas/token.js +48 -48
  282. package/src/state/schemas/use-group.js +102 -102
  283. package/src/state/store.js +176 -176
  284. package/src/state.js +1 -1
  285. package/src/sync-claude-desktop.js +234 -234
  286. package/src/sync-claude.js +152 -152
  287. package/src/sync-codex.js +116 -116
  288. package/src/sync-workbuddy.js +158 -158
  289. package/src/talk-full-redact.js +129 -129
  290. package/src/talk-full.js +131 -131
  291. package/src/talk-log.js +257 -257
  292. package/src/text-tokens.js +52 -52
  293. package/src/thinking-effort.js +67 -67
  294. package/src/time.js +106 -106
  295. package/src/timeline.js +26 -26
  296. package/src/transport/index.js +1 -1
  297. package/src/upstream-engine/index.js +74 -74
  298. package/src/upstream-engine/mode.js +15 -15
  299. package/src/upstream-engine/sdk/attempt.js +221 -221
  300. package/src/upstream-engine/sdk/chat.js +28 -28
  301. package/src/upstream-engine/sdk/convert.js +192 -192
  302. package/src/upstream-engine/sdk/diagnose.js +62 -62
  303. package/src/upstream-engine/sdk/dispatch.js +55 -55
  304. package/src/upstream-engine/sdk/responses.js +176 -176
  305. package/src/upstream-engine/sdk/sse.js +166 -166
  306. package/src/upstream-responses.js +305 -305
  307. package/src/usage/cache-fields.js +58 -0
  308. package/src/usage/record.js +109 -106
  309. package/src/usage/report.js +209 -195
  310. package/src/v6pool/claim.js +157 -157
  311. package/src/v6pool/config.js +65 -65
  312. package/src/v6pool/detect.js +130 -130
  313. package/src/v6pool/index.js +152 -152
  314. package/src/v6pool/prefix.js +140 -140
  315. package/src/v6pool/probe.js +40 -40
  316. package/src/v6pool/report.js +104 -104
  317. package/src/v6pool/rotate.js +263 -263
  318. package/src/v6pool/selftest.js +154 -0
  319. package/src/warp-pool/config.js +74 -74
  320. package/src/warp-pool/health.js +106 -106
  321. package/src/warp-pool/identity.js +141 -141
  322. package/src/warp-pool/index.js +358 -358
  323. package/src/warp-pool/keys.js +29 -29
  324. package/src/warp-pool/nodes.js +95 -95
  325. package/src/warp-pool/ports.js +160 -160
  326. package/src/warp-pool/register.js +113 -113
  327. package/src/warp-pool/report.js +232 -232
  328. package/src/warp-pool/revive.js +168 -168
  329. package/src/warp-pool/rotate.js +214 -214
  330. package/src/warp-pool/strategy.js +43 -43
  331. package/src/warp-pool/verdict.js +69 -69
  332. 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";