这个部分主要介绍 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"]
gpt-3.5-turbo
就是 ChatGPT;这里对 message 增加说明。它是一个 List 类型,每个 List 元素对应一个 Dict。一个Dict里面需要指定两个字段:
我们看一个具体的例子: