Spaces:
Sleeping
Sleeping
File size: 1,587 Bytes
884868c |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
#!/usr/bin/env python3
import os
from dotenv import load_dotenv
load_dotenv()
geminiApiKey = os.getenv("GEMINI_API_KEY")
if not geminiApiKey:
print("❌ GEMINI_API_KEY not found in .env file")
exit(1)
print("🔍 Checking available Gemini models...")
print("=" * 70)
try:
import google.generativeai as genai
genai.configure(api_key=geminiApiKey)
print("\n✓ SDK Version:", genai.__version__)
print("\n📋 Available Models:\n")
models = genai.list_models()
for model in models:
if 'generateContent' in model.supported_generation_methods:
print(f" • {model.name}")
print(f" Display Name: {model.display_name}")
print(f" Description: {model.description[:100]}...")
print()
print("=" * 70)
print("\n✅ Now testing with the first available model...")
test_model_name = None
for model in genai.list_models():
if 'generateContent' in model.supported_generation_methods:
test_model_name = model.name
break
if test_model_name:
print(f"\n🧪 Testing with: {test_model_name}")
test_model = genai.GenerativeModel(test_model_name)
response = test_model.generate_content("Say 'Hello, Legal RAG Backend!'")
print(f"✓ Response: {response.text}")
print(f"\n✅ Gemini API is working! Use model: {test_model_name}")
else:
print("❌ No suitable models found")
except Exception as e:
print(f"❌ Error: {e}")
import traceback
traceback.print_exc()
|