test_model.py

"""
Simple Model Test
Test if the model works with a basic request
"""

import google.generativeai as genai
import os
from dotenv import load_dotenv

load_dotenv()

# Configure API
api_key = os.getenv("GEMINI_API_KEY")
print(f"API Key loaded: {api_key[:10]}...")

genai.configure(api_key=api_key)

print("\n" + "=" * 70)
print("Testing models/gemini-1.5-flash")
print("=" * 70)

try:
    model = genai.GenerativeModel("models/gemini-1.5-flash")
    resp = model.generate_content("Say hello!")
    print(f"✅ Success! Response: {resp.text}")
except Exception as e:
    print(f"❌ Error with models/gemini-1.5-flash: {e}")
    print("\nTrying gemini-1.5-flash without 'models/' prefix...")
    try:
        model = genai.GenerativeModel("gemini-1.5-flash")
        resp = model.generate_content("Say hello!")
        print(f"✅ Success! Response: {resp.text}")
        print("\n⚠️ Note: Use 'gemini-1.5-flash' (without 'models/' prefix)")
    except Exception as e2:
        print(f"❌ Also failed: {e2}")