17 lines
555 B
JavaScript
17 lines
555 B
JavaScript
const agent = require('../src/agent');
|
|
const retriever = require('../src/retriever');
|
|
|
|
beforeAll(async () => {
|
|
await retriever.loadKnowledgeBase('./knowledge');
|
|
});
|
|
|
|
test('retriever returns passages', async () => {
|
|
const passages = await retriever.getRelevantPassages('capital', 2);
|
|
expect(Array.isArray(passages)).toBe(true);
|
|
expect(passages.length).toBeLessThanOrEqual(2);
|
|
});
|
|
|
|
test('agent generates answer', async () => {
|
|
const answer = await agent.ask('What is the capital of France?');
|
|
expect(answer.toLowerCase()).toContain('paris');
|
|
}); |