Added javadocs

Signed-off-by: Owais Kazi <[email protected]>

src/main/java/org/opensearch/flowframework/workflow/DeployModelStep.java

@@ -20,13 +20,20 @@
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 
+/**
+ * Step to deploy a model
+ */
 public class DeployModelStep implements WorkflowStep {
     private static final Logger logger = LogManager.getLogger(DeployModelStep.class);
 
     private NodeClient nodeClient;
     private static final String MODEL_ID = "model_id";
     static final String NAME = "deploy_model";
 
+    /**
+     * Instantiate this class
+     * @param nodeClient client to instantiate MLClient
+     */
     public DeployModelStep(NodeClient nodeClient) {
         this.nodeClient = nodeClient;
     }

src/main/java/org/opensearch/flowframework/workflow/GetTask.java

@@ -17,18 +17,29 @@
 import org.opensearch.ml.common.MLTaskState;
 import org.opensearch.ml.common.transport.task.MLTaskGetResponse;
 
+/**
+ * Step to get modelID of a registered local model
+ */
 @SuppressForbidden(reason = "This class is for the future work of registering local model")
 public class GetTask {
 
     private static final Logger logger = LogManager.getLogger(GetTask.class);
     private MachineLearningNodeClient machineLearningNodeClient;
     private String taskId;
 
+    /**
+     * Instantiate this class
+     * @param machineLearningNodeClient client to instantiate ml-commons APIs
+     * @param taskId taskID of the model
+     */
     public GetTask(MachineLearningNodeClient machineLearningNodeClient, String taskId) {
         this.machineLearningNodeClient = machineLearningNodeClient;
         this.taskId = taskId;
     }
 
+    /**
+     * Invokes get task API of ml-commons
+     */
     public void getTask() {
 
         ActionListener<MLTask> actionListener = new ActionListener<>() {
@@ -51,8 +62,4 @@ public void onFailure(Exception e) {
 
     }
 
-    /*@Override
-    public void run() {
-        getTask();
-    }*/
 }

src/main/java/org/opensearch/flowframework/workflow/RegisterModelStep.java

@@ -28,6 +28,9 @@
 import java.util.concurrent.CompletableFuture;
 import java.util.stream.Stream;
 
+/**
+ * Step to register a remote model
+ */
 public class RegisterModelStep implements WorkflowStep {
 
     private static final Logger logger = LogManager.getLogger(RegisterModelStep.class);
@@ -45,6 +48,10 @@ public class RegisterModelStep implements WorkflowStep {
     private static final String MODEL_FORMAT = "model_format";
     private static final String MODEL_CONFIG = "model_config";
 
+    /**
+     * Instantiate this class
+     * @param nodeClient client to instantiate MLClient
+     */
     public RegisterModelStep(NodeClient nodeClient) {
         this.nodeClient = nodeClient;
     }

src/main/java/org/opensearch/flowframework/workflow/WorkflowStepFactory.java

@@ -29,6 +29,7 @@ public class WorkflowStepFactory {
      * Instantiate this class.
      *
      * @param client The OpenSearch client steps can use
+     * @param nodeClient NodeClient to execute transport calls
      */
     public WorkflowStepFactory(Client client, NodeClient nodeClient) {
         populateMap(client, nodeClient);

src/test/java/org/opensearch/flowframework/workflow/DeployModelStepTests.java

@@ -28,7 +28,8 @@
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 
-import static org.mockito.Mockito.*;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.doAnswer;
 
 @ThreadLeakScope(ThreadLeakScope.Scope.NONE)
 public class DeployModelStepTests extends OpenSearchTestCase {