diff --git a/src/agents/pi-hooks/context-pruning/pruner.test.ts b/src/agents/pi-hooks/context-pruning/pruner.test.ts index 64751364bf5..cb64db13bb1 100644 --- a/src/agents/pi-hooks/context-pruning/pruner.test.ts +++ b/src/agents/pi-hooks/context-pruning/pruner.test.ts @@ -390,7 +390,7 @@ describe("pruneContextMessages", () => { const toolResult = result[1] as Extract; expect(toolResult.content).toHaveLength(1); - expect(toolResult.content[0]).toMatchObject({ type: "text" }); + expect(toolResult.content[0]?.type).toBe("text"); const textBlock = toolResult.content[0] as { type: "text"; text: string }; expect(textBlock.text).toContain("[image removed during context pruning]"); expect(textBlock.text).toContain(