42 lines
1.4 KiB
TypeScript
42 lines
1.4 KiB
TypeScript
import { describe, expect, it, vi } from 'vitest';
|
|
import { createMemoryTools } from './memory-tools.js';
|
|
|
|
describe('createMemoryTools operator retrieval binding', () => {
|
|
const memory = {
|
|
insights: { searchByEmbedding: vi.fn(), create: vi.fn() },
|
|
preferences: { findByUserAndCategory: vi.fn(), findByUser: vi.fn(), upsert: vi.fn() },
|
|
};
|
|
const scope = { tenantId: 'tenant-a', ownerId: 'owner-a', sessionId: 'session-a' };
|
|
|
|
it('uses the configured plugin with the server-derived scope for retrieval and capture', async () => {
|
|
const plugin = {
|
|
search: vi.fn(async () => []),
|
|
capture: vi.fn(async () => ({ id: 'insight-1' })),
|
|
};
|
|
const tools = createMemoryTools(memory as never, null, 'owner-a', {
|
|
plugin: plugin as never,
|
|
scope,
|
|
});
|
|
|
|
await tools
|
|
.find((tool) => tool.name === 'memory_search')!
|
|
.execute('call-1', { query: 'plans' }, undefined, undefined, {} as never);
|
|
await tools
|
|
.find((tool) => tool.name === 'memory_save_insight')!
|
|
.execute(
|
|
'call-2',
|
|
{ content: 'secret', category: 'decision' },
|
|
undefined,
|
|
undefined,
|
|
{} as never,
|
|
);
|
|
|
|
expect(plugin.search).toHaveBeenCalledWith(scope, 'plans', 5);
|
|
expect(plugin.capture).toHaveBeenCalledWith(scope, {
|
|
content: 'secret',
|
|
source: 'agent',
|
|
category: 'decision',
|
|
});
|
|
});
|
|
});
|