Bohdan Triapitsyn
bd68e303d4
Add pin and unpin actions for user and assistant text messages, with clear compaction-survival labels, localized tooltips, status-info active styling, and VS Code gating where the server runtime is unavailable. Persist pinned message IDs, creation timestamps, and roles under the OpenChamber session metadata namespace using fresh-read merge updates so goal, review, and other metadata remain intact. Introduce a server runtime that reacts to OpenCode's dedicated session.compacted event, fetches pinned messages by ID, extracts and chronologically orders their text parts, and injects them as hidden synthetic context through prompt_async. The restoration prompt tells the agent to use the context silently while work remains and limits idle summaries to one short paragraph. Track the last handled compaction summary to avoid replay duplication, tolerate individually missing pinned messages, integrate runtime shutdown, document ownership and limitations, and cover metadata round trips plus compaction injection behavior with focused tests.
78 lines
3.6 KiB
JavaScript
78 lines
3.6 KiB
JavaScript
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
|
|
import { createContextObligatoryRuntime } from './runtime.js';
|
|
|
|
const json = (body) => new Response(JSON.stringify(body), {
|
|
status: 200,
|
|
headers: { 'Content-Type': 'application/json' },
|
|
});
|
|
|
|
describe('context obligatory runtime', () => {
|
|
afterEach(() => {
|
|
vi.unstubAllGlobals();
|
|
});
|
|
|
|
it('injects pinned text in chronological order after compaction and records the summary cursor', async () => {
|
|
const requests = [];
|
|
let sessionReads = 0;
|
|
vi.stubGlobal('fetch', vi.fn(async (input, init = {}) => {
|
|
const url = new URL(typeof input === 'string' ? input : input.url);
|
|
requests.push({ path: url.pathname, method: init.method ?? 'GET', body: init.body });
|
|
if (url.pathname === '/session/ses_1' && init.method === 'PATCH') return json({});
|
|
if (url.pathname === '/session/ses_1') {
|
|
sessionReads += 1;
|
|
return json({
|
|
id: 'ses_1',
|
|
metadata: { openchamber: { context_obligatory_messages: [
|
|
{ id: 'msg_2', createdAt: 20, role: 'assistant' },
|
|
{ id: 'msg_1', createdAt: 10, role: 'user' },
|
|
] } },
|
|
});
|
|
}
|
|
if (url.pathname === '/session/ses_1/message') return json([
|
|
{ info: { id: 'msg_agent', role: 'assistant', providerID: 'provider', modelID: 'model', agent: 'build' } },
|
|
{ info: { id: 'msg_summary', role: 'assistant', summary: true, time: { completed: 30 } } },
|
|
]);
|
|
if (url.pathname === '/session/ses_1/message/msg_1') return json({ parts: [{ type: 'text', text: 'First' }] });
|
|
if (url.pathname === '/session/ses_1/message/msg_2') return json({ parts: [{ type: 'text', text: 'Second' }] });
|
|
if (url.pathname === '/session/ses_1/prompt_async') return json({});
|
|
throw new Error(`Unexpected ${url.pathname}`);
|
|
}));
|
|
const runtime = createContextObligatoryRuntime({
|
|
buildOpenCodeUrl: (path) => `http://opencode.test${path}`,
|
|
getOpenCodeAuthHeaders: () => ({}),
|
|
});
|
|
|
|
await runtime.processPayload({ type: 'session.compacted', properties: { sessionID: 'ses_1' } });
|
|
|
|
const prompt = requests.find((request) => request.path.endsWith('/prompt_async'));
|
|
const payload = JSON.parse(prompt.body);
|
|
expect(payload).toMatchObject({
|
|
model: { providerID: 'provider', modelID: 'model' },
|
|
agent: 'build',
|
|
parts: [{ type: 'text', synthetic: true }],
|
|
});
|
|
expect(payload.parts[0].text.indexOf('First')).toBeLessThan(payload.parts[0].text.indexOf('Second'));
|
|
expect(payload.parts[0].text).toContain('continuing the pre-compaction work');
|
|
expect(payload.parts[0].text).toContain('use it silently as background context');
|
|
expect(payload.parts[0].text).toContain('Only if no tasks or next steps remain');
|
|
expect(payload.parts[0].text).toContain('no more than one short paragraph');
|
|
const patch = requests.find((request) => request.method === 'PATCH');
|
|
expect(JSON.parse(patch.body).metadata.openchamber.context_obligatory_last_compaction_message_id).toBe('msg_summary');
|
|
expect(sessionReads).toBe(2);
|
|
runtime.stop();
|
|
});
|
|
|
|
it('ignores ordinary idle events without making requests', async () => {
|
|
const fetchImpl = vi.fn();
|
|
vi.stubGlobal('fetch', fetchImpl);
|
|
const runtime = createContextObligatoryRuntime({
|
|
buildOpenCodeUrl: (path) => `http://opencode.test${path}`,
|
|
getOpenCodeAuthHeaders: () => ({}),
|
|
});
|
|
await runtime.processPayload({ type: 'session.status', properties: { sessionID: 'ses_1', status: { type: 'idle' } } });
|
|
expect(fetchImpl).not.toHaveBeenCalled();
|
|
runtime.stop();
|
|
});
|
|
});
|