elasticsearch 8.13.1__py3-none-any.whl → 8.13.2__py3-none-any.whl

elasticsearch/_async/client/ml.py

@@ -5043,6 +5043,62 @@ class MlClient(NamespacedClient):
             path_parts=__path_parts,
         )
 
+    @_rewrite_parameters(
+        body_fields=("number_of_allocations",),
+    )
+    async def update_trained_model_deployment(
+        self,
+        *,
+        model_id: str,
+        error_trace: t.Optional[bool] = None,
+        filter_path: t.Optional[t.Union[str, t.Sequence[str]]] = None,
+        human: t.Optional[bool] = None,
+        number_of_allocations: t.Optional[int] = None,
+        pretty: t.Optional[bool] = None,
+        body: t.Optional[t.Dict[str, t.Any]] = None,
+    ) -> ObjectApiResponse[t.Any]:
+        """
+        Updates certain properties of trained model deployment.
+
+        `<https://www.elastic.co/guide/en/elasticsearch/reference/8.13/update-trained-model-deployment.html>`_
+
+        :param model_id: The unique identifier of the trained model. Currently, only
+            PyTorch models are supported.
+        :param number_of_allocations: The number of model allocations on each node where
+            the model is deployed. All allocations on a node share the same copy of the
+            model in memory but use a separate set of threads to evaluate the model.
+            Increasing this value generally increases the throughput. If this setting
+            is greater than the number of hardware threads it will automatically be changed
+            to a value less than the number of hardware threads.
+        """
+        if model_id in SKIP_IN_PATH:
+            raise ValueError("Empty value passed for parameter 'model_id'")
+        __path_parts: t.Dict[str, str] = {"model_id": _quote(model_id)}
+        __path = f'/_ml/trained_models/{__path_parts["model_id"]}/deployment/_update'
+        __query: t.Dict[str, t.Any] = {}
+        __body: t.Dict[str, t.Any] = body if body is not None else {}
+        if error_trace is not None:
+            __query["error_trace"] = error_trace
+        if filter_path is not None:
+            __query["filter_path"] = filter_path
+        if human is not None:
+            __query["human"] = human
+        if pretty is not None:
+            __query["pretty"] = pretty
+        if not __body:
+            if number_of_allocations is not None:
+                __body["number_of_allocations"] = number_of_allocations
+        __headers = {"accept": "application/json", "content-type": "application/json"}
+        return await self.perform_request(  # type: ignore[return-value]
+            "POST",
+            __path,
+            params=__query,
+            headers=__headers,
+            body=__body,
+            endpoint_id="ml.update_trained_model_deployment",
+            path_parts=__path_parts,
+        )
+
     @_rewrite_parameters()
     async def upgrade_job_snapshot(
         self,

elasticsearch/_sync/client/ml.py

@@ -5043,6 +5043,62 @@ class MlClient(NamespacedClient):
             path_parts=__path_parts,
         )
 
+    @_rewrite_parameters(
+        body_fields=("number_of_allocations",),
+    )
+    def update_trained_model_deployment(
+        self,
+        *,
+        model_id: str,
+        error_trace: t.Optional[bool] = None,
+        filter_path: t.Optional[t.Union[str, t.Sequence[str]]] = None,
+        human: t.Optional[bool] = None,
+        number_of_allocations: t.Optional[int] = None,
+        pretty: t.Optional[bool] = None,
+        body: t.Optional[t.Dict[str, t.Any]] = None,
+    ) -> ObjectApiResponse[t.Any]:
+        """
+        Updates certain properties of trained model deployment.
+
+        `<https://www.elastic.co/guide/en/elasticsearch/reference/8.13/update-trained-model-deployment.html>`_
+
+        :param model_id: The unique identifier of the trained model. Currently, only
+            PyTorch models are supported.
+        :param number_of_allocations: The number of model allocations on each node where
+            the model is deployed. All allocations on a node share the same copy of the
+            model in memory but use a separate set of threads to evaluate the model.
+            Increasing this value generally increases the throughput. If this setting
+            is greater than the number of hardware threads it will automatically be changed
+            to a value less than the number of hardware threads.
+        """
+        if model_id in SKIP_IN_PATH:
+            raise ValueError("Empty value passed for parameter 'model_id'")
+        __path_parts: t.Dict[str, str] = {"model_id": _quote(model_id)}
+        __path = f'/_ml/trained_models/{__path_parts["model_id"]}/deployment/_update'
+        __query: t.Dict[str, t.Any] = {}
+        __body: t.Dict[str, t.Any] = body if body is not None else {}
+        if error_trace is not None:
+            __query["error_trace"] = error_trace
+        if filter_path is not None:
+            __query["filter_path"] = filter_path
+        if human is not None:
+            __query["human"] = human
+        if pretty is not None:
+            __query["pretty"] = pretty
+        if not __body:
+            if number_of_allocations is not None:
+                __body["number_of_allocations"] = number_of_allocations
+        __headers = {"accept": "application/json", "content-type": "application/json"}
+        return self.perform_request(  # type: ignore[return-value]
+            "POST",
+            __path,
+            params=__query,
+            headers=__headers,
+            body=__body,
+            endpoint_id="ml.update_trained_model_deployment",
+            path_parts=__path_parts,
+        )
+
     @_rewrite_parameters()
     def upgrade_job_snapshot(
         self,

elasticsearch/_version.py

@@ -15,4 +15,4 @@
 #  specific language governing permissions and limitations
 #  under the License.
 
-__versionstr__ = "8.13.1"
+__versionstr__ = "8.13.2"

{elasticsearch-8.13.1.dist-info → elasticsearch-8.13.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: elasticsearch
-Version: 8.13.1
+Version: 8.13.2
 Summary: Python client for Elasticsearch
 Home-page: https://github.com/elastic/elasticsearch-py
 Author: Elastic Client Library Maintainers
@@ -33,7 +33,7 @@ Requires-Dist: aiohttp <4,>=3 ; extra == 'async'
 Provides-Extra: orjson
 Requires-Dist: orjson >=3 ; extra == 'orjson'
 Provides-Extra: requests
-Requires-Dist: requests <3.0.0,>=2.4.0 ; extra == 'requests'
+Requires-Dist: requests !=2.32.2,<3.0.0,>=2.4.0 ; extra == 'requests'
 Provides-Extra: vectorstore_mmr
 Requires-Dist: numpy >=1 ; extra == 'vectorstore_mmr'
 Requires-Dist: simsimd >=3 ; extra == 'vectorstore_mmr'

{elasticsearch-8.13.1.dist-info → elasticsearch-8.13.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 elasticsearch/__init__.py,sha256=w5YnO16zjOi6loGJ8caUgSXsj3b-Y8OfF0BIddP2BiE,3289
 elasticsearch/_otel.py,sha256=9nMvDwSz_BephEQmkETCmEdA_6YNtj_CJQyxIwOzEGo,3406
 elasticsearch/_utils.py,sha256=Vr_aNG5ddxInE1PgDpCXMYpXBTNUFM9nYrgbw-cjeCc,1419
-elasticsearch/_version.py,sha256=DXfrNbuYRJqhJULnvIrY294V7ZIHYCd3oa4diFaiFbk,814
+elasticsearch/_version.py,sha256=0P-o-HdbMQa0vGhTZYE8ZkuZ9v7zXoV3M6M0wUoqfrs,814
 elasticsearch/client.py,sha256=p4naAgAPs_B9laux-aC-bpZNvbu9fvrpicBvLRlp_yw,5120
 elasticsearch/compat.py,sha256=hL3mtqVxWwxeiFbNGADva5XruAwK-A6xcu-vrpnDXFs,2657
 elasticsearch/exceptions.py,sha256=HHqidMTlLXRcNbo4gjsbhHsLfiV4JGg7tx1lXQndG98,4025
@@ -31,7 +31,7 @@ elasticsearch/_async/client/ingest.py,sha256=5V8fur9z4Vkv4p-3geOfc1ek5N8qJ6CvWbd
 elasticsearch/_async/client/license.py,sha256=kuqW-yfOVdsDHh5puu7_tD_LN7JU0Axt0AtE5ttx8jc,12100
 elasticsearch/_async/client/logstash.py,sha256=JrvbrTi9sCO-GzOe3lBWcaIOPkYFqzenCkYw_B6JW7Q,6127
 elasticsearch/_async/client/migration.py,sha256=It0NTjXaN-UW8keaNURglL4rYbOmAvTbHD7QlFBOJg4,5331
-elasticsearch/_async/client/ml.py,sha256=fklo9AmgLQjtzvjXy5PgVBL9vQgdA_z-N3DM5tMJRA8,238505
+elasticsearch/_async/client/ml.py,sha256=tckls24jxPSOb_OA3f6plc0DDLfKhwddhpAoVfyr0gU,241082
 elasticsearch/_async/client/monitoring.py,sha256=ZQSiQ8AXfu_8_--cHOpfhjpsQkQZekmBlPiOLAYotpw,3822
 elasticsearch/_async/client/nodes.py,sha256=xmLfm0cPsQSjH_WKyzGKHI6IhgT9gKhj3XrsJdTXL5w,23723
 elasticsearch/_async/client/query_ruleset.py,sha256=vBgpKRYL0X7giU05Vw6-pXre10W81zGoALWL_XAnkRM,7606
@@ -73,7 +73,7 @@ elasticsearch/_sync/client/ingest.py,sha256=UY-x1MXjnD2LkbX8O2qa5Ag6eD4lzf4LeZO8
 elasticsearch/_sync/client/license.py,sha256=tEc366OvzPh9OFbgCMAo8dyw7wjajIcNQyfAX-uin3I,12016
 elasticsearch/_sync/client/logstash.py,sha256=2Mpd3y-wNR3H7wsFW2riiB6bbobctNoMW22VOvDbN2s,6091
 elasticsearch/_sync/client/migration.py,sha256=dJ27K1SLcJn4XyBvyPFVocjxHtsAyf2vuOQl52uxK64,5295
-elasticsearch/_sync/client/ml.py,sha256=inanWjUHJK25cEjpo_bZ0bdbw8tZGgmCkqmK-Z7P3AE,237641
+elasticsearch/_sync/client/ml.py,sha256=7YgW9g5MQ4KUhI05X_mp7zDHwmRhDAXc0UahK_7qmwM,240206
 elasticsearch/_sync/client/monitoring.py,sha256=Wh2RTfFOOm5ssbdGv0_gSDS2og1L8ATJamLuUuNrBYU,3810
 elasticsearch/_sync/client/nodes.py,sha256=YloTg5V90lyGeoY_XU_2X2aeLHOpPss1g7Kp0hR2EF4,23639
 elasticsearch/_sync/client/query_ruleset.py,sha256=dRzaVZVYLtlRi6PJVGQTPXCKCJuuf6fkU6wug-zmmXs,7558
@@ -108,9 +108,9 @@ elasticsearch/helpers/vectorstore/_sync/_utils.py,sha256=5pdvNS5XC3wqShjliW9Njl9
 elasticsearch/helpers/vectorstore/_sync/embedding_service.py,sha256=sAw_WKUcmyqOOJRqnNesZCzn7ZyA91v4NvvQszHIWJ8,3582
 elasticsearch/helpers/vectorstore/_sync/strategies.py,sha256=0Q1zoOrO51S6HXjXwkAePPVtCUGQz5lKN9NyRCso-GU,15220
 elasticsearch/helpers/vectorstore/_sync/vectorstore.py,sha256=MqTDeHyG4NS9gInxwljh4J0QVvS499SajQ22fe1UpLs,14970
-elasticsearch-8.13.1.dist-info/LICENSE,sha256=XfKg2H1sVi8OoRxoisUlMqoo10TKvHmU_wU39ks7MyA,10143
-elasticsearch-8.13.1.dist-info/METADATA,sha256=duPLn_1q7QvVNyqQp65ChUjtccEl_NHYT8gyA_Q-CqQ,6455
-elasticsearch-8.13.1.dist-info/NOTICE,sha256=t4IjKAJ_G-0hYaL4AH16CVS_xDel8UXrJVK6x7JDaGA,61
-elasticsearch-8.13.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-elasticsearch-8.13.1.dist-info/top_level.txt,sha256=Jp2bLWq49skvCN4YCZsg1Hfn_NDLgleC-x-Bn01_HgM,14
-elasticsearch-8.13.1.dist-info/RECORD,,
+elasticsearch-8.13.2.dist-info/LICENSE,sha256=XfKg2H1sVi8OoRxoisUlMqoo10TKvHmU_wU39ks7MyA,10143
+elasticsearch-8.13.2.dist-info/METADATA,sha256=Eex24ZRMP8r810PL3irLjbwO0c_KeWgszymGV2Q6k8M,6464
+elasticsearch-8.13.2.dist-info/NOTICE,sha256=t4IjKAJ_G-0hYaL4AH16CVS_xDel8UXrJVK6x7JDaGA,61
+elasticsearch-8.13.2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+elasticsearch-8.13.2.dist-info/top_level.txt,sha256=Jp2bLWq49skvCN4YCZsg1Hfn_NDLgleC-x-Bn01_HgM,14
+elasticsearch-8.13.2.dist-info/RECORD,,