repC option 1 improvement

xinyual · xinyual · commit 0c242ce9ab4f · 2024-02-22T10:01:14.000+08:00
Signed-off-by: xinyual &lt;xinyual@amazon.com&gt;
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");

Original file line number	Diff line number	Diff line change
`@@ -417,7 +417,7 @@ private void getNextStep(`
`417`	`417`	`) {`
`418`	`418`	`String toolSeletctionType = "original";`
`419`	`419`	`String toolSelectionModelId;`
`420`		`- if (agentParameters.containsKey("tool_selection"))`
	`420`	`+ if (agentParameters !=null && agentParameters.containsKey("tool_selection"))`
`421`	`421`	`{`
`422`	`422`	`Map<String, String> toolSelectionConfig = gson.fromJson(agentParameters.get("tool_selection"), Map.class);`
`423`	`423`	`toolSeletctionType = toolSelectionConfig.getOrDefault("type", "original");`