python接入ChatGPT API连续对话脚本演示

使用python接入chatgpt api的小demo
本demo 使用的是3.5版本的
要求：需要在自己电脑安装openai。

pip install openai

没有逐字输出功能，需要的自己去开发。如果回复内容较多较慢，gpt会自动分开部分回复出来。
以下是源码。

import openai

openai.api_key = "密钥"


def ask_gpt(message):
    completion = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=message
    )
    return completion.choices[0].message


if __name__ == '__main__':
    message = []
    gpt_msg = []
    try:
        while True:
            user_input = input("我: ").strip()
            if user_input == "exit" or user_input == "退出":
                exit()
            content = {"role": "user", "content": ""}
            content["content"] = user_input
            message.append(content)
            gpt_res = ask_gpt(message)
            # print(gpt_res)
            gpt_msg = {"content": "", "role": "assistant"}
            gpt_msg["role"] = gpt_res["role"]
            gpt_msg["content"] = gpt_res["content"]
            print("" + gpt_res["content"])
            message.append(gpt_msg)
    except Exception as e:
        print(e)