diff --git a/src/main/java/org/opensearch/agent/tools/RCATool.java b/src/main/java/org/opensearch/agent/tools/RCATool.java index 0a790972..6bc286fd 100644 --- a/src/main/java/org/opensearch/agent/tools/RCATool.java +++ b/src/main/java/org/opensearch/agent/tools/RCATool.java @@ -272,7 +272,7 @@ public void init(Client client) { @Override public RCATool create(Map parameters) { - Boolean isLLMOption = (Boolean) parameters.get(IS_LLM_OPTION); + Boolean isLLMOption = Boolean.valueOf((String) parameters.getOrDefault(IS_LLM_OPTION, "true")); String modelId = (String) parameters.get(MODEL_ID); if (isLLMOption && Strings.isBlank(modelId)) { throw new IllegalArgumentException("model_id cannot be null or blank.");