repC option 1 improvement

Signed-off-by: xinyual <[email protected]>
xinyual · Feb 22, 2024 · 0c242ce · 0c242ce
1 parent b169b2f
commit 0c242ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLChatAgentRunner.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLChatAgentRunner.java
@@ -417,7 +417,7 @@ private void getNextStep(
     ) {
         String toolSeletctionType = "original";
         String toolSelectionModelId;
-        if (agentParameters.containsKey("tool_selection"))
+        if (agentParameters !=null && agentParameters.containsKey("tool_selection"))
         {
             Map<String, String> toolSelectionConfig = gson.fromJson(agentParameters.get("tool_selection"), Map.class);
             toolSeletctionType = toolSelectionConfig.getOrDefault("type", "original");