File size: 860 Bytes
a9e46a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | import google.generativeai as genai
import os
# 替换为你的 Gemini API Key
api_key = os.environ.get("GEMINI_API_KEY")
print("api_key:"+api_key)
os.environ["GEMINI_API_KEY"] = api_key
genai.configure(api_key=os.environ["GEMINI_API_KEY"])
def list_gemini_models():
print("Listing available Gemini models:")
print("-" * 50)
# 遍历所有模型
for model in genai.list_models():
# 过滤掉仅支持旧版或特定功能的模型（可选）
if 'generateContent' in model.supported_generation_methods:
print(f"Model Name: {model.name}")
print(f"Description: {model.description}")
print(f"Input Token Limit: {model.input_token_limit}")
print(f"Output Token Limit: {model.output_token_limit}")
print("-" * 50)
if __name__ == "__main__":
list_gemini_models() |