79 lines
2 KiB
JavaScript
79 lines
2 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
const { GenAI } = ChromeUtils.importESModule(
|
|
"resource:///modules/GenAI.sys.mjs"
|
|
);
|
|
|
|
registerCleanupFunction(() =>
|
|
Services.prefs.clearUserPref("browser.ml.chat.prompts.test")
|
|
);
|
|
|
|
const DEFAULT_CONTEXT = { provider: "" };
|
|
async function promptCount(context = DEFAULT_CONTEXT) {
|
|
return (await GenAI.getContextualPrompts(context)).length;
|
|
}
|
|
|
|
/**
|
|
* Check prompts from prefs are used
|
|
*/
|
|
add_task(async function test_prompt_prefs() {
|
|
const origPrompts = await promptCount();
|
|
|
|
Services.prefs.setStringPref("browser.ml.chat.prompts.test", "{}");
|
|
Assert.equal(await promptCount(), origPrompts + 1, "Added a prompt");
|
|
|
|
Services.prefs.setStringPref(
|
|
"browser.ml.chat.prompts.test",
|
|
JSON.stringify({ targeting: "false" })
|
|
);
|
|
Assert.equal(
|
|
await promptCount(),
|
|
origPrompts,
|
|
"Prompt not added for targeting"
|
|
);
|
|
});
|
|
|
|
/**
|
|
* Check context is used for targeting
|
|
*/
|
|
add_task(async function test_prompt_context() {
|
|
Services.prefs.setStringPref(
|
|
"browser.ml.chat.prompts.test",
|
|
JSON.stringify({ targeting: "provider" })
|
|
);
|
|
const origPrompts = await promptCount();
|
|
|
|
Assert.equal(
|
|
await promptCount({ provider: "localhost" }),
|
|
origPrompts + 1,
|
|
"Added contextual prompt"
|
|
);
|
|
|
|
Assert.equal(
|
|
await promptCount(),
|
|
origPrompts,
|
|
"Prompt not added for contextual targeting"
|
|
);
|
|
});
|
|
|
|
/**
|
|
* Check that prompts can be localized
|
|
*/
|
|
add_task(async function test_prompt_l10n() {
|
|
Services.prefs.setStringPref(
|
|
"browser.ml.chat.prompts.test",
|
|
JSON.stringify({
|
|
l10nId: "genai-prompts-summarize",
|
|
label: "bad",
|
|
test: true,
|
|
})
|
|
);
|
|
const custom = (await GenAI.getContextualPrompts(DEFAULT_CONTEXT)).find(
|
|
prompt => prompt.id == null
|
|
);
|
|
|
|
Assert.ok(custom.test, "Found the custom prompt");
|
|
Assert.notEqual(custom.label, "bad", "Localized label takes priority");
|
|
Assert.ok(custom.value, "Got localized prompt");
|
|
});
|