import os
from dotenv import load_dotenv
from langchain_google_genai import ChatGoogleGenerativeAI

load_dotenv()
api_key = os.getenv("GOOGLE_API_KEY")

models_to_try = ["gemini-1.5-flash", "gemini-1.5-flash-latest", "gemini-pro", "gemini-1.5-pro"]

for model_name in models_to_try:
    print(f"Testing model: {model_name}...")
    try:
        llm = ChatGoogleGenerativeAI(model=model_name, google_api_key=api_key)
        response = llm.invoke("Say hello")
        print(f"  Success with {model_name}!")
        break
    except Exception as e:
        print(f"  Failed with {model_name}: {e}")