chatbot.py

import openai

# add your API key here
openai.api_key = ''

messages = []
system_msg = input("你想要创建什么样的聊天机器人？")
messages.append({"role": "system", "content": system_msg})
print("和你的聊天机器人问好！输入quit来退出。")

while True: 
    message = input('> ')
    if message.strip().lower() == 'quit':
        break
    messages.append({"role": "user", "content": message})
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo-0301",
        messages=messages)
    reply = response["choices"][0]["message"]["content"]
    messages.append({"role": "assistant", "content": reply})
    print("\nChatGPT: " + reply + "\n")