Skip to content

Commit 576bc1f

Browse files
Fix visualization tool
Signed-off-by: Nathalie Jonathan <[email protected]>
1 parent 3196df5 commit 576bc1f

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/tool/MLToolExecutor.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ public void execute(Input input, ActionListener<Output> listener) {
8585
try {
8686
Map<String, String> mutableParams = new HashMap<>(parameters);
8787
Map<String, Object> originalParams = toolMLInput.getOriginalParameters();
88-
Tool tool = toolFactory.create(originalParams != null ? originalParams : new HashMap<>(mutableParams));
88+
Tool tool = toolFactory.create(originalParams);
8989

9090
if (!tool.validate(mutableParams)) {
9191
listener.onFailure(new IllegalArgumentException("Invalid parameters for tool: " + toolName));

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/VisualizationsTool.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,9 @@ public VisualizationsTool create(Map<String, Object> params) {
204204
} catch (NumberFormatException ignored) {
205205
size = DEFAULT_SIZE;
206206
}
207-
return VisualizationsTool.builder().client(client).index(index).size(size).build();
207+
VisualizationsTool tool = VisualizationsTool.builder().client(client).index(index).size(size).build();
208+
tool.maxInputLength = ConfigurationUtils.readIntProperty(TYPE, null, params, MAX_INPUT_LENGTH_FIELD, DEFAULT_MAX_INPUT_LENGTH);
209+
return tool;
208210
}
209211

210212
@Override

0 commit comments

Comments
 (0)