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}")