diff --git a/main/manager-api/src/main/java/xiaozhi/modules/knowledge/service/impl/KnowledgeBaseServiceImpl.java b/main/manager-api/src/main/java/xiaozhi/modules/knowledge/service/impl/KnowledgeBaseServiceImpl.java index 7d37fe4a..80934316 100644 --- a/main/manager-api/src/main/java/xiaozhi/modules/knowledge/service/impl/KnowledgeBaseServiceImpl.java +++ b/main/manager-api/src/main/java/xiaozhi/modules/knowledge/service/impl/KnowledgeBaseServiceImpl.java @@ -1,5 +1,6 @@ package xiaozhi.modules.knowledge.service.impl; +import java.io.IOException; import java.io.Serializable; import java.util.ArrayList; import java.util.HashMap; @@ -150,7 +151,7 @@ public class KnowledgeBaseServiceImpl extends BaseServiceImpl response = restTemplate.exchange(url, HttpMethod.POST, requestEntity, String.class); + ResponseEntity response; + try { + response = restTemplate.exchange(url, HttpMethod.POST, requestEntity, String.class); + } catch (Exception e) { + String errorMessage = url + e.getMessage(); + log.error(errorMessage, e); + throw new RenException(ErrorCode.RAG_API_ERROR, errorMessage); + } log.info("RAGFlow API响应状态码: {}", response.getStatusCode()); log.debug("RAGFlow API响应内容: {}", response.getBody()); @@ -468,6 +476,9 @@ public class KnowledgeBaseServiceImpl extends BaseServiceImpl response = restTemplate.exchange(url, HttpMethod.PUT, requestEntity, String.class); + ResponseEntity response; + try { + response = restTemplate.exchange(url, HttpMethod.PUT, requestEntity, String.class); + } catch (Exception e) { + String errorMessage = url + e.getMessage(); + log.error(errorMessage, e); + throw new RenException(ErrorCode.RAG_API_ERROR, errorMessage); + } log.info("RAGFlow API响应状态码: {}", response.getStatusCode()); log.debug("RAGFlow API响应内容: {}", response.getBody()); @@ -546,6 +564,9 @@ public class KnowledgeBaseServiceImpl extends BaseServiceImpl response = restTemplate.exchange(url, HttpMethod.DELETE, requestEntity, - String.class); + ResponseEntity response; + try { + response = restTemplate.exchange(url, HttpMethod.DELETE, requestEntity, String.class); + } catch (Exception e) { + String errorMessage = url + e.getMessage(); + log.error(errorMessage, e); + throw new RenException(ErrorCode.RAG_API_ERROR, errorMessage); + } log.info("RAGFlow API响应状态码: {}", response.getStatusCode()); log.debug("RAGFlow API响应内容: {}", response.getBody()); @@ -615,6 +642,9 @@ public class KnowledgeBaseServiceImpl extends BaseServiceImpl response = restTemplate.exchange(url, HttpMethod.GET, requestEntity, String.class); + ResponseEntity response; + try { + response = restTemplate.exchange(url, HttpMethod.GET, requestEntity, String.class); + } catch (Exception e) { + String errorMessage = url + e.getMessage(); + log.error(errorMessage, e); + return 0; + } log.info("RAGFlow API响应状态码: {}", response.getStatusCode()); @@ -767,6 +804,8 @@ public class KnowledgeBaseServiceImpl extends BaseServiceImpl