import os
from dotenv import load_dotenv
from langchain_google_genai import ChatGoogleGenerativeAI

GENAI_MODEL = "gemini-2.0-flash"

def get_llm():
    # LangChain 的 Google Generative API 封裝
    llm = ChatGoogleGenerativeAI(model=GENAI_MODEL)
    return llm