使用node和js对接chatgpt,支持流式传输

node代码


	
const express = require('express');
const request = require('request');
const bodyParser = require('body-parser');

const app = express();

app.use(express.json());
app.use(bodyParser.urlencoded({ extended: false }));
app.use(bodyParser.json());
// 前端项目的部署位置
app.use(express.static('front'));


app.post('/chat', (req, res) => {
	const messages = req.body.messages;
  const options = {
    url: 'https://api.openai.com/v1/chat/completions',
    method: 'POST',
    headers: {
			"Authorization": "Bearer sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
			"Content-Type": "application/json",
    },
    json: {
			"model": "gpt-3.5-turbo",
			"stream": true,
			"messages": messages
    }
  };
	const proxyReq = request(options);
  proxyReq.on('response', function(response) {
    response.pipe(res);
  });
});

app.listen(3003, function() {
    console.log('victor server is running on port 3003');
  });
  


js接收代码

  const postStreamList = async (callback) => {
    const requestOptions = {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({
        "model": "gpt-3.5-turbo",
        "messages": [{ "role": "assistant", "content": "how are you" }]
      })
    };
    let count = 0;
    const streamResponse = await fetch('/chat', requestOptions);
    const reader = streamResponse.body.getReader();
    let errText = "";
    const read = () => {
      return reader.read().then(({ done, value }) => {
        count++;
        if (done) {
          console.log("victor react reviced: end");
          callback(ENDTEXT);
          return;
        }

        const textDecoder = new TextDecoder();
        // console.log("返回的数据:", textDecoder.decode(value));
        let text = "";
        const strArr = (errText + textDecoder.decode(value)).split("data: ");
        console.log("解析字符", textDecoder.decode(value))
        if (strArr) {
          for (let i = 0; i < strArr.length; i++) {
            let json = {};
            if (strArr[i] && strArr[i] !== "[DONE]") {
              try {
                json = JSON.parse(strArr[i]);
                if (json.choices.length && json.choices[0].delta.content) {
                  text = text + json.choices[0].delta.content;
                }
                errText = "";
              } catch (e) {
                console.log("出错", strArr[i])
                errText = strArr[i];
              }

            }
          }
          callback(text);
        }
        return read();
      });
    }

    read();
}
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值