51 lines
2.5 KiB
TypeScript
51 lines
2.5 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import { spawnSync } from "node:child_process";
|
|
import { mkdtempSync, mkdirSync, readFileSync, rmSync, writeFileSync } from "node:fs";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { test } from "node:test";
|
|
|
|
test("思考核查：列目录不截断，首次读源码才截断，并保留原始行号", () => {
|
|
const root = mkdtempSync(join(tmpdir(), "inforsi-route-thinking-"));
|
|
try {
|
|
const call = join(root, "calls/r0-entryDraft.researcher-1.entry.none.1");
|
|
const sessions = join(call, "sessions");
|
|
const code = join(call, "code");
|
|
mkdirSync(sessions, { recursive: true });
|
|
mkdirSync(code);
|
|
writeFileSync(join(code, "initial.py"), "print('baseline')\n");
|
|
writeFileSync(join(code, "SOURCES.md"), "reference\n");
|
|
const messages = [
|
|
{ type: "session" },
|
|
{ type: "message", message: { role: "assistant", content: [
|
|
{ type: "thinking", thinking: "列目录前" },
|
|
{ type: "toolCall", name: "read_source", arguments: { path: code } },
|
|
] } },
|
|
{ type: "message", message: { role: "assistant", content: [
|
|
{ type: "thinking", thinking: "目录之后先想两种解法" },
|
|
{ type: "toolCall", name: "read_source", arguments: { path: join(code, "SOURCES.md") } },
|
|
] } },
|
|
{ type: "message", message: { role: "assistant", content: [
|
|
{ type: "thinking", thinking: "看过来源说明，还未读源码" },
|
|
{ type: "toolCall", name: "read_source", arguments: { path: join(code, "initial.py") } },
|
|
] } },
|
|
{ type: "message", message: { role: "assistant", content: [
|
|
{ type: "thinking", thinking: "读源码后的内容不能算先想" },
|
|
] } },
|
|
];
|
|
writeFileSync(join(sessions, "session.jsonl"), messages.map((m) => JSON.stringify(m)).join("\n"));
|
|
const output = join(root, "thinking.txt");
|
|
const result = spawnSync(process.execPath, ["eval/route/thinking.ts", root, output], { encoding: "utf8" });
|
|
assert.equal(result.status, 0, result.stderr);
|
|
const text = readFileSync(output, "utf8");
|
|
assert.match(text, /目录之后先想两种解法/);
|
|
assert.match(text, /看过来源说明，还未读源码/);
|
|
assert.match(text, /原始 JSONL:4/);
|
|
assert.match(text, /仅列目录/);
|
|
assert.match(text, /首次读取基线源码文件/);
|
|
assert.doesNotMatch(text, /读源码后的内容不能算先想/);
|
|
} finally {
|
|
rmSync(root, { recursive: true, force: true });
|
|
}
|
|
});
|