Spotless fixes

Signed-off-by: Owais Kazi <[email protected]>
opensearch-project · Dec 3, 2023 · 9fd114a · 9fd114a
1 parent a8c5cb9
commit 9fd114a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/workflow/RegisterAgentStep.java b/src/main/java/org/opensearch/flowframework/workflow/RegisterAgentStep.java
@@ -186,11 +186,12 @@ private List<MLToolSpec> addTools(Object tools) {
         return mlToolSpecList;
     }
 
+    // TODO https://github.com/opensearch-project/flow-framework/issues/241 - to add multiple tools
     private List<MLToolSpec> getTools(Map<String, String> previousNodeInputs, Map<String, WorkflowData> outputs) {
         List<MLToolSpec> mlToolSpecList = new ArrayList<>();
         List<String> previousNodes = previousNodeInputs.entrySet()
             .stream()
-            .filter(e -> e.getValue().startsWith(TOOLS_FIELD + "."))
+            .filter(e -> TOOLS_FIELD.equals(e.getValue()))
             .map(Map.Entry::getKey)
             .collect(Collectors.toList());
 

diff --git a/src/main/java/org/opensearch/flowframework/workflow/ToolStep.java b/src/main/java/org/opensearch/flowframework/workflow/ToolStep.java
@@ -20,7 +20,14 @@
 import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 
-import static org.opensearch.flowframework.common.CommonValue.*;
+import static org.opensearch.flowframework.common.CommonValue.AGENT_ID;
+import static org.opensearch.flowframework.common.CommonValue.DESCRIPTION_FIELD;
+import static org.opensearch.flowframework.common.CommonValue.INCLUDE_OUTPUT_IN_AGENT_RESPONSE;
+import static org.opensearch.flowframework.common.CommonValue.MODEL_ID;
+import static org.opensearch.flowframework.common.CommonValue.NAME_FIELD;
+import static org.opensearch.flowframework.common.CommonValue.PARAMETERS_FIELD;
+import static org.opensearch.flowframework.common.CommonValue.TOOLS_FIELD;
+import static org.opensearch.flowframework.common.CommonValue.TYPE;
 
 /**
  * Step to register a tool for an agent