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');
});