118 lines
4.0 KiB
TypeScript
118 lines
4.0 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
||
import {
|
||
buildIntakeFollowUpMessage,
|
||
buildIntakeProgress,
|
||
extractIntakeHeuristic,
|
||
intakeFieldsFromInquiry,
|
||
synthesizeDemandText,
|
||
} from "../src/intake/intake.js";
|
||
import { applyEvent, createSession } from "../src/runtime/phase-machine.js";
|
||
import { toActiveSkillSnapshot } from "../src/skills/snapshot.js";
|
||
import { loadSkill } from "../src/skills/loader.js";
|
||
|
||
describe("intake fields", () => {
|
||
it("parses required and optional from roleplay skill", async () => {
|
||
const skill = await loadSkill("roleplay-game-theory");
|
||
const fields = intakeFieldsFromInquiry(skill.startupInquiry);
|
||
expect(fields.filter((f) => f.required).length).toBe(3);
|
||
expect(fields.filter((f) => !f.required).length).toBe(3);
|
||
});
|
||
|
||
it("marks ready when all required filled", () => {
|
||
const fields = intakeFieldsFromInquiry({
|
||
prompt: "p",
|
||
targetKey: "用户.博弈需求",
|
||
requiredFields: ["情境", "角色", "轮次"],
|
||
optionalFields: ["输出偏好"],
|
||
});
|
||
const partial = buildIntakeProgress(fields, {
|
||
[fields[0].id]: "德州扑克",
|
||
});
|
||
expect(partial.ready).toBe(false);
|
||
expect(partial.requiredFilled).toBe(1);
|
||
|
||
const full = buildIntakeProgress(fields, {
|
||
[fields[0].id]: "德州扑克",
|
||
[fields[1].id]: "玩家A算计，玩家B保守",
|
||
[fields[2].id]: "单轮",
|
||
});
|
||
expect(full.ready).toBe(true);
|
||
});
|
||
|
||
it("heuristic fills scenario and rounds keywords", () => {
|
||
const fields = intakeFieldsFromInquiry({
|
||
prompt: "p",
|
||
targetKey: "t",
|
||
requiredFields: ["实验情境", "轮次模式"],
|
||
optionalFields: [],
|
||
});
|
||
const values = extractIntakeHeuristic(
|
||
"德州扑克，单轮定胜负",
|
||
fields,
|
||
{},
|
||
);
|
||
expect(Object.keys(values).length).toBeGreaterThan(0);
|
||
});
|
||
|
||
it("parses roleplay intake fields as 3 required + optional", async () => {
|
||
const skill = await loadSkill("roleplay-game-theory");
|
||
const fields = intakeFieldsFromInquiry(skill.startupInquiry);
|
||
expect(fields.filter((f) => f.required).length).toBe(3);
|
||
expect(fields.filter((f) => !f.required).length).toBe(3);
|
||
expect(fields.some((f) => f.label.includes("情境"))).toBe(true);
|
||
expect(fields.some((f) => f.label.includes("进程"))).toBe(true);
|
||
});
|
||
|
||
it("buildIntakeFollowUpMessage lists missing required fields", () => {
|
||
const fields = intakeFieldsFromInquiry({
|
||
prompt: "p",
|
||
targetKey: "t",
|
||
requiredFields: ["情境", "角色", "进程"],
|
||
optionalFields: [],
|
||
});
|
||
const progress = buildIntakeProgress(fields, {
|
||
[fields[0].id]: "德州扑克",
|
||
});
|
||
const msg = buildIntakeFollowUpMessage(progress);
|
||
expect(msg).toContain("还缺以下必要项");
|
||
expect(msg).toContain("角色");
|
||
expect(msg).toContain("进程");
|
||
});
|
||
|
||
it("confirm intake writes structured demand tag", async () => {
|
||
const snap = toActiveSkillSnapshot(await loadSkill("basic"));
|
||
let session = createSession("default");
|
||
session = applyEvent(session, {
|
||
type: "session_started",
|
||
payload: {
|
||
presetId: "default",
|
||
availableSkills: [{ name: "basic", description: "", category: "novel" }],
|
||
},
|
||
}).session;
|
||
session = applyEvent(session, {
|
||
type: "skill_selected",
|
||
payload: { skill: snap },
|
||
}).session;
|
||
expect(session.waitingReason?.kind).toBe("intake");
|
||
|
||
const fields = snap.intakeFields;
|
||
const values: Record<string, string> = {};
|
||
for (const f of fields) {
|
||
values[f.id] = `值-${f.label}`;
|
||
}
|
||
session = applyEvent(session, {
|
||
type: "user_submitted_input",
|
||
payload: { text: "一次性提交", intakeValues: values },
|
||
}).session;
|
||
expect(session.waitingReason?.kind).toBe("intake");
|
||
|
||
session = applyEvent(session, {
|
||
type: "user_confirmed_intake",
|
||
payload: {},
|
||
}).session;
|
||
expect(session.slots.startupCompleted).toBe(true);
|
||
expect(String(session.slots["book.brief"])).toContain("题材");
|
||
expect(synthesizeDemandText(fields, values)).toContain("值-题材");
|
||
});
|
||
});
|