MerchFlow-AI / check_gemini_clean.py
Gaurav vashistha
Final Release: Updated README and Sync
842ca78
raw
history blame contribute delete
638 Bytes
import os
import google.generativeai as genai
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY")
genai.configure(api_key=api_key)
candidates = [
"gemini-2.0-flash",
"gemini-2.0-flash-exp",
"models/gemini-2.0-flash"
]
for model_name in candidates:
print(f"\nTesting model: {model_name}")
try:
model = genai.GenerativeModel(model_name)
response = model.generate_content("Hello")
print(f"βœ… Success with {model_name}: {response.text}")
break
except Exception as e:
print(f"❌ Failed with {model_name}: {e}")