BioScientist/agent_system/engines/test_gemini.py

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()