| 
 | 1 | +# Licensed to Elasticsearch B.V. under one or more contributor  | 
 | 2 | +# license agreements. See the NOTICE file distributed with  | 
 | 3 | +# this work for additional information regarding copyright  | 
 | 4 | +# ownership. Elasticsearch B.V. licenses this file to you under  | 
 | 5 | +# the Apache License, Version 2.0 (the "License"); you may  | 
 | 6 | +# not use this file except in compliance with the License.  | 
 | 7 | +# You may obtain a copy of the License at  | 
 | 8 | +#  | 
 | 9 | +#   http://www.apache.org/licenses/LICENSE-2.0  | 
 | 10 | +#  | 
 | 11 | +# Unless required by applicable law or agreed to in writing,  | 
 | 12 | +# software distributed under the License is distributed on an  | 
 | 13 | +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY  | 
 | 14 | +# KIND, either express or implied.  See the License for the  | 
 | 15 | +# specific language governing permissions and limitations  | 
 | 16 | +# under the License.  | 
 | 17 | +#  | 
 | 18 | +# This code was automatically generated from the Elasticsearch Specification  | 
 | 19 | +# See https://github.com/elastic/elasticsearch-specification  | 
 | 20 | +# See Elasticsearch::ES_SPECIFICATION_COMMIT for commit hash.  | 
 | 21 | +module Elasticsearch  | 
 | 22 | +  module API  | 
 | 23 | +    module Inference  | 
 | 24 | +      module Actions  | 
 | 25 | +        # Perform inference on the service.  | 
 | 26 | +        # This API enables you to use machine learning models to perform specific tasks on data that you provide as an input.  | 
 | 27 | +        # It returns a response with the results of the tasks.  | 
 | 28 | +        # The inference endpoint you use can perform one specific task that has been defined when the endpoint was created with the create inference API.  | 
 | 29 | +        #  | 
 | 30 | +        # @option arguments [String] :task_type The type of inference task that the model performs.  | 
 | 31 | +        # @option arguments [String] :inference_id The unique identifier for the inference endpoint. (*Required*)  | 
 | 32 | +        # @option arguments [Time] :timeout The amount of time to wait for the inference request to complete. Server default: 30s.  | 
 | 33 | +        # @option arguments [Hash] :headers Custom HTTP headers  | 
 | 34 | +        # @option arguments [Hash] :body request body  | 
 | 35 | +        #  | 
 | 36 | +        # @see https://www.elastic.co/docs/api/doc/elasticsearch/operation/operation-inference-inference  | 
 | 37 | +        #  | 
 | 38 | +        def inference(arguments = {})  | 
 | 39 | +          request_opts = { endpoint: arguments[:endpoint] || 'inference.inference' }  | 
 | 40 | + | 
 | 41 | +          defined_params = [:inference_id, :task_type].each_with_object({}) do |variable, set_variables|  | 
 | 42 | +            set_variables[variable] = arguments[variable] if arguments.key?(variable)  | 
 | 43 | +          end  | 
 | 44 | +          request_opts[:defined_params] = defined_params unless defined_params.empty?  | 
 | 45 | + | 
 | 46 | +          raise ArgumentError, "Required argument 'inference_id' missing" unless arguments[:inference_id]  | 
 | 47 | + | 
 | 48 | +          arguments = arguments.clone  | 
 | 49 | +          headers = arguments.delete(:headers) || {}  | 
 | 50 | + | 
 | 51 | +          body = arguments.delete(:body)  | 
 | 52 | + | 
 | 53 | +          _task_type = arguments.delete(:task_type)  | 
 | 54 | + | 
 | 55 | +          _inference_id = arguments.delete(:inference_id)  | 
 | 56 | + | 
 | 57 | +          method = Elasticsearch::API::HTTP_POST  | 
 | 58 | +          path   = if _task_type && _inference_id  | 
 | 59 | +                     "_inference/#{Utils.listify(_task_type)}/#{Utils.listify(_inference_id)}"  | 
 | 60 | +                   else  | 
 | 61 | +                     "_inference/#{Utils.listify(_inference_id)}"  | 
 | 62 | +                   end  | 
 | 63 | +          params = Utils.process_params(arguments)  | 
 | 64 | + | 
 | 65 | +          Elasticsearch::API::Response.new(  | 
 | 66 | +            perform_request(method, path, params, body, headers, request_opts)  | 
 | 67 | +          )  | 
 | 68 | +        end  | 
 | 69 | +      end  | 
 | 70 | +    end  | 
 | 71 | +  end  | 
 | 72 | +end  | 
0 commit comments