Files
dsProject/dsLightRag/Test/TestGoogleGemini.py
2025-08-15 15:59:09 +08:00

40 lines
1.1 KiB
Python

import requests
import json
# API配置
API_URL = "https://goapi.gptnb.ai/v1/chat/completions"
API_KEY = "sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662"
# 请求头
headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {API_KEY}"
}
# 请求体
payload = {
"model": "gemini-2.5-pro",
"messages": [{"role": "user", "content": "你是谁?"}],
"temperature": 0.7
}
try:
# 发送POST请求
response = requests.post(API_URL, headers=headers, data=json.dumps(payload))
response.raise_for_status() # 检查请求是否成功
# 解析响应
result = response.json()
# 提取并打印返回的文本内容
try:
# 从响应中获取文本内容
text_content = result['choices'][0]['message']['content']
print("\n提取的文本内容:")
print(text_content)
except (KeyError, IndexError) as e:
print(f"\n提取文本内容时出错: {e}")
print("响应结构可能与预期不符。")
except requests.exceptions.RequestException as e:
print(f"请求发生错误: {e}")