@elastic/elasticsearch 8.0.0-beta.1 → 8.2.0-patch.1

package/lib/api/api/ml.js

@@ -30,10 +30,24 @@ class Ml {
     }
     async closeJob(params, options) {
         const acceptedPath = ['job_id'];
+        const acceptedBody = ['allow_no_match', 'force', 'timeout'];
         const querystring = {};
-        const body = undefined;
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
         for (const key in params) {
-            if (acceptedPath.includes(key)) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
@@ -356,6 +370,7 @@ class Ml {
         else {
             body = userBody != null ? { ...userBody } : undefined;
         }
+        params = params !== null && params !== void 0 ? params : {};
         for (const key in params) {
             if (acceptedBody.includes(key)) {
                 body = body !== null && body !== void 0 ? body : {};
@@ -384,7 +399,7 @@ class Ml {
     }
     async flushJob(params, options) {
         const acceptedPath = ['job_id'];
-        const acceptedBody = ['advance_time', 'calc_interim', 'end', 'start'];
+        const acceptedBody = ['advance_time', 'calc_interim', 'end', 'skip_time', 'start'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -446,7 +461,7 @@ class Ml {
     }
     async getBuckets(params, options) {
         const acceptedPath = ['job_id', 'timestamp'];
-        const acceptedBody = ['anomaly_score', 'desc', 'exclude_interim', 'expand', 'sort', 'start', 'end'];
+        const acceptedBody = ['anomaly_score', 'desc', 'end', 'exclude_interim', 'expand', 'page', 'sort', 'start'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -792,9 +807,52 @@ class Ml {
         }
         return await this.transport.request({ path, method, querystring, body }, options);
     }
+    async getMemoryStats(params, options) {
+        const acceptedPath = ['node_id'];
+        const querystring = {};
+        const body = undefined;
+        params = params !== null && params !== void 0 ? params : {};
+        for (const key in params) {
+            if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        let method = '';
+        let path = '';
+        if (params.node_id != null) {
+            method = 'GET';
+            path = `/_ml/memory/${encodeURIComponent(params.node_id.toString())}/_stats`;
+        }
+        else {
+            method = 'GET';
+            path = '/_ml/memory/_stats';
+        }
+        return await this.transport.request({ path, method, querystring, body }, options);
+    }
+    async getModelSnapshotUpgradeStats(params, options) {
+        const acceptedPath = ['job_id', 'snapshot_id'];
+        const querystring = {};
+        const body = undefined;
+        params = params !== null && params !== void 0 ? params : {};
+        for (const key in params) {
+            if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'GET';
+        const path = `/_ml/anomaly_detectors/${encodeURIComponent(params.job_id.toString())}/model_snapshots/${encodeURIComponent(params.snapshot_id.toString())}/_upgrade/_stats`;
+        return await this.transport.request({ path, method, querystring, body }, options);
+    }
     async getModelSnapshots(params, options) {
         const acceptedPath = ['job_id', 'snapshot_id'];
-        const acceptedBody = ['start', 'end'];
+        const acceptedBody = ['desc', 'end', 'page', 'sort', 'start'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -833,10 +891,24 @@ class Ml {
     }
     async getOverallBuckets(params, options) {
         const acceptedPath = ['job_id'];
+        const acceptedBody = ['allow_no_match', 'bucket_span', 'end', 'exclude_interim', 'overall_score', 'start', 'top_n'];
         const querystring = {};
-        const body = undefined;
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
         for (const key in params) {
-            if (acceptedPath.includes(key)) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
@@ -850,7 +922,7 @@ class Ml {
     }
     async getRecords(params, options) {
         const acceptedPath = ['job_id'];
-        const acceptedBody = ['desc', 'exclude_interim', 'page', 'record_score', 'sort', 'start', 'end'];
+        const acceptedBody = ['desc', 'end', 'exclude_interim', 'page', 'record_score', 'sort', 'start'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -879,23 +951,6 @@ class Ml {
         const path = `/_ml/anomaly_detectors/${encodeURIComponent(params.job_id.toString())}/results/records`;
         return await this.transport.request({ path, method, querystring, body }, options);
     }
-    async getTrainedModelDeploymentStats(params, options) {
-        const acceptedPath = ['model_id'];
-        const querystring = {};
-        const body = undefined;
-        params = params !== null && params !== void 0 ? params : {};
-        for (const key in params) {
-            if (acceptedPath.includes(key)) {
-                continue;
-            }
-            else if (key !== 'body') {
-                querystring[key] = params[key];
-            }
-        }
-        const method = 'GET';
-        const path = `/_ml/trained_models/${encodeURIComponent(params.model_id.toString())}/deployment/_stats`;
-        return await this.transport.request({ path, method, querystring, body }, options);
-    }
     async getTrainedModels(params, options) {
         const acceptedPath = ['model_id'];
         const querystring = {};
@@ -950,14 +1005,28 @@ class Ml {
     }
     async inferTrainedModelDeployment(params, options) {
         const acceptedPath = ['model_id'];
+        const acceptedBody = ['docs', 'inference_config'];
         const querystring = {};
-        const body = undefined;
-        params = params !== null && params !== void 0 ? params : {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
         for (const key in params) {
-            if (acceptedPath.includes(key)) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
+                // @ts-expect-error
                 querystring[key] = params[key];
             }
         }
@@ -1151,7 +1220,7 @@ class Ml {
     }
     async putCalendar(params, options) {
         const acceptedPath = ['calendar_id'];
-        const acceptedBody = ['description'];
+        const acceptedBody = ['job_ids', 'description'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -1199,7 +1268,7 @@ class Ml {
     }
     async putDataFrameAnalytics(params, options) {
         const acceptedPath = ['id'];
-        const acceptedBody = ['allow_lazy_start', 'analysis', 'analyzed_fields', 'description', 'dest', 'max_num_threads', 'model_memory_limit', 'source'];
+        const acceptedBody = ['allow_lazy_start', 'analysis', 'analyzed_fields', 'description', 'dest', 'max_num_threads', 'model_memory_limit', 'source', 'headers', 'version'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -1230,7 +1299,7 @@ class Ml {
     }
     async putDatafeed(params, options) {
         const acceptedPath = ['datafeed_id'];
-        const acceptedBody = ['aggregations', 'chunking_config', 'delayed_data_check_config', 'frequency', 'indices', 'indices_options', 'job_id', 'max_empty_searches', 'query', 'query_delay', 'runtime_mappings', 'script_fields', 'scroll_size'];
+        const acceptedBody = ['aggregations', 'chunking_config', 'delayed_data_check_config', 'frequency', 'indices', 'indexes', 'indices_options', 'job_id', 'max_empty_searches', 'query', 'query_delay', 'runtime_mappings', 'script_fields', 'scroll_size', 'headers'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -1323,7 +1392,7 @@ class Ml {
     }
     async putTrainedModel(params, options) {
         const acceptedPath = ['model_id'];
-        const acceptedBody = ['compressed_definition', 'definition', 'description', 'inference_config', 'input', 'metadata', 'tags'];
+        const acceptedBody = ['compressed_definition', 'definition', 'description', 'inference_config', 'input', 'metadata', 'model_type', 'model_size_bytes', 'tags'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -1371,14 +1440,28 @@ class Ml {
     }
     async putTrainedModelDefinitionPart(params, options) {
         const acceptedPath = ['model_id', 'part'];
+        const acceptedBody = ['definition', 'total_definition_length', 'total_parts'];
         const querystring = {};
-        const body = undefined;
-        params = params !== null && params !== void 0 ? params : {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
         for (const key in params) {
-            if (acceptedPath.includes(key)) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
+                // @ts-expect-error
                 querystring[key] = params[key];
             }
         }
@@ -1388,14 +1471,28 @@ class Ml {
     }
     async putTrainedModelVocabulary(params, options) {
         const acceptedPath = ['model_id'];
+        const acceptedBody = ['vocabulary', 'merges'];
         const querystring = {};
-        const body = undefined;
-        params = params !== null && params !== void 0 ? params : {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
         for (const key in params) {
-            if (acceptedPath.includes(key)) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
+                // @ts-expect-error
                 querystring[key] = params[key];
             }
         }
@@ -1521,12 +1618,12 @@ class Ml {
         const acceptedPath = ['model_id'];
         const querystring = {};
         const body = undefined;
-        params = params !== null && params !== void 0 ? params : {};
         for (const key in params) {
             if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
+                // @ts-expect-error
                 querystring[key] = params[key];
             }
         }
@@ -1553,7 +1650,7 @@ class Ml {
     }
     async stopDatafeed(params, options) {
         const acceptedPath = ['datafeed_id'];
-        const acceptedBody = ['force', 'timeout'];
+        const acceptedBody = ['allow_no_match', 'force', 'timeout'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -1586,12 +1683,12 @@ class Ml {
         const acceptedPath = ['model_id'];
         const querystring = {};
         const body = undefined;
-        params = params !== null && params !== void 0 ? params : {};
         for (const key in params) {
             if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
+                // @ts-expect-error
                 querystring[key] = params[key];
             }
         }
@@ -1632,14 +1729,28 @@ class Ml {
     }
     async updateDatafeed(params, options) {
         const acceptedPath = ['datafeed_id'];
+        const acceptedBody = ['aggregations', 'chunking_config', 'delayed_data_check_config', 'frequency', 'indices', 'indexes', 'indices_options', 'max_empty_searches', 'query', 'query_delay', 'runtime_mappings', 'script_fields', 'scroll_size'];
         const querystring = {};
-        const body = undefined;
-        params = params !== null && params !== void 0 ? params : {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
         for (const key in params) {
-            if (acceptedPath.includes(key)) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
                 continue;
             }
             else if (key !== 'body') {
+                // @ts-expect-error
                 querystring[key] = params[key];
             }
         }
@@ -1759,7 +1870,7 @@ class Ml {
     }
     async validate(params, options) {
         const acceptedPath = [];
-        const acceptedBody = ['job_id', 'analysis_config', 'analysis_limits', 'data_description', 'description', 'model_plot', 'model_snapshot_retention_days', 'results_index_name'];
+        const acceptedBody = ['job_id', 'analysis_config', 'analysis_limits', 'data_description', 'description', 'model_plot', 'model_snapshot_id', 'model_snapshot_retention_days', 'results_index_name'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;