MerchFlow-AI / check_gemini.py
Gaurav vashistha
Final Release: Updated README and Sync
842ca78
raw
history blame contribute delete
713 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)
print("Listing available Gemini models...")
try:
for m in genai.list_models():
if 'generateContent' in m.supported_generation_methods:
print(m.name)
except Exception as e:
print(f"List models failed: {e}")
model_name = "gemini-1.5-flash"
print(f"\nTesting model: {model_name}")
try:
model = genai.GenerativeModel(model_name)
response = model.generate_content("Hello, can you see this?")
print("Response:", response.text)
except Exception as e:
print(f"Test failed: {e}")