From 6261beb392f54faaf58d60049bd8d545a29ea298 Mon Sep 17 00:00:00 2001 From: Mike Aizatsky Date: Sat, 28 Mar 2026 19:08:29 -0700 Subject: [PATCH] Fix Messages test for refactored API --- packages/core/test/messages.test.ts | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/packages/core/test/messages.test.ts b/packages/core/test/messages.test.ts index df3fef9..ae0ef6f 100644 --- a/packages/core/test/messages.test.ts +++ b/packages/core/test/messages.test.ts @@ -3,17 +3,18 @@ import { describe, expect, it } from "vitest"; import { Messages } from "../src/messages.ts"; describe("Messages", () => { - it("is iterable and supports in-place replacement", () => { + it("is iterable and supports replacing the internal array in-place", () => { const first: ModelMessage = { role: "user", content: "first" }; const second: ModelMessage = { role: "assistant", content: "second" }; - const messages = new Messages(); - messages.push(first, second); + const messages = new Messages([first]); + messages.pushAll([second]); expect([...messages]).toEqual([first, second]); - expect(messages.slice(1)).toEqual([second]); + expect(messages.get().slice(1)).toEqual([second]); - messages.replace([second]); + const replacement = [second]; + messages.get().splice(0, messages.length, ...replacement); expect(messages.length).toBe(1); expect([...messages]).toEqual([second]);