百度文心一言 流式接口的封装和透传_百度 流式接口

1、项目需要对流式接口进行封装 
 

@PostMapping(value = "/chat-stream", consumes = org.springframework.http.MediaType.APPLICATION_JSON_VALUE, produces = org.springframework.http.MediaType.TEXT_EVENT_STREAM_VALUE)
    public ResponseEntity<StreamingResponseBody> streamChatResponse(@org.springframework.web.bind.annotation.RequestBody ChatRequest chatRequest) {
        OkHttpClient client = new OkHttpClient.Builder().build();
 
        // 构建请求体
        MediaType mediaType = MediaType.parse("application/json; charset=utf-8");
        String requestBodyJson = "{\"messages\":[{\"role\":\"user\",\"content\":\"" + chatRequest.getMessage() + "\"}],\"temperature\":0.2,\"topP\":1,\"penaltyScore\":1,\"stream\":true}";
        RequestBody requestBody = RequestBody.create(mediaType, requestBodyJson);
 
        // 构建请求URL
        String url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + getAccessToken();
        Request request = new Request.Builder()
                .url(url)
                .post(requestBody)
                .addHeader("Content-Type", mediaType.toString())
                .build();
 
        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) {
                            outputStream.write(buffer, 0, bytesRead);
                            outputStream.flush();
                        }
                    } catch (IOException e) {
                        log.error("Error during streaming data: ", e);
                        outputStream.write(("{" +
                                "  \"error_code\": 112," +
                                "  \"error_msg\": \"" + e.getMessage() + "\"" +
                                "}").getBytes(StandardCharsets.UTF_8));
                        outputStream.flush();
                    }
                });
    }

返回效果如下：