Commit dcf61fce authored by zhouwei's avatar zhouwei

新增request id日志

parent 9a8d1364
......@@ -38,6 +38,7 @@ public class AIServiceImpl implements AIService {
AppConfig appConfig;
@Autowired
RestTemplate restTemplate;
private static final String REQUEST_ID = "request_id";
private static final Map<String, String> NEWS_LABEL_MAP = new HashMap<>();
private static final Map<String, String> NATURAL_DISASTER_LABEL_MAP = new HashMap<>();
......@@ -560,7 +561,12 @@ public class AIServiceImpl implements AIService {
HttpEntity<String> entity = new HttpEntity<>(writeValueAsString, header.getHeaders());
ResponseEntity<ChatCompletionResponse> response = restTemplate.postForEntity(DEEP_SEEK_URL, entity, ChatCompletionResponse.class);
if (response.getStatusCode().is2xxSuccessful()) {
log.info("translate response:{}", JSONObject.toJSONString(response.getBody()));
if(response.getHeaders().containsKey(REQUEST_ID)){
log.info("request id:{},translate response:{}",response.getHeaders().get(REQUEST_ID), JSONObject.toJSONString(response.getBody()));
}else {
log.info("translate response:{}",JSONObject.toJSONString(response.getBody()));
}
return response.getBody();
}
throw new BaseException(MsgCode.TRANSLATE_ERROR);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment