Skip to content

Commit 07429ba

Browse files
committed
[API] Updates machine_learning APIs
- machine_learning.put_trained_model adds `wait_for_completion` (Boolean) parameter: Whether to wait for all child operations(e.g. model download) to complete, before returning or not. Default to false. - machine_learning.start_trained_model_deployment adds `deployment_id` (String) The Id of the new deployment. Defaults to the model_id if not set.
1 parent 1613d9a commit 07429ba

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

elasticsearch-api/lib/elasticsearch/api/actions/machine_learning/put_trained_model.rb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ module Actions
2626
#
2727
# @option arguments [String] :model_id The ID of the trained models to store
2828
# @option arguments [Boolean] :defer_definition_decompression If set to `true` and a `compressed_definition` is provided, the request defers definition decompression and skips relevant validations.
29+
# @option arguments [Boolean] :wait_for_completion Whether to wait for all child operations(e.g. model download) to complete, before returning or not. Default to false
2930
# @option arguments [Hash] :headers Custom HTTP headers
3031
# @option arguments [Hash] :body The trained model configuration (*Required*)
3132
#

elasticsearch-api/lib/elasticsearch/api/actions/machine_learning/start_trained_model_deployment.rb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ module Actions
2626
#
2727
# @option arguments [String] :model_id The unique identifier of the trained model. (*Required*)
2828
# @option arguments [String] :cache_size A byte-size value for configuring the inference cache size. For example, 20mb.
29+
# @option arguments [String] :deployment_id The Id of the new deployment. Defaults to the model_id if not set.
2930
# @option arguments [Integer] :number_of_allocations The total number of allocations this model is assigned across machine learning nodes.
3031
# @option arguments [Integer] :threads_per_allocation The number of threads used by each model allocation during inference.
3132
# @option arguments [String] :priority The deployment priority.

0 commit comments

Comments
 (0)