From 777a114fe0fb5c57b57d4ded40de4a46a478286a Mon Sep 17 00:00:00 2001
From: Amit Galitzky <amgalitz@amazon.com>
Date: Thu, 4 Apr 2024 16:12:42 +0000
Subject: [PATCH] Revert "[Backport 2.13] Fixing register agent tests (#597)"

This reverts commit 79ecb6f73fccb43cd8d57d10343860df70fb1103.
---
 .../opensearch/flowframework/workflow/RegisterAgentTests.java | 3 +--
 src/test/resources/template/agent-framework.json              | 4 ++--
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java b/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java
index 9005052ae..a2e0807b3 100644
--- a/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java
+++ b/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java
@@ -16,7 +16,6 @@
 import org.opensearch.flowframework.exception.FlowFrameworkException;
 import org.opensearch.flowframework.indices.FlowFrameworkIndicesHandler;
 import org.opensearch.ml.client.MachineLearningNodeClient;
-import org.opensearch.ml.common.MLAgentType;
 import org.opensearch.ml.common.agent.LLMSpec;
 import org.opensearch.ml.common.agent.MLAgent;
 import org.opensearch.ml.common.agent.MLMemorySpec;
@@ -70,7 +69,7 @@ public void setUp() throws Exception {
             Map.ofEntries(
                 Map.entry("name", "test"),
                 Map.entry("description", "description"),
-                Map.entry("type", MLAgentType.FLOW.name()),
+                Map.entry("type", "type"),
                 Map.entry("llm.model_id", "xyz"),
                 Map.entry("llm.parameters", Collections.emptyMap()),
                 Map.entry("tools", tools),
diff --git a/src/test/resources/template/agent-framework.json b/src/test/resources/template/agent-framework.json
index 5c4ce6ef0..2a12938a4 100644
--- a/src/test/resources/template/agent-framework.json
+++ b/src/test/resources/template/agent-framework.json
@@ -96,7 +96,7 @@
           },
           "user_inputs": {
             "name": "Sub Agent",
-            "type": "CONVERSATIONAL",
+            "type": "cot",
             "description": "this is a test agent",
             "parameters": {
               "hello": "world"
@@ -139,7 +139,7 @@
           },
           "user_inputs": {
             "name": "DEMO-Test_Agent_For_CoT",
-            "type": "CONVERSATIONAL",
+            "type": "cot",
             "description": "this is a test agent",
             "parameters": {
               "hello": "world"