test_Conversation.js

Enable keyboard shortcuts

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

const { Conversation } = ChromeUtils.importESModule(

  "moz-src:///browser/components/aiwindow/models/Conversation.sys.mjs"

);

// Values mirror ui/modules/ChatEnums.sys.mjs.

const USER = 0;

const ASSISTANT = 1;

const SYS = 2;

add_task(function test_Conversation_starts_empty() {

  const c = new Conversation();

  Assert.equal(c.messages.length, 0);

  Assert.equal(c.messageCount, 0);

  Assert.equal(c.currentTurnIndex(), 0);

  Assert.equal(c.systemPromptVersion, "");

});

add_task(function test_addUserMessage_then_assistant_orders() {

  const c = new Conversation();

  c.addUserMessage("hello");

  c.addAssistantMessage("hi back");

  const messages = c.messages;

  Assert.equal(messages.length, 2);

  Assert.equal(messages[0].role, USER);

  Assert.equal(messages[0].content, "hello");

  Assert.equal(messages[0].ordinal, 1);

  Assert.equal(messages[1].role, ASSISTANT);

  Assert.equal(messages[1].ordinal, 2);

  Assert.equal(messages[1].parentMessageId, messages[0].id);

});

add_task(function test_setSystemMessage_idempotent_upsert() {

  const c = new Conversation();

  c.setSystemMessage({ body: "v1", version: "0.1" });

  c.addUserMessage("hello");

  Assert.equal(c.messages[0].role, SYS);

  Assert.equal(c.messages[0].content.body, "v1");

  Assert.equal(c.systemPromptVersion, "0.1");

  c.setSystemMessage({ body: "v2", version: "0.2" });

  Assert.equal(c.messages.length, 2);

  Assert.equal(c.messages[0].content.body, "v2");

  Assert.equal(c.systemPromptVersion, "0.2");

});

add_task(function test_setSystemMessage_string_wraps_into_body() {

  const c = new Conversation();

  c.setSystemMessage("hello");

  Assert.equal(c.messages[0].content.body, "hello");

  // No version provided — getter returns empty.

  Assert.equal(c.systemPromptVersion, "");

});

add_task(function test_getMessagesInChatCompletionsFormat_basic() {

  const c = new Conversation();

  c.setSystemMessage("sys");

  c.addUserMessage("u");

  c.addAssistantMessage("a");

  const wire = c.getMessagesInChatCompletionsFormat();

  Assert.equal(wire.length, 3);

  Assert.equal(wire[0].role, "system");

  Assert.equal(wire[0].content, "sys");

  Assert.equal(wire[1].role, "user");

  Assert.equal(wire[1].content, "u");

  Assert.equal(wire[2].role, "assistant");

  Assert.equal(wire[2].content, "a");

});

add_task(function test_addToolMessage_carries_tool_linkage() {

  const c = new Conversation();

  c.addToolMessage({ tool_call_id: "tc-1", content: "result", name: "search" });

  const wire = c.getMessagesInChatCompletionsFormat();

  Assert.equal(wire[0].role, "tool");

  Assert.equal(wire[0].tool_call_id, "tc-1");

  Assert.equal(wire[0].name, "search");

});

add_task(function test_retryMessage_truncates_and_bumps_floor() {

  const c = new Conversation();

  c.addUserMessage("u1");

  c.addAssistantMessage("a1");

  c.addUserMessage("u2");

  const removed = c.retryMessage(c.messages[1]);

  Assert.equal(removed.length, 2);

  Assert.equal(c.messages.length, 1);

  // Next ordinal floor must not reuse the removed ordinals (2 and 3).

  c.addUserMessage("u3");

  Assert.equal(c.messages.at(-1).ordinal, 4);

});

add_task(async function test_run_delegates_to_engine_with_parameters_merged() {

  let receivedArgs = null;

  const fakeEngine = {

    run(payload) {

      receivedArgs = payload;

      return Promise.resolve({ finalOutput: "ok" });

},

};

  const c = new Conversation({

    engine: fakeEngine,

    parameters: { temperature: 0.5 },

});

  c.setSystemMessage("sys");

  c.addUserMessage("hi");

  const resp = await c.run({ fxAccountToken: "tk" });

  Assert.equal(resp.finalOutput, "ok");

  Assert.equal(receivedArgs.temperature, 0.5);

  Assert.equal(receivedArgs.fxAccountToken, "tk");

  Assert.equal(receivedArgs.args.length, 2);

  Assert.equal(receivedArgs.args[0].role, "system");

});