File size: 894 Bytes
984400f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import { GoogleGenerativeAI } from "@google/generative-ai";
import dotenv from "dotenv";

dotenv.config();

const GoogleAPIKey = process.env.GOOGLE_API_KEY;

if (!GoogleAPIKey) {
    throw new Error("Missing variables GOOGLE_API_KEY");
}

const genAI = new GoogleGenerativeAI(GoogleAPIKey);

/**
 * Generate a response using the Gemini AI model.
 * @param {string} prompt - The input string for the model.
 * @returns {Promise<string>} The generated response text.
 */
async function GeminiResponse(prompt) {
    try {
        const model = genAI.getGenerativeModel({
            model: "gemini-1.5-flash",
        });
        const result = await model.generateContent(prompt);
        console.log("Response", result)
        return result.response.text();
    } catch (e) {
        console.error(`Error in GeminiResponse: ${e.message}`);
        return "";
    }
}

export { GeminiResponse };