@livekit/agents 1.1.0 → 1.2.0

package/dist/llm/chat_context.test.js

@@ -1,4 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { initializeLogger } from "../log.js";
+import { FakeLLM } from "../voice/testing/fake_llm.js";
 import {
   ChatContext,
   ChatMessage,
@@ -6,6 +8,7 @@ import {
   FunctionCallOutput,
   ReadonlyChatContext
 } from "./chat_context.js";
+initializeLogger({ pretty: false, level: "error" });
 describe("ChatContext.toJSON", () => {
   it("should match snapshot for empty context", () => {
     const context = new ChatContext();
@@ -237,6 +240,46 @@ describe("ChatContext.toJSON", () => {
     ).toMatchSnapshot("message-properties-full");
   });
 });
+describe("ChatContext._summarize", () => {
+  it("keeps chronological timestamps with summary + tail", async () => {
+    const ctx = new ChatContext();
+    ctx.addMessage({ role: "system", content: "System prompt", createdAt: 0 });
+    ctx.addMessage({ role: "user", content: "hello", createdAt: 1e3 });
+    ctx.addMessage({ role: "assistant", content: "hi there", createdAt: 2e3 });
+    ctx.insert(
+      new FunctionCallOutput({
+        callId: "call_1",
+        name: "lookup",
+        output: '{"ok":true}',
+        isError: false,
+        createdAt: 3500
+      })
+    );
+    ctx.addMessage({ role: "user", content: "my color is blue", createdAt: 3e3 });
+    ctx.addMessage({ role: "assistant", content: "noted", createdAt: 4e3 });
+    const fake = new FakeLLM([
+      {
+        input: "Conversation to summarize:\n\nuser: hello\nassistant: hi there",
+        content: "condensed head"
+      }
+    ]);
+    await ctx._summarize(fake, { keepLastTurns: 1 });
+    const summary = ctx.items.find(
+      (item) => {
+        var _a;
+        return item.type === "message" && item.role === "assistant" && ((_a = item.extra) == null ? void 0 : _a.is_summary) === true;
+      }
+    );
+    expect(summary).toBeDefined();
+    if (!summary || summary.type !== "message") {
+      throw new Error("summary message is missing");
+    }
+    expect(summary.createdAt).toBeCloseTo(2999.999, 6);
+    const createdAts = ctx.items.map((item) => item.createdAt);
+    const sorted = [...createdAts].sort((a, b) => a - b);
+    expect(createdAts).toEqual(sorted);
+  });
+});
 describe("ReadonlyChatContext with immutable array", () => {
   it("should have readonly property set to true", () => {
     const items = [
@@ -382,4 +425,535 @@ describe("ReadonlyChatContext with immutable array", () => {
     expect(ids).toEqual(["msg_1", "msg_2"]);
   });
 });
+describe("ChatContext.isEquivalent", () => {
+  it("should return true for same reference", () => {
+    const ctx = new ChatContext();
+    ctx.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    expect(ctx.isEquivalent(ctx)).toBe(true);
+  });
+  it("should return true for identical empty contexts", () => {
+    const ctx1 = new ChatContext();
+    const ctx2 = new ChatContext();
+    expect(ctx1.isEquivalent(ctx2)).toBe(true);
+  });
+  it("should return false for contexts with different lengths", () => {
+    const ctx1 = new ChatContext();
+    ctx1.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    const ctx2 = new ChatContext();
+    ctx2.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    ctx2.addMessage({
+      id: "msg_2",
+      role: "assistant",
+      content: "Hi"
+    });
+    expect(ctx1.isEquivalent(ctx2)).toBe(false);
+  });
+  it("should return false for contexts with different item IDs", () => {
+    const ctx1 = new ChatContext();
+    ctx1.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    const ctx2 = new ChatContext();
+    ctx2.addMessage({
+      id: "msg_2",
+      role: "user",
+      content: "Hello"
+    });
+    expect(ctx1.isEquivalent(ctx2)).toBe(false);
+  });
+  it("should return false for contexts with different item types", () => {
+    const ctx1 = new ChatContext();
+    ctx1.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    const ctx2 = new ChatContext();
+    ctx2.insert(
+      new FunctionCall({
+        id: "msg_1",
+        callId: "call_1",
+        name: "test",
+        args: "{}"
+      })
+    );
+    expect(ctx1.isEquivalent(ctx2)).toBe(false);
+  });
+  describe("message comparison", () => {
+    it("should return true for identical messages", () => {
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: false
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: false
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    it("should return false for messages with different roles", () => {
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello"
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "assistant",
+        content: "Hello"
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return false for messages with different interrupted flags", () => {
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: false
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: true
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return false for messages with different content", () => {
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello"
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "World"
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return true for messages with identical array content", () => {
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello", "World"]
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello", "World"]
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    it("should return false for messages with different array content", () => {
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello", "World"]
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello"]
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return true for messages with identical image content", () => {
+      const imageContent = {
+        id: "img_1",
+        type: "image_content",
+        image: "https://example.com/image.jpg",
+        inferenceDetail: "high",
+        inferenceWidth: 1024,
+        inferenceHeight: 768,
+        mimeType: "image/jpeg",
+        _cache: {}
+      };
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", imageContent]
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", { ...imageContent }]
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    it("should return false for messages with different image content", () => {
+      const imageContent1 = {
+        id: "img_1",
+        type: "image_content",
+        image: "https://example.com/image1.jpg",
+        inferenceDetail: "high",
+        _cache: {}
+      };
+      const imageContent2 = {
+        id: "img_2",
+        type: "image_content",
+        image: "https://example.com/image2.jpg",
+        inferenceDetail: "high",
+        _cache: {}
+      };
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", imageContent1]
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", imageContent2]
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+  });
+  describe("function call comparison", () => {
+    it("should return true for identical function calls", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    it("should return false for function calls with different names", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}"
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_time",
+          args: "{}"
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return false for function calls with different call IDs", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}"
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_2",
+          name: "get_weather",
+          args: "{}"
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return false for function calls with different arguments", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "London"}'
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should ignore timestamps", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}",
+          createdAt: 1e3
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}",
+          createdAt: 2e3
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+  });
+  describe("function call output comparison", () => {
+    it("should return true for identical function call outputs", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    it("should return false for function call outputs with different names", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_time",
+          output: "{}",
+          isError: false
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return false for function call outputs with different call IDs", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_2",
+          name: "get_weather",
+          output: "{}",
+          isError: false
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return false for function call outputs with different output values", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 25}',
+          isError: false
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should return false for function call outputs with different error flags", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "Error occurred",
+          isError: false
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "Error occurred",
+          isError: true
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    it("should ignore timestamps", () => {
+      const ctx1 = new ChatContext();
+      ctx1.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false,
+          createdAt: 1e3
+        })
+      );
+      const ctx2 = new ChatContext();
+      ctx2.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false,
+          createdAt: 2e3
+        })
+      );
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+  });
+  describe("complex context comparison", () => {
+    it("should return true for identical complex contexts", () => {
+      const ctx1 = new ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "What is the weather?"
+      });
+      ctx1.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      ctx1.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      ctx1.addMessage({
+        id: "msg_2",
+        role: "assistant",
+        content: "The weather is 22\xB0C"
+      });
+      const ctx2 = new ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "What is the weather?"
+      });
+      ctx2.insert(
+        new FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      ctx2.insert(
+        new FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      ctx2.addMessage({
+        id: "msg_2",
+        role: "assistant",
+        content: "The weather is 22\xB0C"
+      });
+      expect(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+  });
+});
 //# sourceMappingURL=chat_context.test.js.map