1. 问题描述
平台接入Kimi服务API,技术实现通过RestTemplate框架调用Kimi的流式对话接口,发现修改的请求体参数,一直无法生效,导致输出的结果不完整。
线上临时切换成 okhttp3.OkHttpClient处理
2.问题排查
2.1 问题原因
定义RestCallback接口时,设置请求体的方式不正确。
注:RequestCallback是Spring RestTemplate中用来定制HTTP请求的一个接口,可以设置请求头、请求体、查询字符串参数。
2.2 问题源代码
// 设置请求头
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.APPLICATION_JSON);
headers.set("Authorization", "Bearer " + appKey);
// 构建请求体
HttpEntity<String> requestEntity = new HttpEntity(dto, headers);
// 定制请求接口
RequestCallback requestCallback = restTemplate.httpEntityCallback(requestEntity);
3. 问题解决
3.1 解决方案
通过ClientHttpRequest对象的方法来定制Request请求,设置请求体信息
3.2 问题修复的源代码
HttpEntity<String> requestEntity = new HttpEntity(dto, headers);RestTemplate restTemplate = new RestTemplate();RequestCallback requestCallback = request -> {request.getHeaders().setContentType(org.springframework.http.MediaType.APPLICATION_JSON);request.getHeaders().set("Authorization", "Bearer " + API_KEY);request.getBody().write(JSON.toJSONBytes(requestEntity.getBody()));};