Retrieve remote model id from registration response in IT to avoid flaky (opensearch-project#3244)

Signed-off-by: zane-neo <[email protected]>

Author: zane-neo

plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java

@@ -1002,15 +1002,11 @@ public String registerRemoteModel(String createConnectorInput, String modelName,
         String connectorId = (String) responseMap.get("connector_id");
         response = RestMLRemoteInferenceIT.registerRemoteModel(modelName, modelName, connectorId);
         responseMap = parseResponseToMap(response);
-        String taskId = (String) responseMap.get("task_id");
-        waitForTask(taskId, MLTaskState.COMPLETED);
-        response = RestMLRemoteInferenceIT.getTask(taskId);
-        responseMap = parseResponseToMap(response);
         String modelId = (String) responseMap.get("model_id");
         if (deploy) {
             response = RestMLRemoteInferenceIT.deployRemoteModel(modelId);
             responseMap = parseResponseToMap(response);
-            taskId = (String) responseMap.get("task_id");
+            String taskId = (String) responseMap.get("task_id");
             waitForTask(taskId, MLTaskState.COMPLETED);
         }
         return modelId;