参考资料:

这个部分主要介绍 OpenAI API 的使用。API 的参数也会影响返回的效果。

OpenAI 提供了 Python API:

import openai

想要使用它,必须提供一个 Key 以获取权限:

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

之后就可以使用 openai 的 API 了,这里封装了一个调用函数:

def get_completion_from_messages(messages, model="gpt-3.5-turbo", temperature=0):
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        temperature=temperature, # this is the degree of randomness of the model's output
    )
    return response.choices[0].message["content"]

这里对 message 增加说明。它是一个 List 类型,每个 List 元素对应一个 Dict。一个Dict里面需要指定两个字段:

我们看一个具体的例子: