用Python调用OpenAI API

安装openai库

由于openai限制国内注册，所以需要自行解决获取 API KEY问题。

第一步安装openai库，Win+R，输入cmd或者powershell，打开窗口，输入下列命令，安装openai库

pip install openai

生成文本

OpenAI的接口有多种不同的类型，这里介绍一种常见的文本生成类型的接口。直接上代码：

import os
import openai

openai.api_key = "OPENAI_API_KEY"

response = openai.Completion.create(
engine="text-davinci-003",
prompt="二月二，龙抬头，结合学校的美景，创作一首诗",
temperature=0.7,
max_tokens=800,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)

print(response.choices[0].text)

关于参数说明，API Reference - OpenAI API

运行上面的代码，获得一个 API 响应，其中包含 AI 模型根据您的提示自动生成的响应。

{
  "choices": [
    {
      "finish_reason": "stop",
      "index": 0,
      "logprobs": null,
      "text": "\n\n\u5c71\u666f\u6e05\u5e7d\u6625\u7a7a\u6ee1\uff0c\n\u6e56\u5149\u63a9\u62a4\u8349\u9752\u7eff\uff0c\n\u4f46\u89c1\u9f99\u5934\u9ad8\u6602\u8d77\uff0c\n\u7a7a\u4e2d\u70df\u96fe\u6735\u6735\u7f2d\u3002\n\u7fa4\u5cf0\u79c0\u51fa\u79cb\u8272\u6df1\uff0c\n\u6821\u56ed\u6d53\u90c1\u82b3\u534e\u591a\uff0c\n\u98ce\u96e8\u4ea4\u52a0\u666f\u81f4\u53d8\uff0c\n\u6668\u66e6\u5f69\u971e\u6620\u84dd\u5929\u3002"
    }
  ],
  "created": 1676951216,
  "id": "cmpl-6mDuCLS7Nn1mvH37Lz0CFuUTw3TZU",
  "model": "text-davinci-003",
  "object": "text_completion",
  "usage": {
    "completion_tokens": 167,
    "prompt_tokens": 47,
    "total_tokens": 214
  }
}

其中的text内容如下：

山景清幽春空满，
湖光掩护草青绿，
但见龙头高昂起，
空中烟雾朵朵缭。
群峰秀出秋色深，
校园浓郁芳华多，
风雨交加景致变，
晨曦彩霞映蓝天。