Python调用ChatGPT API 程序示例

代码

代码如下：

# coding=utf-8
import requests
import json
import os

def send_prompt(prompt=''):
    API_KEY = '密钥'

    if not prompt:
        print("Prompt is empty")
        return

    headers = {
        'Content-Type': 'application/json',
        'Authorization': 'Bearer ' + API_KEY
    }

    data = {
        "model": "gpt-3.5-turbo",  # 修改为合适的模型名称
        "messages": [{"role": "system", "content": "You are a helpful assistant."},  # 系统消息，定义聊天模型的角色
                     {"role": "user", "content": prompt}]  # 用户消息
    }

    try:
        print("Sending data:", json.dumps(data))  # 打印发送的数据
        response = requests.post('https://api.openai.com/v1/chat/completions', headers=headers, data=json.dumps(data))
        response.raise_for_status()  # 检查响应状态
        response_data = response.json()
        print("Received data:", response_data)  # 打印接收的数据
        if response_data.get('choices'):
            print(response_data['choices'][0].get('message', {}).get('content', ''))
    except requests.exceptions.RequestException as e:
        print("请求出错:", e)

prompt = '如何学习Python?'
send_prompt(prompt)

更多技术服务

下方卡片添加咨询