Add inference.stream_inference

Author: lcawl

compiler/src/model/utils.ts

@@ -670,7 +670,7 @@ export function hoistRequestAnnotations (
         'manage_enrich', 'manage_ilm', 'manage_index_templates', 'manage_inference', 'manage_ingest_pipelines', 'manage_logstash_pipelines',
         'manage_ml', 'manage_oidc', 'manage_own_api_key', 'manage_pipeline', 'manage_rollup', 'manage_saml',
         'manage_security', 'manage_service_account', 'manage_slm', 'manage_token', 'manage_transform', 'manage_user_profile',
-        'manage_watcher', 'monitor', 'monitor_ml', 'monitor_rollup', 'monitor_snapshot', 'monitor_text_structure',
+        'manage_watcher', 'monitor', 'monitor_inference', 'monitor_ml', 'monitor_rollup', 'monitor_snapshot', 'monitor_text_structure',
         'monitor_transform', 'monitor_watcher', 'read_ccr', 'read_ilm', 'read_pipeline', 'read_security', 'read_slm', 'transport_client'
       ]
       const values = parseCommaSeparated(value)

docs/overlays/elasticsearch-openapi-overlays.yaml

@@ -391,4 +391,12 @@ actions:
         application/json:
           examples:
             indicesLegacyPutTemplateRequestExample1:
-              $ref: "../../specification/indices/put_template/indicesPutTemplateRequestExample1.yaml"
+              $ref: "../../specification/indices/put_template/indicesPutTemplateRequestExample1.yaml"
+  - target: "$.components['requestBodies']['inference.stream_inference']"
+    description: "Add example for inference stream request"
+    update: 
+      content:
+        application/json:
+          examples:
+            streamInferenceRequestExample1:
+              $ref: "../../specification/inference/stream_inference/StreamInferenceRequestExample1.yaml"

output/openapi/elasticsearch-openapi.json

@@ -450,12 +450,6 @@
       ],
       "response": []
     },
-    "inference.stream_inference": {
-      "request": [
-        "Missing request & response"
-      ],
-      "response": []
-    },
     "ingest.delete_geoip_database": {
       "request": [
         "Request: query parameter 'master_timeout' does not exist in the json spec",

output/schema/schema.json

@@ -257,6 +257,7 @@ indices-stats,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/i
 indices-templates,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/indices-templates.html
 indices-update-settings,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/indices-update-settings.html
 infer-trained-model-deployment,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/infer-trained-model-deployment.html
+inference-api-stream,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/stream-inference-api.html
 inference-processor,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/inference-processor.html
 info-api,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/info-api.html
 ingest,https://www.elastic.co/guide/en/elasticsearch/reference/{branch}/ingest.html

output/schema/validation-errors.json

@@ -0,0 +1,57 @@
+/*
+ * Licensed to Elasticsearch B.V. under one or more contributor
+ * license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright
+ * ownership. Elasticsearch B.V. licenses this file to you under
+ * the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+import { TaskType } from '@inference/_types/TaskType'
+import { RequestBase } from '@_types/Base'
+import { Id } from '@_types/common'
+
+/**
+ * Perform streaming inference.
+ * Get real-time responses for completion tasks by delivering answers incrementally, reducing response times during computation.
+ * This API works only with the completion task type.
+ *
+ * IMPORTANT: The inference APIs enable you to use certain services, such as built-in machine learning models (ELSER, E5), models uploaded through Eland, Cohere, OpenAI, Azure, Google AI Studio, Google Vertex AI, Anthropic, Watsonx.ai, or Hugging Face. For built-in models and models uploaded through Eland, the inference APIs offer an alternative way to use and manage trained models. However, if you do not plan to use the inference APIs to use these models or if you want to use non-NLP models, use the machine learning trained model APIs.
+ *
+ * This API requires the `monitor_inference` cluster privilege (the built-in `inference_admin` and `inference_user` roles grant this privilege). You must use a client that supports streaming.
+ * @rest_spec_name inference.stream_inference
+ * @availability stack since=8.16.0 stability=stable visibility=public
+ * @cluster_privileges monitor_inference
+ * @doc_id inference-api-stream
+ */
+export interface Request extends RequestBase {
+  path_parts: {
+    /**
+     * The unique identifier for the inference endpoint.
+     */
+    inference_id: Id
+    /**
+     * The type of task that the model performs.
+     */
+    task_type?: TaskType
+  }
+  body: {
+    /**
+     * The text on which you want to perform the inference task.
+     * It can be a single string or an array.
+     *
+     * NOTE: Inference endpoints for the completion task type currently only support a single string as input.
+     */
+    input: string
+  }
+}

output/typescript/types.ts

@@ -0,0 +1,6 @@
+summary: Perform a completion task
+description: Run `POST _inference/completion/openai-completion/_stream` to perform a completion on the example question with streaming.
+# method_request: "POST _inference/completion/openai-completion/_stream"
+# type: "request"
+value:
+  input: 'What is Elastic?'

specification/_doc_ids/table.csv

@@ -0,0 +1,22 @@
+/*
+ * Licensed to Elasticsearch B.V. under one or more contributor
+ * license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright
+ * ownership. Elasticsearch B.V. licenses this file to you under
+ * the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+export class Response {
+  body: {}
+}