LLM_helper.py

from llama_index.llms.gemini import Gemini
from dotenv import load_dotenv
import os

load_dotenv()
def llm_builder():
    llm = Gemini(
        api_key=os.environ.get('GOOGLE_API_KEY'),
        model_name="models/gemini-pro"
    )
    return llm