import requests
import json
# 配置 API 参数
API_KEY = "your_api_key_here"
BASE_URL = "https://api.jeniya.top/v1"
# 支持的模型列表
MODELS = [
"gemini-2.5-pro",
"gemini-2.5-pro-preview-03-25",
"gemini-2.5-pro-exp-03-25"
]
def call_gemini_api(model, prompt):
url = f"{BASE_URL}/models/{model}:generateContent"
headers = {
"Authorization": f"Bearer {API_KEY}",
"Content-Type": "application/json"
}
data = {
"contents": [
{
"parts": [{"text": prompt}]
}
],
"generationConfig": {
"temperature": 0.7,
"maxOutputTokens": 2048
}
}
response = requests.post(url, headers=headers, json=data)
if response.status_code == 200:
return response.json()
else:
return {"error": f"API调用失败: {response.status_code}"}
# 使用示例
if __name__ == "__main__":
# 测试不同模型
prompt = "请介绍一下 Gemini 2.5 Pro 的主要特点"
for model in MODELS:
print(f"\n=== 使用模型: {model} ===")
result = call_gemini_api(model, prompt)
print(json.dumps(result, indent=2, ensure_ascii=False))