779ca4872b
- Removed Prettier configuration files (.prettierignore and .prettierrc) and migrated formatting to Biome. - Updated Biome configuration from version 2.3.11 to 2.3.12 and changed arrowParentheses rule from 'always' to 'asNeeded'. - Pinned @biomejs/biome dependency to exact version 2.3.12 in package.json and bun.lock. - Applied consistent arrow function formatting across 489 files by removing unnecessary parentheses around single parameters. - Removed blank lines after comment blocks and reorganized imports for consistency across the codebase.
71 lines
2.4 KiB
TypeScript
71 lines
2.4 KiB
TypeScript
/**
|
|
* Verify the documentation example from hooks.md compiles and works.
|
|
*/
|
|
|
|
import { describe, expect, it } from "bun:test";
|
|
import type {
|
|
HookAPI,
|
|
SessionBeforeCompactEvent,
|
|
SessionCompactEvent,
|
|
} from "@oh-my-pi/pi-coding-agent/extensibility/hooks";
|
|
|
|
describe("Documentation example", () => {
|
|
it("custom compaction example should type-check correctly", () => {
|
|
// This is the example from hooks.md - verify it compiles
|
|
const exampleHook = (pi: HookAPI) => {
|
|
pi.on("session_before_compact", async (event: SessionBeforeCompactEvent, ctx) => {
|
|
// All these should be accessible on the event
|
|
const { preparation, branchEntries } = event;
|
|
// sessionManager, modelRegistry, and model come from ctx
|
|
const { sessionManager, modelRegistry } = ctx;
|
|
const { messagesToSummarize, turnPrefixMessages, tokensBefore, firstKeptEntryId, isSplitTurn } =
|
|
preparation;
|
|
|
|
// Verify types
|
|
expect(Array.isArray(messagesToSummarize)).toBe(true);
|
|
expect(Array.isArray(turnPrefixMessages)).toBe(true);
|
|
expect(typeof isSplitTurn).toBe("boolean");
|
|
expect(typeof tokensBefore).toBe("number");
|
|
expect(typeof sessionManager.getEntries).toBe("function");
|
|
expect(typeof modelRegistry.getApiKey).toBe("function");
|
|
expect(typeof firstKeptEntryId).toBe("string");
|
|
expect(Array.isArray(branchEntries)).toBe(true);
|
|
|
|
const summary = messagesToSummarize
|
|
.filter(m => m.role === "user")
|
|
.map(m => `- ${typeof m.content === "string" ? m.content.slice(0, 100) : "[complex]"}`)
|
|
.join("\n");
|
|
|
|
// Hooks return compaction content - SessionManager adds id/parentId
|
|
return {
|
|
compaction: {
|
|
summary: `User requests:\n${summary}`,
|
|
firstKeptEntryId,
|
|
tokensBefore,
|
|
},
|
|
};
|
|
});
|
|
};
|
|
|
|
// Just verify the function exists and is callable
|
|
expect(typeof exampleHook).toBe("function");
|
|
});
|
|
|
|
it("compact event should have correct fields", () => {
|
|
const checkCompactEvent = (pi: HookAPI) => {
|
|
pi.on("session_compact", async (event: SessionCompactEvent) => {
|
|
// These should all be accessible
|
|
const entry = event.compactionEntry;
|
|
const fromExtension = event.fromExtension;
|
|
|
|
expect(entry.type).toBe("compaction");
|
|
expect(typeof entry.summary).toBe("string");
|
|
expect(typeof entry.tokensBefore).toBe("number");
|
|
expect(typeof fromExtension).toBe("boolean");
|
|
});
|
|
};
|
|
|
|
expect(typeof checkCompactEvent).toBe("function");
|
|
});
|
|
});
|