Gemini Deno

GOOGLE_AI_API_KEY=xxxxxxxxxxxx

import { GoogleGenerativeAI } from "npm:@google/generative-ai";
 
export const genAI = new GoogleGenerativeAI(Deno.env.get("GOOGLE_AI_API_KEY") || "");
 
export async function aiPredictUtil(request: Request): Promise<string> {
 
    try{
        if (request.method !== "POST") {
            throw new Error("Method not allow")
        }
 
        const body = await request.text();
        const { prompt } = JSON.parse(body);
 
        if(!prompt){
            throw new Error("Invalid input params")
        }
 
        const generationConfig = {
            maxOutputTokens: 1024,
            temperature: 0.9,
        };
 
        const model = genAI.getGenerativeModel({ 
            model: "gemini-pro",
            generationConfig
        });
    
        const result = await model.generateContent(prompt);
        const response = await result.response;
        const text = response.text();
 
        return new Response(JSON.stringify({ status: true, data: text }), {
            status: 200,
            headers: {
                "content-type": "application/json",
            },
        })
    }
    catch(error: any){
        return new Response(JSON.stringify({ status: false, msg: error.message }), {
            status: 500,
            headers: {
                "content-type": "application/json",
            },
        })
    }
 
}
 
Deno.serve((request) => aiPredictUtil(request));