Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

// META: title=Language Model Response Regex
// META: script=/resources/testdriver.js
// META: script=/resources/testdriver-vendor.js
// META: script=../resources/util.js
// META: timeout=long
'use strict';
promise_test(async t => {
await ensureLanguageModel();
const session = await createLanguageModel();
const response =
await session.prompt(kTestPrompt, {responseConstraint: /hello/});
assert_true(typeof response === 'string');
}, 'Prompt should work with a valid regex constraint.');
promise_test(async t => {
await ensureLanguageModel();
const session = await createLanguageModel();
const goodPrefix = 'Greetings';
const response = await session.prompt(
[
{role: 'user', content: 'hello'},
{role: 'assistant', content: goodPrefix, prefix: true}
],
{responseConstraint: /^Greetings and salutations.*/});
assert_true(typeof response === 'string');
}, 'Prompt should work with a valid regex constraint and matching prefix.');
promise_test(async t => {
await ensureLanguageModel();
const session = await createLanguageModel();
const badPrefix = 'invalid';
await promise_rejects_dom(
t, 'NotSupportedError',
session.prompt(
[
{role: 'user', content: 'hello'},
{role: 'assistant', content: badPrefix, prefix: true}
],
{responseConstraint: /^Greetings and salutations.*/}),
'The request is invalid - the input or options could not be processed.');
}, 'Prompt should reject if the prefix deviates from the regex constraint.');