106 lines
4.4 KiB
JavaScript
106 lines
4.4 KiB
JavaScript
// 让本仓库发同一次真实请求，落到 capture-server 上。
|
|
// 用法: node drive-wa.mjs --preset "C:/Users/zed/Downloads/linux_1.3.json" --message 你好 --capture-port 8802
|
|
// 用独立 WRITING_AGENT_DATA_DIR，不碰 ~/.writing-agent 里的配置。
|
|
// 预设走本仓库自己的 /api/presets/import（纯导入 json），之后走直聊的 ephemeral complete 真发一次。
|
|
import { spawn } from "node:child_process";
|
|
import { mkdirSync, readFileSync, rmSync, writeFileSync } from "node:fs";
|
|
import path from "node:path";
|
|
import { fileURLToPath } from "node:url";
|
|
import { listCaptures, makeLogger, parseArgs, waitForNewCapture, waitHttp } from "./lib.mjs";
|
|
|
|
const HERE = path.dirname(fileURLToPath(import.meta.url));
|
|
const REPO = path.resolve(HERE, "../..");
|
|
const log = makeLogger("drive-wa");
|
|
|
|
const DEFAULTS = {
|
|
preset: "",
|
|
message: "你好",
|
|
capturePort: 8802,
|
|
port: 23377,
|
|
out: path.join(HERE, "captures"),
|
|
dataRoot: path.join(HERE, ".wa-data"),
|
|
timeoutMs: 120000,
|
|
};
|
|
|
|
async function postJson(url, body) {
|
|
const res = await fetch(url, { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify(body) });
|
|
const text = await res.text();
|
|
let data = null;
|
|
try { data = JSON.parse(text); } catch (e) { data = { raw: text.slice(0, 800) }; }
|
|
if (!res.ok) throw new Error("HTTP " + res.status + " " + JSON.stringify(data).slice(0, 800));
|
|
return data;
|
|
}
|
|
|
|
async function main() {
|
|
const args = parseArgs(process.argv.slice(2), DEFAULTS);
|
|
if (!args.preset) throw new Error("缺少 --preset");
|
|
const presetName = path.basename(args.preset, ".json");
|
|
const raw = JSON.parse(readFileSync(args.preset, "utf8"));
|
|
mkdirSync(args.dataRoot, { recursive: true });
|
|
mkdirSync(args.out, { recursive: true });
|
|
|
|
// 一条指向假上游的 API 配置。直接写盘，避免能力探测的额外请求混进抓包。
|
|
const now = new Date().toISOString();
|
|
const profiles = {
|
|
version: 1,
|
|
profiles: [
|
|
{
|
|
id: "probe-mock",
|
|
name: "mock",
|
|
baseUrl: "http://127.0.0.1:" + args.capturePort + "/v1",
|
|
apiKey: "sk-mock",
|
|
model: "mock-model",
|
|
createdAt: now,
|
|
updatedAt: now,
|
|
},
|
|
],
|
|
};
|
|
writeFileSync(path.join(args.dataRoot, "profiles.json"), JSON.stringify(profiles, null, 2), "utf8");
|
|
|
|
try { rmSync(path.join(args.dataRoot, "presets"), { recursive: true, force: true }); } catch (e) {} const known = new Set(listCaptures(args.out, "wa"));
|
|
const env = { ...process.env, PORT: String(args.port), WRITING_AGENT_DATA_DIR: args.dataRoot };
|
|
const server = spawn(process.execPath, [path.join(REPO, "node_modules/tsx/dist/cli.mjs"), path.join(REPO, "src/server/web-server.ts")], { cwd: REPO, env, stdio: ["ignore", "pipe", "pipe"] });
|
|
const logs = [];
|
|
server.stdout.on("data", (d) => logs.push(String(d)));
|
|
server.stderr.on("data", (d) => logs.push(String(d)));
|
|
const base = "http://127.0.0.1:" + args.port;
|
|
try {
|
|
await waitHttp(base + "/api/health", args.timeoutMs, "本仓库服务起来");
|
|
log("服务已启动: " + base);
|
|
|
|
const imported = await postJson(base + "/api/presets/import", { raw, name: presetName, activate: true });
|
|
const presetId = imported.preset.id;
|
|
log("预设已导入: " + presetName + " (id=" + presetId + ")");
|
|
if (imported.warnings && imported.warnings.length) log("导入提示: " + JSON.stringify(imported.warnings).slice(0, 600));
|
|
|
|
const body = {
|
|
text: args.message,
|
|
presetId,
|
|
complete: true,
|
|
turns: [],
|
|
windows: [{ id: "w1", profileId: "probe-mock" }],
|
|
systemPrompt: "",
|
|
loreBefore: "",
|
|
seedHistory: "",
|
|
loreAfter: "",
|
|
postTurn: "",
|
|
};
|
|
const result = await postJson(base + "/api/direct-chat/ephemeral/complete", body);
|
|
const first = result.multi && result.multi.results && result.multi.results[0];
|
|
if (first && first.error) throw new Error("本轮请求失败: " + first.error);
|
|
log("本轮完成, reply=" + JSON.stringify(String((first && first.reply) || "").slice(0, 40)));
|
|
|
|
const captureFile = await waitForNewCapture(args.out, "wa", known, args.timeoutMs);
|
|
log("抓到请求: " + captureFile);
|
|
console.log(JSON.stringify({ side: "wa", preset: presetName, capture: path.join(args.out, captureFile) }));
|
|
} catch (err) {
|
|
console.error("[drive-wa] 失败: " + err.message);
|
|
if (logs.length) console.error(logs.join("").slice(-4000));
|
|
process.exitCode = 1;
|
|
} finally {
|
|
try { server.kill(); } catch (e) {}
|
|
}
|
|
}
|
|
|
|
await main();
|