File size: 321 Bytes
685adc8
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from google import genai
from dotenv import load_dotenv
import os
load_dotenv()

client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))

def get_response(prompt: str) -> str:
    response = client.models.generate_content(
        model="gemini-2.5-flash-preview-05-20", contents=prompt
    )
    return(response.text)