openai gpt-3.5-turbo模型流式传输 python聊天测试

蓝色心情QQ

已于 2023-09-13 15:52:04 修改

阅读量331

点赞数

文章标签： python

于 2023-09-13 15:36:39 首次发布

本文链接：https://blog.csdn.net/qq_30717687/article/details/132854289

版权

本文展示了如何利用Python进行OpenAI GPT-3.5-Turbo模型的流式传输聊天测试。首先，需要安装`sseclient`库，然后可以参考OpenAI的官方文档来调用API创建聊天交互。

摘要由CSDN通过智能技术生成

import datetime

import requests
import json
import sseclient

class OpenaiBase:
    api_url = 'https://api.openai.com/v1/completions'
    api_key = "you openai key"

    @staticmethod
    def get_headers():
        return {
            'Accept': 'text/event-stream',
            'Authorization': 'Bearer ' + OpenaiBaseapi_key
        }

    @staticmethod
    def get_body(content):
        return {
            "model": "gpt-3.5-turbo",
            "messages": [{"role": "user", "content": content}],
            "max_tokens": 100,
            "stream": True
        }

    @staticmethod
    def chat_completions(content):
        req_headers = OpenaiBase.get_headers()
        req_body = OpenaiBase.get_body(content)
        res = requests.post(OpenaiBase.api_url, stream=True, headers=req_headers, json=