文心一言的流式接口数据进行处理增加属性

原创

已于 2025-01-17 14:40:15 修改 · 601 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

于 2024-07-10 11:13:19 首次发布

需求：需要对文心一言的流式接口数据进行处理增加属性

return ResponseEntity.ok()
                .header("Access-Control-Allow-Origin", "*")
                .contentType(org.springframework.http.MediaType.TEXT_EVENT_STREAM)
                .cacheControl(org.springframework.http.CacheControl.noCache())
                .body(outputStream -> {
                        try (Response response = client.newCall(request).execute();
                             ResponseBody responseBody = response.body();
                             InputStream inputStream = responseBody.byteStream()) {
                            if (!response.isSuccessful()) {
                                throw new IOException("Failed to fetch streaming data, HTTP error code: " + response.code());
                            }
                            byte[] buffer = new byte[4096];
                            int bytesRead;
                            while ((bytesRead = inputStream.read(buffer)) != -1) {