anyscale 0.26.69__py3-none-any.whl → 0.26.71__py3-none-any.whl

anyscale/job_queue/_private/job_queue_sdk.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 from anyscale._private.models.model_base import ResultIterator
 from anyscale._private.workload import WorkloadSDK
@@ -12,6 +12,9 @@ from anyscale.client.openapi_client.models.job_queue_sort_directive import (
 from anyscale.client.openapi_client.models.list_response_metadata import (
     ListResponseMetadata,
 )
+from anyscale.client.openapi_client.models.resource_tag_resource_type import (
+    ResourceTagResourceType,
+)
 from anyscale.client.openapi_client.models.session_state import SessionState
 from anyscale.job_queue.models import JobQueueStatus
 
@@ -28,6 +31,7 @@ class PrivateJobQueueSDK(WorkloadSDK):
         cloud: Optional[str] = None,
         project: Optional[str] = None,
         cluster_status: Optional[SessionState] = None,
+        tags_filter: Optional[Dict[str, List[str]]] = None,
         page_size: Optional[int] = None,
         max_items: Optional[int] = None,
         sorting_directives: Optional[List[JobQueueSortDirective]] = None,
@@ -69,6 +73,7 @@ class PrivateJobQueueSDK(WorkloadSDK):
                 cloud=cloud,
                 project=project,
                 cluster_status=cluster_status,
+                tags_filter=tags_filter,
                 count=page_size,
                 paging_token=token,
                 sorting_directives=sorting_directives,
@@ -112,6 +117,28 @@ class PrivateJobQueueSDK(WorkloadSDK):
 
         return _parse_decorated_jq_to_status(updated_jq)
 
+    def add_tags(
+        self,
+        *,
+        job_queue_id: Optional[str] = None,
+        name: Optional[str] = None,
+        tags: Dict[str, str],
+    ) -> None:
+        jq = self._resolve_to_job_queue_model(job_queue_id=job_queue_id, name=name)
+        assert jq.id is not None
+        self.client.upsert_resource_tags(ResourceTagResourceType.JOB_QUEUE, jq.id, tags)
+
+    def remove_tags(
+        self,
+        *,
+        job_queue_id: Optional[str] = None,
+        name: Optional[str] = None,
+        keys: List[str],
+    ) -> None:
+        jq = self._resolve_to_job_queue_model(job_queue_id=job_queue_id, name=name)
+        assert jq.id is not None
+        self.client.delete_resource_tags(ResourceTagResourceType.JOB_QUEUE, jq.id, keys)
+
     def _resolve_to_job_queue_model(
         self, *, job_queue_id: Optional[str] = None, name: Optional[str] = None,
     ) -> DecoratedJobQueue:

anyscale/job_queue/commands.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 from anyscale._private.models.model_base import ResultIterator
 from anyscale._private.sdk import sdk_command
@@ -58,6 +58,30 @@ _UPDATE_ARG_DOCSTRINGS = {
     "idle_timeout_s": "New idle timeout in seconds.",
 }
 
+_TAGS_ADD_EXAMPLE = """
+import anyscale
+
+anyscale.job_queue.add_tags(job_queue_id="jobq_abc123", tags={"team": "mlops"})
+"""
+
+_TAGS_ADD_ARG_DOCSTRINGS = {
+    "job_queue_id": "ID of the job queue to tag (alternative to name).",
+    "name": "Name of the job queue to tag (alternative to ID).",
+    "tags": "Key/value tags to upsert as a map {key: value}.",
+}
+
+_TAGS_REMOVE_EXAMPLE = """
+import anyscale
+
+anyscale.job_queue.remove_tags(job_queue_id="jobq_abc123", keys=["team"])
+"""
+
+_TAGS_REMOVE_ARG_DOCSTRINGS = {
+    "job_queue_id": "ID of the job queue to modify (alternative to name).",
+    "name": "Name of the job queue to modify (alternative to ID).",
+    "keys": "List of tag keys to remove.",
+}
+
 
 @sdk_command(
     _JOB_QUEUE_SDK_SINGLETON_KEY,
@@ -73,6 +97,7 @@ def list(  # noqa: A001
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     cluster_status: Optional[SessionState] = None,
+    tags_filter: Optional[Dict[str, List[str]]] = None,
     page_size: Optional[int] = None,
     max_items: Optional[int] = None,
     sorting_directives: Optional[List[JobQueueSortDirective]] = None,
@@ -86,6 +111,7 @@ def list(  # noqa: A001
         cloud=cloud,
         project=project,
         cluster_status=cluster_status,
+        tags_filter=tags_filter,
         page_size=page_size,
         max_items=max_items,
         sorting_directives=sorting_directives,
@@ -128,3 +154,37 @@ def update(
         max_concurrency=max_concurrency,
         idle_timeout_s=idle_timeout_s,
     )
+
+
+@sdk_command(
+    _JOB_QUEUE_SDK_SINGLETON_KEY,
+    PrivateJobQueueSDK,
+    doc_py_example=_TAGS_ADD_EXAMPLE,
+    arg_docstrings=_TAGS_ADD_ARG_DOCSTRINGS,
+)
+def add_tags(
+    *,
+    job_queue_id: Optional[str] = None,
+    name: Optional[str] = None,
+    tags: Dict[str, str],
+    _private_sdk: Optional[PrivateJobQueueSDK] = None,
+) -> None:
+    """Upsert tags for a job queue."""
+    return _private_sdk.add_tags(job_queue_id=job_queue_id, name=name, tags=tags)  # type: ignore
+
+
+@sdk_command(
+    _JOB_QUEUE_SDK_SINGLETON_KEY,
+    PrivateJobQueueSDK,
+    doc_py_example=_TAGS_REMOVE_EXAMPLE,
+    arg_docstrings=_TAGS_REMOVE_ARG_DOCSTRINGS,
+)
+def remove_tags(
+    *,
+    job_queue_id: Optional[str] = None,
+    name: Optional[str] = None,
+    keys: List[str],
+    _private_sdk: Optional[PrivateJobQueueSDK] = None,
+) -> None:
+    """Remove tags by key from a job queue."""
+    return _private_sdk.remove_tags(job_queue_id=job_queue_id, name=name, keys=keys)  # type: ignore

anyscale/sdk/anyscale_client/__init__.py

@@ -133,6 +133,7 @@ from anyscale_client.models.organization import Organization
 from anyscale_client.models.organization_response import OrganizationResponse
 from anyscale_client.models.page_query import PageQuery
 from anyscale_client.models.pause_schedule import PauseSchedule
+from anyscale_client.models.physical_resources import PhysicalResources
 from anyscale_client.models.production_job import ProductionJob
 from anyscale_client.models.production_job_config import ProductionJobConfig
 from anyscale_client.models.production_job_state_transition import ProductionJobStateTransition

anyscale/sdk/anyscale_client/api/default_api.py

@@ -2696,6 +2696,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cloud_id: The cloud id whose default cluster compute you want to fetch. If None, will use the default cloud.
+        :param str cloud_resource_id: The cloud resource ID to fetch the default cluster compute for. If None, will use the primary cloud resource of the cloud.
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -2721,6 +2722,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cloud_id: The cloud id whose default cluster compute you want to fetch. If None, will use the default cloud.
+        :param str cloud_resource_id: The cloud resource ID to fetch the default cluster compute for. If None, will use the primary cloud resource of the cloud.
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -2738,7 +2740,8 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'cloud_id'
+            'cloud_id',
+            'cloud_resource_id'
         ]
         all_params.extend(
             [
@@ -2765,6 +2768,8 @@ class DefaultApi(object):
         query_params = []
         if 'cloud_id' in local_var_params and local_var_params['cloud_id'] is not None:  # noqa: E501
             query_params.append(('cloud_id', local_var_params['cloud_id']))  # noqa: E501
+        if 'cloud_resource_id' in local_var_params and local_var_params['cloud_resource_id'] is not None:  # noqa: E501
+            query_params.append(('cloud_resource_id', local_var_params['cloud_resource_id']))  # noqa: E501
 
         header_params = {}
 
@@ -2929,6 +2934,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cloud_id: (required)
+        :param str cloud_resource_id:
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -2954,6 +2960,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cloud_id: (required)
+        :param str cloud_resource_id:
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -2971,7 +2978,8 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'cloud_id'
+            'cloud_id',
+            'cloud_resource_id'
         ]
         all_params.extend(
             [
@@ -3002,6 +3010,8 @@ class DefaultApi(object):
             path_params['cloud_id'] = local_var_params['cloud_id']  # noqa: E501
 
         query_params = []
+        if 'cloud_resource_id' in local_var_params and local_var_params['cloud_resource_id'] is not None:  # noqa: E501
+            query_params.append(('cloud_resource_id', local_var_params['cloud_resource_id']))  # noqa: E501
 
         header_params = {}
 

anyscale/sdk/anyscale_client/models/__init__.py

@@ -119,6 +119,7 @@ from anyscale_client.models.organization import Organization
 from anyscale_client.models.organization_response import OrganizationResponse
 from anyscale_client.models.page_query import PageQuery
 from anyscale_client.models.pause_schedule import PauseSchedule
+from anyscale_client.models.physical_resources import PhysicalResources
 from anyscale_client.models.production_job import ProductionJob
 from anyscale_client.models.production_job_config import ProductionJobConfig
 from anyscale_client.models.production_job_state_transition import ProductionJobStateTransition

anyscale/sdk/anyscale_client/models/apply_production_service_v2_model.py

@@ -47,7 +47,8 @@ class ApplyProductionServiceV2Model(object):
         'tracing_config': 'TracingConfig',
         'auto_complete_rollout': 'bool',
         'max_surge_percent': 'int',
-        'tags': 'dict(str, str)'
+        'tags': 'dict(str, str)',
+        'traffic_percent': 'int'
     }
 
     attribute_map = {
@@ -65,10 +66,11 @@ class ApplyProductionServiceV2Model(object):
         'tracing_config': 'tracing_config',
         'auto_complete_rollout': 'auto_complete_rollout',
         'max_surge_percent': 'max_surge_percent',
-        'tags': 'tags'
+        'tags': 'tags',
+        'traffic_percent': 'traffic_percent'
     }
 
-    def __init__(self, name=None, description=None, project_id=None, version=None, canary_percent=None, ray_serve_config=None, build_id=None, compute_config_id=None, config=None, rollout_strategy=None, ray_gcs_external_storage_config=None, tracing_config=None, auto_complete_rollout=True, max_surge_percent=None, tags=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, description=None, project_id=None, version=None, canary_percent=None, ray_serve_config=None, build_id=None, compute_config_id=None, config=None, rollout_strategy=None, ray_gcs_external_storage_config=None, tracing_config=None, auto_complete_rollout=True, max_surge_percent=None, tags=None, traffic_percent=None, local_vars_configuration=None):  # noqa: E501
         """ApplyProductionServiceV2Model - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -89,6 +91,7 @@ class ApplyProductionServiceV2Model(object):
         self._auto_complete_rollout = None
         self._max_surge_percent = None
         self._tags = None
+        self._traffic_percent = None
         self.discriminator = None
 
         self.name = name
@@ -117,6 +120,8 @@ class ApplyProductionServiceV2Model(object):
             self.max_surge_percent = max_surge_percent
         if tags is not None:
             self.tags = tags
+        if traffic_percent is not None:
+            self.traffic_percent = traffic_percent
 
     @property
     def name(self):
@@ -471,6 +476,29 @@ class ApplyProductionServiceV2Model(object):
 
         self._tags = tags
 
+    @property
+    def traffic_percent(self):
+        """Gets the traffic_percent of this ApplyProductionServiceV2Model.  # noqa: E501
+
+         Percentage of traffic forwarded to a particular service version from the ALB.  # noqa: E501
+
+        :return: The traffic_percent of this ApplyProductionServiceV2Model.  # noqa: E501
+        :rtype: int
+        """
+        return self._traffic_percent
+
+    @traffic_percent.setter
+    def traffic_percent(self, traffic_percent):
+        """Sets the traffic_percent of this ApplyProductionServiceV2Model.
+
+         Percentage of traffic forwarded to a particular service version from the ALB.  # noqa: E501
+
+        :param traffic_percent: The traffic_percent of this ApplyProductionServiceV2Model.  # noqa: E501
+        :type: int
+        """
+
+        self._traffic_percent = traffic_percent
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

anyscale/sdk/anyscale_client/models/apply_service_model.py

@@ -47,7 +47,8 @@ class ApplyServiceModel(object):
         'tracing_config': 'TracingConfig',
         'auto_complete_rollout': 'bool',
         'max_surge_percent': 'int',
-        'tags': 'dict(str, str)'
+        'tags': 'dict(str, str)',
+        'traffic_percent': 'int'
     }
 
     attribute_map = {
@@ -65,10 +66,11 @@ class ApplyServiceModel(object):
         'tracing_config': 'tracing_config',
         'auto_complete_rollout': 'auto_complete_rollout',
         'max_surge_percent': 'max_surge_percent',
-        'tags': 'tags'
+        'tags': 'tags',
+        'traffic_percent': 'traffic_percent'
     }
 
-    def __init__(self, name=None, description=None, project_id=None, version=None, canary_percent=None, ray_serve_config=None, build_id=None, compute_config_id=None, config=None, rollout_strategy=None, ray_gcs_external_storage_config=None, tracing_config=None, auto_complete_rollout=True, max_surge_percent=None, tags=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, description=None, project_id=None, version=None, canary_percent=None, ray_serve_config=None, build_id=None, compute_config_id=None, config=None, rollout_strategy=None, ray_gcs_external_storage_config=None, tracing_config=None, auto_complete_rollout=True, max_surge_percent=None, tags=None, traffic_percent=None, local_vars_configuration=None):  # noqa: E501
         """ApplyServiceModel - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -89,6 +91,7 @@ class ApplyServiceModel(object):
         self._auto_complete_rollout = None
         self._max_surge_percent = None
         self._tags = None
+        self._traffic_percent = None
         self.discriminator = None
 
         self.name = name
@@ -117,6 +120,8 @@ class ApplyServiceModel(object):
             self.max_surge_percent = max_surge_percent
         if tags is not None:
             self.tags = tags
+        if traffic_percent is not None:
+            self.traffic_percent = traffic_percent
 
     @property
     def name(self):
@@ -471,6 +476,29 @@ class ApplyServiceModel(object):
 
         self._tags = tags
 
+    @property
+    def traffic_percent(self):
+        """Gets the traffic_percent of this ApplyServiceModel.  # noqa: E501
+
+         Percentage of traffic forwarded to a particular service version from the ALB.  # noqa: E501
+
+        :return: The traffic_percent of this ApplyServiceModel.  # noqa: E501
+        :rtype: int
+        """
+        return self._traffic_percent
+
+    @traffic_percent.setter
+    def traffic_percent(self, traffic_percent):
+        """Sets the traffic_percent of this ApplyServiceModel.
+
+         Percentage of traffic forwarded to a particular service version from the ALB.  # noqa: E501
+
+        :param traffic_percent: The traffic_percent of this ApplyServiceModel.  # noqa: E501
+        :type: int
+        """
+
+        self._traffic_percent = traffic_percent
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}