使用 JupyterLab 在 Colab 上进行OpenAI API简单开发

603 阅读2分钟

图片

如果你不是程序员或不想在本地搭建开发环境,你可以选择使用Google提供的在线Python Notebook环境,Colab(colab.research.google.com/)。 即使你已经有了本地开发环境,我也建议你注册一个Colab账号,因为Colab还提供了以下几个好处:

  • 集成了Python Notebook 的环境

  • 免费的GPU资源,当你需要使用GPU来训练一些深度学习模型时,Colab可以直接为你提供这个资源。

  • Colab方便将你自己编写的Python代码在网络上分享给其他人。

图片

Colab 已经是一个 Python Notebook 的环境了。所以我们不需要再去安装 Python 和 JupyterLab了。不过我们还是需要安装 OpenAI 的库,以及设定我们的 API Key。你只需要在 Notebook 开始的时候,执行下面这样一小段代码就可以配置好必要环境了。

pip install openai
%env OPENAI_API_KEY=你的OPENAI ApiKey

体验 OpenAI 的 API

现在,你已经成功地搭建好了开发环境。不管你是使用本地的JupyterLab环境,还是Google免费提供的 Colab 环境,让我们一起体验OpenAI的魅力吧。在这里,我提供了一段简单代码,你只需将其复制粘贴到你的Notebook中,就可以直接运行了。

import openai

import os

openai.api_key = os.environ.get("OPENAI_API_KEY")

COMPLETION_MODEL = "gpt-3.5-turbo"

prompt = """

Consideration product : 2023夏装新款短袖女白色t恤韩版纯棉印花字母修身大码宽松圆领潮

1. Compose human readable product title used on Amazon in english within 20 words.

2. Write 5 selling points for the products in Amazon.

3. Evaluate a price range for this product in U.S.

Output the result in json format with three properties called title, selling_points and price_range

"""

def get_response(prompt):

    completions = openai.ChatCompletion.create(

    model=COMPLETION_MODEL,

    temperature=0.8,

    max_tokens=1000,

    top_p=1.0,

    frequency_penalty=0.5,

    presence_penalty=0.0,

    messages=[

        {"role": "user", "content": prompt}

        ]

    )

    message = completions["choices"][0]["message"]["content"]

    return message
    
print(get_response(prompt))

来看看返回结果。

{

    "title": "2023 Summer New Short Sleeve Women's White T-Shirt Korean Version of Pure Cotton Printing Letters Slim Plus Size Loose Round Neck Trendy",

    "selling_points": [

        "Made of pure cotton material for comfortable wear",

        "Stylish print design with letters",

        "Slim fit design for a flattering look",

        "Available in plus sizes for a variety of body types",

        "Perfect for summer wear and casual occasions"

    ],

    "price_range": "$15.99 - $25.99"

}

图片

这段代码里面,我们调用了 OpenAI 的 gpt-3.5-turbo Completion 接口,之后我向它提出了一个要求,即针对在淘宝上找到的中文商品名称做三件事情:为该商品编写一个适用于亚马逊的英文标题,撰写五个卖点,最后估算一下该商品在美国的最佳售价。

搞定,这样你的第一个简单调用OPENAI API代码实例就做出来了,是不是很简单!!有兴趣的小伙伴也试试吧! 

接下来的文章,我们也会探讨更多OPENAI API其他接口的用法。

原文关注: 云原生SRE