60 lines
1.8 KiB
TypeScript
60 lines
1.8 KiB
TypeScript
import { describe, it, expect, vi } from 'vitest';
|
|
import type { HybridSearch, MemoryFrame } from '@waggle/core';
|
|
import { MemoryLinker } from '../src/memory-linker.js';
|
|
|
|
describe('MemoryLinker', () => {
|
|
const mockSearch = {
|
|
search: vi.fn(),
|
|
} as unknown as HybridSearch;
|
|
|
|
it('finds related frames above threshold', async () => {
|
|
const linker = new MemoryLinker({ search: mockSearch, threshold: 0.1 });
|
|
|
|
vi.mocked(mockSearch.search).mockResolvedValueOnce([
|
|
{
|
|
frame: { id: 1, content: 'TypeScript generics guide' } as unknown as MemoryFrame,
|
|
rrfScore: 0.5,
|
|
relevanceScore: 0.4,
|
|
finalScore: 0.45,
|
|
},
|
|
{
|
|
frame: { id: 2, content: 'Advanced type inference' } as unknown as MemoryFrame,
|
|
rrfScore: 0.3,
|
|
relevanceScore: 0.2,
|
|
finalScore: 0.25,
|
|
},
|
|
]);
|
|
|
|
const links = await linker.findRelated('TypeScript types');
|
|
|
|
expect(mockSearch.search).toHaveBeenCalledWith('TypeScript types', { limit: 5 });
|
|
expect(links).toEqual([
|
|
{ frameId: 1, content: 'TypeScript generics guide', score: 0.45 },
|
|
{ frameId: 2, content: 'Advanced type inference', score: 0.25 },
|
|
]);
|
|
});
|
|
|
|
it('filters out frames below threshold', async () => {
|
|
const linker = new MemoryLinker({ search: mockSearch, threshold: 0.3 });
|
|
|
|
vi.mocked(mockSearch.search).mockResolvedValueOnce([
|
|
{
|
|
frame: { id: 3, content: 'Cooking recipes' } as unknown as MemoryFrame,
|
|
rrfScore: 0.1,
|
|
relevanceScore: 0.05,
|
|
finalScore: 0.08,
|
|
},
|
|
{
|
|
frame: { id: 4, content: 'Grocery list' } as unknown as MemoryFrame,
|
|
rrfScore: 0.05,
|
|
relevanceScore: 0.02,
|
|
finalScore: 0.03,
|
|
},
|
|
]);
|
|
|
|
const links = await linker.findRelated('quantum physics');
|
|
|
|
expect(links).toEqual([]);
|
|
});
|
|
});
|