memind/chat-skills.test.mjs

import test from 'node:test';
import assert from 'node:assert/strict';
import {
  buildAutoChatSkillPrefix,
  buildChatSkillPrompt,
  CHAT_SKILL_DEFINITIONS,
  filterChatSkills,
  isPageDataIntent,
  isPageGenerationIntent,
} from './chat-skills.mjs';

test('filterChatSkills shows summarize and analyze without granted skills', () => {
  const visible = filterChatSkills(CHAT_SKILL_DEFINITIONS, { canPublish: false });
  assert.ok(visible.some((item) => item.id === 'summarize'));
  assert.ok(visible.some((item) => item.id === 'analyze'));
  assert.equal(visible.some((item) => item.id === 'generate-page'), false);
});

test('filterChatSkills gates platform skills by grantedSkills', () => {
  const visible = filterChatSkills(CHAT_SKILL_DEFINITIONS, {
    canPublish: false,
    grantedSkills: ['web', 'search'],
  });
  assert.ok(visible.some((item) => item.id === 'web-search'));
  assert.ok(visible.some((item) => item.id === 'code-search'));
  assert.equal(visible.some((item) => item.id === 'form-collect'), false);
  assert.equal(visible.some((item) => item.id === 'service-integration-smoke'), false);
  assert.equal(visible.some((item) => item.id === 'product-campaign-page'), false);
});

test('filterChatSkills shows product campaign page when granted', () => {
  const visible = filterChatSkills(CHAT_SKILL_DEFINITIONS, {
    canPublish: false,
    grantedSkills: ['product-campaign-page'],
  });
  assert.ok(visible.some((item) => item.id === 'product-campaign-page'));
});

test('filterChatSkills shows service integration smoke when granted', () => {
  const visible = filterChatSkills(CHAT_SKILL_DEFINITIONS, {
    canPublish: false,
    grantedSkills: ['service-integration-smoke'],
  });
  assert.ok(visible.some((item) => item.id === 'service-integration-smoke'));
});

test('filterChatSkills shows page-data-collect when granted without static publish', () => {
  const visible = filterChatSkills(CHAT_SKILL_DEFINITIONS, {
    canPublish: false,
    grantedSkills: ['page-data-collect'],
  });
  assert.ok(visible.some((item) => item.id === 'page-data-collect'));
});

test('filterChatSkills shows generate-page when publish is allowed', () => {
  const visible = filterChatSkills(CHAT_SKILL_DEFINITIONS, { canPublish: true });
  assert.ok(visible.some((item) => item.id === 'generate-page'));
});

test('buildChatSkillPrompt includes skill name for platform skills', () => {
  assert.match(buildChatSkillPrompt('web', 'web'), /请使用 web 技能/);
  assert.match(buildChatSkillPrompt('service-integration-smoke'), /标准联调流程/);
  assert.match(buildChatSkillPrompt('product-campaign-page'), /商品宣传 \/ 活动页/);
  assert.match(buildChatSkillPrompt('generate-page'), /static-page-publish/);
  assert.match(buildChatSkillPrompt('generate-page'), /docx-generate/);
  assert.match(buildChatSkillPrompt('generate-page'), /public\/\*\.docx/);
  assert.match(buildChatSkillPrompt('generate-page'), /long-image-download/);
  assert.match(buildChatSkillPrompt('generate-page'), /generate_long_image/);
  assert.match(buildChatSkillPrompt('page-data-collect'), /Page Data API/);
  assert.match(buildChatSkillPrompt('page-data-collect'), /禁止自建 Express/);
});

test('prefillOnly is set for open-ended chat skills', () => {
  assert.equal(CHAT_SKILL_DEFINITIONS.find((item) => item.id === 'web-search')?.prefillOnly, true);
  assert.equal(CHAT_SKILL_DEFINITIONS.find((item) => item.id === 'generate-page')?.prefillOnly, undefined);
});

test('buildAutoChatSkillPrefix enables web for news-like queries', () => {
  assert.match(
    buildAutoChatSkillPrefix('帮我搜索今天热点新闻', ['web', 'search']),
    /先搜索今天\/最新相关的新闻与热点/,
  );
  assert.equal(buildAutoChatSkillPrefix('帮我总结一下这段话', ['web', 'search']), '');
  assert.equal(buildAutoChatSkillPrefix('帮我搜索今天热点新闻', ['search']), '');
});

test('buildAutoChatSkillPrefix enables publish for page generation requests', () => {
  const prefix = buildAutoChatSkillPrefix('帮我做一个 Hello 糖的 H5 页面', ['static-page-publish']);
  assert.match(prefix, /static-page-publish/);
  assert.match(prefix, /write_file\/edit_file/);
  assert.match(prefix, /禁止只给本地路径/);
  assert.equal(buildAutoChatSkillPrefix('帮我做一个 Hello 糖的 H5 页面', []), '');
});

test('isPageGenerationIntent matches implicit travel guide page requests', () => {
  assert.equal(isPageGenerationIntent('苏州攻略页面'), true);
  assert.equal(isPageGenerationIntent('帮我看看苏州攻略，1日攻略，做一个页面'), true);
  assert.equal(isPageGenerationIntent('你好'), false);
});

test('buildAutoChatSkillPrefix prefers page-data-collect for survey requests', () => {
  const text = '帮我在这个页面增加一个调查问卷，出三个问题，密码 888 查看提交记录';
  const prefix = buildAutoChatSkillPrefix(text, ['page-data-collect', 'static-page-publish']);
  assert.match(prefix, /page-data-collect/);
  assert.doesNotMatch(prefix, /static-page-publish 技能：在我的专属 MindSpace 发布目录生成静态 HTML/);
});

test('isPageDataIntent matches survey and backend keywords', () => {
  assert.equal(isPageDataIntent('增加调查问卷和后台入口，密码 888'), true);
  assert.equal(isPageDataIntent('帮我做一个苏州攻略页面'), false);
});

test('buildAutoChatSkillPrefix enables publish for implicit page requests', () => {
  const prefix = buildAutoChatSkillPrefix('苏州攻略页面', ['static-page-publish']);
  assert.match(prefix, /static-page-publish/);
  assert.match(prefix, /禁止询问是否还要发布/);
});