@elastic/elasticsearch 8.17.1 → 8.18.1

package/lib/api/api/inference.js

@@ -28,6 +28,73 @@ class Inference {
         });
         this.transport = transport;
     }
+    async chatCompletionUnified(params, options) {
+        var _a;
+        const acceptedPath = ['inference_id'];
+        const acceptedBody = ['chat_completion_request'];
+        const querystring = {};
+        // @ts-expect-error
+        let body = (_a = params.body) !== null && _a !== void 0 ? _a : undefined;
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                // @ts-expect-error
+                body = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'POST';
+        const path = `/_inference/chat_completion/${encodeURIComponent(params.inference_id.toString())}/_stream`;
+        const meta = {
+            name: 'inference.chat_completion_unified',
+            pathParts: {
+                inference_id: params.inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async completion(params, options) {
+        const acceptedPath = ['inference_id'];
+        const acceptedBody = ['input', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'POST';
+        const path = `/_inference/completion/${encodeURIComponent(params.inference_id.toString())}`;
+        const meta = {
+            name: 'inference.completion',
+            pathParts: {
+                inference_id: params.inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
     async delete(params, options) {
         const acceptedPath = ['task_type', 'inference_id'];
         const querystring = {};
@@ -182,9 +249,9 @@ class Inference {
         };
         return await this.transport.request({ path, method, querystring, body, meta }, options);
     }
-    async streamInference(params, options) {
-        const acceptedPath = ['inference_id', 'task_type'];
-        const acceptedBody = ['input'];
+    async putAlibabacloud(params, options) {
+        const acceptedPath = ['task_type', 'alibabacloud_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
         const querystring = {};
         // @ts-expect-error
         const userBody = params === null || params === void 0 ? void 0 : params.body;
@@ -209,36 +276,73 @@ class Inference {
                 querystring[key] = params[key];
             }
         }
-        let method = '';
-        let path = '';
-        if (params.task_type != null && params.inference_id != null) {
-            method = 'POST';
-            path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.inference_id.toString())}/_stream`;
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.alibabacloud_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_alibabacloud',
+            pathParts: {
+                task_type: params.task_type,
+                alibabacloud_inference_id: params.alibabacloud_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putAmazonbedrock(params, options) {
+        const acceptedPath = ['task_type', 'amazonbedrock_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
         }
         else {
-            method = 'POST';
-            path = `/_inference/${encodeURIComponent(params.inference_id.toString())}/_stream`;
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
         }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.amazonbedrock_inference_id.toString())}`;
         const meta = {
-            name: 'inference.stream_inference',
+            name: 'inference.put_amazonbedrock',
             pathParts: {
-                inference_id: params.inference_id,
-                task_type: params.task_type
+                task_type: params.task_type,
+                amazonbedrock_inference_id: params.amazonbedrock_inference_id
             }
         };
         return await this.transport.request({ path, method, querystring, body, meta }, options);
     }
-    async update(params, options) {
-        var _a;
-        const acceptedPath = ['inference_id', 'task_type'];
-        const acceptedBody = ['inference_config'];
+    async putAnthropic(params, options) {
+        const acceptedPath = ['task_type', 'anthropic_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
         const querystring = {};
         // @ts-expect-error
-        let body = (_a = params.body) !== null && _a !== void 0 ? _a : undefined;
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
         for (const key in params) {
             if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
                 // @ts-expect-error
-                body = params[key];
+                body[key] = params[key];
             }
             else if (acceptedPath.includes(key)) {
                 continue;
@@ -248,14 +352,687 @@ class Inference {
                 querystring[key] = params[key];
             }
         }
-        let method = '';
-        let path = '';
-        if (params.task_type != null && params.inference_id != null) {
-            method = 'POST';
-            path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.inference_id.toString())}/_update`;
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.anthropic_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_anthropic',
+            pathParts: {
+                task_type: params.task_type,
+                anthropic_inference_id: params.anthropic_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putAzureaistudio(params, options) {
+        const acceptedPath = ['task_type', 'azureaistudio_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
         }
         else {
-            method = 'POST';
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.azureaistudio_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_azureaistudio',
+            pathParts: {
+                task_type: params.task_type,
+                azureaistudio_inference_id: params.azureaistudio_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putAzureopenai(params, options) {
+        const acceptedPath = ['task_type', 'azureopenai_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.azureopenai_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_azureopenai',
+            pathParts: {
+                task_type: params.task_type,
+                azureopenai_inference_id: params.azureopenai_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putCohere(params, options) {
+        const acceptedPath = ['task_type', 'cohere_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.cohere_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_cohere',
+            pathParts: {
+                task_type: params.task_type,
+                cohere_inference_id: params.cohere_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putElasticsearch(params, options) {
+        const acceptedPath = ['task_type', 'elasticsearch_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.elasticsearch_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_elasticsearch',
+            pathParts: {
+                task_type: params.task_type,
+                elasticsearch_inference_id: params.elasticsearch_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putElser(params, options) {
+        const acceptedPath = ['task_type', 'elser_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.elser_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_elser',
+            pathParts: {
+                task_type: params.task_type,
+                elser_inference_id: params.elser_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putGoogleaistudio(params, options) {
+        const acceptedPath = ['task_type', 'googleaistudio_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.googleaistudio_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_googleaistudio',
+            pathParts: {
+                task_type: params.task_type,
+                googleaistudio_inference_id: params.googleaistudio_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putGooglevertexai(params, options) {
+        const acceptedPath = ['task_type', 'googlevertexai_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.googlevertexai_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_googlevertexai',
+            pathParts: {
+                task_type: params.task_type,
+                googlevertexai_inference_id: params.googlevertexai_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putHuggingFace(params, options) {
+        const acceptedPath = ['task_type', 'huggingface_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.huggingface_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_hugging_face',
+            pathParts: {
+                task_type: params.task_type,
+                huggingface_inference_id: params.huggingface_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putJinaai(params, options) {
+        const acceptedPath = ['task_type', 'jinaai_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.jinaai_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_jinaai',
+            pathParts: {
+                task_type: params.task_type,
+                jinaai_inference_id: params.jinaai_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putMistral(params, options) {
+        const acceptedPath = ['task_type', 'mistral_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.mistral_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_mistral',
+            pathParts: {
+                task_type: params.task_type,
+                mistral_inference_id: params.mistral_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putOpenai(params, options) {
+        const acceptedPath = ['task_type', 'openai_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.openai_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_openai',
+            pathParts: {
+                task_type: params.task_type,
+                openai_inference_id: params.openai_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putVoyageai(params, options) {
+        const acceptedPath = ['task_type', 'voyageai_inference_id'];
+        const acceptedBody = ['chunking_settings', 'service', 'service_settings', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.voyageai_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_voyageai',
+            pathParts: {
+                task_type: params.task_type,
+                voyageai_inference_id: params.voyageai_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async putWatsonx(params, options) {
+        const acceptedPath = ['task_type', 'watsonx_inference_id'];
+        const acceptedBody = ['service', 'service_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'PUT';
+        const path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.watsonx_inference_id.toString())}`;
+        const meta = {
+            name: 'inference.put_watsonx',
+            pathParts: {
+                task_type: params.task_type,
+                watsonx_inference_id: params.watsonx_inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async rerank(params, options) {
+        const acceptedPath = ['inference_id'];
+        const acceptedBody = ['query', 'input', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'POST';
+        const path = `/_inference/rerank/${encodeURIComponent(params.inference_id.toString())}`;
+        const meta = {
+            name: 'inference.rerank',
+            pathParts: {
+                inference_id: params.inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async sparseEmbedding(params, options) {
+        const acceptedPath = ['inference_id'];
+        const acceptedBody = ['input', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'POST';
+        const path = `/_inference/sparse_embedding/${encodeURIComponent(params.inference_id.toString())}`;
+        const meta = {
+            name: 'inference.sparse_embedding',
+            pathParts: {
+                inference_id: params.inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async streamCompletion(params, options) {
+        const acceptedPath = ['inference_id'];
+        const acceptedBody = ['input', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'POST';
+        const path = `/_inference/completion/${encodeURIComponent(params.inference_id.toString())}/_stream`;
+        const meta = {
+            name: 'inference.stream_completion',
+            pathParts: {
+                inference_id: params.inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async textEmbedding(params, options) {
+        const acceptedPath = ['inference_id'];
+        const acceptedBody = ['input', 'task_settings'];
+        const querystring = {};
+        // @ts-expect-error
+        const userBody = params === null || params === void 0 ? void 0 : params.body;
+        let body;
+        if (typeof userBody === 'string') {
+            body = userBody;
+        }
+        else {
+            body = userBody != null ? { ...userBody } : undefined;
+        }
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                body = body !== null && body !== void 0 ? body : {};
+                // @ts-expect-error
+                body[key] = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        const method = 'POST';
+        const path = `/_inference/text_embedding/${encodeURIComponent(params.inference_id.toString())}`;
+        const meta = {
+            name: 'inference.text_embedding',
+            pathParts: {
+                inference_id: params.inference_id
+            }
+        };
+        return await this.transport.request({ path, method, querystring, body, meta }, options);
+    }
+    async update(params, options) {
+        var _a;
+        const acceptedPath = ['inference_id', 'task_type'];
+        const acceptedBody = ['inference_config'];
+        const querystring = {};
+        // @ts-expect-error
+        let body = (_a = params.body) !== null && _a !== void 0 ? _a : undefined;
+        for (const key in params) {
+            if (acceptedBody.includes(key)) {
+                // @ts-expect-error
+                body = params[key];
+            }
+            else if (acceptedPath.includes(key)) {
+                continue;
+            }
+            else if (key !== 'body') {
+                // @ts-expect-error
+                querystring[key] = params[key];
+            }
+        }
+        let method = '';
+        let path = '';
+        if (params.task_type != null && params.inference_id != null) {
+            method = 'PUT';
+            path = `/_inference/${encodeURIComponent(params.task_type.toString())}/${encodeURIComponent(params.inference_id.toString())}/_update`;
+        }
+        else {
+            method = 'PUT';
             path = `/_inference/${encodeURIComponent(params.inference_id.toString())}/_update`;
         }
         const meta = {