seraph-agent/dist/llm/gemini.js

An extremely lightweight, SRE autonomous AI agent for seamless integration with common observability tasks.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.GeminiProvider = void 0;
const generative_ai_1 = require("@google/generative-ai");
class GeminiProvider {
    genAI;
    model;
    constructor(config) {
        if (!config.apiKey) {
            throw new Error('Gemini API key not found in config.');
        }
        this.genAI = new generative_ai_1.GoogleGenerativeAI(config.apiKey);
        this.model = config.llm?.model || 'gemini-1.5-pro-latest';
    }
    async generate(prompt) {
        const model = this.genAI.getGenerativeModel({ model: this.model });
        const result = await model.generateContent(prompt);
        const response = await result.response;
        return response.text();
    }
}
exports.GeminiProvider = GeminiProvider;