初识Python -- 使用Python完成chatgpt聊天功能

小菜鸟正洋洋

已于 2023-05-24 16:25:54 修改

阅读量638

点赞数

文章标签： python openai

于 2023-05-24 16:22:34 首次发布

本文链接：https://blog.csdn.net/qq_32080545/article/details/130849784

版权

运行前需设置环境变量：

export OPENAI_API_ORIGAN=org-你的组织id
export OPENAI_API_KEY=sk-IlhlSea你的apikey

更多参数说明：https://platform.openai.com/docs/api-reference/chat

运行效果：

基于stream流式返回，逐字打印回答内容
在这里插入图片描述

实现代码：

# chatgpt聊天
import openai
import sys

def doChat(quest):
    completion = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "user", "content": quest}
        ],
        stream=True
    )

    for i,response in enumerate(completion):
        objcontent = response.choices[0].delta
        if hasattr(objcontent, 'content'):
            msg = objcontent.content
            sys.stdout.write(msg)
            sys.stdout.flush()

        else:
            continue

while True:
    print('（请输入你的问题：）')
    quest = input()
    doChat(quest)