File size: 615 Bytes
4416e3b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | 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}")
|