"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.updateDeploymentParamsSchema = exports.threadingParamsSchema = exports.stopDeploymentSchema = exports.putTrainedModelQuerySchema = exports.pipelineSimulateBody = exports.pipelineDocs = exports.optionalModelIdSchema = exports.modelIdSchema = exports.modelAndDeploymentIdSchema = exports.inferTrainedModelQuery = exports.inferTrainedModelBody = exports.getInferenceQuerySchema = exports.deleteTrainedModelQuerySchema = exports.createIngestPipelineSchema = void 0;
var _configSchema = require("@kbn/config-schema");
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0; you may not use this file except in compliance with the Elastic License
 * 2.0.
 */

const modelIdSchema = _configSchema.schema.object({
  /**
   * Model ID
   */
  modelId: _configSchema.schema.string()
});
exports.modelIdSchema = modelIdSchema;
const modelAndDeploymentIdSchema = _configSchema.schema.object({
  /**
   * Model ID
   */
  modelId: _configSchema.schema.string(),
  /**
   * Deployment ID
   */
  deploymentId: _configSchema.schema.string()
});
exports.modelAndDeploymentIdSchema = modelAndDeploymentIdSchema;
const threadingParamsSchema = _configSchema.schema.maybe(_configSchema.schema.object({
  number_of_allocations: _configSchema.schema.number(),
  threads_per_allocation: _configSchema.schema.number(),
  priority: _configSchema.schema.oneOf([_configSchema.schema.literal('low'), _configSchema.schema.literal('normal')]),
  deployment_id: _configSchema.schema.maybe(_configSchema.schema.string())
}));
exports.threadingParamsSchema = threadingParamsSchema;
const updateDeploymentParamsSchema = _configSchema.schema.object({
  number_of_allocations: _configSchema.schema.number()
});
exports.updateDeploymentParamsSchema = updateDeploymentParamsSchema;
const optionalModelIdSchema = _configSchema.schema.object({
  /**
   * Model ID
   */
  modelId: _configSchema.schema.maybe(_configSchema.schema.string())
});
exports.optionalModelIdSchema = optionalModelIdSchema;
const getInferenceQuerySchema = _configSchema.schema.object({
  size: _configSchema.schema.maybe(_configSchema.schema.string()),
  with_pipelines: _configSchema.schema.maybe(_configSchema.schema.string()),
  include: _configSchema.schema.maybe(_configSchema.schema.string())
});
exports.getInferenceQuerySchema = getInferenceQuerySchema;
const putTrainedModelQuerySchema = _configSchema.schema.object({
  defer_definition_decompression: _configSchema.schema.maybe(_configSchema.schema.boolean())
});
exports.putTrainedModelQuerySchema = putTrainedModelQuerySchema;
const inferTrainedModelQuery = _configSchema.schema.object({
  timeout: _configSchema.schema.maybe(_configSchema.schema.string())
});
exports.inferTrainedModelQuery = inferTrainedModelQuery;
const inferTrainedModelBody = _configSchema.schema.object({
  docs: _configSchema.schema.any(),
  inference_config: _configSchema.schema.maybe(_configSchema.schema.any())
});
exports.inferTrainedModelBody = inferTrainedModelBody;
const pipelineSimulateBody = _configSchema.schema.object({
  pipeline: _configSchema.schema.any(),
  docs: _configSchema.schema.arrayOf(_configSchema.schema.any())
});
exports.pipelineSimulateBody = pipelineSimulateBody;
const pipelineDocs = _configSchema.schema.arrayOf(_configSchema.schema.string());
exports.pipelineDocs = pipelineDocs;
const stopDeploymentSchema = _configSchema.schema.object({
  modelId: _configSchema.schema.string(),
  /** force stop */
  force: _configSchema.schema.maybe(_configSchema.schema.boolean())
});
exports.stopDeploymentSchema = stopDeploymentSchema;
const deleteTrainedModelQuerySchema = _configSchema.schema.object({
  with_pipelines: _configSchema.schema.maybe(_configSchema.schema.boolean({
    defaultValue: false
  })),
  force: _configSchema.schema.maybe(_configSchema.schema.boolean({
    defaultValue: false
  }))
});
exports.deleteTrainedModelQuerySchema = deleteTrainedModelQuerySchema;
const createIngestPipelineSchema = _configSchema.schema.object({
  pipelineName: _configSchema.schema.string(),
  pipeline: _configSchema.schema.maybe(_configSchema.schema.object({
    processors: _configSchema.schema.arrayOf(_configSchema.schema.any()),
    description: _configSchema.schema.maybe(_configSchema.schema.string())
  }))
});
exports.createIngestPipelineSchema = createIngestPipelineSchema;