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"