70 lines
2.3 KiB
TypeScript
70 lines
2.3 KiB
TypeScript
/// <reference types="jest-fetch-mock" />
|
|
import 'jest-fetch-mock';
|
|
import { FetchMock } from 'jest-fetch-mock';
|
|
import { LLMService } from '../src/services/LLMService';
|
|
require('dotenv').config();
|
|
|
|
const fetchMock = fetch as FetchMock;
|
|
|
|
describe('LLMService Refactor', () => {
|
|
let llmService: LLMService;
|
|
|
|
beforeAll(() => {
|
|
fetchMock.enableMocks(); // Enable mocks
|
|
const apiKey = process.env.GEMINI_API_KEY;
|
|
if (!apiKey) {
|
|
throw new Error('GEMINI_API_KEY is not defined in .env file');
|
|
}
|
|
llmService = new LLMService(apiKey);
|
|
});
|
|
|
|
afterEach(() => {
|
|
fetchMock.resetMocks(); // Reset mocks after each test
|
|
});
|
|
|
|
it('should correctly categorize desires based on the rules', async () => {
|
|
fetchMock.mockResponseOnce(JSON.stringify({
|
|
candidates: [{
|
|
content: {
|
|
parts: [{
|
|
text: JSON.stringify({
|
|
goTo: ['Pizza'],
|
|
alsoGood: ['Pasta'],
|
|
noGoes: ['Salad', 'Tacos'],
|
|
needsDiscussion: [],
|
|
considerable: []
|
|
})
|
|
}]
|
|
}
|
|
}]
|
|
}));
|
|
|
|
const desireSets = [
|
|
{ participantId: '1', wants: ['Pizza'], accepts: ['Pasta'], noGoes: ['Salad'] },
|
|
{ participantId: '2', wants: ['Pizza'], accepts: ['Pasta'], noGoes: ['Tacos'] },
|
|
];
|
|
|
|
const result = await llmService.analyzeDesires(desireSets as any);
|
|
|
|
expect(result.goTo).toContain('Pizza');
|
|
expect(result.alsoGood).toContain('Pasta');
|
|
expect(result.noGoes).toEqual(expect.arrayContaining(['Salad', 'Tacos']));
|
|
});
|
|
|
|
it('should detect inner contradictions in a desire set', async () => {
|
|
fetchMock.mockResponseOnce(JSON.stringify({
|
|
candidates: [{
|
|
content: {
|
|
parts: [{
|
|
text: 'true'
|
|
}]
|
|
}
|
|
}]
|
|
}));
|
|
|
|
const desireSet = { wants: ['Ice Cream', 'No desserts'], accepts: [], noGoes: [] };
|
|
const hasContradictions = await llmService.checkForInnerContradictions(desireSet as any);
|
|
expect(hasContradictions).toBe(true);
|
|
});
|
|
});
|