fakeshield-api / debug_gemini.py
Akash4911's picture
Initial Deploy: FakeShield Backend v2.0 (Sovereign Vanguard)
89e8242
import google.generativeai as genai
import os
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY")
print(f"API Key found: {api_key[:10]}...")
if api_key:
genai.configure(api_key=api_key)
# Try multiple common model names for robustness
models = ["gemini-2.0-flash-exp", "gemini-1.5-flash", "gemini-pro"]
for model_name in models:
try:
print(f"Testing model: {model_name}")
model = genai.GenerativeModel(model_name)
response = model.generate_content("hello", generation_config={"max_output_tokens": 5})
print(f"Success with {model_name}: {response.text}")
break
except Exception as e:
print(f"Failed with {model_name}: {e}")
else:
print("No API Key found in .env")