OpenAI API 调用

发表于 分类于

前言

本文将使用 OpenAI 提供的 API 进行简单调用，有个初步了解。在使用 API 前请确认网络状况，以及 OpenAI 账户余额及到期时间。
建议在 Colaboratory 上运行代码。本文不会讲述如何申请 OpenAI 账户，也不会提供 key。

比较

本文将使用 text-davinci-003 模型，那么他与我们常用的 GPT 3.5-turbo的区别如下：

  1. 性能：GPT-3.5-turbo是OpenAI推出的一种改进型语言模型，相对于”text-davinci-003”具有更高的性能。GPT-3.5-turbo在处理自然语言任务时更快且更经济高效，但在某些情况下，它可能在生成长文本时略有限制。
  2. API访问成本：使用GPT-3.5-turbo相对于”text-davinci-003”更经济。OpenAI为GPT-3.5-turbo提供了更低的API访问成本，使得更多的开发者能够使用这个模型。
  3. 上下文敏感性：相对于GPT-3.5-turbo，”text-davinci-003”在处理上下文敏感的任务时可能具有一些优势。”text-davinci-003”模型可以更好地记住之前的对话或文本上下文，并根据上下文进行生成。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import openai

openai.api_key = '你的Key'
prompt = '写一首关于春天，关于生命的诗，使用五言绝句'
COMPLETION_MODEL = "text-davinci-003"

def get_response(prompt, temperature = 1.0):
    completions = openai.Completion.create (
        engine=COMPLETION_MODEL,
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=temperature,
    )
    message = completions.choices[0].text
    return message
  
print(get_response(prompt))

temperature：在文本生成任务中，”temperature”是一个用于控制生成文本多样性的参数。它控制着模型在生成文本时对下一个字符或单词进行采样的随机性程度。取值范围 0-1，数值越小回答的内容越稳定越，反之回答内容越随机。

返回值

1
2
3
4
春风满面笑，花蔓簇蕊开；
人间已百花，生命更灿彩。
思古流年深，女娲创地脉；
飞絮洋洋去，芳意徘徊来。