Added TODO

Signed-off-by: Owais Kazi <[email protected]>
opensearch-project · Nov 30, 2023 · e4fb376 · e4fb376
1 parent cc069de
commit e4fb376
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/util/ParseUtils.java b/src/main/java/org/opensearch/flowframework/util/ParseUtils.java
@@ -117,6 +117,7 @@ public static Map<String, String> parseStringToStringMap(XContentParser parser)
         return map;
     }
 
+    // TODO Figure out a way to use the parse method of LLMSpec of ml-commons
     /**
      * Parses an XContent object representing the object of LLMSpec
      * @param parser An XContent parser whose position is at the start of the map object to parse

diff --git a/src/main/java/org/opensearch/flowframework/workflow/RegisterAgentStep.java b/src/main/java/org/opensearch/flowframework/workflow/RegisterAgentStep.java
@@ -69,10 +69,10 @@ public RegisterAgentStep(MachineLearningNodeClient mlClient) {
 
     @Override
     public CompletableFuture<WorkflowData> execute(
-            String currentNodeId,
-            WorkflowData currentNodeInputs,
-            Map<String, WorkflowData> outputs,
-            Map<String, String> previousNodeInputs
+        String currentNodeId,
+        WorkflowData currentNodeInputs,
+        Map<String, WorkflowData> outputs,
+        Map<String, String> previousNodeInputs
     ) throws IOException {
 
         CompletableFuture<WorkflowData> registerAgentModelFuture = new CompletableFuture<>();
@@ -82,11 +82,11 @@ public CompletableFuture<WorkflowData> execute(
             public void onResponse(MLRegisterAgentResponse mlRegisterAgentResponse) {
                 logger.info("Remote Agent registration successful");
                 registerAgentModelFuture.complete(
-                        new WorkflowData(
-                                Map.ofEntries(Map.entry(AGENT_ID, mlRegisterAgentResponse.getAgentId())),
-                                currentNodeInputs.getWorkflowId(),
-                                currentNodeInputs.getNodeId()
-                        )
+                    new WorkflowData(
+                        Map.ofEntries(Map.entry(AGENT_ID, mlRegisterAgentResponse.getAgentId())),
+                        currentNodeInputs.getWorkflowId(),
+                        currentNodeInputs.getNodeId()
+                    )
                 );
             }