【记录】OpenAI｜Python调用GPT API的开发环境及代码（2024/06/21实测）

import os
import openai
 
# 设置OpenAI API的密钥
openai.api_key = os.getenv("OPENAI_API_KEY")
 
# 调用GPT-3 API的示例代码
def get_completion(prompt, model="text-davinci-002", temperature=0.7):
    response = openai.Completion.create(
        engine=model,
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=temperature
    )
    return response.data["choices"][0]["text"]
 
# 使用示例
prompt = "写一个Python脚本，生成一个包含10个随机数的列表"
completion = get_completion(prompt)
print(completion)

这段代码首先导入了必要的模块，并从环境变量中加载了OpenAI API的密钥。然后定义了一个函数get_completion，它接受一个提示和一些参数，并使用OpenAI的Completion.create方法来生成文本。最后，我们提供了一个使用示例，展示了如何使用这个函数来生成一个简单的Python脚本。