GA the update trained model action (#2576)

* GA the update trained model action * Fix lint
elastic · Jun 10, 2024 · f0da131 · f0da131
1 parent b7580b9
commit f0da131
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 5 deletions.
diff --git a/output/schema/schema.json b/output/schema/schema.json
diff --git a/output/typescript/types.ts b/output/typescript/types.ts
diff --git a/specification/_json_spec/ml.update_trained_model_deployment.json b/specification/_json_spec/ml.update_trained_model_deployment.json
@@ -4,7 +4,7 @@
       "url": "https://www.elastic.co/guide/en/elasticsearch/reference/current/update-trained-model-deployment.html",
       "description": "Updates certain properties of trained model deployment."
     },
-    "stability": "beta",
+    "stability": "stable",
     "visibility": "public",
     "headers": {
       "accept": ["application/json"],
@@ -24,9 +24,16 @@
         }
       ]
     },
+    "params": {
+      "number_of_allocations": {
+        "type": "int",
+        "required": false,
+        "description": "Update the model deployment to this number of allocations."
+      }
+    },
     "body": {
       "description": "The updated trained model deployment settings",
-      "required": true
+      "required": false
     }
   }
 }
diff --git a/specification/ml/update_trained_model_deployment/MlUpdateTrainedModelDeploymentRequest.ts b/specification/ml/update_trained_model_deployment/MlUpdateTrainedModelDeploymentRequest.ts
@@ -35,6 +35,18 @@ export interface Request extends RequestBase {
      */
     model_id: Id
   }
+  query_parameters: {
+    /**
+     * The number of model allocations on each node where the model is deployed.
+     * All allocations on a node share the same copy of the model in memory but use
+     * a separate set of threads to evaluate the model.
+     * Increasing this value generally increases the throughput.
+     * If this setting is greater than the number of hardware threads
+     * it will automatically be changed to a value less than the number of hardware threads.
+     * @server_default 1
+     */
+    number_of_allocations?: integer
+  }
   body: {
     /**
      * The number of model allocations on each node where the model is deployed.