python实现ChatGPT连续多轮对话

一位安分的码农

已于 2023-03-26 15:13:34 修改

阅读量4.9k

点赞数 3

分类专栏：大语言模型文章标签： chatgpt

于 2023-03-26 14:48:36 首次发布

本文链接：https://blog.csdn.net/weixin_44612221/article/details/129779356

版权

大语言模型专栏收录该内容

14 篇文章

订阅专栏

这段代码展示了如何在Python中设置OpenAI的API密钥并使用gpt-3.5-turbo模型与ChatGPT进行对话。用户可以输入文本，程序会通过API获取ChatGPT的响应，并在遇到错误时处理异常。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码如下。注意，api_key需要改成自己的密钥才能成功运行。

import openai

api_key = "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"

openai.api_key = api_key

def askChatGPT(messages):
    MODEL = "gpt-3.5-turbo"
    response = openai.ChatCompletion.create(
        model=MODEL,
        messages = messages,
        temperature=1)
    return response['choices'][0]['message']['content']

def main():
    messages = [{"role": "user","content":""}]
    while 1:
        try:
            text = input('你：')
            if text == 'quit':
                break

            d = {"role":"user","content":text}
            messages.append(d)

            text = askChatGPT(messages)
            d = {"role":"assistant","content":text}
            print('ChatGPT：'+text+'\n')
            messages.append(d)
        except:
            messages.pop()
            print('ChatGPT：error\n')
main()