This commit is contained in:
@@ -0,0 +1,17 @@
|
||||
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');
|
||||
});
|
||||
Reference in New Issue
Block a user