mirror of
https://bitbucket.org/siakitem/my-pi.git
synced 2026-10-26 18:55:22 +00:00
云服务部-叶林立
312bb2ba5a
从官方 Git 仓库导入 chandra447/pi-hermes-memory 在 tag v0.9.10 （df624056a2e8c68bc89ba0a81e1c120856ec9665，2026-10-02）的源码快照。 本次仅导入源码，尚未接入组合包：package.json、加载顺序、依赖与权限策略 均未改动。 - 原样导入 src/、tests/、docs/、scripts/、README、CHANGELOG、PLAN、 AGENTS 与 LICENSE；该快照共 156 个上游文件。 - 剔除嵌套 .git、上游 .github 工作流、.ralph/、.claude/、node_modules、 覆盖率与构建产物。 - 新增中文 UPSTREAM.md，记录来源，以及启用前必须完成的待决事项： 在 pi-compaction 之前注册加载顺序、better-sqlite3 版本与根 allowScripts 对齐、strip-ansi 解析确认。 - 说明选择 tag 而非 main 的原因：导入时上游 main 已领先至 24fcd8ad26fc27badb8d7dbaf8034645de07b893（2026-10-11），选 tag 是为 对应一个已发布、可复现的版本。 尚未接线。启用前还需反转 tests/extension-load.test.ts 中"不得加载 Hermes"的断言，并同步 pi-tool-search 分组、权限策略与文档。 验证：快照 156 个上游文件全部导入，目录内仅多出 UPSTREAM.md 一份； 已确认与上游 main 内容存在差异，证明取的是 tag 而非 main。
73 lines
3.0 KiB
JavaScript
73 lines
3.0 KiB
JavaScript
// Run from a development checkout: node --import tsx scripts/benchmark-memory-startup.mjs [--lazy] [entry.ts]
|
|
// Measures the extension lifecycle, not Pi's TUI. No model calls or user data.
|
|
import { mkdtemp, mkdir, writeFile, rm } from "node:fs/promises";
|
|
import { existsSync } from "node:fs";
|
|
import { tmpdir } from "node:os";
|
|
import { dirname, join, resolve } from "node:path";
|
|
import { fileURLToPath, pathToFileURL } from "node:url";
|
|
|
|
const lazy = process.argv.includes("--lazy");
|
|
const entry = process.argv.slice(2).find((arg) => arg !== "--lazy")
|
|
?? join(dirname(fileURLToPath(import.meta.url)), "../src/index.ts");
|
|
const root = await mkdtemp(join(tmpdir(), "pi-memory-startup-bench-"));
|
|
const globalDir = join(root, "pi-hermes-memory");
|
|
const handlers = new Map();
|
|
const tools = new Map();
|
|
const ctx = {
|
|
cwd: join(root, "workspace"), hasUI: false,
|
|
sessionManager: { getBranch: () => [], getSessionFile: () => undefined },
|
|
ui: { notify() {} },
|
|
};
|
|
const emit = async (event, data = {}) => {
|
|
for (const handler of handlers.get(event) ?? []) await handler(data, ctx);
|
|
};
|
|
|
|
try {
|
|
process.env.PI_CODING_AGENT_DIR = root;
|
|
await mkdir(globalDir);
|
|
await mkdir(ctx.cwd);
|
|
await writeFile(join(root, "hermes-memory-config.json"), JSON.stringify({
|
|
lazyInitialization: lazy, memoryMode: "policy-only",
|
|
reviewEnabled: false, correctionDetection: false,
|
|
flushOnShutdown: false, flushOnCompact: false,
|
|
}));
|
|
await writeFile(join(globalDir, "MEMORY.md"), "durable global preference");
|
|
for (let i = 0; i < 20; i++) {
|
|
const dir = join(root, "projects-memory", `project-${i}`);
|
|
await mkdir(dir, { recursive: true });
|
|
await writeFile(join(dir, "MEMORY.md"), `durable project ${i} convention`);
|
|
}
|
|
|
|
const started = performance.now();
|
|
const { default: register } = await import(pathToFileURL(resolve(entry)).href);
|
|
const imported = performance.now();
|
|
register({
|
|
on(event, handler) {
|
|
if (!handlers.has(event)) handlers.set(event, []);
|
|
handlers.get(event).push(handler);
|
|
},
|
|
registerTool(tool) { tools.set(tool.name, tool); },
|
|
registerCommand() {},
|
|
});
|
|
const registered = performance.now();
|
|
await emit("session_start");
|
|
await emit("resources_discover", { cwd: ctx.cwd, reason: "startup" });
|
|
const ready = performance.now();
|
|
const databaseAtStartup = existsSync(join(globalDir, "sessions.db"));
|
|
const search = await tools.get("memory_search").execute("bench", { query: "durable" }, undefined, undefined, ctx);
|
|
const searched = performance.now();
|
|
if (!search.details.success || search.details.count === 0) throw new Error("First-use search failed");
|
|
console.log(JSON.stringify({
|
|
lazy, entry: resolve(entry),
|
|
importMs: Math.round(imported - started),
|
|
registrationMs: Math.round(registered - imported),
|
|
sessionStartMs: Math.round(ready - registered),
|
|
totalStartupMs: Math.round(ready - started),
|
|
firstSearchMs: Math.round(searched - ready),
|
|
databaseAtStartup,
|
|
}));
|
|
} finally {
|
|
await emit("session_shutdown", { reason: "quit" });
|
|
await rm(root, { recursive: true, force: true });
|
|
}
|