Infrastructure: Implement master/dev/vX branching, save git path, and fix username casing
This commit is contained in:
20
backend/check_models.py
Normal file
20
backend/check_models.py
Normal file
@@ -0,0 +1,20 @@
|
||||
import os
|
||||
import google.generativeai as genai
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
API_KEY = os.environ.get("GEMINI_API_KEY")
|
||||
|
||||
if not API_KEY:
|
||||
print("Error: GEMINI_API_KEY not found in .env")
|
||||
exit(1)
|
||||
|
||||
genai.configure(api_key=API_KEY)
|
||||
|
||||
print(f"Checking available models for your API key...")
|
||||
try:
|
||||
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}")
|
||||
Reference in New Issue
Block a user