File size: 3,023 Bytes
fc93158 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 | import { beforeEach, describe, expect, it } from "vitest";
import { createHookRunner } from "./hooks.js";
import { createEmptyPluginRegistry, type PluginRegistry } from "./registry.js";
import type {
PluginHookBeforeModelResolveResult,
PluginHookBeforePromptBuildResult,
PluginHookRegistration,
} from "./types.js";
function addTypedHook(
registry: PluginRegistry,
hookName: "before_model_resolve" | "before_prompt_build",
pluginId: string,
handler: () =>
| PluginHookBeforeModelResolveResult
| PluginHookBeforePromptBuildResult
| Promise<PluginHookBeforeModelResolveResult | PluginHookBeforePromptBuildResult>,
priority?: number,
) {
registry.typedHooks.push({
pluginId,
hookName,
handler,
priority,
source: "test",
} as PluginHookRegistration);
}
describe("phase hooks merger", () => {
let registry: PluginRegistry;
beforeEach(() => {
registry = createEmptyPluginRegistry();
});
it("before_model_resolve keeps higher-priority override values", async () => {
addTypedHook(registry, "before_model_resolve", "low", () => ({ modelOverride: "gpt-4o" }), 1);
addTypedHook(
registry,
"before_model_resolve",
"high",
() => ({ modelOverride: "llama3.3:8b", providerOverride: "ollama" }),
10,
);
const runner = createHookRunner(registry);
const result = await runner.runBeforeModelResolve({ prompt: "test" }, {});
expect(result?.modelOverride).toBe("llama3.3:8b");
expect(result?.providerOverride).toBe("ollama");
});
it("before_prompt_build concatenates prependContext and preserves systemPrompt precedence", async () => {
addTypedHook(
registry,
"before_prompt_build",
"high",
() => ({ prependContext: "context A", systemPrompt: "system A" }),
10,
);
addTypedHook(
registry,
"before_prompt_build",
"low",
() => ({ prependContext: "context B" }),
1,
);
const runner = createHookRunner(registry);
const result = await runner.runBeforePromptBuild({ prompt: "test", messages: [] }, {});
expect(result?.prependContext).toBe("context A\n\ncontext B");
expect(result?.systemPrompt).toBe("system A");
});
it("before_prompt_build concatenates prependSystemContext and appendSystemContext", async () => {
addTypedHook(
registry,
"before_prompt_build",
"first",
() => ({
prependSystemContext: "prepend A",
appendSystemContext: "append A",
}),
10,
);
addTypedHook(
registry,
"before_prompt_build",
"second",
() => ({
prependSystemContext: "prepend B",
appendSystemContext: "append B",
}),
1,
);
const runner = createHookRunner(registry);
const result = await runner.runBeforePromptBuild({ prompt: "test", messages: [] }, {});
expect(result?.prependSystemContext).toBe("prepend A\n\nprepend B");
expect(result?.appendSystemContext).toBe("append A\n\nappend B");
});
});
|