data_science_agent / utils /google_genai_llm.py
Master-warrier's picture
Add requirements gathering flow
685adc8
raw
history blame
321 Bytes
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)