test

Signed-off-by: Jackie Han <[email protected]>
jackiehanyang · Feb 1, 2024 · 84c6a68 · 84c6a68
1 parent 0ec4407
commit 84c6a68
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/common/CommonValue.java b/src/main/java/org/opensearch/flowframework/common/CommonValue.java
@@ -65,6 +65,8 @@ private CommonValue() {}
     /** The field name for provision workflow within a use case template*/
     public static final String PROVISION_WORKFLOW = "provision";
 
+    public static final String SEARCH_WORKFLOW = "search";
+
     /*
      * Constants associated with plugin configuration
      */

diff --git a/src/main/java/org/opensearch/flowframework/processor/FlowFrameworkResponseProcessor.java b/src/main/java/org/opensearch/flowframework/processor/FlowFrameworkResponseProcessor.java
@@ -9,6 +9,8 @@
 import org.opensearch.client.Client;
 import org.opensearch.core.action.ActionListener;
 import org.opensearch.flowframework.model.Template;
+import org.opensearch.flowframework.model.Workflow;
+import org.opensearch.flowframework.model.WorkflowNode;
 import org.opensearch.flowframework.transport.GetWorkflowAction;
 import org.opensearch.flowframework.transport.WorkflowRequest;
 import org.opensearch.search.pipeline.AbstractProcessor;
@@ -18,6 +20,8 @@
 import java.util.Map;
 
 import static org.opensearch.flowframework.common.CommonValue.GLOBAL_CONTEXT_INDEX;
+import static org.opensearch.flowframework.common.CommonValue.PROVISION_WORKFLOW;
+import static org.opensearch.flowframework.common.CommonValue.SEARCH_WORKFLOW;
 import static org.opensearch.ingest.ConfigurationUtils.readStringProperty;
 
 public class FlowFrameworkResponseProcessor extends AbstractProcessor implements SearchResponseProcessor {
@@ -59,6 +63,9 @@ public SearchResponse processResponse(SearchRequest request, SearchResponse resp
         GetResponse getResponse = client.get(getRequest).get();
         System.out.println("source: " + getResponse.getSourceAsString());
         Template template = Template.parse(getResponse.getSourceAsString());
+        Workflow searchWorkflow  = template.workflows().get(SEARCH_WORKFLOW);
+        WorkflowNode workflowNode = searchWorkflow.nodes().get(0);
+        System.out.println("node: " + workflowNode.type());
         System.out.println("template: " + template.toJson());
 
         return response;