API generation

This commit is contained in:
delvedor
2020-01-13 12:23:10 +01:00
parent 436f1c891d
commit 7478913d6c
5 changed files with 144 additions and 8 deletions

View File

@ -0,0 +1,81 @@
// Licensed to Elasticsearch B.V under one or more agreements.
// Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
// See the LICENSE file in the project root for more information
'use strict'
/* eslint camelcase: 0 */
/* eslint no-unused-vars: 0 */
function buildMlPutTrainedModel (opts) {
// eslint-disable-next-line no-unused-vars
const { makeRequest, ConfigurationError, handleError, snakeCaseKeys } = opts
const acceptedQuerystring = [
]
const snakeCase = {
}
/**
* Perform a ml.put_trained_model request
* TODO
*/
return function mlPutTrainedModel (params, options, callback) {
options = options || {}
if (typeof options === 'function') {
callback = options
options = {}
}
if (typeof params === 'function' || params == null) {
callback = params
params = {}
options = {}
}
// check required parameters
if (params['model_id'] == null && params['modelId'] == null) {
const err = new ConfigurationError('Missing required parameter: model_id or modelId')
return handleError(err, callback)
}
if (params['body'] == null) {
const err = new ConfigurationError('Missing required parameter: body')
return handleError(err, callback)
}
// validate headers object
if (options.headers != null && typeof options.headers !== 'object') {
const err = new ConfigurationError(`Headers should be an object, instead got: ${typeof options.headers}`)
return handleError(err, callback)
}
var warnings = []
var { method, body, modelId, model_id, ...querystring } = params
querystring = snakeCaseKeys(acceptedQuerystring, snakeCase, querystring, warnings)
var ignore = options.ignore
if (typeof ignore === 'number') {
options.ignore = [ignore]
}
var path = ''
if (method == null) method = 'PUT'
path = '/' + '_ml' + '/' + 'inference' + '/' + encodeURIComponent(model_id || modelId)
// build request object
const request = {
method,
path,
body: body || '',
querystring
}
options.warnings = warnings.length === 0 ? null : warnings
return makeRequest(request, options, callback)
}
}
module.exports = buildMlPutTrainedModel

View File

@ -360,6 +360,8 @@ function ESAPI (opts) {
putFilter: lazyLoad('ml.put_filter', opts),
put_job: lazyLoad('ml.put_job', opts),
putJob: lazyLoad('ml.put_job', opts),
put_trained_model: lazyLoad('ml.put_trained_model', opts),
putTrainedModel: lazyLoad('ml.put_trained_model', opts),
revert_model_snapshot: lazyLoad('ml.revert_model_snapshot', opts),
revertModelSnapshot: lazyLoad('ml.revert_model_snapshot', opts),
set_upgrade_mode: lazyLoad('ml.set_upgrade_mode', opts),

View File

@ -1855,6 +1855,11 @@ export interface MlPutJob<T = any> extends Generic {
body: T;
}
export interface MlPutTrainedModel<T = any> extends Generic {
model_id: string;
body: T;
}
export interface MlRevertModelSnapshot<T = any> extends Generic {
job_id: string;
snapshot_id: string;