在使用ChatGPT时,模型的回复内容是一个字一个字蹦出来的,而不是整段话直接出现,因为模型需要不断预测接下来要回复什么内容,如果等整段回复生成之后再输出到网页,用户体验就会很差,一直以为这种流式输出效果是用WebSocket实现的,后来接入open ai接口,发现接口是http协议,才了解到SSE技术。
Server-Sent Events (SSE) 是一种基于 HTTP 协议的服务器推送技术,它允许服务器向客户端发送数据和信息。与 WebSocket 不同,SSE 是一种单向通信方式,只有服务器可以向客户端推送消息。SSE 是 HTML5 规范的一部分,使用非常简单,主要由服务端与浏览器端的通讯协议(HTTP协议)和 EventSource 接口来处理 Server-sent events 组成,服务器端的响应的内容类型是“text/event-stream”。
使用postman给open ai接口发个请求,验证一下。
可以看到响应头中的Content-Type为text/event-stream:
再看响应体,可以看到有很多个以data: 开头的片段,每个片段包含一个choices数组,数组元素的delta.content属性的值是一个单词,每个片段中的单词连起来就是ChatGPT回答的内容,最后一个片段内容是[done],表示输出结束。
由此可验证open ai的接口是通过SSE技术来实现的。
接下来我们写一个简单的demo来模拟一下ChatGPT的输出效果,更直观的学习和体验SSE技术。
先看一下效果:
新建springboot项目,包含一个原生的index.html页面和一个http接口,页面请求接口,接口读取一个txt文件并将文件内容流式输出到页面。点击查看github源码
页面代码:
<body>
<button type="button" onclick="output()">输出文章</button>
<div id="message"></div>
<script>function output() {let source = new EventSource('http://localhost:8080/article');let innerHTML = '';source.onmessage = function (e) {if (e.data == '[done]') {source.close();} else {innerHTML += e.data;document.getElementById("message").innerHTML = innerHTML;}};}
</script>
</body>
服务端代码:
@Controller
public class SseController {@Autowiredprivate ResourceLoader resourceLoader;
@RequestMapping("/article")public void sendArticle(HttpServletResponse res) throws Exception {res.setContentType("text/event-stream;charset=UTF-8");Resource resource = resourceLoader.getResource("classpath:1.txt");InputStream inputStream = resource.getInputStream();BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));String line;while ((line = reader.readLine()) != null) {// 这里用空格来将一行内容分为多个单词输出String delimiter = " ";String[] words = line.split(delimiter);// 为了支持将中文句子分为单个汉子输出if (words.length == 1) {delimiter = "";words = line.split(delimiter);}for (int i = 0; i < words.length; i++) {// 每次只输出一个词,每个片段以data: 开头,以\n\n结尾res.getWriter().write("data: " + words[i] + delimiter + "\n\n");res.getWriter().flush();// 睡眠100ms,便于观察效果Thread.sleep(100);}// 由于每次读的是一行数据,因此输出一个换行res.getWriter().write("data: </br>\n\n");}// 也用[done]来标识结束res.getWriter().write("data: [done]\n\n");res.getWriter().flush();}
}