diff --git a/docs/reference.asciidoc b/docs/reference.asciidoc
index 91fa09692..2414962c0 100644
--- a/docs/reference.asciidoc
+++ b/docs/reference.asciidoc
@@ -6409,7 +6409,11 @@ or `windows-x86_64`. For portable models (those that work independent of process
 architecture or OS features), leave this field unset.
 ** *`tags` (Optional, string[])*: An array of tags to organize the model.
 ** *`prefix_strings` (Optional, { ingest, search })*: Optional prefix strings applied at inference
-** *`defer_definition_decompression` (Optional, boolean)*: If set to `true` and a `compressed_definition` is provided, the request defers definition decompression and skips relevant validations.
+** *`defer_definition_decompression` (Optional, boolean)*: If set to `true` and a `compressed_definition` is provided,
+the request defers definition decompression and skips relevant
+validations.
+** *`wait_for_completion` (Optional, boolean)*: Whether to wait for all child operations (e.g. model download)
+to complete.
 
 [discrete]
 ==== put_trained_model_alias
diff --git a/src/api/types.ts b/src/api/types.ts
index ac59a9410..c1912352f 100644
--- a/src/api/types.ts
+++ b/src/api/types.ts
@@ -14087,6 +14087,7 @@ export interface MlPutTrainedModelPreprocessor {
 export interface MlPutTrainedModelRequest extends RequestBase {
   model_id: Id
   defer_definition_decompression?: boolean
+  wait_for_completion?: boolean
   compressed_definition?: string
   definition?: MlPutTrainedModelDefinition
   description?: string
diff --git a/src/api/typesWithBodyKey.ts b/src/api/typesWithBodyKey.ts
index a617e64e3..938dc3328 100644
--- a/src/api/typesWithBodyKey.ts
+++ b/src/api/typesWithBodyKey.ts
@@ -14348,6 +14348,7 @@ export interface MlPutTrainedModelPreprocessor {
 export interface MlPutTrainedModelRequest extends RequestBase {
   model_id: Id
   defer_definition_decompression?: boolean
+  wait_for_completion?: boolean
   /** @deprecated The use of the 'body' key has been deprecated, move the nested keys to the top level object. */
   body?: {
     compressed_definition?: string