软件测试/人工智能|教你如何使用ChatGPT的API

简介

自从有了ChatGPT之后，我每天都想去调戏它一番，自从开放了ChatGPT的API，我就想着通过API来使用ChatGPT，这样的话，速度上的体验应该会更好，本文就来介绍一下如何使用ChatGPT的API。

环境准备

在调用API之前，我们首先需要准备好我们的环境，一是我们需要安装openAI第三方库，OpenAI 提供了一个封装的非常完备的 Python 库，可以直接使用 pip 安装，pip install openai。然后，我们需要创建好自己的Token。

注意：API 调用是收费的，但是 OpenAI 已经为我们免费提供了5美元的用量。如果超出限额，需要及时充值。

请求示例

在我们准备好环境之后，就可以使用脚本去调用ChatGPT的API了，下面是我们的一个请求示例：

import openai

# 配置请求openai的基础url地址，默认是直接访问openai官方的API接口
# openai.api_base = 'https://api.openai.com/v1'
# 配置请求openai的API Key，需要拥有openai的账号，从账号管理中生成自己的API Key
openai.api_key = 'sk-xxx'

# 发送请求
response = openai.ChatCompletion.create(
    # 指定请求模型
    model='gpt-3.5-turbo',
    # 请求消息体，列表中支持多条消息，可以将旧消息作为维持上下文信息的数据一并传入本次请求
    messages=[
        {"role": "user", "content": "我在百米飞人大战中超过了第二名，现在我是第几名？"},
    ]
)
answer = response['choices'][0]['message']['content']
print(answer)

响应如下：

{
"choices":[
{
"finish_reason":"stop",
"index":0,
"message":{
"content":"如果你超过了第二名，那么你目前是排在第二名之前，也就是第一名位置上。",
"role":"assistant"
}
}
],
"created":1693916326,
"id":"chatcmpl-7vPJ0oJjgJZFuMdQHD5un3HnvQ0Nv",
"model":"gpt-3.5-turbo-0613",
"object":"chat.completion",
"usage":{
"completion_tokens":7,
"prompt_tokens":25,
"total_tokens":32
}
}

总结

本文主要介绍了如何使用代码调用ChatGPT的API，我们需要注意的就是API调用并不免费，一旦超出额度，就需要我们支付相关费用，这对我们来说会比较麻烦。希望本文能够帮到大家。