mslxdff 0.1.159 → 0.1.160
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/bin/mslxdff.js +4 -4
- package/package.json +1 -1
- package/src/auto.js +254 -254
- package/src/bench/cline-bench.js +42 -42
- package/src/bench/probe.js +70 -70
- package/src/bench/report.js +162 -162
- package/src/bench/runner.js +77 -77
- package/src/bench/via-probe.js +124 -124
- package/src/bench/via-routes.js +87 -87
- package/src/bench/workbuddy-bench.js +54 -54
- package/src/chat/engine.js +160 -160
- package/src/chat/gateway.js +163 -163
- package/src/chat/orchestrator.js +234 -234
- package/src/chat/prompt.js +70 -70
- package/src/chat/repl.js +88 -88
- package/src/chat/terminal.js +135 -135
- package/src/chat/tools.js +306 -306
- package/src/chat-pipeline/auto-race.js +1 -1
- package/src/chat-pipeline/index.js +123 -123
- package/src/chat-pipeline/policy.js +76 -76
- package/src/chat-pipeline/serial-trial.js +210 -210
- package/src/cli/commands/group.js +249 -249
- package/src/cli/commands/model/picks.js +50 -50
- package/src/cli/commands/provider/bench-via.js +247 -247
- package/src/cli/commands/provider/bench.js +141 -141
- package/src/cli/commands/provider/index.js +124 -124
- package/src/cli/commands/provider/models.js +139 -139
- package/src/cli/commands/provider/qwenwork-login.js +119 -119
- package/src/cli/commands/stats.js +20 -5
- package/src/cli/commands/sync.js +232 -232
- package/src/cli/provider-row.js +2 -2
- package/src/cli/status.js +279 -279
- package/src/daemon.js +96 -96
- package/src/model-capabilities/enrich.js +86 -86
- package/src/model-capabilities/index.js +183 -183
- package/src/model-capabilities/parse.js +70 -70
- package/src/models.js +225 -225
- package/src/providers/cline/auth.js +228 -228
- package/src/providers/cline/chat.js +307 -307
- package/src/providers/cline.js +2 -2
- package/src/providers/keyring.js +60 -60
- package/src/providers/qoder/chat.js +183 -183
- package/src/providers/qoder/index.js +230 -230
- package/src/providers/qoder/sse.js +103 -103
- package/src/providers/qwenwork/account-store.js +133 -133
- package/src/providers/qwenwork/constants.js +67 -67
- package/src/providers/qwenwork/cosy.js +120 -120
- package/src/providers/qwenwork/crypto.js +218 -218
- package/src/providers/qwenwork/http.js +20 -20
- package/src/providers/qwenwork/index.js +327 -327
- package/src/providers/qwenwork/payload.js +142 -142
- package/src/providers/qwenwork/rsa.js +54 -54
- package/src/providers/qwenwork/sse.js +268 -268
- package/src/providers/qwenwork/stream.js +130 -130
- package/src/providers/qwenwork/upstream.js +120 -120
- package/src/providers/qwenwork.js +1 -1
- package/src/providers/registry.js +66 -66
- package/src/providers/workbuddy/chat.js +248 -248
- package/src/providers/workbuddy/reshape.js +152 -152
- package/src/providers/workbuddy.js +2 -2
- package/src/providers/zcode/chat.js +11 -10
- package/src/providers/zcode/const.js +1 -1
- package/src/providers/zcode/context-shape.js +162 -0
- package/src/providers/zcode/headers.js +38 -0
- package/src/reasoning.js +32 -32
- package/src/routes/chat/broadband-handler.js +3 -2
- package/src/routes/chat/gateway.js +46 -46
- package/src/routes/chat/hedge-handler.js +3 -0
- package/src/routes/chat/local-handler.js +2 -0
- package/src/routes/chat/peer-handler.js +2 -0
- package/src/routes/chat/relay-pipeline.js +264 -250
- package/src/routes/chat/via-route-handler.js +146 -144
- package/src/routes/hedge.js +255 -255
- package/src/routes/models-route.js +167 -167
- package/src/routes/peers.js +273 -273
- package/src/routes/stream-scan.js +86 -0
- package/src/routes/stream.js +393 -438
- package/src/runtime/bootstrap.js +45 -45
- package/src/runtime/provider-gate.js +33 -33
- package/src/runtime/providers-setup.js +165 -165
- package/src/server.js +64 -64
- package/src/state/schemas/allowlist.js +92 -92
- package/src/sync-opencode.js +280 -280
- package/src/transport/index.js +244 -244
- package/src/transport/pool.js +56 -56
- package/src/transport/retry.js +24 -24
- package/src/transport/sse.js +93 -93
- package/src/upstream-probe/display.js +52 -52
- package/src/upstream-probe/probe.js +49 -49
- package/src/upstream-probe/rotate.js +110 -110
- package/src/upstream-probe/start.js +45 -45
- package/src/upstream.js +289 -289
- package/src/usage/record.js +8 -1
- package/src/usage/report.js +39 -2
- package/docs/cli_help_mini.md +0 -133
package/src/chat/tools.js
CHANGED
|
@@ -1,307 +1,307 @@
|
|
|
1
|
-
import { execFile } from "node:child_process";
|
|
2
|
-
import { readFileSync, statSync, existsSync } from "node:fs";
|
|
3
|
-
import { resolve, join, dirname } from "node:path";
|
|
4
|
-
import { fileURLToPath } from "node:url";
|
|
5
|
-
import { performance } from "node:perf_hooks";
|
|
6
|
-
import { FORBIDDEN } from "./config.js";
|
|
7
|
-
import { logDir } from "../logs.js";
|
|
8
|
-
import { defaultStateFile, loadProviderKeys, loadProviderConfigs } from "../state.js";
|
|
9
|
-
import { compatFetch } from "../compat.js";
|
|
1
|
+
import { execFile } from "node:child_process";
|
|
2
|
+
import { readFileSync, statSync, existsSync } from "node:fs";
|
|
3
|
+
import { resolve, join, dirname } from "node:path";
|
|
4
|
+
import { fileURLToPath } from "node:url";
|
|
5
|
+
import { performance } from "node:perf_hooks";
|
|
6
|
+
import { FORBIDDEN } from "./config.js";
|
|
7
|
+
import { logDir } from "../logs.js";
|
|
8
|
+
import { defaultStateFile, loadProviderKeys, loadProviderConfigs } from "../state.js";
|
|
9
|
+
import { compatFetch } from "../compat.js";
|
|
10
10
|
import { opencodeClientIdentity } from "../opencode-identity.js";
|
|
11
|
-
|
|
12
|
-
const pkgRoot = resolve(dirname(fileURLToPath(import.meta.url)), "../..");
|
|
13
|
-
const allowedRoots = [
|
|
14
|
-
resolve(pkgRoot),
|
|
15
|
-
resolve(logDir()),
|
|
16
|
-
resolve(dirname(defaultStateFile())),
|
|
17
|
-
];
|
|
18
|
-
|
|
19
|
-
function isAllowed(p) {
|
|
20
|
-
const abs = resolve(p);
|
|
21
|
-
// 展开 ~/ 形式
|
|
22
|
-
const expanded = abs.startsWith("~/") ? join(process.env.HOME || process.env.USERPROFILE || "", abs.slice(2)) : abs;
|
|
23
|
-
const chk = resolve(expanded);
|
|
24
|
-
return allowedRoots.some((r) => chk === r || chk.startsWith(r + "/") || chk.startsWith(r + "\\"));
|
|
25
|
-
}
|
|
26
|
-
|
|
27
|
-
function parseCommand(str) {
|
|
28
|
-
// 支持引号包裹
|
|
29
|
-
const out = [];
|
|
30
|
-
let cur = "";
|
|
31
|
-
let q = null;
|
|
32
|
-
for (let i = 0; i < str.length; i++) {
|
|
33
|
-
const c = str[i];
|
|
34
|
-
if (q) {
|
|
35
|
-
if (c === q) q = null;
|
|
36
|
-
else cur += c;
|
|
37
|
-
} else if (c === '"' || c === "'") {
|
|
38
|
-
q = c;
|
|
39
|
-
} else if (c === " " || c === "\t") {
|
|
40
|
-
if (cur) { out.push(cur); cur = ""; }
|
|
41
|
-
} else {
|
|
42
|
-
cur += c;
|
|
43
|
-
}
|
|
44
|
-
}
|
|
45
|
-
if (cur) out.push(cur);
|
|
46
|
-
return out;
|
|
47
|
-
}
|
|
48
|
-
|
|
49
|
-
export function getToolDefs() {
|
|
50
|
-
return [
|
|
51
|
-
{
|
|
52
|
-
type: "function",
|
|
53
|
-
function: {
|
|
54
|
-
name: "run_command",
|
|
55
|
-
description: "执行一条 mslxdff CLI 命令(不含 mslxdff 前缀)。仅限 cli_help_mini 所列命令,禁止 -uninstall;禁止 mslxdff \"hi\" --model X / --model X \"hi\" / -chat --model X 等幻觉命令,探活模型必须用 curl 工具 POST 本机 /v1/chat/completions。",
|
|
56
|
-
parameters: {
|
|
57
|
-
type: "object",
|
|
58
|
-
properties: {
|
|
59
|
-
command: { type: "string", description: "例如: -model set hy3-free 或 -group list 或 -log 20;模型探活禁止用此工具,必须用 curl POST http://localhost:8989/v1/chat/completions" },
|
|
60
|
-
},
|
|
61
|
-
required: ["command"],
|
|
62
|
-
},
|
|
63
|
-
},
|
|
64
|
-
},
|
|
65
|
-
{
|
|
66
|
-
type: "function",
|
|
67
|
-
function: {
|
|
68
|
-
name: "read_file",
|
|
69
|
-
description: "读取项目内的文件内容(src/ docs/ logs/ 等),用于查看日志或配置。禁止读取项目外文件。",
|
|
70
|
-
parameters: {
|
|
71
|
-
type: "object",
|
|
72
|
-
properties: {
|
|
73
|
-
path: { type: "string", description: "相对项目根或绝对路径,如 src/logs.js 或 ~/.config/mslxdff/events.log" },
|
|
74
|
-
limit: { type: "number", description: "最多读取字符数,默认 8000" },
|
|
75
|
-
},
|
|
76
|
-
required: ["path"],
|
|
77
|
-
},
|
|
78
|
-
},
|
|
79
|
-
},
|
|
80
|
-
{
|
|
81
|
-
type: "function",
|
|
82
|
-
function: {
|
|
83
|
-
name: "curl",
|
|
84
|
-
description: "网络/HTTP 探活,检测上游或本机服务可用性。支持任意 http(s) URL,返回状态码、耗时、响应头与前几千字符。常用: upstream(上游模型列表)、local/health(本机健康)、local/models(本机模型列表)。探活指定模型必须用 POST http://localhost:8989/v1/chat/completions body {\"model\":\"<provider/模型>\",\"messages\":[{\"role\":\"user\",\"content\":\"hi\"}]}。简写会自动补全为完整 URL。",
|
|
85
|
-
parameters: {
|
|
86
|
-
type: "object",
|
|
87
|
-
properties: {
|
|
88
|
-
url: { type: "string", description: "要请求的 URL 或简写:upstream、upstream/models、upstream/chat、local、local/health、local/models、health、models,或完整 http(s) URL" },
|
|
89
|
-
method: { type: "string", description: "HTTP 方法,默认 GET", enum: ["GET", "POST", "PUT", "DELETE", "HEAD", "PATCH", "OPTIONS"] },
|
|
90
|
-
headers: { type: "object", description: "附加请求头,键值均为字符串", additionalProperties: { type: "string" } },
|
|
91
|
-
body: { type: "string", description: "请求体(POST/PUT 时),字符串,如 JSON" },
|
|
92
|
-
timeoutMs: { type: "number", description: "超时毫秒,默认 8000,最大 15000" },
|
|
93
|
-
},
|
|
94
|
-
required: ["url"],
|
|
95
|
-
},
|
|
96
|
-
},
|
|
97
|
-
},
|
|
98
|
-
];
|
|
99
|
-
}
|
|
100
|
-
|
|
101
|
-
export function validateCommand(cmd) {
|
|
102
|
-
const low = String(cmd || "").toLowerCase();
|
|
103
|
-
for (const f of FORBIDDEN) if (low.includes(f)) return `forbidden: ${f} 已拦截,禁止执行`;
|
|
104
|
-
return null;
|
|
105
|
-
}
|
|
106
|
-
|
|
107
|
-
export async function execCommand(command) {
|
|
108
|
-
const err = validateCommand(command);
|
|
109
|
-
if (err) return { ok: false, output: err };
|
|
110
|
-
const args = parseCommand(String(command).trim());
|
|
111
|
-
if (!args.length) return { ok: false, output: "empty command" };
|
|
112
|
-
// 关键路径打点:exec 本身耗时(fork+加载)是 -chat 最大的本地开销,后续可改直调以省 300-500ms
|
|
113
|
-
const bin = join(pkgRoot, "bin/mslxdff.js");
|
|
114
|
-
const t0 = performance.now();
|
|
115
|
-
const res = await new Promise((resolve) => {
|
|
116
|
-
execFile(process.execPath, [bin, ...args], { timeout: 15000, maxBuffer: 1024 * 500 }, (e, stdout, stderr) => {
|
|
117
|
-
const out = String(stdout || "") + (stderr ? `\n${stderr}` : "");
|
|
118
|
-
if (e) {
|
|
119
|
-
resolve({ ok: false, output: out.slice(0, 8000) || String(e.message).slice(0, 2000) });
|
|
120
|
-
} else {
|
|
121
|
-
resolve({ ok: true, output: out.slice(0, 8000) || "(no output)" });
|
|
122
|
-
}
|
|
123
|
-
});
|
|
124
|
-
});
|
|
125
|
-
res._ms = Math.round(performance.now() - t0);
|
|
126
|
-
return res;
|
|
127
|
-
}
|
|
128
|
-
|
|
129
|
-
export async function readFileTool({ path, limit }) {
|
|
130
|
-
const raw = String(path || "").trim();
|
|
131
|
-
if (!raw) return { ok: false, output: "empty path" };
|
|
132
|
-
let target = raw;
|
|
133
|
-
if (target.startsWith("~/")) {
|
|
134
|
-
const home = process.env.HOME || process.env.USERPROFILE || "";
|
|
135
|
-
target = join(home, target.slice(2));
|
|
136
|
-
} else if (!target.startsWith("/") && !/^[A-Za-z]:/.test(target)) {
|
|
137
|
-
// 相对路径视作项目根相对
|
|
138
|
-
target = join(pkgRoot, target);
|
|
139
|
-
}
|
|
140
|
-
if (!isAllowed(target)) return { ok: false, output: `not allowed: ${raw}(仅限项目目录或日志目录)` };
|
|
141
|
-
try {
|
|
142
|
-
if (!existsSync(target)) return { ok: false, output: `not found: ${raw}` };
|
|
143
|
-
const st = statSync(target);
|
|
144
|
-
if (st.isDirectory()) {
|
|
145
|
-
// 列目录
|
|
146
|
-
const { readdirSync } = await import("node:fs");
|
|
147
|
-
const files = readdirSync(target).slice(0, 80);
|
|
148
|
-
return { ok: true, output: `dir ${raw}:\n` + files.join("\n") };
|
|
149
|
-
}
|
|
150
|
-
const cap = Number(limit) > 0 ? Math.min(Number(limit), 20000) : 8000;
|
|
151
|
-
const data = readFileSync(target, "utf8");
|
|
152
|
-
const txt = data.length > cap ? data.slice(0, cap) + `\n... (truncated ${data.length - cap} chars)` : data;
|
|
153
|
-
return { ok: true, output: txt || "(empty file)" };
|
|
154
|
-
} catch (e) {
|
|
155
|
-
return { ok: false, output: String(e.message).slice(0, 2000) };
|
|
156
|
-
}
|
|
157
|
-
}
|
|
158
|
-
|
|
159
|
-
function resolveLocalPort() {
|
|
160
|
-
// 复用 state 的端口优先级:state.port > MSLXDFF_PORT env > 8989
|
|
161
|
-
try {
|
|
162
|
-
const f = defaultStateFile();
|
|
163
|
-
const j = JSON.parse(readFileSync(f, "utf8"));
|
|
164
|
-
const p = j?.port;
|
|
165
|
-
if (Number.isInteger(p) && p > 0) return p;
|
|
166
|
-
} catch {}
|
|
167
|
-
const env = Number(process.env.MSLXDFF_PORT);
|
|
168
|
-
if (Number.isInteger(env) && env > 0) return env;
|
|
169
|
-
return 8989;
|
|
170
|
-
}
|
|
171
|
-
|
|
172
|
-
function resolveLocalToken() {
|
|
173
|
-
try {
|
|
174
|
-
const f = defaultStateFile();
|
|
175
|
-
const j = JSON.parse(readFileSync(f, "utf8"));
|
|
176
|
-
const t = j?.token;
|
|
177
|
-
if (typeof t === "string" && t.trim()) return t.trim();
|
|
178
|
-
} catch {}
|
|
179
|
-
return "";
|
|
180
|
-
}
|
|
181
|
-
|
|
182
|
-
function expandCurlUrl(raw) {
|
|
183
|
-
const s = String(raw || "").trim();
|
|
184
|
-
const low = s.toLowerCase();
|
|
185
|
-
const port = resolveLocalPort();
|
|
186
|
-
const localBase = `http://127.0.0.1:${port}`;
|
|
187
|
-
// 完整 http(s) 直接用
|
|
188
|
-
if (/^https?:\/\//i.test(s)) return s;
|
|
189
|
-
// 简写映射
|
|
190
|
-
if (low === "upstream" || low === "upstream/models" || low === "upstream/models/") return "https://opencode.ai/zen/v1/models";
|
|
191
|
-
if (low === "upstream/chat" || low === "upstream/chat/completions") return "https://opencode.ai/zen/v1/chat/completions";
|
|
192
|
-
if (low === "local" || low === "local/health" || low === "health") return `${localBase}/health`;
|
|
193
|
-
if (low === "local/models" || low === "models" || low === "v1/models") return `${localBase}/v1/models`;
|
|
194
|
-
if (low === "local/chat" || low === "v1/chat/completions") return `${localBase}/v1/chat/completions`;
|
|
195
|
-
// 供应商简写:bai/models、openrouter/models 等 → 该供应商 baseUrl + /models
|
|
196
|
-
const provMatch = low.match(/^([a-z0-9_-]+)\/models\/?$/);
|
|
197
|
-
if (provMatch) {
|
|
198
|
-
try {
|
|
199
|
-
const pid = provMatch[1];
|
|
200
|
-
const cfgs = loadProviderConfigs();
|
|
201
|
-
const cfg = cfgs[pid];
|
|
202
|
-
if (cfg?.baseUrl) return `${cfg.baseUrl.replace(/\/+$/, "")}/models`;
|
|
203
|
-
} catch {}
|
|
204
|
-
}
|
|
205
|
-
// 裸路径视作本机相对
|
|
206
|
-
if (s.startsWith("/")) return `${localBase}${s}`;
|
|
207
|
-
return s;
|
|
208
|
-
}
|
|
209
|
-
|
|
210
|
-
export async function curlTool({ url, method, headers, body, timeoutMs }) {
|
|
211
|
-
const raw = String(url || "").trim();
|
|
212
|
-
if (!raw) return { ok: false, output: "empty url" };
|
|
213
|
-
let target = expandCurlUrl(raw);
|
|
214
|
-
// 仅允许 http(s)
|
|
215
|
-
if (!/^https?:\/\//i.test(target)) return { ok: false, output: `only http(s) allowed: ${raw} -> ${target}` };
|
|
216
|
-
let m = String(method || "GET").toUpperCase();
|
|
217
|
-
if (!["GET", "POST", "PUT", "DELETE", "HEAD", "PATCH", "OPTIONS"].includes(m)) return { ok: false, output: `unsupported method: ${m}` };
|
|
218
|
-
const timeout = Math.min(Math.max(Number(timeoutMs) || 8000, 500), 15000);
|
|
219
|
-
// 组装头
|
|
220
|
-
const h = {};
|
|
221
|
-
if (headers && typeof headers === "object") {
|
|
222
|
-
for (const [k, v] of Object.entries(headers)) if (typeof v === "string") h[String(k)] = v;
|
|
223
|
-
}
|
|
224
|
-
// 上游自动补最小可用头(zen 免费层门禁:UA 带版本 + opencode 形状 session/request)
|
|
225
|
-
if (/opencode\.ai/i.test(target)) {
|
|
226
|
-
const ident = opencodeClientIdentity();
|
|
227
|
-
if (!h["x-opencode-client"]) h["x-opencode-client"] = "desktop";
|
|
228
|
-
if (!h["Authorization"] && !h["authorization"]) h["Authorization"] = "Bearer public";
|
|
229
|
-
if (!h["Accept"]) h["Accept"] = m === "GET" ? "*/*" : "application/json";
|
|
230
|
-
if (!h["Content-Type"] && m !== "GET" && m !== "HEAD") h["Content-Type"] = "application/json";
|
|
231
|
-
if (!h["User-Agent"] && !h["user-agent"]) h["User-Agent"] = ident["User-Agent"];
|
|
232
|
-
if (!h["x-opencode-session"]) h["x-opencode-session"] = ident["x-opencode-session"];
|
|
233
|
-
if (!h["x-opencode-request"]) h["x-opencode-request"] = ident["x-opencode-request"];
|
|
234
|
-
if (!h["x-opencode-project"]) h["x-opencode-project"] = "global";
|
|
235
|
-
}
|
|
236
|
-
// 本机自动带 token(/v1/* 需要鉴权);/v1/models 必须是 GET,模型常误用 POST 直接纠正
|
|
237
|
-
if (/127\.0\.0\.1|localhost/i.test(target) && /\/v1\//i.test(target)) {
|
|
238
|
-
const hasAuth = !!(h["Authorization"] || h["authorization"]);
|
|
239
|
-
if (!hasAuth) {
|
|
240
|
-
const tok = resolveLocalToken();
|
|
241
|
-
if (tok) h["Authorization"] = `Bearer ${tok}`;
|
|
242
|
-
}
|
|
243
|
-
if (/\/v1\/models/i.test(target) && m === "POST") m = "GET";
|
|
244
|
-
}
|
|
245
|
-
// 已配置供应商自动带 key(curl 直连供应商时免手动传头)
|
|
246
|
-
if (!h["Authorization"] && !h["authorization"]) {
|
|
247
|
-
try {
|
|
248
|
-
const cfgs = loadProviderConfigs();
|
|
249
|
-
for (const [pid, cfg] of Object.entries(cfgs)) {
|
|
250
|
-
const base = String(cfg?.baseUrl || "").replace(/\/+$/, "");
|
|
251
|
-
if (!base) continue;
|
|
252
|
-
if (target.toLowerCase().startsWith(base.toLowerCase())) {
|
|
253
|
-
const keys = loadProviderKeys(pid);
|
|
254
|
-
if (keys.length) { h["Authorization"] = `Bearer ${keys[0]}`; break; }
|
|
255
|
-
}
|
|
256
|
-
}
|
|
257
|
-
// 兜底:api.b.ai 即 bai 供应商(env 或 state)
|
|
258
|
-
if (!h["Authorization"] && /api\.b\.ai/i.test(target)) {
|
|
259
|
-
const k = loadProviderKeys("bai")[0];
|
|
260
|
-
if (k) h["Authorization"] = `Bearer ${k}`;
|
|
261
|
-
}
|
|
262
|
-
} catch {}
|
|
263
|
-
}
|
|
264
|
-
// fetch 走兼容层(undici 优先,老 Node 兜底)
|
|
265
|
-
const fetchImpl = compatFetch;
|
|
266
|
-
const controller = new AbortController();
|
|
267
|
-
const timer = setTimeout(() => controller.abort(new Error(`curl timed out after ${timeout}ms`)), timeout);
|
|
268
|
-
const t0 = performance.now();
|
|
269
|
-
try {
|
|
270
|
-
const opts = { method: m, headers: h, signal: controller.signal };
|
|
271
|
-
if (body != null && m !== "GET" && m !== "HEAD") opts.body = String(body);
|
|
272
|
-
const res = await fetchImpl(target, opts);
|
|
273
|
-
const ms = Math.round(performance.now() - t0);
|
|
274
|
-
// 抓响应头(过滤成可读)
|
|
275
|
-
const rh = {};
|
|
276
|
-
try {
|
|
277
|
-
for (const [k, v] of res.headers.entries()) rh[k] = v;
|
|
278
|
-
} catch {}
|
|
279
|
-
let txt = "";
|
|
280
|
-
try {
|
|
281
|
-
// 限制读取大小,避免超大 body
|
|
282
|
-
const ab = await res.arrayBuffer();
|
|
283
|
-
const buf = Buffer.from(ab);
|
|
284
|
-
const lim = 6000;
|
|
285
|
-
txt = buf.toString("utf8").slice(0, lim);
|
|
286
|
-
if (buf.length > lim) txt += `\n... (truncated ${buf.length - lim} bytes)`;
|
|
287
|
-
} catch (e) {
|
|
288
|
-
txt = String(e.message || e).slice(0, 1000);
|
|
289
|
-
}
|
|
290
|
-
const headerLines = Object.entries(rh).slice(0, 20).map(([k, v]) => `${k}: ${v}`).join("\n");
|
|
291
|
-
const out = [
|
|
292
|
-
`${m} ${target}`,
|
|
293
|
-
`status: ${res.status} ${res.statusText || ""}`.trim(),
|
|
294
|
-
`latency: ${ms}ms`,
|
|
295
|
-
headerLines ? `headers:\n${headerLines}` : "headers: (none)",
|
|
296
|
-
`body (${txt.length} chars):`,
|
|
297
|
-
txt || "(empty body)",
|
|
298
|
-
].join("\n");
|
|
299
|
-
// 2xx 视为 ok,其余仍返回 ok:true 但带状态,便于模型判断(网络层面成功)
|
|
300
|
-
return { ok: true, output: out.slice(0, 8000) };
|
|
301
|
-
} catch (e) {
|
|
302
|
-
const ms = Math.round(performance.now() - t0);
|
|
303
|
-
return { ok: false, output: `curl failed: ${String(e.message || e).slice(0, 1200)}\nurl: ${target}\nlatency: ${ms}ms` };
|
|
304
|
-
} finally {
|
|
305
|
-
clearTimeout(timer);
|
|
306
|
-
}
|
|
307
|
-
}
|
|
11
|
+
|
|
12
|
+
const pkgRoot = resolve(dirname(fileURLToPath(import.meta.url)), "../..");
|
|
13
|
+
const allowedRoots = [
|
|
14
|
+
resolve(pkgRoot),
|
|
15
|
+
resolve(logDir()),
|
|
16
|
+
resolve(dirname(defaultStateFile())),
|
|
17
|
+
];
|
|
18
|
+
|
|
19
|
+
function isAllowed(p) {
|
|
20
|
+
const abs = resolve(p);
|
|
21
|
+
// 展开 ~/ 形式
|
|
22
|
+
const expanded = abs.startsWith("~/") ? join(process.env.HOME || process.env.USERPROFILE || "", abs.slice(2)) : abs;
|
|
23
|
+
const chk = resolve(expanded);
|
|
24
|
+
return allowedRoots.some((r) => chk === r || chk.startsWith(r + "/") || chk.startsWith(r + "\\"));
|
|
25
|
+
}
|
|
26
|
+
|
|
27
|
+
function parseCommand(str) {
|
|
28
|
+
// 支持引号包裹
|
|
29
|
+
const out = [];
|
|
30
|
+
let cur = "";
|
|
31
|
+
let q = null;
|
|
32
|
+
for (let i = 0; i < str.length; i++) {
|
|
33
|
+
const c = str[i];
|
|
34
|
+
if (q) {
|
|
35
|
+
if (c === q) q = null;
|
|
36
|
+
else cur += c;
|
|
37
|
+
} else if (c === '"' || c === "'") {
|
|
38
|
+
q = c;
|
|
39
|
+
} else if (c === " " || c === "\t") {
|
|
40
|
+
if (cur) { out.push(cur); cur = ""; }
|
|
41
|
+
} else {
|
|
42
|
+
cur += c;
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
if (cur) out.push(cur);
|
|
46
|
+
return out;
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
export function getToolDefs() {
|
|
50
|
+
return [
|
|
51
|
+
{
|
|
52
|
+
type: "function",
|
|
53
|
+
function: {
|
|
54
|
+
name: "run_command",
|
|
55
|
+
description: "执行一条 mslxdff CLI 命令(不含 mslxdff 前缀)。仅限 cli_help_mini 所列命令,禁止 -uninstall;禁止 mslxdff \"hi\" --model X / --model X \"hi\" / -chat --model X 等幻觉命令,探活模型必须用 curl 工具 POST 本机 /v1/chat/completions。",
|
|
56
|
+
parameters: {
|
|
57
|
+
type: "object",
|
|
58
|
+
properties: {
|
|
59
|
+
command: { type: "string", description: "例如: -model set hy3-free 或 -group list 或 -log 20;模型探活禁止用此工具,必须用 curl POST http://localhost:8989/v1/chat/completions" },
|
|
60
|
+
},
|
|
61
|
+
required: ["command"],
|
|
62
|
+
},
|
|
63
|
+
},
|
|
64
|
+
},
|
|
65
|
+
{
|
|
66
|
+
type: "function",
|
|
67
|
+
function: {
|
|
68
|
+
name: "read_file",
|
|
69
|
+
description: "读取项目内的文件内容(src/ docs/ logs/ 等),用于查看日志或配置。禁止读取项目外文件。",
|
|
70
|
+
parameters: {
|
|
71
|
+
type: "object",
|
|
72
|
+
properties: {
|
|
73
|
+
path: { type: "string", description: "相对项目根或绝对路径,如 src/logs.js 或 ~/.config/mslxdff/events.log" },
|
|
74
|
+
limit: { type: "number", description: "最多读取字符数,默认 8000" },
|
|
75
|
+
},
|
|
76
|
+
required: ["path"],
|
|
77
|
+
},
|
|
78
|
+
},
|
|
79
|
+
},
|
|
80
|
+
{
|
|
81
|
+
type: "function",
|
|
82
|
+
function: {
|
|
83
|
+
name: "curl",
|
|
84
|
+
description: "网络/HTTP 探活,检测上游或本机服务可用性。支持任意 http(s) URL,返回状态码、耗时、响应头与前几千字符。常用: upstream(上游模型列表)、local/health(本机健康)、local/models(本机模型列表)。探活指定模型必须用 POST http://localhost:8989/v1/chat/completions body {\"model\":\"<provider/模型>\",\"messages\":[{\"role\":\"user\",\"content\":\"hi\"}]}。简写会自动补全为完整 URL。",
|
|
85
|
+
parameters: {
|
|
86
|
+
type: "object",
|
|
87
|
+
properties: {
|
|
88
|
+
url: { type: "string", description: "要请求的 URL 或简写:upstream、upstream/models、upstream/chat、local、local/health、local/models、health、models,或完整 http(s) URL" },
|
|
89
|
+
method: { type: "string", description: "HTTP 方法,默认 GET", enum: ["GET", "POST", "PUT", "DELETE", "HEAD", "PATCH", "OPTIONS"] },
|
|
90
|
+
headers: { type: "object", description: "附加请求头,键值均为字符串", additionalProperties: { type: "string" } },
|
|
91
|
+
body: { type: "string", description: "请求体(POST/PUT 时),字符串,如 JSON" },
|
|
92
|
+
timeoutMs: { type: "number", description: "超时毫秒,默认 8000,最大 15000" },
|
|
93
|
+
},
|
|
94
|
+
required: ["url"],
|
|
95
|
+
},
|
|
96
|
+
},
|
|
97
|
+
},
|
|
98
|
+
];
|
|
99
|
+
}
|
|
100
|
+
|
|
101
|
+
export function validateCommand(cmd) {
|
|
102
|
+
const low = String(cmd || "").toLowerCase();
|
|
103
|
+
for (const f of FORBIDDEN) if (low.includes(f)) return `forbidden: ${f} 已拦截,禁止执行`;
|
|
104
|
+
return null;
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
export async function execCommand(command) {
|
|
108
|
+
const err = validateCommand(command);
|
|
109
|
+
if (err) return { ok: false, output: err };
|
|
110
|
+
const args = parseCommand(String(command).trim());
|
|
111
|
+
if (!args.length) return { ok: false, output: "empty command" };
|
|
112
|
+
// 关键路径打点:exec 本身耗时(fork+加载)是 -chat 最大的本地开销,后续可改直调以省 300-500ms
|
|
113
|
+
const bin = join(pkgRoot, "bin/mslxdff.js");
|
|
114
|
+
const t0 = performance.now();
|
|
115
|
+
const res = await new Promise((resolve) => {
|
|
116
|
+
execFile(process.execPath, [bin, ...args], { timeout: 15000, maxBuffer: 1024 * 500 }, (e, stdout, stderr) => {
|
|
117
|
+
const out = String(stdout || "") + (stderr ? `\n${stderr}` : "");
|
|
118
|
+
if (e) {
|
|
119
|
+
resolve({ ok: false, output: out.slice(0, 8000) || String(e.message).slice(0, 2000) });
|
|
120
|
+
} else {
|
|
121
|
+
resolve({ ok: true, output: out.slice(0, 8000) || "(no output)" });
|
|
122
|
+
}
|
|
123
|
+
});
|
|
124
|
+
});
|
|
125
|
+
res._ms = Math.round(performance.now() - t0);
|
|
126
|
+
return res;
|
|
127
|
+
}
|
|
128
|
+
|
|
129
|
+
export async function readFileTool({ path, limit }) {
|
|
130
|
+
const raw = String(path || "").trim();
|
|
131
|
+
if (!raw) return { ok: false, output: "empty path" };
|
|
132
|
+
let target = raw;
|
|
133
|
+
if (target.startsWith("~/")) {
|
|
134
|
+
const home = process.env.HOME || process.env.USERPROFILE || "";
|
|
135
|
+
target = join(home, target.slice(2));
|
|
136
|
+
} else if (!target.startsWith("/") && !/^[A-Za-z]:/.test(target)) {
|
|
137
|
+
// 相对路径视作项目根相对
|
|
138
|
+
target = join(pkgRoot, target);
|
|
139
|
+
}
|
|
140
|
+
if (!isAllowed(target)) return { ok: false, output: `not allowed: ${raw}(仅限项目目录或日志目录)` };
|
|
141
|
+
try {
|
|
142
|
+
if (!existsSync(target)) return { ok: false, output: `not found: ${raw}` };
|
|
143
|
+
const st = statSync(target);
|
|
144
|
+
if (st.isDirectory()) {
|
|
145
|
+
// 列目录
|
|
146
|
+
const { readdirSync } = await import("node:fs");
|
|
147
|
+
const files = readdirSync(target).slice(0, 80);
|
|
148
|
+
return { ok: true, output: `dir ${raw}:\n` + files.join("\n") };
|
|
149
|
+
}
|
|
150
|
+
const cap = Number(limit) > 0 ? Math.min(Number(limit), 20000) : 8000;
|
|
151
|
+
const data = readFileSync(target, "utf8");
|
|
152
|
+
const txt = data.length > cap ? data.slice(0, cap) + `\n... (truncated ${data.length - cap} chars)` : data;
|
|
153
|
+
return { ok: true, output: txt || "(empty file)" };
|
|
154
|
+
} catch (e) {
|
|
155
|
+
return { ok: false, output: String(e.message).slice(0, 2000) };
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
|
|
159
|
+
function resolveLocalPort() {
|
|
160
|
+
// 复用 state 的端口优先级:state.port > MSLXDFF_PORT env > 8989
|
|
161
|
+
try {
|
|
162
|
+
const f = defaultStateFile();
|
|
163
|
+
const j = JSON.parse(readFileSync(f, "utf8"));
|
|
164
|
+
const p = j?.port;
|
|
165
|
+
if (Number.isInteger(p) && p > 0) return p;
|
|
166
|
+
} catch {}
|
|
167
|
+
const env = Number(process.env.MSLXDFF_PORT);
|
|
168
|
+
if (Number.isInteger(env) && env > 0) return env;
|
|
169
|
+
return 8989;
|
|
170
|
+
}
|
|
171
|
+
|
|
172
|
+
function resolveLocalToken() {
|
|
173
|
+
try {
|
|
174
|
+
const f = defaultStateFile();
|
|
175
|
+
const j = JSON.parse(readFileSync(f, "utf8"));
|
|
176
|
+
const t = j?.token;
|
|
177
|
+
if (typeof t === "string" && t.trim()) return t.trim();
|
|
178
|
+
} catch {}
|
|
179
|
+
return "";
|
|
180
|
+
}
|
|
181
|
+
|
|
182
|
+
function expandCurlUrl(raw) {
|
|
183
|
+
const s = String(raw || "").trim();
|
|
184
|
+
const low = s.toLowerCase();
|
|
185
|
+
const port = resolveLocalPort();
|
|
186
|
+
const localBase = `http://127.0.0.1:${port}`;
|
|
187
|
+
// 完整 http(s) 直接用
|
|
188
|
+
if (/^https?:\/\//i.test(s)) return s;
|
|
189
|
+
// 简写映射
|
|
190
|
+
if (low === "upstream" || low === "upstream/models" || low === "upstream/models/") return "https://opencode.ai/zen/v1/models";
|
|
191
|
+
if (low === "upstream/chat" || low === "upstream/chat/completions") return "https://opencode.ai/zen/v1/chat/completions";
|
|
192
|
+
if (low === "local" || low === "local/health" || low === "health") return `${localBase}/health`;
|
|
193
|
+
if (low === "local/models" || low === "models" || low === "v1/models") return `${localBase}/v1/models`;
|
|
194
|
+
if (low === "local/chat" || low === "v1/chat/completions") return `${localBase}/v1/chat/completions`;
|
|
195
|
+
// 供应商简写:bai/models、openrouter/models 等 → 该供应商 baseUrl + /models
|
|
196
|
+
const provMatch = low.match(/^([a-z0-9_-]+)\/models\/?$/);
|
|
197
|
+
if (provMatch) {
|
|
198
|
+
try {
|
|
199
|
+
const pid = provMatch[1];
|
|
200
|
+
const cfgs = loadProviderConfigs();
|
|
201
|
+
const cfg = cfgs[pid];
|
|
202
|
+
if (cfg?.baseUrl) return `${cfg.baseUrl.replace(/\/+$/, "")}/models`;
|
|
203
|
+
} catch {}
|
|
204
|
+
}
|
|
205
|
+
// 裸路径视作本机相对
|
|
206
|
+
if (s.startsWith("/")) return `${localBase}${s}`;
|
|
207
|
+
return s;
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
export async function curlTool({ url, method, headers, body, timeoutMs }) {
|
|
211
|
+
const raw = String(url || "").trim();
|
|
212
|
+
if (!raw) return { ok: false, output: "empty url" };
|
|
213
|
+
let target = expandCurlUrl(raw);
|
|
214
|
+
// 仅允许 http(s)
|
|
215
|
+
if (!/^https?:\/\//i.test(target)) return { ok: false, output: `only http(s) allowed: ${raw} -> ${target}` };
|
|
216
|
+
let m = String(method || "GET").toUpperCase();
|
|
217
|
+
if (!["GET", "POST", "PUT", "DELETE", "HEAD", "PATCH", "OPTIONS"].includes(m)) return { ok: false, output: `unsupported method: ${m}` };
|
|
218
|
+
const timeout = Math.min(Math.max(Number(timeoutMs) || 8000, 500), 15000);
|
|
219
|
+
// 组装头
|
|
220
|
+
const h = {};
|
|
221
|
+
if (headers && typeof headers === "object") {
|
|
222
|
+
for (const [k, v] of Object.entries(headers)) if (typeof v === "string") h[String(k)] = v;
|
|
223
|
+
}
|
|
224
|
+
// 上游自动补最小可用头(zen 免费层门禁:UA 带版本 + opencode 形状 session/request)
|
|
225
|
+
if (/opencode\.ai/i.test(target)) {
|
|
226
|
+
const ident = opencodeClientIdentity();
|
|
227
|
+
if (!h["x-opencode-client"]) h["x-opencode-client"] = "desktop";
|
|
228
|
+
if (!h["Authorization"] && !h["authorization"]) h["Authorization"] = "Bearer public";
|
|
229
|
+
if (!h["Accept"]) h["Accept"] = m === "GET" ? "*/*" : "application/json";
|
|
230
|
+
if (!h["Content-Type"] && m !== "GET" && m !== "HEAD") h["Content-Type"] = "application/json";
|
|
231
|
+
if (!h["User-Agent"] && !h["user-agent"]) h["User-Agent"] = ident["User-Agent"];
|
|
232
|
+
if (!h["x-opencode-session"]) h["x-opencode-session"] = ident["x-opencode-session"];
|
|
233
|
+
if (!h["x-opencode-request"]) h["x-opencode-request"] = ident["x-opencode-request"];
|
|
234
|
+
if (!h["x-opencode-project"]) h["x-opencode-project"] = "global";
|
|
235
|
+
}
|
|
236
|
+
// 本机自动带 token(/v1/* 需要鉴权);/v1/models 必须是 GET,模型常误用 POST 直接纠正
|
|
237
|
+
if (/127\.0\.0\.1|localhost/i.test(target) && /\/v1\//i.test(target)) {
|
|
238
|
+
const hasAuth = !!(h["Authorization"] || h["authorization"]);
|
|
239
|
+
if (!hasAuth) {
|
|
240
|
+
const tok = resolveLocalToken();
|
|
241
|
+
if (tok) h["Authorization"] = `Bearer ${tok}`;
|
|
242
|
+
}
|
|
243
|
+
if (/\/v1\/models/i.test(target) && m === "POST") m = "GET";
|
|
244
|
+
}
|
|
245
|
+
// 已配置供应商自动带 key(curl 直连供应商时免手动传头)
|
|
246
|
+
if (!h["Authorization"] && !h["authorization"]) {
|
|
247
|
+
try {
|
|
248
|
+
const cfgs = loadProviderConfigs();
|
|
249
|
+
for (const [pid, cfg] of Object.entries(cfgs)) {
|
|
250
|
+
const base = String(cfg?.baseUrl || "").replace(/\/+$/, "");
|
|
251
|
+
if (!base) continue;
|
|
252
|
+
if (target.toLowerCase().startsWith(base.toLowerCase())) {
|
|
253
|
+
const keys = loadProviderKeys(pid);
|
|
254
|
+
if (keys.length) { h["Authorization"] = `Bearer ${keys[0]}`; break; }
|
|
255
|
+
}
|
|
256
|
+
}
|
|
257
|
+
// 兜底:api.b.ai 即 bai 供应商(env 或 state)
|
|
258
|
+
if (!h["Authorization"] && /api\.b\.ai/i.test(target)) {
|
|
259
|
+
const k = loadProviderKeys("bai")[0];
|
|
260
|
+
if (k) h["Authorization"] = `Bearer ${k}`;
|
|
261
|
+
}
|
|
262
|
+
} catch {}
|
|
263
|
+
}
|
|
264
|
+
// fetch 走兼容层(undici 优先,老 Node 兜底)
|
|
265
|
+
const fetchImpl = compatFetch;
|
|
266
|
+
const controller = new AbortController();
|
|
267
|
+
const timer = setTimeout(() => controller.abort(new Error(`curl timed out after ${timeout}ms`)), timeout);
|
|
268
|
+
const t0 = performance.now();
|
|
269
|
+
try {
|
|
270
|
+
const opts = { method: m, headers: h, signal: controller.signal };
|
|
271
|
+
if (body != null && m !== "GET" && m !== "HEAD") opts.body = String(body);
|
|
272
|
+
const res = await fetchImpl(target, opts);
|
|
273
|
+
const ms = Math.round(performance.now() - t0);
|
|
274
|
+
// 抓响应头(过滤成可读)
|
|
275
|
+
const rh = {};
|
|
276
|
+
try {
|
|
277
|
+
for (const [k, v] of res.headers.entries()) rh[k] = v;
|
|
278
|
+
} catch {}
|
|
279
|
+
let txt = "";
|
|
280
|
+
try {
|
|
281
|
+
// 限制读取大小,避免超大 body
|
|
282
|
+
const ab = await res.arrayBuffer();
|
|
283
|
+
const buf = Buffer.from(ab);
|
|
284
|
+
const lim = 6000;
|
|
285
|
+
txt = buf.toString("utf8").slice(0, lim);
|
|
286
|
+
if (buf.length > lim) txt += `\n... (truncated ${buf.length - lim} bytes)`;
|
|
287
|
+
} catch (e) {
|
|
288
|
+
txt = String(e.message || e).slice(0, 1000);
|
|
289
|
+
}
|
|
290
|
+
const headerLines = Object.entries(rh).slice(0, 20).map(([k, v]) => `${k}: ${v}`).join("\n");
|
|
291
|
+
const out = [
|
|
292
|
+
`${m} ${target}`,
|
|
293
|
+
`status: ${res.status} ${res.statusText || ""}`.trim(),
|
|
294
|
+
`latency: ${ms}ms`,
|
|
295
|
+
headerLines ? `headers:\n${headerLines}` : "headers: (none)",
|
|
296
|
+
`body (${txt.length} chars):`,
|
|
297
|
+
txt || "(empty body)",
|
|
298
|
+
].join("\n");
|
|
299
|
+
// 2xx 视为 ok,其余仍返回 ok:true 但带状态,便于模型判断(网络层面成功)
|
|
300
|
+
return { ok: true, output: out.slice(0, 8000) };
|
|
301
|
+
} catch (e) {
|
|
302
|
+
const ms = Math.round(performance.now() - t0);
|
|
303
|
+
return { ok: false, output: `curl failed: ${String(e.message || e).slice(0, 1200)}\nurl: ${target}\nlatency: ${ms}ms` };
|
|
304
|
+
} finally {
|
|
305
|
+
clearTimeout(timer);
|
|
306
|
+
}
|
|
307
|
+
}
|
|
@@ -101,7 +101,7 @@ export async function runAutoRace(ctx, deps = {}) {
|
|
|
101
101
|
else if (r.status === "rejected") await auto.recordError(m, { status: 502 });
|
|
102
102
|
}
|
|
103
103
|
handlerCtx.model = winModel;
|
|
104
|
-
const lr = await localRelay({ upRes: best.val.res, model: winModel, body, order: raceModels, idx: best.idx, lastErr: null, requested, useAuto, lockModel, auto, handlerCtx, evt, logCall, logError, mark, perf0, stages, startedAt, plugins, res });
|
|
104
|
+
const lr = await localRelay({ upRes: best.val.res, model: winModel, body, order: raceModels, idx: best.idx, lastErr: null, requested, useAuto, lockModel, auto, handlerCtx, evt, logCall, logError, mark, perf0, attemptStartMs: raceStart, stages, startedAt, plugins, res });
|
|
105
105
|
if (lr.handled) return { done: true };
|
|
106
106
|
if (!lr.lastErr) return { done: true };
|
|
107
107
|
} else {
|