john
42 lines
1.6 KiB
JavaScript
42 lines
1.6 KiB
JavaScript
import test from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import { filterMemoriesByQuery } from './memory-legacy-fallback.mjs';
|
|
import { isMemoryRecallQuestion } from './chat-intent-router.mjs';
|
|
|
|
test('filterMemoriesByQuery matches query tokens in memory text', () => {
|
|
const memories = [
|
|
{ text: '用户的测试别名是蓝鲸42' },
|
|
{ text: 'TKMind H5 聊天助手默认回复' },
|
|
{ text: '无关记忆条目' },
|
|
];
|
|
const matched = filterMemoriesByQuery(memories, '我之前让你记住的测试别名是什么');
|
|
assert.equal(matched.length, 1);
|
|
assert.match(matched[0].text, /蓝鲸42/);
|
|
});
|
|
|
|
test('isMemoryRecallQuestion matches remember-alias recall prompts', () => {
|
|
assert.equal(isMemoryRecallQuestion('我之前让你记住的测试别名是什么'), true);
|
|
assert.equal(isMemoryRecallQuestion('记住的测试别名是什么'), true);
|
|
});
|
|
|
|
test('resolveMemoriesWithLegacyFallback prefers legacy on recall questions', async () => {
|
|
const { resolveMemoriesWithLegacyFallback } = await import('./memory-legacy-fallback.mjs');
|
|
const memories = await resolveMemoriesWithLegacyFallback({
|
|
memoryV2: {
|
|
async resolve() {
|
|
return { memories: [{ text: '用户当前登录称呼为John' }] };
|
|
},
|
|
},
|
|
conversationMemoryService: {
|
|
async listMemories() {
|
|
return [{ text: '用户的测试别名是蓝鲸42' }];
|
|
},
|
|
},
|
|
userId: 'user-1',
|
|
query: '我之前让你记住的测试别名是什么？',
|
|
limit: 5,
|
|
recallQuestion: true,
|
|
});
|
|
assert.match(memories[0].text, /蓝鲸42/);
|
|
});
|