[Python]：OPENAI API调用

原创已于 2023-07-01 16:52:20 修改 · 735 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python

于 2023-07-01 14:53:32 首次发布

[Python]：OPENAI API使用

安装

使用conda安装，python版本最好在3.9.x。

使用

Python API

基于Python体验一下OPENAI，代码很简单：

import openai
import json

openai.api_key = "替换为你的key"

if __name__ == '__main__':
    print("GPT 3.5 Example")

    while True:
        user_input = input("请输入内容（输入 'exit' 结束循环）：")

        if user_input == "exit":
            break

        #处理处理输入
        completion = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "user", "content": user_input}
            ]
        )

        message = completion.choices[0].message
        data = json.loads(str(message))

        content = data["content"]
        print(content)

简单测试一下，还是可以，提前是要使用魔法。

在这里插入图片描述

HTTP请求


import openai
import json
import requests


if __name__ == '__main__':
    # 设置 OpenAI API 认证信息
    headers = {
        'Authorization': 'Bearer 你的key',
        'Content-Type': 'application/json'
    }

    while True:
        user_input = input("请输入内容（输入 'exit' 结束循环）：")

        if user_input == "exit":
            break

        # 构建请求的数据
        data = {
            'model': 'gpt-3.5-turbo',
            'messages': [
                {'role': 'user', 'content': user_input}
            ]
        }

        # 将请求数据转换为 JSON 字符串
        data_json = json.dumps(data)

        # 发送 HTTP POST 请求
        response = requests.post('https://api.openai.com/v1/chat/completions', headers=headers, data=data_json)
        response_data = response.json()

        # 解析响应数据
        completion = response_data['choices'][0]['message']

        print(completion["content"])