TRIEM_AI / debug_flash_models.py
rzvn1's picture
Initial commit of TRIEM AI
3017127
Raw
History Blame Contribute Delete
293 Bytes
import os
from dotenv import load_dotenv
from google import genai
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY")
client = genai.Client(api_key=api_key)
print("Listing FLASH models...")
for model in client.models.list():
if 'flash' in model.name:
print(f" - {model.name}")