在 Python 中调用 GPT-3 API

〇、效果展示

本文讲解如何调用 openAI 模型 GPT-3 官方 API，各位可以先熟悉熟悉，等 ChatGPT API 出来以后无非就是换一个模型名吧。
而且ChatGPT 是从 GPT-3.5 系列中的模型进行微调来的，而 GPT-3.5 是基于 GPT-3 训练得到的。使用 GPT-3 中的模型能体验初代版的 ChatGPT。

一、官方代码展示

官方给出了很多样例：Examples

以第一个“Q&A”为例（语言：Python；IDE：PyCharm）：

import os
import openai

openai.api_key = os.getenv("OPENAI_API_KEY")

response = openai.Completion.create(
  model="text-davinci-003",
  prompt="...",
  temperature=0,
  max_tokens=100,
  top_p=1,
  frequency_penalty=0.0,
  presence_penalty=0.0,
  stop=["\n"]
)

二、代码详解和调整

# 0、在Terminal中输入：pip install openai
import os
import openai

# 1、输入你的API KEY
openai.api_key = os.getenv("OPENAI_API_KEY") # 方式一：设置环境变量OPENAI_API_KEY
# openai.api_key = "将API KEY复制进来" # 方式二：直接赋值

# 2 定义相关参数
response = openai.Completion.create(
  model="text-davinci-003", # 模型的选择
  prompt="...", # 输入。显然输入来自于键盘
  temperature=0, # 模型将承担风险的高低
  max_tokens=100, # 输出生成的最大token数
  top_p=1, # 默认值就是1，可以省去
  frequency_penalty=0.0,
  presence_penalty=0.0,
  stop=["\n"] # API将停止生成进一步的token。返回的文本将不包含停止序列
)

模型的选择：GPT-3 模型特点
参数的详解：【OpenAI API】Request body中参数含义

相关调整后：

import os
import openai

# openai.api_key = os.getenv("OPENAI_API_KEY")
openai.api_key = "sk-..."

Q = "\n\nQ: "
A = "A: "

while True:
    prompt = input(Q)
    if prompt == 'exit':
        break;
    else:
        try:
            response = openai.Completion.create(
                model="text-davinci-003",
                prompt=prompt,
                temperature=1,
                max_tokens=500,
                frequency_penalty=0.0,
                presence_penalty=0.0,
            )
            print(A, response["choices"][0]["text"].strip())
        except Exception as exc:
            print(exc)