From 3af51d8e36e60ddef7d3abdd59246d41e9cd89a6 Mon Sep 17 00:00:00 2001 From: Joshua Palis Date: Tue, 19 Mar 2024 16:32:59 +0000 Subject: [PATCH] Fixing register agent tests Signed-off-by: Joshua Palis --- .../opensearch/flowframework/workflow/RegisterAgentTests.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java b/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java index a2e0807b3..9005052ae 100644 --- a/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java +++ b/src/test/java/org/opensearch/flowframework/workflow/RegisterAgentTests.java @@ -16,6 +16,7 @@ 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; @@ -69,7 +70,7 @@ public void setUp() throws Exception { Map.ofEntries( Map.entry("name", "test"), Map.entry("description", "description"), - Map.entry("type", "type"), + Map.entry("type", MLAgentType.FLOW.name()), Map.entry("llm.model_id", "xyz"), Map.entry("llm.parameters", Collections.emptyMap()), Map.entry("tools", tools),