memind/chat-skills.test.mjs

import test from 'node:test';
import assert from 'node:assert/strict';
import {
  buildAutoChatSkillPrefix,
  buildChatSkillPrompt,
  CHAT_SKILL_DEFINITIONS,
  filterChatSkills,
  isExcelAnalysisIntent,
  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', () => {
  const webPrompt = buildChatSkillPrompt('web', 'web');
  assert.match(webPrompt, /请使用 web 技能/);
  assert.match(webPrompt, /tkmind_search/);
  assert.match(webPrompt, /web_search/);
  const enhancedPrompt = buildChatSkillPrompt('search-enhanced', 'search-enhanced');
  assert.match(enhancedPrompt, /tkmind_search/);
  assert.match(enhancedPrompt, /web_search/);
  assert.match(buildChatSkillPrompt('service-integration-smoke'), /标准联调流程/);
  assert.match(buildChatSkillPrompt('product-campaign-page'), /商品宣传 \/ 活动页/);
  assert.match(buildChatSkillPrompt('image-generation'), /asset\.htmlSrc/);
  assert.match(buildChatSkillPrompt('image-generation'), /workspaceRelativePath 只用于文件操作/);
  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 routes an uploaded xlsx only when Excel Analyst is granted', () => {
  const text = '帮我分析一下\n\n[文件1: sales.xlsx]: /api/mindspace/v1/assets/asset-1/download';
  assert.match(
    buildAutoChatSkillPrefix(text, ['excel-analyst']),
    /excel_inspect/,
  );
  assert.equal(buildAutoChatSkillPrefix(text, []), '');
  assert.equal(isExcelAnalysisIntent('请分析这份工作簿的销售趋势'), true);
  assert.equal(isExcelAnalysisIntent('请总结这段普通文字'), false);
});

test('manifest enhanced search route is opt-in and uses the MindSearch skill prompt', () => {
  const routes = [{ skillName: 'search-enhanced', promptKey: 'search-enhanced', keywords: ['最新资料'], priority: 30 }];
  assert.match(buildAutoChatSkillPrefix('请搜索最新资料：Goose MCP', ['search-enhanced'], { skillRouterV2: true, manifestRoutes: routes }), /tkmind-search/);
  assert.equal(buildAutoChatSkillPrefix('请搜索最新资料：Goose MCP', [], { skillRouterV2: true, manifestRoutes: routes }), '');
});

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('帮我做一个家庭记账页面，我每天可以记录日期、收入、支出、分类和备注，日期默认当天。再做一个管理页面，可以查看所有记录和收支汇总。'),
    true,
  );
  assert.equal(isPageDataIntent('帮我做一个苏州攻略页面'), false);
});

test('isPageDataIntent matches implicit interactive sticky-note app requests', () => {
  const text = '帮我设计一个便签提醒，可以写便签提交，时间轴来显示';
  assert.equal(isPageDataIntent(text), true);
  assert.match(
    buildAutoChatSkillPrefix(text, ['page-data-collect', 'static-page-publish']),
    /page-data-collect/,
  );
});

test('isPageDataIntent does not steal a simple schedule reminder', () => {
  assert.equal(isPageDataIntent('明天上午9点提醒我开会'), false);
  assert.equal(isPageDataIntent('创建一个待办'), false);
});

test('buildAutoChatSkillPrefix routes ordinary household ledger language to Page Data', () => {
  const text = '帮我做一个家庭记账页面，我每天可以记录日期、收入、支出、分类和备注，日期默认当天。再做一个管理页面，可以查看所有记录和收支汇总。';
  const prefix = buildAutoChatSkillPrefix(text, ['page-data-collect', 'static-page-publish']);
  assert.match(prefix, /page-data-collect/);
  assert.match(prefix, /同一轮继续执行/);
  assert.match(prefix, /禁止 localStorage/);
  assert.doesNotMatch(prefix, /static-page-publish 技能/);
});

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