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,225 @@
|
||||
import { describe, it, beforeEach, afterEach } from 'node:test';
|
||||
import assert from 'node:assert/strict';
|
||||
import fs from 'node:fs';
|
||||
import path from 'node:path';
|
||||
import os from 'node:os';
|
||||
import { DatabaseManager } from '../../src/store/db.js';
|
||||
import {
|
||||
scheduleLiveSessionIndex,
|
||||
waitForLiveSessionIndex,
|
||||
type SessionLiveIndexState,
|
||||
} from '../../src/handlers/session-live-index.js';
|
||||
|
||||
describe('session live indexing handler', () => {
|
||||
let tmpDir: string;
|
||||
let dbManager: DatabaseManager;
|
||||
|
||||
beforeEach(() => {
|
||||
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'session-live-index-test-'));
|
||||
dbManager = new DatabaseManager(path.join(tmpDir, 'memory'));
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
dbManager.close();
|
||||
fs.rmSync(tmpDir, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
function createSnapshot(entries: unknown[]) {
|
||||
return {
|
||||
getHeader: () => ({ id: 'live-session', timestamp: '2026-05-03T00:00:00Z', cwd: '/work/live-project' }),
|
||||
getEntries: () => entries,
|
||||
};
|
||||
}
|
||||
|
||||
it('defers indexing so message_end does not block and then indexes live messages', async () => {
|
||||
const entries = [{
|
||||
type: 'message',
|
||||
id: 'entry-1',
|
||||
timestamp: '2026-05-03T00:01:00Z',
|
||||
message: { role: 'user', content: 'hello after message_end' },
|
||||
}];
|
||||
const callbacks: (() => void)[] = [];
|
||||
const state: SessionLiveIndexState = { inProgress: false, promise: null };
|
||||
|
||||
const scheduled = scheduleLiveSessionIndex(dbManager, createSnapshot(entries), {
|
||||
state,
|
||||
delayMs: 0,
|
||||
setTimeoutFn: (callback) => {
|
||||
callbacks.push(callback);
|
||||
return 0;
|
||||
},
|
||||
});
|
||||
|
||||
assert.equal(scheduled, true);
|
||||
assert.equal(callbacks.length, 1);
|
||||
assert.equal(dbManager.getStats().messages, 0, 'message_end handler should not index synchronously');
|
||||
|
||||
const promise = state.promise;
|
||||
assert.ok(promise);
|
||||
callbacks[0]();
|
||||
await promise;
|
||||
|
||||
assert.equal(dbManager.getStats().sessions, 1);
|
||||
assert.equal(dbManager.getStats().messages, 1);
|
||||
});
|
||||
|
||||
it('coalesces multiple scheduled message_end events and indexes all missing entries', async () => {
|
||||
const entries = [{
|
||||
type: 'message',
|
||||
id: 'entry-1',
|
||||
timestamp: '2026-05-03T00:01:00Z',
|
||||
message: { role: 'user', content: 'first' },
|
||||
}];
|
||||
const callbacks: (() => void)[] = [];
|
||||
const state: SessionLiveIndexState = { inProgress: false, promise: null };
|
||||
const snapshot = createSnapshot(entries);
|
||||
|
||||
assert.equal(scheduleLiveSessionIndex(dbManager, snapshot, {
|
||||
state,
|
||||
delayMs: 0,
|
||||
setTimeoutFn: (callback) => {
|
||||
callbacks.push(callback);
|
||||
return 0;
|
||||
},
|
||||
}), true);
|
||||
|
||||
entries.push({
|
||||
type: 'message',
|
||||
id: 'entry-2',
|
||||
timestamp: '2026-05-03T00:02:00Z',
|
||||
message: { role: 'assistant', content: [{ type: 'text', text: 'second' }] },
|
||||
});
|
||||
assert.equal(scheduleLiveSessionIndex(dbManager, snapshot, { state, delayMs: 0 }), false);
|
||||
|
||||
const promise = state.promise;
|
||||
callbacks[0]();
|
||||
await promise;
|
||||
|
||||
assert.equal(dbManager.getStats().sessions, 1);
|
||||
assert.equal(dbManager.getStats().messages, 2);
|
||||
});
|
||||
|
||||
it('indexes appended messages for an already indexed resumed session', async () => {
|
||||
const entries = [{
|
||||
type: 'message',
|
||||
id: 'entry-1',
|
||||
timestamp: '2026-05-03T00:01:00Z',
|
||||
message: { role: 'user', content: 'before resume' },
|
||||
}];
|
||||
const snapshot = createSnapshot(entries);
|
||||
const state: SessionLiveIndexState = { inProgress: false, promise: null };
|
||||
|
||||
scheduleLiveSessionIndex(dbManager, snapshot, {
|
||||
state,
|
||||
delayMs: 0,
|
||||
setTimeoutFn: (callback) => {
|
||||
queueMicrotask(callback);
|
||||
return 0;
|
||||
},
|
||||
});
|
||||
await state.promise;
|
||||
assert.equal(dbManager.getStats().messages, 1);
|
||||
|
||||
entries.push({
|
||||
type: 'message',
|
||||
id: 'entry-2',
|
||||
timestamp: '2026-05-03T00:02:00Z',
|
||||
message: { role: 'user', content: 'after resume' },
|
||||
});
|
||||
scheduleLiveSessionIndex(dbManager, snapshot, {
|
||||
state,
|
||||
delayMs: 0,
|
||||
setTimeoutFn: (callback) => {
|
||||
queueMicrotask(callback);
|
||||
return 0;
|
||||
},
|
||||
});
|
||||
await state.promise;
|
||||
|
||||
assert.equal(dbManager.getStats().sessions, 1);
|
||||
assert.equal(dbManager.getStats().messages, 2);
|
||||
});
|
||||
|
||||
it('scheduled live indexing is best-effort and does not reject on errors', async () => {
|
||||
const state: SessionLiveIndexState = { inProgress: false, promise: null };
|
||||
const errors: unknown[] = [];
|
||||
|
||||
const scheduled = scheduleLiveSessionIndex(dbManager, createSnapshot([]), {
|
||||
state,
|
||||
indexLiveSessionFn: () => {
|
||||
throw new Error('boom');
|
||||
},
|
||||
onError: (err) => errors.push(err),
|
||||
delayMs: 0,
|
||||
setTimeoutFn: (callback) => {
|
||||
queueMicrotask(callback);
|
||||
return 0;
|
||||
},
|
||||
});
|
||||
|
||||
assert.equal(scheduled, true);
|
||||
await state.promise;
|
||||
assert.equal(state.inProgress, false);
|
||||
assert.equal(errors.length, 1);
|
||||
assert.match(errors[0] instanceof Error ? errors[0].message : String(errors[0]), /boom/);
|
||||
});
|
||||
|
||||
it('repairs corruption and retries live indexing once before reporting an error', async () => {
|
||||
dbManager.getDb();
|
||||
const state: SessionLiveIndexState = { inProgress: false, promise: null };
|
||||
const errors: unknown[] = [];
|
||||
let attempts = 0;
|
||||
|
||||
const scheduled = scheduleLiveSessionIndex(dbManager, createSnapshot([]), {
|
||||
state,
|
||||
indexLiveSessionFn: () => {
|
||||
attempts++;
|
||||
if (attempts === 1) {
|
||||
const err = new Error('SQLITE_CORRUPT: database disk image is malformed') as Error & { code: string };
|
||||
err.code = 'SQLITE_CORRUPT';
|
||||
throw err;
|
||||
}
|
||||
return null;
|
||||
},
|
||||
onError: (err) => errors.push(err),
|
||||
delayMs: 0,
|
||||
setTimeoutFn: (callback) => {
|
||||
queueMicrotask(callback);
|
||||
return 0;
|
||||
},
|
||||
});
|
||||
|
||||
assert.equal(scheduled, true);
|
||||
await state.promise;
|
||||
|
||||
assert.equal(attempts, 2);
|
||||
assert.equal(errors.length, 0);
|
||||
assert.equal(dbManager.getLastRecovery()?.strategy, 'reused');
|
||||
});
|
||||
|
||||
it('shutdown wait resolves true when live indexing completes before timeout', async () => {
|
||||
let resolveIndex!: () => void;
|
||||
const state: SessionLiveIndexState = {
|
||||
inProgress: true,
|
||||
promise: new Promise<void>((resolve) => {
|
||||
resolveIndex = resolve;
|
||||
}),
|
||||
};
|
||||
|
||||
setTimeout(resolveIndex, 5);
|
||||
const completed = await waitForLiveSessionIndex(100, state);
|
||||
|
||||
assert.equal(completed, true);
|
||||
});
|
||||
|
||||
it('shutdown wait resolves false when live indexing exceeds timeout', async () => {
|
||||
const state: SessionLiveIndexState = {
|
||||
inProgress: true,
|
||||
promise: new Promise<void>(() => {}),
|
||||
};
|
||||
|
||||
const completed = await waitForLiveSessionIndex(5, state);
|
||||
|
||||
assert.equal(completed, false);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user