90 lines
4.0 KiB
JavaScript
90 lines
4.0 KiB
JavaScript
import { describe, it } from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import {
|
|
userAsksLook,
|
|
userAsksNoGenerate,
|
|
userAsksGenerate,
|
|
resolveTurnIntent,
|
|
packWantsVision,
|
|
} from '../src/intent.js';
|
|
|
|
describe('intent.js', () => {
|
|
it('userAsksLook detects critique RU', () => {
|
|
assert.equal(userAsksLook('посмотри на результат'), true);
|
|
assert.equal(userAsksLook('нарисуй лису'), false);
|
|
});
|
|
|
|
it('userAsksNoGenerate vetoes remember / no-gen', () => {
|
|
assert.equal(userAsksNoGenerate('только запомни промпт'), true);
|
|
assert.equal(userAsksNoGenerate('не генерируй'), true);
|
|
assert.equal(userAsksNoGenerate('нарисуй лису'), false);
|
|
});
|
|
|
|
it('userAsksGenerate detects сгенерируй', () => {
|
|
assert.equal(userAsksGenerate('сгенерируй'), true);
|
|
assert.equal(userAsksGenerate('нарисуй лису'), true);
|
|
assert.equal(userAsksGenerate('Давай сделаем изображение красивой рыжеволосой девушки'), true);
|
|
assert.equal(userAsksGenerate('Привет, давай фотографию 19-летней девушки на кастинге'), true);
|
|
assert.equal(userAsksGenerate('создай фото как на референсе — рыжая в кожанке'), true);
|
|
assert.equal(userAsksGenerate('сделай фото портрет'), true);
|
|
assert.equal(userAsksGenerate('create a photo of a redhead'), true);
|
|
assert.equal(userAsksGenerate('как сгенерировать изображение в swarm'), false);
|
|
assert.equal(userAsksGenerate('а ты знаешь какие то промпты с civitai'), false);
|
|
assert.equal(userAsksGenerate('что ты умеешь'), false);
|
|
assert.equal(userAsksGenerate('не генерируй'), false);
|
|
});
|
|
|
|
it('resolveTurnIntent uses user generate ask + veto, not model flag alone', () => {
|
|
const lookOnly = resolveTurnIntent({ look_at: ['generate'] }, 'что не так с кадром?', {});
|
|
assert.equal(lookOnly.generate, false);
|
|
assert.equal(lookOnly.look, true);
|
|
|
|
const gen = resolveTurnIntent({ prompt: 'fox', generate: true }, 'нарисуй лису', {});
|
|
assert.equal(gen.generate, true);
|
|
|
|
const userGen = resolveTurnIntent({ prompt: 'A slender redhead in leather' }, 'сгенерируй', {});
|
|
assert.equal(userGen.generate, true);
|
|
|
|
const createPhoto = resolveTurnIntent(
|
|
{ prompt: '19yo redhead at casting, studio light' },
|
|
'создай фото рыжей девушки на кастинге',
|
|
{},
|
|
);
|
|
assert.equal(createPhoto.generate, true);
|
|
|
|
const promptOnly = resolveTurnIntent(
|
|
{ prompt: 'fox in snow' },
|
|
'создай фото лисы',
|
|
{},
|
|
);
|
|
assert.equal(promptOnly.generate, true);
|
|
|
|
const vetoed = resolveTurnIntent({ prompt: 'fox', generate: true }, 'только запомни промпт', {});
|
|
assert.equal(vetoed.generate, false);
|
|
assert.equal(vetoed.vetoed, true);
|
|
|
|
const chatOnly = resolveTurnIntent({ prompt: 'fox' }, 'что ты умеешь про генерацию?', {});
|
|
assert.equal(chatOnly.generate, false);
|
|
|
|
const falseDump = resolveTurnIntent(
|
|
{ prompt: 'civitai example', generate: true },
|
|
'а ты знаешь какие то промпты с civitai',
|
|
{},
|
|
);
|
|
assert.equal(falseDump.generate, false);
|
|
assert.equal(falseDump.modelAsked, true);
|
|
|
|
const sdelаем = resolveTurnIntent(
|
|
{ prompt: 'A 19-year-old redhead', generate: true },
|
|
'Давай сделаем изображение красивой рыжеволосой девушки',
|
|
{},
|
|
);
|
|
assert.equal(sdelаем.generate, true);
|
|
});
|
|
|
|
it('packWantsVision for critique pack', () => {
|
|
assert.equal(packWantsVision('critique_image'), true);
|
|
assert.equal(packWantsVision('ordinary'), false);
|
|
});
|
|
});
|