55 lines
1.8 KiB
TypeScript
55 lines
1.8 KiB
TypeScript
import { GoogleGenerativeAI } from '@google/generative-ai';
|
|
import { LLMService } from '../src/services/LLMService';
|
|
|
|
// Mock the GoogleGenerativeAI class and its methods
|
|
jest.mock('@google/generative-ai', () => ({
|
|
GoogleGenerativeAI: jest.fn().mockImplementation(() => ({
|
|
getGenerativeModel: jest.fn().mockReturnValue({
|
|
generateContent: jest.fn().mockResolvedValue({
|
|
response: {
|
|
text: jest.fn().mockReturnValue(
|
|
JSON.stringify({
|
|
"item1": "Concept A",
|
|
"item2": "Concept A",
|
|
"item3": "Concept B"
|
|
})
|
|
),
|
|
},
|
|
}),
|
|
}),
|
|
})),
|
|
}));
|
|
|
|
describe('LLMService', () => {
|
|
let llmService: LLMService;
|
|
const mockApiKey = 'test-api-key';
|
|
|
|
beforeEach(() => {
|
|
llmService = new LLMService(mockApiKey);
|
|
});
|
|
|
|
it('should call the Gemini API with the correct prompt and return parsed content', async () => {
|
|
const desires = [
|
|
{ wants: ['item1'], accepts: [], noGoes: [] },
|
|
{ wants: ['item2'], accepts: [], noGoes: [] },
|
|
{ wants: [], accepts: ['item3'], noGoes: [] },
|
|
];
|
|
|
|
const result = await llmService.analyzeDesires(desires);
|
|
|
|
expect(GoogleGenerativeAI).toHaveBeenCalledWith(mockApiKey);
|
|
expect(llmService['model'].generateContent).toHaveBeenCalled();
|
|
expect(result).toEqual({
|
|
"item1": "Concept A",
|
|
"item2": "Concept A",
|
|
"item3": "Concept B"
|
|
});
|
|
});
|
|
|
|
it('should handle API errors gracefully', async () => {
|
|
llmService['model'].generateContent.mockRejectedValueOnce(new Error('API Error'));
|
|
|
|
await expect(llmService.analyzeDesires([])).rejects.toThrow('API Error');
|
|
});
|
|
});
|