diff --git a/src/test/java/org/opensearch/neuralsearch/common/BaseNeuralSearchIT.java b/src/test/java/org/opensearch/neuralsearch/common/BaseNeuralSearchIT.java index 60454e6bc..33cdff9a0 100644 --- a/src/test/java/org/opensearch/neuralsearch/common/BaseNeuralSearchIT.java +++ b/src/test/java/org/opensearch/neuralsearch/common/BaseNeuralSearchIT.java @@ -752,20 +752,20 @@ protected String getDeployedModelId() { @SneakyThrows private String registerModelGroup() { String modelGroupRegisterRequestBody = Files.readString( - Path.of(classLoader.getResource("processor/CreateModelGroupRequestBody.json").toURI()) + Path.of(classLoader.getResource("processor/CreateModelGroupRequestBody.json").toURI()) ).replace("", "public_model_" + RandomizedTest.randomAsciiAlphanumOfLength(8)); Response modelGroupResponse = makeRequest( - client(), - "POST", - "/_plugins/_ml/model_groups/_register", - null, - toHttpEntity(modelGroupRegisterRequestBody), - ImmutableList.of(new BasicHeader(HttpHeaders.USER_AGENT, "Kibana")) + client(), + "POST", + "/_plugins/_ml/model_groups/_register", + null, + toHttpEntity(modelGroupRegisterRequestBody), + ImmutableList.of(new BasicHeader(HttpHeaders.USER_AGENT, "Kibana")) ); Map modelGroupResJson = XContentHelper.convertToMap( - XContentType.JSON.xContent(), - EntityUtils.toString(modelGroupResponse.getEntity()), - false + XContentType.JSON.xContent(), + EntityUtils.toString(modelGroupResponse.getEntity()), + false ); String modelGroupId = modelGroupResJson.get("model_group_id").toString(); assertNotNull(modelGroupId); diff --git a/src/test/java/org/opensearch/neuralsearch/processor/TextImageEmbeddingProcessorIT.java b/src/test/java/org/opensearch/neuralsearch/processor/TextImageEmbeddingProcessorIT.java index 59d964b9c..321cb1dc3 100644 --- a/src/test/java/org/opensearch/neuralsearch/processor/TextImageEmbeddingProcessorIT.java +++ b/src/test/java/org/opensearch/neuralsearch/processor/TextImageEmbeddingProcessorIT.java @@ -11,9 +11,10 @@ import lombok.SneakyThrows; -import org.apache.http.HttpHeaders; -import org.apache.http.message.BasicHeader; -import org.apache.http.util.EntityUtils; + +import org.apache.hc.core5.http.HttpHeaders; +import org.apache.hc.core5.http.io.entity.EntityUtils; +import org.apache.hc.core5.http.message.BasicHeader; import org.junit.After; import org.opensearch.client.Response; import org.opensearch.common.xcontent.XContentHelper;