Spaces:
Paused
Paused
File size: 778 Bytes
4156c57 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import os
import google.generativeai as genai
api_key = os.getenv("GOOGLE_API_KEY")
if not api_key:
print("Error: GOOGLE_API_KEY not set")
exit(1)
print(f"Checking models for key ending in ...{api_key[-4:]}")
genai.configure(api_key=api_key)
try:
print("Listing available models...")
for m in genai.list_models():
if 'generateContent' in m.supported_generation_methods:
print(f"- {m.name}")
except Exception as e:
print(f"Error listing models: {e}")
try:
print("\nAttempting generation with 'gemini-1.5-flash'...")
model = genai.GenerativeModel('gemini-1.5-flash')
response = model.generate_content("Hello")
print(f"Success! Response: {response.text}")
except Exception as e:
print(f"Test generation failed: {e}")
|