Fix parsing boolean string issue

Signed-off-by: Songkan Tang <[email protected]>
yuye-aws · Aug 5, 2024 · 13323fc · 13323fc
1 parent 70a5482
commit 13323fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<String, Object> 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.");