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.
- package/README.md +3 -3
- package/bin/mslxdff.js +1 -1
- package/docs/cli_help_mini.md +147 -0
- package/package.json +38 -38
- package/plugins/README.md +14 -14
- package/plugins/prefer-model.mjs.example +14 -14
- package/src/anthropic/canonical-json.js +55 -0
- package/src/anthropic/count-tokens.js +61 -61
- package/src/anthropic/request-body.js +183 -126
- package/src/anthropic/request-shape.js +60 -0
- package/src/anthropic/session-id.js +58 -0
- package/src/anthropic/shared.js +18 -18
- package/src/anthropic/sse-aggregate.js +56 -56
- package/src/anthropic/tool-media.js +75 -0
- package/src/anthropic/translate.js +296 -287
- package/src/anthropic/web-search.js +361 -361
- package/src/api-kind.js +31 -31
- package/src/autostart.js +288 -288
- package/src/bench/probe.js +1 -1
- package/src/bench/runner.js +1 -1
- package/src/bench/via-probe.js +1 -1
- package/src/bench/via.js +144 -144
- package/src/bench/workbuddy-bench.js +1 -1
- package/src/chat/config.js +11 -11
- package/src/chat/cooling.js +81 -81
- package/src/chat/direct.js +58 -58
- package/src/chat/index.js +6 -6
- package/src/chat/spinner.js +38 -38
- package/src/chat/sse.js +69 -69
- package/src/chat/stats.js +247 -247
- package/src/chat/store.js +48 -48
- package/src/chat/tool-handlers.js +67 -67
- package/src/chat/tools.js +1 -1
- package/src/chat/upstream.js +59 -59
- package/src/chat-pipeline/auto-race.js +158 -158
- package/src/chat-pipeline/empty-turn.js +147 -147
- package/src/chat-pipeline/engine.js +17 -17
- package/src/chat-pipeline/index.js +8 -40
- package/src/chat-pipeline/serial-trial.js +279 -283
- package/src/chat-pipeline/terminal-verdict.js +132 -147
- package/src/chooser.js +54 -54
- package/src/claude-desktop/apply.js +101 -101
- package/src/claude-desktop/context1m.js +38 -38
- package/src/claude-desktop/doctor.js +146 -146
- package/src/claude-desktop/slots.js +152 -152
- package/src/cli/bootstrap.js +1 -1
- package/src/cli/commands/claude-desktop.js +284 -284
- package/src/cli/commands/daemon.js +231 -231
- package/src/cli/commands/model/list-live.js +22 -22
- package/src/cli/commands/model/list-providers.js +76 -76
- package/src/cli/commands/model/list-render.js +77 -77
- package/src/cli/commands/model/list-sort.js +52 -52
- package/src/cli/commands/model/list-tty.js +76 -76
- package/src/cli/commands/model/list.js +166 -166
- package/src/cli/commands/model/live-models.js +24 -24
- package/src/cli/commands/model/stats.js +43 -43
- package/src/cli/commands/model/status.js +47 -47
- package/src/cli/commands/model.js +51 -51
- package/src/cli/commands/provider/add.js +108 -108
- package/src/cli/commands/provider/allowlist.js +99 -99
- package/src/cli/commands/provider/cline-free.js +99 -99
- package/src/cli/commands/provider/cline-login.js +218 -218
- package/src/cli/commands/provider/cline-quota.js +65 -65
- package/src/cli/commands/provider/codearts-login.js +45 -45
- package/src/cli/commands/provider/config.js +119 -119
- package/src/cli/commands/provider/globalqwenwork-login.js +126 -126
- package/src/cli/commands/provider/keys-guard.js +50 -50
- package/src/cli/commands/provider/keys.js +161 -161
- package/src/cli/commands/provider/qoder-checkin.js +81 -81
- package/src/cli/commands/provider/qoder-login.js +102 -102
- package/src/cli/commands/provider/raccoon-login.js +180 -180
- package/src/cli/commands/provider/raccoon-password.js +162 -162
- package/src/cli/commands/provider/raccoon-quota.js +109 -109
- package/src/cli/commands/provider/traework-checkin.js +144 -144
- package/src/cli/commands/provider/traework-login.js +164 -164
- package/src/cli/commands/provider/workbuddy-login.js +153 -153
- package/src/cli/commands/provider/zcode-login.js +77 -77
- package/src/cli/commands/provider/zcode-quota.js +55 -55
- package/src/cli/commands/provider.js +1 -1
- package/src/cli/commands/stats.js +158 -149
- package/src/cli/commands/system.js +261 -261
- package/src/cli/commands/timezone.js +47 -47
- package/src/cli/commands/use-group.js +60 -60
- package/src/cli/commands/v6-pool.js +123 -0
- package/src/cli/commands/workbuddy.js +189 -189
- package/src/cli/format.js +141 -141
- package/src/cli/group-helpers.js +70 -70
- package/src/cli/help.js +80 -80
- package/src/cli/index.js +73 -70
- package/src/cli/interactive.js +121 -121
- package/src/cli/join-core.js +99 -99
- package/src/cli/join-wizard.js +175 -175
- package/src/cli/policy.js +136 -136
- package/src/cli/provider-row.js +126 -126
- package/src/cli/util.js +24 -24
- package/src/compat.js +53 -53
- package/src/events.js +38 -38
- package/src/free-lane.js +131 -131
- package/src/free-watcher.js +63 -63
- package/src/groups.js +262 -262
- package/src/logs.js +195 -195
- package/src/metrics.js +71 -62
- package/src/model-capabilities/context-lookup.js +65 -65
- package/src/model-capabilities/merge.js +142 -142
- package/src/model-trace.js +187 -187
- package/src/models.js +1 -1
- package/src/opencode-identity.js +72 -72
- package/src/peers.js +254 -254
- package/src/plugins.js +107 -107
- package/src/providers/AGENTS.md +49 -0
- package/src/providers/base.js +164 -164
- package/src/providers/classify.js +22 -22
- package/src/providers/cline/allowlist-sync.js +93 -93
- package/src/providers/cline/free-catalog.js +71 -71
- package/src/providers/cline/headers.js +37 -37
- package/src/providers/cline/index.js +130 -130
- package/src/providers/cline/models.js +150 -150
- package/src/providers/cline/usage.js +274 -274
- package/src/providers/codearts/auth-pool.js +148 -148
- package/src/providers/codearts/concurrency.js +34 -34
- package/src/providers/codearts/const.js +65 -65
- package/src/providers/codearts/dpop.js +95 -95
- package/src/providers/codearts/index.js +175 -175
- package/src/providers/codearts/login.js +214 -214
- package/src/providers/codearts/models.js +182 -182
- package/src/providers/codearts/sign.js +96 -96
- package/src/providers/codearts/sse.js +204 -204
- package/src/providers/codearts/stream.js +209 -209
- package/src/providers/codearts/sts.js +124 -124
- package/src/providers/codearts.js +1 -1
- package/src/providers/dispatcher.js +146 -146
- package/src/providers/generic.js +156 -156
- package/src/providers/globalqwenwork/account-store.js +141 -141
- package/src/providers/globalqwenwork/constants.js +73 -73
- package/src/providers/globalqwenwork/crypto.js +2 -2
- package/src/providers/globalqwenwork/http.js +2 -2
- package/src/providers/globalqwenwork/index.js +331 -331
- package/src/providers/globalqwenwork/payload.js +147 -147
- package/src/providers/globalqwenwork/rsa.js +2 -2
- package/src/providers/globalqwenwork/sse.js +2 -2
- package/src/providers/globalqwenwork/stream.js +2 -2
- package/src/providers/globalqwenwork.js +1 -1
- package/src/providers/globalworkbuddy/account-store.js +198 -198
- package/src/providers/globalworkbuddy/auth.js +97 -97
- package/src/providers/globalworkbuddy/catalog.js +138 -138
- package/src/providers/globalworkbuddy/constants.js +73 -73
- package/src/providers/globalworkbuddy/device-auth.js +120 -120
- package/src/providers/globalworkbuddy/errors.js +98 -98
- package/src/providers/globalworkbuddy/headers.js +89 -89
- package/src/providers/globalworkbuddy/models.js +96 -96
- package/src/providers/globalworkbuddy/payload.js +80 -80
- package/src/providers/index.js +5 -5
- package/src/providers/model-id.js +172 -172
- package/src/providers/opencode.js +19 -19
- package/src/providers/openrouter.js +31 -31
- package/src/providers/qoder/account-store.js +106 -106
- package/src/providers/qoder/aggregate.js +60 -60
- package/src/providers/qoder/baseprompt.json +591 -591
- package/src/providers/qoder/checkin.js +160 -160
- package/src/providers/qoder/constants.js +44 -44
- package/src/providers/qoder/encode.js +41 -41
- package/src/providers/qoder/fingerprint.js +38 -38
- package/src/providers/qoder/models.js +77 -77
- package/src/providers/qoder/oauth.js +84 -84
- package/src/providers/qoder/payload.js +162 -162
- package/src/providers/qoder/request.js +27 -27
- package/src/providers/qoder/session.js +134 -134
- package/src/providers/qoder/sticky.js +38 -38
- package/src/providers/qoder/stream.js +116 -116
- package/src/providers/qoder.js +1 -1
- package/src/providers/raccoon/account-store.js +216 -216
- package/src/providers/raccoon/auth.js +67 -67
- package/src/providers/raccoon/chat.js +228 -228
- package/src/providers/raccoon/const.js +180 -180
- package/src/providers/raccoon/credits.js +185 -185
- package/src/providers/raccoon/crypto.js +43 -43
- package/src/providers/raccoon/envelope.js +31 -31
- package/src/providers/raccoon/headers.js +45 -45
- package/src/providers/raccoon/index.js +216 -216
- package/src/providers/raccoon/login.js +279 -279
- package/src/providers/raccoon/models.js +160 -160
- package/src/providers/raccoon/password-login.js +102 -102
- package/src/providers/raccoon/qr-codewords.js +115 -115
- package/src/providers/raccoon/qr.js +236 -236
- package/src/providers/raccoon/sse.js +135 -135
- package/src/providers/responses-channel.js +120 -120
- package/src/providers/share-keys.js +53 -53
- package/src/providers/traework/account-store.js +234 -234
- package/src/providers/traework/chat.js +217 -217
- package/src/providers/traework/checkin.js +77 -77
- package/src/providers/traework/constants.js +41 -41
- package/src/providers/traework/errors.js +52 -52
- package/src/providers/traework/headers.js +79 -79
- package/src/providers/traework/index.js +132 -132
- package/src/providers/traework/market-identity.js +116 -116
- package/src/providers/traework/models.js +98 -98
- package/src/providers/traework/payload.js +86 -86
- package/src/providers/traework/sse.js +184 -184
- package/src/providers/traework/token.js +52 -52
- package/src/providers/traework.js +2 -2
- package/src/providers/workbuddy/account-store.js +186 -186
- package/src/providers/workbuddy/auth.js +158 -158
- package/src/providers/workbuddy/balance.js +86 -86
- package/src/providers/workbuddy/cat-travel.js +125 -125
- package/src/providers/workbuddy/checkin.js +75 -75
- package/src/providers/workbuddy/desktop-info.js +78 -78
- package/src/providers/workbuddy/growth-api.js +67 -67
- package/src/providers/workbuddy/growth-plans.js +63 -63
- package/src/providers/workbuddy/growth.js +181 -181
- package/src/providers/workbuddy/index.js +217 -217
- package/src/providers/workbuddy/models.js +126 -126
- package/src/providers/workbuddy/payload.js +141 -141
- package/src/providers/workbuddy/rotation-log.js +83 -83
- package/src/providers/workbuddy/sanitize-tools.js +61 -61
- package/src/providers/workbuddy/sdk-chat.js +26 -26
- package/src/providers/workbuddy-balance.js +78 -78
- package/src/providers/zcode/account-store.js +129 -129
- package/src/providers/zcode/auth.js +28 -28
- package/src/providers/zcode/chat.js +179 -179
- package/src/providers/zcode/const.js +54 -54
- package/src/providers/zcode/context-shape.js +162 -162
- package/src/providers/zcode/headers.js +93 -93
- package/src/providers/zcode/index.js +124 -124
- package/src/providers/zcode/models.js +65 -65
- package/src/providers/zcode/oauth.js +120 -120
- package/src/providers/zcode/quota.js +176 -176
- package/src/providers/zcode/sse.js +196 -196
- package/src/readline-compat.js +25 -25
- package/src/responses/translate.js +311 -311
- package/src/routes/AGENTS.md +37 -0
- package/src/routes/chat/broadband-handler.js +72 -72
- package/src/routes/chat/empty-turn-report.js +31 -31
- package/src/routes/chat/exhausted-handler.js +87 -87
- package/src/routes/chat/hedge-handler.js +130 -130
- package/src/routes/chat/index.js +1 -1
- package/src/routes/chat/local-handler.js +63 -63
- package/src/routes/chat/peer-handler.js +81 -81
- package/src/routes/chat.js +3 -3
- package/src/routes/fallback.js +88 -88
- package/src/routes/groups-relay.js +214 -214
- package/src/routes/groups.js +122 -122
- package/src/routes/helpers.js +93 -83
- package/src/routes/index.js +147 -147
- package/src/routes/messages-route.js +375 -365
- package/src/routes/peers.js +10 -2
- package/src/routes/relay-queue.js +181 -181
- package/src/routes/relay.js +60 -60
- package/src/routes/responses-route.js +213 -213
- package/src/routes/stream-body.js +84 -84
- package/src/routes/stream-hold.js +337 -337
- package/src/routes/stream-scan.js +366 -366
- package/src/routes.js +3 -3
- package/src/runtime/auto-update.js +113 -113
- package/src/runtime/broadband-stream.js +79 -79
- package/src/runtime/broadband.js +103 -103
- package/src/runtime/group-sync.js +81 -81
- package/src/runtime/keep-warm.js +120 -120
- package/src/runtime/lifecycle-forensics.js +116 -116
- package/src/runtime/lifecycle-log.js +87 -87
- package/src/runtime/log-stamp.js +73 -73
- package/src/runtime/proc-info.js +137 -137
- package/src/runtime/qoder-checkin.js +93 -93
- package/src/runtime/raccoon-checkin.js +126 -126
- package/src/runtime/server-lifecycle.js +189 -189
- package/src/runtime/traework-checkin.js +118 -118
- package/src/runtime/workbuddy-checkin.js +120 -120
- package/src/runtime/workbuddy-growth.js +133 -133
- package/src/state/facade.js +58 -58
- package/src/state/memory.js +162 -162
- package/src/state/merge.js +25 -25
- package/src/state/migrations/cline-unify.js +94 -94
- package/src/state/migrations.js +33 -33
- package/src/state/persist.js +42 -42
- package/src/state/provider-config.js +142 -142
- package/src/state/schemas/group.js +31 -31
- package/src/state/schemas/model.js +94 -94
- package/src/state/schemas/peer.js +31 -31
- package/src/state/schemas/port.js +10 -10
- package/src/state/schemas/provider.js +202 -202
- package/src/state/schemas/timezone.js +70 -70
- package/src/state/schemas/token.js +48 -48
- package/src/state/schemas/use-group.js +102 -102
- package/src/state/store.js +176 -176
- package/src/state.js +1 -1
- package/src/sync-claude-desktop.js +234 -234
- package/src/sync-claude.js +152 -152
- package/src/sync-codex.js +116 -116
- package/src/sync-workbuddy.js +158 -158
- package/src/talk-full-redact.js +129 -129
- package/src/talk-full.js +131 -131
- package/src/talk-log.js +257 -257
- package/src/text-tokens.js +52 -52
- package/src/thinking-effort.js +67 -67
- package/src/time.js +106 -106
- package/src/timeline.js +26 -26
- package/src/transport/index.js +1 -1
- package/src/upstream-engine/index.js +74 -74
- package/src/upstream-engine/mode.js +15 -15
- package/src/upstream-engine/sdk/attempt.js +221 -221
- package/src/upstream-engine/sdk/chat.js +28 -28
- package/src/upstream-engine/sdk/convert.js +192 -192
- package/src/upstream-engine/sdk/diagnose.js +62 -62
- package/src/upstream-engine/sdk/dispatch.js +55 -55
- package/src/upstream-engine/sdk/responses.js +176 -176
- package/src/upstream-engine/sdk/sse.js +166 -166
- package/src/upstream-responses.js +305 -305
- package/src/usage/cache-fields.js +58 -0
- package/src/usage/record.js +109 -106
- package/src/usage/report.js +209 -195
- package/src/v6pool/claim.js +157 -157
- package/src/v6pool/config.js +65 -65
- package/src/v6pool/detect.js +130 -130
- package/src/v6pool/index.js +152 -152
- package/src/v6pool/prefix.js +140 -140
- package/src/v6pool/probe.js +40 -40
- package/src/v6pool/report.js +104 -104
- package/src/v6pool/rotate.js +263 -263
- package/src/v6pool/selftest.js +154 -0
- package/src/warp-pool/config.js +74 -74
- package/src/warp-pool/health.js +106 -106
- package/src/warp-pool/identity.js +141 -141
- package/src/warp-pool/index.js +358 -358
- package/src/warp-pool/keys.js +29 -29
- package/src/warp-pool/nodes.js +95 -95
- package/src/warp-pool/ports.js +160 -160
- package/src/warp-pool/register.js +113 -113
- package/src/warp-pool/report.js +232 -232
- package/src/warp-pool/revive.js +168 -168
- package/src/warp-pool/rotate.js +214 -214
- package/src/warp-pool/strategy.js +43 -43
- package/src/warp-pool/verdict.js +69 -69
- package/src/warp-pool/wireproxy.js +162 -162
|
@@ -1,124 +1,124 @@
|
|
|
1
|
-
// STS /v1/oauth2/tokens:授权码换凭证 + refresh_token 续期(DPoP 必带)。
|
|
2
|
-
// 终态失效(refresh_token 过期/已用/DPoP 或 client_id 不匹配)→ relogin=true,
|
|
3
|
-
// 与网络/5xx 瞬时失败分开,避免好号被误杀或死号无限刷。
|
|
4
|
-
import { STS_HOST, EP_OAUTH_TOKENS, CLIENT_ID } from "./const.js";
|
|
5
|
-
import { signDpopProof } from "./dpop.js";
|
|
6
|
-
import { compatFetch } from "../../compat.js";
|
|
7
|
-
|
|
8
|
-
function tokenErr(rawText) {
|
|
9
|
-
try { return JSON.parse(rawText) || {}; } catch { return {}; }
|
|
10
|
-
}
|
|
11
|
-
|
|
12
|
-
// 对齐 codearts2api:invalid_grant / ExpiredRefreshToken / InvalidDPoPHeader /
|
|
13
|
-
// invalid client id / has been used → 终态,必须重新登录。
|
|
14
|
-
export function isReLoginRejected(status, rawText) {
|
|
15
|
-
const t = tokenErr(rawText);
|
|
16
|
-
const code = String(t.error_code || "");
|
|
17
|
-
const msg = String(t.error_msg || "");
|
|
18
|
-
const low = (t.error || msg).toLowerCase();
|
|
19
|
-
return t.error === "invalid_grant"
|
|
20
|
-
|| code.includes("ExpiredRefreshToken")
|
|
21
|
-
|| code.includes("InvalidDPoPHeader")
|
|
22
|
-
|| low.includes("invalid client id")
|
|
23
|
-
|| low.includes("has been used");
|
|
24
|
-
}
|
|
25
|
-
|
|
26
|
-
/** 归一化 STS 响应(新 credentials / 旧 credential 双形状)。 */
|
|
27
|
-
export function normalizeTokenResponse(json) {
|
|
28
|
-
const c = json?.credentials || {};
|
|
29
|
-
const legacy = json?.credential || {};
|
|
30
|
-
const securityToken = c.security_token || legacy.securitytoken || "";
|
|
31
|
-
const accessKeyId = c.access_key_id || legacy.access || "";
|
|
32
|
-
const secretAccessKey = c.secret_access_key || legacy.secret || "";
|
|
33
|
-
const expiration = c.expiration || legacy.expires_at || "";
|
|
34
|
-
if (!securityToken || !accessKeyId || !secretAccessKey) return null;
|
|
35
|
-
const expiresAtMs = expiration ? Date.parse(expiration) : NaN;
|
|
36
|
-
return {
|
|
37
|
-
userId: String(json.user_id || ""),
|
|
38
|
-
userName: String(json.user_name || ""),
|
|
39
|
-
domainId: String(json.domain_id || ""),
|
|
40
|
-
refreshToken: String(json.refresh_token || ""),
|
|
41
|
-
accessKeyId,
|
|
42
|
-
secretAccessKey,
|
|
43
|
-
securityToken,
|
|
44
|
-
expiration,
|
|
45
|
-
expiresAtMs: Number.isFinite(expiresAtMs) ? expiresAtMs : 0,
|
|
46
|
-
};
|
|
47
|
-
}
|
|
48
|
-
|
|
49
|
-
async function postToken({ url, form, dpopJwk, fetchImpl }) {
|
|
50
|
-
const doFetch = fetchImpl || compatFetch;
|
|
51
|
-
const proof = signDpopProof(dpopJwk, url);
|
|
52
|
-
const res = await doFetch(url, {
|
|
53
|
-
method: "POST",
|
|
54
|
-
headers: { "content-type": "application/x-www-form-urlencoded", dpop: proof },
|
|
55
|
-
body: form.toString(),
|
|
56
|
-
});
|
|
57
|
-
const raw = await res.text().catch(() => "");
|
|
58
|
-
return { res, raw };
|
|
59
|
-
}
|
|
60
|
-
|
|
61
|
-
/**
|
|
62
|
-
* 授权码换凭证(登录用)。
|
|
63
|
-
* @param {object} p { code, verifier, redirectUri, clientId?, dpopJwk, fetchImpl?, stsHost? }
|
|
64
|
-
*/
|
|
65
|
-
export async function exchangeAuthorizationCode({ code, verifier, redirectUri, clientId = CLIENT_ID, dpopJwk, fetchImpl, stsHost } = {}) {
|
|
66
|
-
const url = (stsHost || STS_HOST) + EP_OAUTH_TOKENS;
|
|
67
|
-
const form = new URLSearchParams({
|
|
68
|
-
client_id: clientId,
|
|
69
|
-
code: String(code || ""),
|
|
70
|
-
code_verifier: String(verifier || ""),
|
|
71
|
-
grant_type: "authorization_code",
|
|
72
|
-
redirect_uri: String(redirectUri || ""),
|
|
73
|
-
});
|
|
74
|
-
const { res, raw } = await postToken({ url, form, dpopJwk, fetchImpl });
|
|
75
|
-
if (res.status >= 400) throw new Error(`codearts sts exchange http ${res.status}: ${String(raw).slice(0, 200)}`);
|
|
76
|
-
const json = JSON.parse(raw);
|
|
77
|
-
const account = normalizeTokenResponse(json);
|
|
78
|
-
if (!account) throw new Error("codearts sts exchange: response missing credentials");
|
|
79
|
-
account.clientId = clientId;
|
|
80
|
-
return account;
|
|
81
|
-
}
|
|
82
|
-
|
|
83
|
-
/**
|
|
84
|
-
* 用 refresh_token 换新 STS 凭证(refresh_token 单次轮换,响应里取新值写回)。
|
|
85
|
-
* @param {object} blob 账号凭证 blob(refreshToken/codeVerifier/dpopJwk/clientId)
|
|
86
|
-
* @param {object} [opts] { fetchImpl, stsHost }
|
|
87
|
-
* @returns {{account, refreshToken?, relogin?, error?}}
|
|
88
|
-
*/
|
|
89
|
-
export async function refreshStsCredential(blob, { fetchImpl, stsHost } = {}) {
|
|
90
|
-
const refreshToken = String(blob?.refreshToken || "");
|
|
91
|
-
const verifier = String(blob?.codeVerifier || "");
|
|
92
|
-
const jwk = blob?.dpopJwk;
|
|
93
|
-
const clientId = String(blob?.clientId || CLIENT_ID);
|
|
94
|
-
if (!refreshToken) return { account: null, relogin: true, error: "missing refresh_token" };
|
|
95
|
-
if (!verifier) return { account: null, relogin: true, error: "missing code_verifier" };
|
|
96
|
-
if (!jwk || !jwk.d) return { account: null, relogin: true, error: "missing DPoP private key — re-login required" };
|
|
97
|
-
|
|
98
|
-
const url = (stsHost || STS_HOST) + EP_OAUTH_TOKENS;
|
|
99
|
-
const form = new URLSearchParams({
|
|
100
|
-
client_id: clientId,
|
|
101
|
-
code_verifier: verifier,
|
|
102
|
-
grant_type: "refresh_token",
|
|
103
|
-
refresh_token: refreshToken,
|
|
104
|
-
});
|
|
105
|
-
let res, raw;
|
|
106
|
-
try {
|
|
107
|
-
({ res, raw } = await postToken({ url, form, dpopJwk: jwk, fetchImpl }));
|
|
108
|
-
} catch (err) {
|
|
109
|
-
return { account: null, error: `network: ${err?.message || err}` };
|
|
110
|
-
}
|
|
111
|
-
if (res.status >= 400) {
|
|
112
|
-
if (isReLoginRejected(res.status, raw)) {
|
|
113
|
-
const t = tokenErr(raw);
|
|
114
|
-
return { account: null, relogin: true, status: res.status, code: t.error_code, error: `refresh rejected: ${String(raw).slice(0, 200)}` };
|
|
115
|
-
}
|
|
116
|
-
return { account: null, status: res.status, error: `sts http ${res.status}: ${String(raw).slice(0, 200)}` };
|
|
117
|
-
}
|
|
118
|
-
let json;
|
|
119
|
-
try { json = JSON.parse(raw); } catch { return { account: null, error: "sts returned non-JSON body" }; }
|
|
120
|
-
const account = normalizeTokenResponse(json);
|
|
121
|
-
if (!account) return { account: null, error: "sts response missing credentials" };
|
|
122
|
-
account.clientId = clientId;
|
|
123
|
-
return { account, refreshToken: account.refreshToken || null };
|
|
124
|
-
}
|
|
1
|
+
// STS /v1/oauth2/tokens:授权码换凭证 + refresh_token 续期(DPoP 必带)。
|
|
2
|
+
// 终态失效(refresh_token 过期/已用/DPoP 或 client_id 不匹配)→ relogin=true,
|
|
3
|
+
// 与网络/5xx 瞬时失败分开,避免好号被误杀或死号无限刷。
|
|
4
|
+
import { STS_HOST, EP_OAUTH_TOKENS, CLIENT_ID } from "./const.js";
|
|
5
|
+
import { signDpopProof } from "./dpop.js";
|
|
6
|
+
import { compatFetch } from "../../compat.js";
|
|
7
|
+
|
|
8
|
+
function tokenErr(rawText) {
|
|
9
|
+
try { return JSON.parse(rawText) || {}; } catch { return {}; }
|
|
10
|
+
}
|
|
11
|
+
|
|
12
|
+
// 对齐 codearts2api:invalid_grant / ExpiredRefreshToken / InvalidDPoPHeader /
|
|
13
|
+
// invalid client id / has been used → 终态,必须重新登录。
|
|
14
|
+
export function isReLoginRejected(status, rawText) {
|
|
15
|
+
const t = tokenErr(rawText);
|
|
16
|
+
const code = String(t.error_code || "");
|
|
17
|
+
const msg = String(t.error_msg || "");
|
|
18
|
+
const low = (t.error || msg).toLowerCase();
|
|
19
|
+
return t.error === "invalid_grant"
|
|
20
|
+
|| code.includes("ExpiredRefreshToken")
|
|
21
|
+
|| code.includes("InvalidDPoPHeader")
|
|
22
|
+
|| low.includes("invalid client id")
|
|
23
|
+
|| low.includes("has been used");
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
/** 归一化 STS 响应(新 credentials / 旧 credential 双形状)。 */
|
|
27
|
+
export function normalizeTokenResponse(json) {
|
|
28
|
+
const c = json?.credentials || {};
|
|
29
|
+
const legacy = json?.credential || {};
|
|
30
|
+
const securityToken = c.security_token || legacy.securitytoken || "";
|
|
31
|
+
const accessKeyId = c.access_key_id || legacy.access || "";
|
|
32
|
+
const secretAccessKey = c.secret_access_key || legacy.secret || "";
|
|
33
|
+
const expiration = c.expiration || legacy.expires_at || "";
|
|
34
|
+
if (!securityToken || !accessKeyId || !secretAccessKey) return null;
|
|
35
|
+
const expiresAtMs = expiration ? Date.parse(expiration) : NaN;
|
|
36
|
+
return {
|
|
37
|
+
userId: String(json.user_id || ""),
|
|
38
|
+
userName: String(json.user_name || ""),
|
|
39
|
+
domainId: String(json.domain_id || ""),
|
|
40
|
+
refreshToken: String(json.refresh_token || ""),
|
|
41
|
+
accessKeyId,
|
|
42
|
+
secretAccessKey,
|
|
43
|
+
securityToken,
|
|
44
|
+
expiration,
|
|
45
|
+
expiresAtMs: Number.isFinite(expiresAtMs) ? expiresAtMs : 0,
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
async function postToken({ url, form, dpopJwk, fetchImpl }) {
|
|
50
|
+
const doFetch = fetchImpl || compatFetch;
|
|
51
|
+
const proof = signDpopProof(dpopJwk, url);
|
|
52
|
+
const res = await doFetch(url, {
|
|
53
|
+
method: "POST",
|
|
54
|
+
headers: { "content-type": "application/x-www-form-urlencoded", dpop: proof },
|
|
55
|
+
body: form.toString(),
|
|
56
|
+
});
|
|
57
|
+
const raw = await res.text().catch(() => "");
|
|
58
|
+
return { res, raw };
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
/**
|
|
62
|
+
* 授权码换凭证(登录用)。
|
|
63
|
+
* @param {object} p { code, verifier, redirectUri, clientId?, dpopJwk, fetchImpl?, stsHost? }
|
|
64
|
+
*/
|
|
65
|
+
export async function exchangeAuthorizationCode({ code, verifier, redirectUri, clientId = CLIENT_ID, dpopJwk, fetchImpl, stsHost } = {}) {
|
|
66
|
+
const url = (stsHost || STS_HOST) + EP_OAUTH_TOKENS;
|
|
67
|
+
const form = new URLSearchParams({
|
|
68
|
+
client_id: clientId,
|
|
69
|
+
code: String(code || ""),
|
|
70
|
+
code_verifier: String(verifier || ""),
|
|
71
|
+
grant_type: "authorization_code",
|
|
72
|
+
redirect_uri: String(redirectUri || ""),
|
|
73
|
+
});
|
|
74
|
+
const { res, raw } = await postToken({ url, form, dpopJwk, fetchImpl });
|
|
75
|
+
if (res.status >= 400) throw new Error(`codearts sts exchange http ${res.status}: ${String(raw).slice(0, 200)}`);
|
|
76
|
+
const json = JSON.parse(raw);
|
|
77
|
+
const account = normalizeTokenResponse(json);
|
|
78
|
+
if (!account) throw new Error("codearts sts exchange: response missing credentials");
|
|
79
|
+
account.clientId = clientId;
|
|
80
|
+
return account;
|
|
81
|
+
}
|
|
82
|
+
|
|
83
|
+
/**
|
|
84
|
+
* 用 refresh_token 换新 STS 凭证(refresh_token 单次轮换,响应里取新值写回)。
|
|
85
|
+
* @param {object} blob 账号凭证 blob(refreshToken/codeVerifier/dpopJwk/clientId)
|
|
86
|
+
* @param {object} [opts] { fetchImpl, stsHost }
|
|
87
|
+
* @returns {{account, refreshToken?, relogin?, error?}}
|
|
88
|
+
*/
|
|
89
|
+
export async function refreshStsCredential(blob, { fetchImpl, stsHost } = {}) {
|
|
90
|
+
const refreshToken = String(blob?.refreshToken || "");
|
|
91
|
+
const verifier = String(blob?.codeVerifier || "");
|
|
92
|
+
const jwk = blob?.dpopJwk;
|
|
93
|
+
const clientId = String(blob?.clientId || CLIENT_ID);
|
|
94
|
+
if (!refreshToken) return { account: null, relogin: true, error: "missing refresh_token" };
|
|
95
|
+
if (!verifier) return { account: null, relogin: true, error: "missing code_verifier" };
|
|
96
|
+
if (!jwk || !jwk.d) return { account: null, relogin: true, error: "missing DPoP private key — re-login required" };
|
|
97
|
+
|
|
98
|
+
const url = (stsHost || STS_HOST) + EP_OAUTH_TOKENS;
|
|
99
|
+
const form = new URLSearchParams({
|
|
100
|
+
client_id: clientId,
|
|
101
|
+
code_verifier: verifier,
|
|
102
|
+
grant_type: "refresh_token",
|
|
103
|
+
refresh_token: refreshToken,
|
|
104
|
+
});
|
|
105
|
+
let res, raw;
|
|
106
|
+
try {
|
|
107
|
+
({ res, raw } = await postToken({ url, form, dpopJwk: jwk, fetchImpl }));
|
|
108
|
+
} catch (err) {
|
|
109
|
+
return { account: null, error: `network: ${err?.message || err}` };
|
|
110
|
+
}
|
|
111
|
+
if (res.status >= 400) {
|
|
112
|
+
if (isReLoginRejected(res.status, raw)) {
|
|
113
|
+
const t = tokenErr(raw);
|
|
114
|
+
return { account: null, relogin: true, status: res.status, code: t.error_code, error: `refresh rejected: ${String(raw).slice(0, 200)}` };
|
|
115
|
+
}
|
|
116
|
+
return { account: null, status: res.status, error: `sts http ${res.status}: ${String(raw).slice(0, 200)}` };
|
|
117
|
+
}
|
|
118
|
+
let json;
|
|
119
|
+
try { json = JSON.parse(raw); } catch { return { account: null, error: "sts returned non-JSON body" }; }
|
|
120
|
+
const account = normalizeTokenResponse(json);
|
|
121
|
+
if (!account) return { account: null, error: "sts response missing credentials" };
|
|
122
|
+
account.clientId = clientId;
|
|
123
|
+
return { account, refreshToken: account.refreshToken || null };
|
|
124
|
+
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
export * from "./codearts/index.js";
|
|
1
|
+
export * from "./codearts/index.js";
|
|
@@ -1,147 +1,147 @@
|
|
|
1
|
-
import { appendEvent } from "../logs.js";
|
|
2
|
-
import { splitModelId, DEFAULT_PROVIDER, joinModelId } from "./model-id.js";
|
|
3
|
-
import { isModelAllowed as stateIsAllowed, loadProviderAllowedModels as stateLoadAllowed, loadProviderAllowAnyModels as stateLoadAllowAny } from "../state.js";
|
|
4
|
-
// 定制预热名单:registry customProviders 9 家 + 默认 opencode;openrouter/自配 generic 不预热(按需自拉)
|
|
5
|
-
const CUSTOM_PREHEAT_IDS = new Set(["opencode", "workbuddy", "traework", "cline", "codearts", "qoder", "globalqwenwork", "qwenwork", "zcode", "raccoon"]);
|
|
6
|
-
function isPreheatDisabled() {
|
|
7
|
-
const raw = process.env.MSLXDFF_PREHEAT;
|
|
8
|
-
if (raw === undefined || raw === null || raw === "") return false;
|
|
9
|
-
const s = String(raw).trim().toLowerCase();
|
|
10
|
-
return s === "0" || s === "off" || s === "false" || s === "no" || s === "disable" || s === "disabled";
|
|
11
|
-
}
|
|
12
|
-
|
|
13
|
-
// 多供应商 dispatcher:把多个 Provider 聚合成一个 `upstream` 形状(chat/preheat/close),
|
|
14
|
-
// 按 body.model 的前缀路由到对应供应商,转发上游前剥掉前缀只发原始 id。
|
|
15
|
-
// 纯化:第二参可注入纯函数 isAllowed/getAllowed/getAllowAny,便于单测不读盘
|
|
16
|
-
export function createProviderDispatcher(providers = [], opts = {}) {
|
|
17
|
-
const byId = new Map(providers.map((p) => [p.id, p]));
|
|
18
|
-
// 纯化注入:第二参可为 {isAllowed, getAllowedModels, getAllowAny} 或直接函数 isAllowed
|
|
19
|
-
let isAllowedFn, getAllowedFn, getAllowAnyFn;
|
|
20
|
-
if (typeof opts === "function") {
|
|
21
|
-
isAllowedFn = opts;
|
|
22
|
-
getAllowedFn = stateLoadAllowed;
|
|
23
|
-
getAllowAnyFn = stateLoadAllowAny;
|
|
24
|
-
} else {
|
|
25
|
-
isAllowedFn = opts.isAllowed || stateIsAllowed;
|
|
26
|
-
getAllowedFn = opts.getAllowedModels || stateLoadAllowed;
|
|
27
|
-
getAllowAnyFn = opts.getAllowAny || stateLoadAllowAny;
|
|
28
|
-
}
|
|
29
|
-
|
|
30
|
-
function resolve(model) {
|
|
31
|
-
const split = splitModelId(model, providers.map((p) => p.id));
|
|
32
|
-
return {
|
|
33
|
-
provider: byId.get(split.provider) ?? byId.get(DEFAULT_PROVIDER) ?? providers[0],
|
|
34
|
-
raw: split.raw,
|
|
35
|
-
prefixed: split.prefixed,
|
|
36
|
-
};
|
|
37
|
-
}
|
|
38
|
-
|
|
39
|
-
async function chat(body, opts = {}) {
|
|
40
|
-
let { provider, raw } = resolve(body?.model);
|
|
41
|
-
if (!provider) throw new Error(`no provider for model ${body?.model ?? "(empty)"}`);
|
|
42
|
-
let workbuddyUid = opts?.workbuddyUid ? String(opts.workbuddyUid).trim() : "";
|
|
43
|
-
// model 前缀钉死:workbuddy/<uid>:<rawId> -> 剥 uid,rawId 为真实模型
|
|
44
|
-
if (provider.id === "workbuddy" && typeof raw === "string" && raw.includes(":")) {
|
|
45
|
-
const idx = raw.indexOf(":");
|
|
46
|
-
const uidPart = raw.slice(0, idx).trim();
|
|
47
|
-
const modelPart = raw.slice(idx + 1).trim();
|
|
48
|
-
if (uidPart && modelPart) {
|
|
49
|
-
if (!workbuddyUid) workbuddyUid = uidPart;
|
|
50
|
-
raw = modelPart;
|
|
51
|
-
}
|
|
52
|
-
}
|
|
53
|
-
if (!isAllowedFn(provider.id, raw)) {
|
|
54
|
-
const allowed = getAllowedFn(provider.id) || [];
|
|
55
|
-
const msg = `model not allowed for provider "${provider.id}": "${raw}" — allowed: ${allowed.join(", ") || "(none)"} (use: mslxdff -provider ${provider.id} allowlist add <model>)`;
|
|
56
|
-
appendEvent({ type: "provider-model-state", provider: provider.id, model: body?.model, rawModel: raw, state: "blocked", reason: "allowlist", status: 403 });
|
|
57
|
-
return new Response(JSON.stringify({ error: msg }), { status: 403, headers: { "Content-Type": "application/json", "x-mslxdff-allowlist": "1" } });
|
|
58
|
-
}
|
|
59
|
-
const forwarded = raw === body?.model ? body : { ...body, model: raw };
|
|
60
|
-
// ADR-0008:本请求携带瞬时共享 key(shareKeys 由组员侧按 header 解析后传入)。
|
|
61
|
-
const sharedKeys = opts?.shareKeys?.[provider.id];
|
|
62
|
-
const startedAt = Date.now();
|
|
63
|
-
let res;
|
|
64
|
-
try {
|
|
65
|
-
if (sharedKeys && sharedKeys.length && typeof provider.chatWithKeys === "function") {
|
|
66
|
-
res = await provider.chatWithKeys(forwarded, sharedKeys, opts);
|
|
67
|
-
} else if (provider.id === "workbuddy" && workbuddyUid) {
|
|
68
|
-
res = await provider.chat(forwarded, { ...opts, workbuddyUid });
|
|
69
|
-
} else {
|
|
70
|
-
res = await provider.chat(forwarded, opts);
|
|
71
|
-
}
|
|
72
|
-
} catch (err) {
|
|
73
|
-
appendEvent({ type: "provider-model-state", provider: provider.id, model: body?.model, rawModel: raw, state: "upstream-error", error: String(err?.message || err), durationMs: Date.now() - startedAt });
|
|
74
|
-
throw err;
|
|
75
|
-
}
|
|
76
|
-
appendEvent({
|
|
77
|
-
type: "provider-model-state",
|
|
78
|
-
provider: provider.id,
|
|
79
|
-
model: body?.model,
|
|
80
|
-
rawModel: raw,
|
|
81
|
-
status: res?.status,
|
|
82
|
-
state: res?.status === 429 ? "limited" : res?.ok ? "ok" : "upstream-error",
|
|
83
|
-
durationMs: Date.now() - startedAt,
|
|
84
|
-
});
|
|
85
|
-
return res;
|
|
86
|
-
}
|
|
87
|
-
|
|
88
|
-
// 聚合所有供应商的模型列表;默认供应商(opencode)裸 id,其它带前缀
|
|
89
|
-
// 若某供应商设置了 allowlist(非空),则仅暴露白名单内的模型(按 raw id 匹配)
|
|
90
|
-
async function listModels() {
|
|
91
|
-
const out = [];
|
|
92
|
-
const seen = new Set();
|
|
93
|
-
for (const p of providers) {
|
|
94
|
-
let list;
|
|
95
|
-
try {
|
|
96
|
-
list = (await p.listModels?.()) ?? [];
|
|
97
|
-
} catch {
|
|
98
|
-
list = [];
|
|
99
|
-
}
|
|
100
|
-
const allowed = getAllowedFn(p.id) || [];
|
|
101
|
-
const allowAny = getAllowAnyFn(p.id);
|
|
102
|
-
const allowedSet = allowed.length ? new Set(allowed) : null;
|
|
103
|
-
// 空名单且不允许任意模型 => 该供应商不暴露任何模型(安全默认)
|
|
104
|
-
if (!allowedSet && !allowAny) continue;
|
|
105
|
-
for (const m of list) {
|
|
106
|
-
if (!m || !m.id) continue;
|
|
107
|
-
if (seen.has(m.id)) continue;
|
|
108
|
-
if (allowedSet) {
|
|
109
|
-
// m.id 已是带前缀的对外 id,需剥回 raw 再比对
|
|
110
|
-
const { raw } = splitModelId(m.id, providers.map((x) => x.id));
|
|
111
|
-
const rawNorm = String(raw || "").trim();
|
|
112
|
-
if (!allowedSet.has(rawNorm)) continue;
|
|
113
|
-
// 对于 opencode 裸 id,rawNorm 即 m.id 本身
|
|
114
|
-
}
|
|
115
|
-
seen.add(m.id);
|
|
116
|
-
out.push(m);
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
return out;
|
|
120
|
-
}
|
|
121
|
-
|
|
122
|
-
// 预热定制供应商 + 默认供应商(opencode):registry 名单 8 家 + opencode 共 9 家并发各打一次;
|
|
123
|
-
// openrouter/用户自配 generic 不在此列(按需首次请求自拉)。单家失败静默,不拖累启动。
|
|
124
|
-
async function preheat() {
|
|
125
|
-
if (isPreheatDisabled()) return { ok: true, skipped: true };
|
|
126
|
-
const targets = providers.filter((p) => p && typeof p.preheat === "function" && CUSTOM_PREHEAT_IDS.has(p.id));
|
|
127
|
-
if (!targets.length) return { ok: false, skipped: true };
|
|
128
|
-
const settled = await Promise.all(targets.map(async (p) => {
|
|
129
|
-
try {
|
|
130
|
-
const r = await p.preheat();
|
|
131
|
-
return { id: p.id, ...(r && typeof r === "object" ? r : { ok: false }) };
|
|
132
|
-
} catch {
|
|
133
|
-
return { id: p.id, ok: false, error: "preheat failed" };
|
|
134
|
-
}
|
|
135
|
-
}));
|
|
136
|
-
const okCount = settled.filter((r) => r && r.ok).length;
|
|
137
|
-
return { ok: okCount > 0, results: settled, okCount, total: settled.length };
|
|
138
|
-
}
|
|
139
|
-
|
|
140
|
-
async function close() {
|
|
141
|
-
for (const p of providers) {
|
|
142
|
-
try { await p.close?.(); } catch {}
|
|
143
|
-
}
|
|
144
|
-
}
|
|
145
|
-
|
|
146
|
-
return { providers, byId, resolve, listModels, chat, preheat, close, joinModelId };
|
|
1
|
+
import { appendEvent } from "../logs.js";
|
|
2
|
+
import { splitModelId, DEFAULT_PROVIDER, joinModelId } from "./model-id.js";
|
|
3
|
+
import { isModelAllowed as stateIsAllowed, loadProviderAllowedModels as stateLoadAllowed, loadProviderAllowAnyModels as stateLoadAllowAny } from "../state.js";
|
|
4
|
+
// 定制预热名单:registry customProviders 9 家 + 默认 opencode;openrouter/自配 generic 不预热(按需自拉)
|
|
5
|
+
const CUSTOM_PREHEAT_IDS = new Set(["opencode", "workbuddy", "traework", "cline", "codearts", "qoder", "globalqwenwork", "qwenwork", "zcode", "raccoon"]);
|
|
6
|
+
function isPreheatDisabled() {
|
|
7
|
+
const raw = process.env.MSLXDFF_PREHEAT;
|
|
8
|
+
if (raw === undefined || raw === null || raw === "") return false;
|
|
9
|
+
const s = String(raw).trim().toLowerCase();
|
|
10
|
+
return s === "0" || s === "off" || s === "false" || s === "no" || s === "disable" || s === "disabled";
|
|
11
|
+
}
|
|
12
|
+
|
|
13
|
+
// 多供应商 dispatcher:把多个 Provider 聚合成一个 `upstream` 形状(chat/preheat/close),
|
|
14
|
+
// 按 body.model 的前缀路由到对应供应商,转发上游前剥掉前缀只发原始 id。
|
|
15
|
+
// 纯化:第二参可注入纯函数 isAllowed/getAllowed/getAllowAny,便于单测不读盘
|
|
16
|
+
export function createProviderDispatcher(providers = [], opts = {}) {
|
|
17
|
+
const byId = new Map(providers.map((p) => [p.id, p]));
|
|
18
|
+
// 纯化注入:第二参可为 {isAllowed, getAllowedModels, getAllowAny} 或直接函数 isAllowed
|
|
19
|
+
let isAllowedFn, getAllowedFn, getAllowAnyFn;
|
|
20
|
+
if (typeof opts === "function") {
|
|
21
|
+
isAllowedFn = opts;
|
|
22
|
+
getAllowedFn = stateLoadAllowed;
|
|
23
|
+
getAllowAnyFn = stateLoadAllowAny;
|
|
24
|
+
} else {
|
|
25
|
+
isAllowedFn = opts.isAllowed || stateIsAllowed;
|
|
26
|
+
getAllowedFn = opts.getAllowedModels || stateLoadAllowed;
|
|
27
|
+
getAllowAnyFn = opts.getAllowAny || stateLoadAllowAny;
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
function resolve(model) {
|
|
31
|
+
const split = splitModelId(model, providers.map((p) => p.id));
|
|
32
|
+
return {
|
|
33
|
+
provider: byId.get(split.provider) ?? byId.get(DEFAULT_PROVIDER) ?? providers[0],
|
|
34
|
+
raw: split.raw,
|
|
35
|
+
prefixed: split.prefixed,
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
|
|
39
|
+
async function chat(body, opts = {}) {
|
|
40
|
+
let { provider, raw } = resolve(body?.model);
|
|
41
|
+
if (!provider) throw new Error(`no provider for model ${body?.model ?? "(empty)"}`);
|
|
42
|
+
let workbuddyUid = opts?.workbuddyUid ? String(opts.workbuddyUid).trim() : "";
|
|
43
|
+
// model 前缀钉死:workbuddy/<uid>:<rawId> -> 剥 uid,rawId 为真实模型
|
|
44
|
+
if (provider.id === "workbuddy" && typeof raw === "string" && raw.includes(":")) {
|
|
45
|
+
const idx = raw.indexOf(":");
|
|
46
|
+
const uidPart = raw.slice(0, idx).trim();
|
|
47
|
+
const modelPart = raw.slice(idx + 1).trim();
|
|
48
|
+
if (uidPart && modelPart) {
|
|
49
|
+
if (!workbuddyUid) workbuddyUid = uidPart;
|
|
50
|
+
raw = modelPart;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
if (!isAllowedFn(provider.id, raw)) {
|
|
54
|
+
const allowed = getAllowedFn(provider.id) || [];
|
|
55
|
+
const msg = `model not allowed for provider "${provider.id}": "${raw}" — allowed: ${allowed.join(", ") || "(none)"} (use: mslxdff -provider ${provider.id} allowlist add <model>)`;
|
|
56
|
+
appendEvent({ type: "provider-model-state", provider: provider.id, model: body?.model, rawModel: raw, state: "blocked", reason: "allowlist", status: 403 });
|
|
57
|
+
return new Response(JSON.stringify({ error: msg }), { status: 403, headers: { "Content-Type": "application/json", "x-mslxdff-allowlist": "1" } });
|
|
58
|
+
}
|
|
59
|
+
const forwarded = raw === body?.model ? body : { ...body, model: raw };
|
|
60
|
+
// ADR-0008:本请求携带瞬时共享 key(shareKeys 由组员侧按 header 解析后传入)。
|
|
61
|
+
const sharedKeys = opts?.shareKeys?.[provider.id];
|
|
62
|
+
const startedAt = Date.now();
|
|
63
|
+
let res;
|
|
64
|
+
try {
|
|
65
|
+
if (sharedKeys && sharedKeys.length && typeof provider.chatWithKeys === "function") {
|
|
66
|
+
res = await provider.chatWithKeys(forwarded, sharedKeys, opts);
|
|
67
|
+
} else if (provider.id === "workbuddy" && workbuddyUid) {
|
|
68
|
+
res = await provider.chat(forwarded, { ...opts, workbuddyUid });
|
|
69
|
+
} else {
|
|
70
|
+
res = await provider.chat(forwarded, opts);
|
|
71
|
+
}
|
|
72
|
+
} catch (err) {
|
|
73
|
+
appendEvent({ type: "provider-model-state", provider: provider.id, model: body?.model, rawModel: raw, state: "upstream-error", error: String(err?.message || err), durationMs: Date.now() - startedAt });
|
|
74
|
+
throw err;
|
|
75
|
+
}
|
|
76
|
+
appendEvent({
|
|
77
|
+
type: "provider-model-state",
|
|
78
|
+
provider: provider.id,
|
|
79
|
+
model: body?.model,
|
|
80
|
+
rawModel: raw,
|
|
81
|
+
status: res?.status,
|
|
82
|
+
state: res?.status === 429 ? "limited" : res?.ok ? "ok" : "upstream-error",
|
|
83
|
+
durationMs: Date.now() - startedAt,
|
|
84
|
+
});
|
|
85
|
+
return res;
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
// 聚合所有供应商的模型列表;默认供应商(opencode)裸 id,其它带前缀
|
|
89
|
+
// 若某供应商设置了 allowlist(非空),则仅暴露白名单内的模型(按 raw id 匹配)
|
|
90
|
+
async function listModels() {
|
|
91
|
+
const out = [];
|
|
92
|
+
const seen = new Set();
|
|
93
|
+
for (const p of providers) {
|
|
94
|
+
let list;
|
|
95
|
+
try {
|
|
96
|
+
list = (await p.listModels?.()) ?? [];
|
|
97
|
+
} catch {
|
|
98
|
+
list = [];
|
|
99
|
+
}
|
|
100
|
+
const allowed = getAllowedFn(p.id) || [];
|
|
101
|
+
const allowAny = getAllowAnyFn(p.id);
|
|
102
|
+
const allowedSet = allowed.length ? new Set(allowed) : null;
|
|
103
|
+
// 空名单且不允许任意模型 => 该供应商不暴露任何模型(安全默认)
|
|
104
|
+
if (!allowedSet && !allowAny) continue;
|
|
105
|
+
for (const m of list) {
|
|
106
|
+
if (!m || !m.id) continue;
|
|
107
|
+
if (seen.has(m.id)) continue;
|
|
108
|
+
if (allowedSet) {
|
|
109
|
+
// m.id 已是带前缀的对外 id,需剥回 raw 再比对
|
|
110
|
+
const { raw } = splitModelId(m.id, providers.map((x) => x.id));
|
|
111
|
+
const rawNorm = String(raw || "").trim();
|
|
112
|
+
if (!allowedSet.has(rawNorm)) continue;
|
|
113
|
+
// 对于 opencode 裸 id,rawNorm 即 m.id 本身
|
|
114
|
+
}
|
|
115
|
+
seen.add(m.id);
|
|
116
|
+
out.push(m);
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
return out;
|
|
120
|
+
}
|
|
121
|
+
|
|
122
|
+
// 预热定制供应商 + 默认供应商(opencode):registry 名单 8 家 + opencode 共 9 家并发各打一次;
|
|
123
|
+
// openrouter/用户自配 generic 不在此列(按需首次请求自拉)。单家失败静默,不拖累启动。
|
|
124
|
+
async function preheat() {
|
|
125
|
+
if (isPreheatDisabled()) return { ok: true, skipped: true };
|
|
126
|
+
const targets = providers.filter((p) => p && typeof p.preheat === "function" && CUSTOM_PREHEAT_IDS.has(p.id));
|
|
127
|
+
if (!targets.length) return { ok: false, skipped: true };
|
|
128
|
+
const settled = await Promise.all(targets.map(async (p) => {
|
|
129
|
+
try {
|
|
130
|
+
const r = await p.preheat();
|
|
131
|
+
return { id: p.id, ...(r && typeof r === "object" ? r : { ok: false }) };
|
|
132
|
+
} catch {
|
|
133
|
+
return { id: p.id, ok: false, error: "preheat failed" };
|
|
134
|
+
}
|
|
135
|
+
}));
|
|
136
|
+
const okCount = settled.filter((r) => r && r.ok).length;
|
|
137
|
+
return { ok: okCount > 0, results: settled, okCount, total: settled.length };
|
|
138
|
+
}
|
|
139
|
+
|
|
140
|
+
async function close() {
|
|
141
|
+
for (const p of providers) {
|
|
142
|
+
try { await p.close?.(); } catch {}
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
|
|
146
|
+
return { providers, byId, resolve, listModels, chat, preheat, close, joinModelId };
|
|
147
147
|
}
|