import os
from google import genai

# Set up API key
os.environ['GEMINI_API_KEY'] = 'AIzaSyC2dTGU667C7l8mzaCjI5S18txWtf0aHPc'

client = genai.Client()

print("Available models:")
models = client.models.list()
for model in models:
    print(f"- {model.name}")
    print(f"  Supported methods: {getattr(model, 'supported_generation_methods', 'Unknown')}")
    print(f"  Input token limit: {getattr(model, 'input_token_limit', 'Unknown')}")
    print()