mirror of
https://github.com/Kaelio/ktx.git
synced 2026-06-28 08:49:38 +02:00
94 lines
3.8 KiB
TypeScript
94 lines
3.8 KiB
TypeScript
import { readFile } from 'node:fs/promises';
|
|
import { join } from 'node:path';
|
|
import { fileURLToPath } from 'node:url';
|
|
import { describe, expect, it } from 'vitest';
|
|
import { PromptService } from '../prompts/index.js';
|
|
import { SkillsRegistryService } from '../skills/index.js';
|
|
|
|
const promptsDir = fileURLToPath(new URL('../../prompts', import.meta.url));
|
|
const skillsDir = fileURLToPath(new URL('../../skills', import.meta.url));
|
|
|
|
const adapterSkillNames = [
|
|
'live_database_ingest',
|
|
'lookml_ingest',
|
|
'metabase_ingest',
|
|
'metricflow_ingest',
|
|
'notion_synthesize',
|
|
'historic_sql_table_digest',
|
|
'historic_sql_patterns',
|
|
'ingest_triage',
|
|
'knowledge_capture',
|
|
'sl_capture',
|
|
] as const;
|
|
|
|
const adapterReconcileSkillNames = [
|
|
'ingest_triage',
|
|
'knowledge_capture',
|
|
'sl_capture',
|
|
] as const;
|
|
|
|
const pageTriagePromptNames = ['skills/page_triage_classifier', 'skills/light_extraction'] as const;
|
|
|
|
function forbiddenProductPattern() {
|
|
return new RegExp([['Kae', 'lio'].join(''), ['kae', 'lio'].join(''), ['KAE', 'LIO_'].join('')].join('|'));
|
|
}
|
|
|
|
describe('ingest runtime assets', () => {
|
|
it('resolves every reusable ingest skill from packaged KTX assets without server fallback', async () => {
|
|
const registry = new SkillsRegistryService({ skillsDir });
|
|
const expected = [...new Set([...adapterSkillNames, ...adapterReconcileSkillNames])].sort();
|
|
|
|
const skills = await registry.listSkills(expected, 'memory_agent');
|
|
|
|
expect(skills.map((skill) => skill.name).sort()).toEqual(expected);
|
|
for (const skill of skills) {
|
|
expect(skill.path.startsWith(skillsDir)).toBe(true);
|
|
const body = await readFile(join(skill.path, 'SKILL.md'), 'utf-8');
|
|
expect(body).not.toMatch(forbiddenProductPattern());
|
|
}
|
|
});
|
|
|
|
it('loads page-triage and light-extraction prompts from packaged KTX prompt assets', async () => {
|
|
const prompts = new PromptService({ promptsDir, partials: [] });
|
|
|
|
for (const promptName of pageTriagePromptNames) {
|
|
const prompt = await prompts.loadPrompt(promptName);
|
|
expect(prompt.trim().length).toBeGreaterThan(100);
|
|
expect(prompt).not.toMatch(forbiddenProductPattern());
|
|
}
|
|
|
|
await expect(prompts.loadPrompt('skills/page_triage_classifier')).resolves.toContain('# Page Triage Classifier');
|
|
await expect(prompts.loadPrompt('skills/light_extraction')).resolves.toContain('# Light Context Extraction');
|
|
});
|
|
|
|
it('packages historic-SQL table digest guidance from KTX assets', async () => {
|
|
const registry = new SkillsRegistryService({ skillsDir });
|
|
const skills = await registry.listSkills(['historic_sql_table_digest'], 'memory_agent');
|
|
|
|
expect(skills.map((skill) => skill.name)).toEqual(['historic_sql_table_digest']);
|
|
|
|
const body = await readFile(join(skills[0]!.path, 'SKILL.md'), 'utf-8');
|
|
expect(body).toContain('# Historic SQL Table Digest');
|
|
expect(body).toContain('tables/<schema>.<name>.json');
|
|
expect(body).toContain('tableUsageOutputSchema');
|
|
expect(body).toContain('emit_historic_sql_evidence');
|
|
expect(body).toContain('Do not call wiki_write');
|
|
expect(body).toContain('Do not call sl_write_source');
|
|
expect(body).not.toMatch(forbiddenProductPattern());
|
|
});
|
|
|
|
it('packages historic-SQL patterns guidance from KTX assets', async () => {
|
|
const registry = new SkillsRegistryService({ skillsDir });
|
|
const skills = await registry.listSkills(['historic_sql_patterns'], 'memory_agent');
|
|
|
|
expect(skills.map((skill) => skill.name)).toEqual(['historic_sql_patterns']);
|
|
|
|
const body = await readFile(join(skills[0]!.path, 'SKILL.md'), 'utf-8');
|
|
expect(body).toContain('# Historic SQL Patterns');
|
|
expect(body).toContain('patterns-input/part-0001.json');
|
|
expect(body).toContain('patternsArraySchema');
|
|
expect(body).toContain('emit_historic_sql_evidence');
|
|
expect(body).toContain('cross-table');
|
|
expect(body).not.toMatch(forbiddenProductPattern());
|
|
});
|
|
});
|