Python实战：使用requests库高效调用ChatGPT API的完整指南

2次阅读

共计 1986 个字符，预计需要花费 5 分钟才能阅读完成。

ChatGPT API 是 OpenAI 提供的人工智能接口服务，允许开发者将强大的自然语言处理能力集成到自己的应用中。无论是构建智能客服、内容生成工具，还是开发编程助手，ChatGPT API 都能提供灵活且高质量的文本交互能力。通过 Python 的 requests 库调用该 API，是最直接、高效的方式之一。

在开始之前，你需要完成以下准备工作：

获取 API 密钥 ：前往 OpenAI 官网注册账号，并在 API 密钥管理页面创建新的密钥。
安装 Python 环境 ：确保你的 Python 版本在 3.6 以上，并安装 requests 库。

pip install requests

ChatGPT API 使用 Bearer Token 进行认证，需要在请求头中添加 Authorization 字段。

import requests

API_KEY = "your-api-key"
headers = {"Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json"
}

API 的主要参数包括 model（指定使用的模型，如 gpt-3.5-turbo）和 messages（对话内容）。

payload = {
    "model": "gpt-3.5-turbo",
    "messages": [{"role": "user", "content": "Hello, ChatGPT!"}
    ]
}

API 返回的响应是 JSON 格式，包含生成的回复内容。

response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
data = response.json()
print(data["choices"][0]["message"]["content"])

以下是一个包含错误处理和重试机制的完整示例：

import requests
import time

API_KEY = "your-api-key"
MAX_RETRIES = 3
RETRY_DELAY = 2

def call_chatgpt(prompt):
    headers = {"Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    payload = {
        "model": "gpt-3.5-turbo",
        "messages": [{"role": "user", "content": prompt}]
    }

    for attempt in range(MAX_RETRIES):
        try:
            response = requests.post(
                "https://api.openai.com/v1/chat/completions",
                headers=headers,
                json=payload,
                timeout=10
            )
            response.raise_for_status()
            return response.json()["choices"][0]["message"]["content"]
        except requests.exceptions.RequestException as e:
            print(f"Attempt {attempt + 1} failed: {e}")
            if attempt < MAX_RETRIES - 1:
                time.sleep(RETRY_DELAY)
            else:
                raise

print(call_chatgpt("Hello, ChatGPT!"))