Addressed PR comments and removed the map

Signed-off-by: Owais Kazi <[email protected]>
opensearch-project · Feb 19, 2024 · a6662e5 · a6662e5
1 parent 990804f
commit a6662e5
Show file tree

Hide file tree

Showing 5 changed files with 326 additions and 141 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/common/CommonValue.java b/src/main/java/org/opensearch/flowframework/common/CommonValue.java
@@ -188,4 +188,6 @@ private CommonValue() {}
     public static final String RESOURCE_TYPE = "resource_type";
     /** The field name for the resource id */
     public static final String RESOURCE_ID = "resource_id";
+    /** The field name for the opensearch-ml plugin */
+    public static final String OPENSEARCH_ML = "opensearch-ml";
 }
diff --git a/src/main/java/org/opensearch/flowframework/workflow/ToolStep.java b/src/main/java/org/opensearch/flowframework/workflow/ToolStep.java
@@ -31,7 +31,6 @@
 /**
  * Step to register a tool for an agent
  */
-@SuppressWarnings("unchecked")
 public class ToolStep implements WorkflowStep {
 
     private static final Logger logger = LogManager.getLogger(ToolStep.class);
@@ -107,6 +106,7 @@ private Map<String, String> getToolsParametersMap(
         Map<String, String> previousNodeInputs,
         Map<String, WorkflowData> outputs
     ) {
+        @SuppressWarnings("unchecked")
         Map<String, String> parametersMap = (Map<String, String>) parameters;
         Optional<String> previousNodeModel = previousNodeInputs.entrySet()
             .stream()