Skip to content

Commit 5bee845

Browse files
committed
[API] Adds back inference.inference endpoint
1 parent 9ad8959 commit 5bee845

File tree

3 files changed

+109
-1
lines changed

3 files changed

+109
-1
lines changed
Lines changed: 72 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,72 @@
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

elasticsearch-api/lib/elasticsearch/api/version.rb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,6 @@
1818
module Elasticsearch
1919
module API
2020
VERSION = '9.0.0'.freeze
21-
ES_SPECIFICATION_COMMIT = '60a81659be928bfe6cec53708c7f7613555a5eaf'.freeze
21+
ES_SPECIFICATION_COMMIT = 'f2651fcb540f55100a80629192c021fd2e7a019c'.freeze
2222
end
2323
end
Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
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+
require 'spec_helper'
19+
20+
describe 'client#inference.inference' do
21+
let(:expected_args) do
22+
[
23+
'POST',
24+
'_inference/bar',
25+
{},
26+
nil,
27+
{},
28+
{ defined_params: { inference_id: 'bar' },
29+
endpoint: 'inference.inference' }
30+
]
31+
end
32+
33+
it 'performs the request' do
34+
expect(client_double.inference.inference(inference_id: 'bar')).to be_a Elasticsearch::API::Response
35+
end
36+
end

0 commit comments

Comments
 (0)