From 64d9f7e6cf846e2d8128891b16baa7bc0000ab23 Mon Sep 17 00:00:00 2001 From: Ryan Bogan Date: Thu, 13 Jul 2023 22:36:15 +0000 Subject: [PATCH] Reenable OpenAI remote inference tests Signed-off-by: Ryan Bogan --- .../java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java | 4 ---- 1 file changed, 4 deletions(-) diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java index 3d7165710a..6230451f2d 100644 --- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java +++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java @@ -131,7 +131,6 @@ public void testDeployRemoteModel() throws IOException, InterruptedException { waitForTask(taskId, MLTaskState.COMPLETED); } - @Ignore public void testPredictRemoteModel() throws IOException, InterruptedException { Response response = createConnector(completionModelConnectorEntity); Map responseMap = parseResponseToMap(response); @@ -184,7 +183,6 @@ public void testUndeployRemoteModel() throws IOException, InterruptedException { assertTrue(responseMap.toString().contains("undeployed")); } - @Ignore public void testOpenAIChatCompletionModel() throws IOException, InterruptedException { String entity = "{\n" + " \"name\": \"OpenAI chat model Connector\",\n" @@ -241,7 +239,6 @@ public void testOpenAIChatCompletionModel() throws IOException, InterruptedExcep assertNotNull(responseMap); } - @Ignore public void testOpenAIEditsModel() throws IOException, InterruptedException { String entity = "{\n" + " \"name\": \"OpenAI Edit model Connector\",\n" @@ -307,7 +304,6 @@ public void testOpenAIEditsModel() throws IOException, InterruptedException { assertFalse(((String) responseMap.get("text")).isEmpty()); } - @Ignore public void testOpenAIModerationsModel() throws IOException, InterruptedException { String entity = "{\n" + " \"name\": \"OpenAI moderations model Connector\",\n"