mirror of
https://bitbucket.org/siakitem/my-pi.git
synced 2026-10-26 18:55:22 +00:00
feat(pi-hermes-memory): 导入上游 v0.9.10 源码快照
从官方 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。
This commit is contained in:
@@ -0,0 +1,128 @@
|
||||
import assert from "node:assert/strict";
|
||||
import { describe, it } from "node:test";
|
||||
import { setupBackgroundReview } from "../../src/handlers/background-review.js";
|
||||
import { setupCorrectionDetector } from "../../src/handlers/correction-detector.js";
|
||||
import { setupSessionFlush } from "../../src/handlers/session-flush.js";
|
||||
import { loadConfig } from "../../src/config.js";
|
||||
|
||||
function fixture() {
|
||||
const config = { ...loadConfig("/nonexistent/pi-memory-hook-test.json"), nudgeInterval: 1, flushMinTurns: 2 };
|
||||
const handlers: Record<string, Function[]> = {};
|
||||
const pi = { on(name: string, fn: Function) { (handlers[name] ??= []).push(fn); } } as any;
|
||||
let ready = false;
|
||||
let loads = 0;
|
||||
let calls = 0;
|
||||
const store = {
|
||||
getMemoryEntries() { assert.ok(ready); return ["existing fact"]; },
|
||||
getUserEntries() { assert.ok(ready); return []; },
|
||||
async addFailure() { assert.ok(ready); return { success: true }; },
|
||||
} as any;
|
||||
const ctx = {
|
||||
cwd: "/project",
|
||||
sessionManager: {
|
||||
getBranch: () => Array.from({ length: 8 }, (_, i) => ({
|
||||
type: "message", message: { role: i % 2 ? "assistant" : "user", content: "no, use pnpm" },
|
||||
})),
|
||||
},
|
||||
ui: { notify() {} },
|
||||
};
|
||||
const ensureMemoryReady = async (context: { cwd: string }) => {
|
||||
assert.equal(context.cwd, ctx.cwd);
|
||||
loads++;
|
||||
ready = true;
|
||||
};
|
||||
const runDirectMemoryCompletion = async () => {
|
||||
assert.ok(ready);
|
||||
calls++;
|
||||
return { ok: true, appliedCount: 0 };
|
||||
};
|
||||
const emit = async (name: string, event: any = {}) => {
|
||||
for (const handler of handlers[name] ?? []) await handler(event, ctx);
|
||||
};
|
||||
return { pi, config, store, emit, ensureMemoryReady, runDirectMemoryCompletion,
|
||||
loads: () => loads, calls: () => calls };
|
||||
}
|
||||
|
||||
describe("lazy memory automatic operations", () => {
|
||||
it("initializes only after a background review actually becomes due", async () => {
|
||||
const f = fixture();
|
||||
const settled = Promise.withResolvers<void>();
|
||||
setupBackgroundReview(f.pi, f.store, null, f.config, {
|
||||
ensureMemoryReady: f.ensureMemoryReady,
|
||||
deps: { runDirectReview: f.runDirectMemoryCompletion, onReviewSettled: settled.resolve },
|
||||
});
|
||||
await f.emit("message_end", { message: { role: "user" } });
|
||||
await f.emit("turn_end");
|
||||
assert.equal(f.loads(), 0);
|
||||
await f.emit("message_end", { message: { role: "user" } });
|
||||
await f.emit("message_end", { message: { role: "user" } });
|
||||
await f.emit("turn_end");
|
||||
await settled.promise;
|
||||
assert.equal(f.loads(), 1);
|
||||
assert.equal(f.calls(), 1);
|
||||
await f.emit("session_shutdown");
|
||||
});
|
||||
|
||||
it("initializes before correction capture, not on ordinary messages", async () => {
|
||||
const f = fixture();
|
||||
setupCorrectionDetector(f.pi, f.store, null, f.config, null, null, {
|
||||
ensureMemoryReady: f.ensureMemoryReady, runDirectMemoryCompletion: f.runDirectMemoryCompletion,
|
||||
});
|
||||
await f.emit("message_end", { message: { role: "user", content: "hello" } });
|
||||
await f.emit("turn_end");
|
||||
assert.equal(f.loads(), 0);
|
||||
await f.emit("message_end", { message: { role: "user", content: "no, use pnpm" } });
|
||||
await f.emit("turn_end");
|
||||
assert.equal(f.loads(), 1);
|
||||
assert.equal(f.calls(), 1);
|
||||
});
|
||||
|
||||
it("initializes for a qualifying flush, not a short session or reload", async () => {
|
||||
const f = fixture();
|
||||
setupSessionFlush(f.pi, f.store, null, f.config, null, null, {
|
||||
ensureMemoryReady: f.ensureMemoryReady, runDirectMemoryCompletion: f.runDirectMemoryCompletion,
|
||||
});
|
||||
await f.emit("message_end", { message: { role: "user" } });
|
||||
await f.emit("session_before_compact");
|
||||
assert.equal(f.loads(), 0);
|
||||
await f.emit("message_end", { message: { role: "user" } });
|
||||
await f.emit("session_shutdown", { reason: "reload" });
|
||||
assert.equal(f.loads(), 0);
|
||||
await f.emit("session_before_compact");
|
||||
assert.equal(f.loads(), 1);
|
||||
assert.equal(f.calls(), 1);
|
||||
});
|
||||
|
||||
it("does not flush against unloaded memory after an initialization error", async () => {
|
||||
const f = fixture();
|
||||
setupSessionFlush(f.pi, f.store, null, { ...f.config, flushMinTurns: 0 }, null, null, {
|
||||
ensureMemoryReady: async () => { throw new Error("load failed"); },
|
||||
runDirectMemoryCompletion: f.runDirectMemoryCompletion,
|
||||
});
|
||||
await f.emit("session_before_compact");
|
||||
assert.equal(f.calls(), 0);
|
||||
});
|
||||
|
||||
it("retries a pending correction without consuming its rate limit on initialization failure", async () => {
|
||||
const f = fixture();
|
||||
let attempts = 0;
|
||||
setupCorrectionDetector(f.pi, f.store, null, f.config, null, null, {
|
||||
ensureMemoryReady: async (ctx) => {
|
||||
if (++attempts === 1) throw new Error("transient load failure");
|
||||
await f.ensureMemoryReady(ctx);
|
||||
},
|
||||
runDirectMemoryCompletion: f.runDirectMemoryCompletion,
|
||||
});
|
||||
await f.emit("message_end", { message: { role: "user", content: "no, use pnpm" } });
|
||||
await f.emit("turn_end");
|
||||
assert.equal(f.calls(), 0);
|
||||
let saved = "";
|
||||
f.store.addFailure = async (text: string) => { saved = text; return { success: true }; };
|
||||
await f.emit("message_end", { message: { role: "user", content: "continue with the work" } });
|
||||
await f.emit("turn_end");
|
||||
assert.equal(attempts, 2);
|
||||
assert.equal(f.calls(), 1);
|
||||
assert.match(saved, /use pnpm/);
|
||||
assert.doesNotMatch(saved, /continue with the work/);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user