test: trim reply label generator mock

This commit is contained in:
Peter Steinberger 2026-04-03 20:25:38 +01:00
parent 58d6c16d12
commit d9af49a7af
No known key found for this signature in database
1 changed files with 3 additions and 2 deletions

View File

@ -7,8 +7,9 @@ const resolveDefaultModelForAgent = vi.hoisted(() => vi.fn());
const resolveModelAsync = vi.hoisted(() => vi.fn());
const prepareModelForSimpleCompletion = vi.hoisted(() => vi.fn());
vi.mock("@mariozechner/pi-ai", async (importOriginal) => {
const original = await importOriginal<typeof import("@mariozechner/pi-ai")>();
vi.mock("@mariozechner/pi-ai", async () => {
const original =
await vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai");
return {
...original,
completeSimple,