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();
}
});
}
返回效果如下:
Content-Type | text/event-stream |