Files
swarm-assistent/test/intent.test.js
T

54 lines
2.2 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('что ты умеешь'), false);
assert.equal(userAsksGenerate('не генерируй'), false);
});
it('resolveTurnIntent uses model generate + user generate ask + veto', () => {
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 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);
});
it('packWantsVision for critique pack', () => {
assert.equal(packWantsVision('critique_image'), true);
assert.equal(packWantsVision('ordinary'), false);
});
});