@livekit/agents 1.1.0 → 1.2.0

package/dist/llm/chat_context.test.cjs

@@ -1,6 +1,9 @@
 "use strict";
 var import_vitest = require("vitest");
+var import_log = require("../log.cjs");
+var import_fake_llm = require("../voice/testing/fake_llm.cjs");
 var import_chat_context = require("./chat_context.cjs");
+(0, import_log.initializeLogger)({ pretty: false, level: "error" });
 (0, import_vitest.describe)("ChatContext.toJSON", () => {
   (0, import_vitest.it)("should match snapshot for empty context", () => {
     const context = new import_chat_context.ChatContext();
@@ -232,6 +235,46 @@ var import_chat_context = require("./chat_context.cjs");
     ).toMatchSnapshot("message-properties-full");
   });
 });
+(0, import_vitest.describe)("ChatContext._summarize", () => {
+  (0, import_vitest.it)("keeps chronological timestamps with summary + tail", async () => {
+    const ctx = new import_chat_context.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 import_chat_context.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 import_fake_llm.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;
+      }
+    );
+    (0, import_vitest.expect)(summary).toBeDefined();
+    if (!summary || summary.type !== "message") {
+      throw new Error("summary message is missing");
+    }
+    (0, import_vitest.expect)(summary.createdAt).toBeCloseTo(2999.999, 6);
+    const createdAts = ctx.items.map((item) => item.createdAt);
+    const sorted = [...createdAts].sort((a, b) => a - b);
+    (0, import_vitest.expect)(createdAts).toEqual(sorted);
+  });
+});
 (0, import_vitest.describe)("ReadonlyChatContext with immutable array", () => {
   (0, import_vitest.it)("should have readonly property set to true", () => {
     const items = [
@@ -377,4 +420,535 @@ var import_chat_context = require("./chat_context.cjs");
     (0, import_vitest.expect)(ids).toEqual(["msg_1", "msg_2"]);
   });
 });
+(0, import_vitest.describe)("ChatContext.isEquivalent", () => {
+  (0, import_vitest.it)("should return true for same reference", () => {
+    const ctx = new import_chat_context.ChatContext();
+    ctx.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    (0, import_vitest.expect)(ctx.isEquivalent(ctx)).toBe(true);
+  });
+  (0, import_vitest.it)("should return true for identical empty contexts", () => {
+    const ctx1 = new import_chat_context.ChatContext();
+    const ctx2 = new import_chat_context.ChatContext();
+    (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+  });
+  (0, import_vitest.it)("should return false for contexts with different lengths", () => {
+    const ctx1 = new import_chat_context.ChatContext();
+    ctx1.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    const ctx2 = new import_chat_context.ChatContext();
+    ctx2.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    ctx2.addMessage({
+      id: "msg_2",
+      role: "assistant",
+      content: "Hi"
+    });
+    (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+  });
+  (0, import_vitest.it)("should return false for contexts with different item IDs", () => {
+    const ctx1 = new import_chat_context.ChatContext();
+    ctx1.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    const ctx2 = new import_chat_context.ChatContext();
+    ctx2.addMessage({
+      id: "msg_2",
+      role: "user",
+      content: "Hello"
+    });
+    (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+  });
+  (0, import_vitest.it)("should return false for contexts with different item types", () => {
+    const ctx1 = new import_chat_context.ChatContext();
+    ctx1.addMessage({
+      id: "msg_1",
+      role: "user",
+      content: "Hello"
+    });
+    const ctx2 = new import_chat_context.ChatContext();
+    ctx2.insert(
+      new import_chat_context.FunctionCall({
+        id: "msg_1",
+        callId: "call_1",
+        name: "test",
+        args: "{}"
+      })
+    );
+    (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+  });
+  (0, import_vitest.describe)("message comparison", () => {
+    (0, import_vitest.it)("should return true for identical messages", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: false
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: false
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    (0, import_vitest.it)("should return false for messages with different roles", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello"
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "assistant",
+        content: "Hello"
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return false for messages with different interrupted flags", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: false
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello",
+        interrupted: true
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return false for messages with different content", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "Hello"
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "World"
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return true for messages with identical array content", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello", "World"]
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello", "World"]
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    (0, import_vitest.it)("should return false for messages with different array content", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello", "World"]
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Hello"]
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.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 import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", imageContent]
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", { ...imageContent }]
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    (0, import_vitest.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 import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", imageContent1]
+      });
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: ["Check this:", imageContent2]
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+  });
+  (0, import_vitest.describe)("function call comparison", () => {
+    (0, import_vitest.it)("should return true for identical function calls", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    (0, import_vitest.it)("should return false for function calls with different names", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}"
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_time",
+          args: "{}"
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return false for function calls with different call IDs", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}"
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_2",
+          name: "get_weather",
+          args: "{}"
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return false for function calls with different arguments", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "London"}'
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should ignore timestamps", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}",
+          createdAt: 1e3
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: "{}",
+          createdAt: 2e3
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+  });
+  (0, import_vitest.describe)("function call output comparison", () => {
+    (0, import_vitest.it)("should return true for identical function call outputs", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+    (0, import_vitest.it)("should return false for function call outputs with different names", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_time",
+          output: "{}",
+          isError: false
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return false for function call outputs with different call IDs", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_2",
+          name: "get_weather",
+          output: "{}",
+          isError: false
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return false for function call outputs with different output values", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 22}',
+          isError: false
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: '{"temperature": 25}',
+          isError: false
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should return false for function call outputs with different error flags", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "Error occurred",
+          isError: false
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "Error occurred",
+          isError: true
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(false);
+    });
+    (0, import_vitest.it)("should ignore timestamps", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false,
+          createdAt: 1e3
+        })
+      );
+      const ctx2 = new import_chat_context.ChatContext();
+      ctx2.insert(
+        new import_chat_context.FunctionCallOutput({
+          id: "output_1",
+          callId: "call_1",
+          name: "get_weather",
+          output: "{}",
+          isError: false,
+          createdAt: 2e3
+        })
+      );
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+  });
+  (0, import_vitest.describe)("complex context comparison", () => {
+    (0, import_vitest.it)("should return true for identical complex contexts", () => {
+      const ctx1 = new import_chat_context.ChatContext();
+      ctx1.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "What is the weather?"
+      });
+      ctx1.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      ctx1.insert(
+        new import_chat_context.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 import_chat_context.ChatContext();
+      ctx2.addMessage({
+        id: "msg_1",
+        role: "user",
+        content: "What is the weather?"
+      });
+      ctx2.insert(
+        new import_chat_context.FunctionCall({
+          id: "func_1",
+          callId: "call_1",
+          name: "get_weather",
+          args: '{"location": "Paris"}'
+        })
+      );
+      ctx2.insert(
+        new import_chat_context.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"
+      });
+      (0, import_vitest.expect)(ctx1.isEquivalent(ctx2)).toBe(true);
+    });
+  });
+});
 //# sourceMappingURL=chat_context.test.cjs.map