seraph-agent
Version:
An extremely lightweight, SRE autonomous AI agent for seamless integration with common observability tasks.
32 lines (31 loc) • 1.24 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
const gemini_1 = require("../gemini");
const generative_ai_1 = require("@google/generative-ai");
jest.mock('@google/generative-ai');
describe('GeminiProvider', () => {
const mockedGoogleGenerativeAI = generative_ai_1.GoogleGenerativeAI;
it('should handle errors from the Gemini API', async () => {
const generateContent = jest.fn().mockRejectedValue(new Error('Gemini API error'));
mockedGoogleGenerativeAI.prototype.getGenerativeModel = jest.fn().mockReturnValue({
generateContent,
});
const config = {
port: 8080,
workers: 4,
apiKey: 'test-key',
serverApiKey: null,
};
const provider = new gemini_1.GeminiProvider(config);
await expect(provider.generate('test prompt')).rejects.toThrow('Gemini API error');
});
it('should throw an error if no API key is provided', () => {
const config = {
port: 8080,
workers: 4,
apiKey: null,
serverApiKey: null,
};
expect(() => new gemini_1.GeminiProvider(config)).toThrow('Gemini API key not found in config.');
});
});