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

anyscale/sdk/anyscale_client/models/worker_node_type.py

@@ -36,6 +36,7 @@ class WorkerNodeType(object):
         'name': 'str',
         'instance_type': 'str',
         'resources': 'Resources',
+        'physical_resources': 'PhysicalResources',
         'labels': 'dict(str, str)',
         'aws_advanced_configurations_json': 'object',
         'gcp_advanced_configurations_json': 'object',
@@ -51,6 +52,7 @@ class WorkerNodeType(object):
         'name': 'name',
         'instance_type': 'instance_type',
         'resources': 'resources',
+        'physical_resources': 'physical_resources',
         'labels': 'labels',
         'aws_advanced_configurations_json': 'aws_advanced_configurations_json',
         'gcp_advanced_configurations_json': 'gcp_advanced_configurations_json',
@@ -62,7 +64,7 @@ class WorkerNodeType(object):
         'fallback_to_ondemand': 'fallback_to_ondemand'
     }
 
-    def __init__(self, name=None, instance_type=None, resources=None, labels=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, min_workers=None, max_workers=None, use_spot=False, fallback_to_ondemand=False, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, instance_type=None, resources=None, physical_resources=None, labels=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, min_workers=None, max_workers=None, use_spot=False, fallback_to_ondemand=False, local_vars_configuration=None):  # noqa: E501
         """WorkerNodeType - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -71,6 +73,7 @@ class WorkerNodeType(object):
         self._name = None
         self._instance_type = None
         self._resources = None
+        self._physical_resources = None
         self._labels = None
         self._aws_advanced_configurations_json = None
         self._gcp_advanced_configurations_json = None
@@ -86,6 +89,8 @@ class WorkerNodeType(object):
         self.instance_type = instance_type
         if resources is not None:
             self.resources = resources
+        if physical_resources is not None:
+            self.physical_resources = physical_resources
         if labels is not None:
             self.labels = labels
         if aws_advanced_configurations_json is not None:
@@ -178,6 +183,29 @@ class WorkerNodeType(object):
 
         self._resources = resources
 
+    @property
+    def physical_resources(self):
+        """Gets the physical_resources of this WorkerNodeType.  # noqa: E501
+
+        Physical resources for compute node type which specifies the actual CPU, memory, and GPU resources that should be allocated for this node type.  # noqa: E501
+
+        :return: The physical_resources of this WorkerNodeType.  # noqa: E501
+        :rtype: PhysicalResources
+        """
+        return self._physical_resources
+
+    @physical_resources.setter
+    def physical_resources(self, physical_resources):
+        """Sets the physical_resources of this WorkerNodeType.
+
+        Physical resources for compute node type which specifies the actual CPU, memory, and GPU resources that should be allocated for this node type.  # noqa: E501
+
+        :param physical_resources: The physical_resources of this WorkerNodeType.  # noqa: E501
+        :type: PhysicalResources
+        """
+
+        self._physical_resources = physical_resources
+
     @property
     def labels(self):
         """Gets the labels of this WorkerNodeType.  # noqa: E501

anyscale/service/__init__.py

@@ -22,14 +22,20 @@ from anyscale.service.commands import (
     _ROLLBACK_EXAMPLE,
     _STATUS_ARG_DOCSTRINGS,
     _STATUS_EXAMPLE,
+    _TAGS_ADD_ARG_DOCSTRINGS,
+    _TAGS_ADD_EXAMPLE,
+    _TAGS_REMOVE_ARG_DOCSTRINGS,
+    _TAGS_REMOVE_EXAMPLE,
     _TERMINATE_ARG_DOCSTRINGS,
     _TERMINATE_EXAMPLE,
     _WAIT_ARG_DOCSTRINGS,
     _WAIT_EXAMPLE,
+    add_tags,
     archive,
     delete,
     deploy,
     list,
+    remove_tags,
     rollback,
     status,
     terminate,
@@ -181,6 +187,7 @@ class ServiceSDK:
         # Filters
         name: Optional[str] = None,
         state_filter: Optional[Union[List[ServiceState], List[str]]] = None,
+        tags_filter: Optional[Dict[str, List[str]]] = None,
         creator_id: Optional[str] = None,
         cloud: Optional[str] = None,
         project: Optional[str] = None,
@@ -205,6 +212,7 @@ class ServiceSDK:
             service_id=service_id,
             name=name,
             state_filter=state_filter,
+            tags_filter=tags_filter,
             creator_id=creator_id,
             cloud=cloud,
             project=project,
@@ -270,3 +278,35 @@ class ServiceSDK:
             mode=mode,
             max_lines=max_lines,
         )
+
+    @sdk_docs(doc_py_example=_TAGS_ADD_EXAMPLE, arg_docstrings=_TAGS_ADD_ARG_DOCSTRINGS)
+    def add_tags(  # noqa: F811
+        self,
+        *,
+        id: Optional[str] = None,  # noqa: A002
+        name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        project: Optional[str] = None,
+        tags: Dict[str, str],
+    ) -> None:
+        """Upsert (add/update) tag key/value pairs for a service."""
+        return self._private_sdk.add_tags(
+            id=id, name=name, cloud=cloud, project=project, tags=tags
+        )
+
+    @sdk_docs(
+        doc_py_example=_TAGS_REMOVE_EXAMPLE, arg_docstrings=_TAGS_REMOVE_ARG_DOCSTRINGS
+    )
+    def remove_tags(  # noqa: F811
+        self,
+        *,
+        id: Optional[str] = None,  # noqa: A002
+        name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        project: Optional[str] = None,
+        keys: List[str],
+    ) -> None:
+        """Remove tags by key from a service."""
+        return self._private_sdk.remove_tags(
+            id=id, name=name, cloud=cloud, project=project, keys=keys
+        )

anyscale/service/_private/service_sdk.py

@@ -20,6 +20,9 @@ from anyscale.client.openapi_client.models.decoratedlistserviceapimodel_list_res
 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.compute_config.models import ComputeConfig
 from anyscale.sdk.anyscale_client.models import (
     AccessConfig,
@@ -296,6 +299,7 @@ class PrivateServiceSDK(WorkloadSDK):
             ),
             ray_gcs_external_storage_config=existing_config.ray_gcs_external_storage_config,
             tracing_config=existing_config.tracing_config,
+            tags=getattr(config, "tags", None),
         )
 
     def _build_apply_service_model_for_rollout(  # noqa: PLR0912
@@ -431,6 +435,7 @@ class PrivateServiceSDK(WorkloadSDK):
             ray_gcs_external_storage_config=ray_gcs_external_storage_config,
             tracing_config=tracing_config,
             traffic_percent=traffic_percent,
+            tags=getattr(config, "tags", None),
         )
 
     def _build_apply_service_model_for_multi_version(
@@ -454,13 +459,20 @@ class PrivateServiceSDK(WorkloadSDK):
         if existing_service is not None:
             existing_versions = {
                 v.version: v
-                for v in self.client.get_service_versions(existing_service.id)
+                for v in self.client.get_service_versions(
+                    existing_service.id, read_all_versions=True
+                )
             }
 
         service_versions: List[ApplyProductionServiceV2Model] = []
         for _, (config, version) in version_service_map.items():
             version_name = version["name"]
             traffic_percent = version["traffic_percent"]
+            if traffic_percent == 0:
+                self.logger.warning(
+                    f"Setting traffic for version {version_name} to 0 will terminate it."
+                )
+
             # If config is None, it means the version is already deployed, and the user is likely to be only updating the traffic/capacity percent without serve config update.
             if config is None:
                 if version_name not in existing_versions:
@@ -945,14 +957,18 @@ class PrivateServiceSDK(WorkloadSDK):
                 sampling_ratio=model.tracing_config.sampling_ratio,
             )
 
+        version_name = model.version or self._get_user_facing_service_version_id(model)
+
         return ServiceVersionStatus(
             id=self._get_user_facing_service_version_id(model),
+            name=version_name,
             created_at=model.created_at,
             state=model.current_state,
             # NOTE(edoakes): there is also a "current_weight" field but it does not match the UI.
             weight=model.weight,
             config=ServiceConfig(
                 name=service_name,
+                version_name=version_name,
                 applications=model.ray_serve_config["applications"],
                 image_uri=str(image_uri),
                 compute_config=compute_config,
@@ -996,35 +1012,63 @@ class PrivateServiceSDK(WorkloadSDK):
         # which means that the per-version `query_auth_token_enabled` field will lie if
         # it's changed.
         query_auth_token_enabled = model.auth_token is not None
+        all_versions = None
+        primary_version = None
+        canary_version = None
+        project_name = None
 
-        primary_version_task = None
-        if model.primary_version is not None:
-            primary_version_task = asyncio.create_task(
-                self._service_version_model_to_status_async(
-                    model.primary_version,
-                    service_name=model.name,
-                    project_id=model.project_id,
-                    query_auth_token_enabled=query_auth_token_enabled,
+        if model.is_multi_version:
+            # For multi-version services, format all versions
+            version_tasks = [
+                asyncio.create_task(
+                    self._service_version_model_to_status_async(
+                        version,
+                        service_name=model.name,
+                        project_id=model.project_id,
+                        query_auth_token_enabled=query_auth_token_enabled,
+                    )
                 )
-            )
+                for version in model.versions
+            ]
+            all_versions = await asyncio.gather(*version_tasks)
+
+            if (
+                all_versions
+                and all_versions[0]
+                and isinstance(all_versions[0].config, ServiceConfig)
+            ):
+                project_name = all_versions[0].config.project
 
-        canary_version_task = None
-        if model.canary_version is not None:
-            canary_version_task = asyncio.create_task(
-                self._service_version_model_to_status_async(
-                    model.canary_version,
-                    service_name=model.name,
-                    project_id=model.project_id,
-                    query_auth_token_enabled=query_auth_token_enabled,
+        else:
+            primary_version_task = None
+            if model.primary_version is not None:
+                primary_version_task = asyncio.create_task(
+                    self._service_version_model_to_status_async(
+                        model.primary_version,
+                        service_name=model.name,
+                        project_id=model.project_id,
+                        query_auth_token_enabled=query_auth_token_enabled,
+                    )
+                )
+
+            canary_version_task = None
+            if model.canary_version is not None:
+                canary_version_task = asyncio.create_task(
+                    self._service_version_model_to_status_async(
+                        model.canary_version,
+                        service_name=model.name,
+                        project_id=model.project_id,
+                        query_auth_token_enabled=query_auth_token_enabled,
+                    )
                 )
-            )
 
-        primary_version = await primary_version_task if primary_version_task else None
-        canary_version = await canary_version_task if canary_version_task else None
+            primary_version = (
+                await primary_version_task if primary_version_task else None
+            )
+            canary_version = await canary_version_task if canary_version_task else None
 
-        project_name = None
-        if primary_version and isinstance(primary_version.config, ServiceConfig):
-            project_name = primary_version.config.project
+            if primary_version and isinstance(primary_version.config, ServiceConfig):
+                project_name = primary_version.config.project
 
         return ServiceStatus(
             id=model.id,
@@ -1036,6 +1080,7 @@ class PrivateServiceSDK(WorkloadSDK):
             primary_version=primary_version,
             canary_version=canary_version,
             project=project_name,
+            versions=all_versions,
         )
 
     def status(
@@ -1056,6 +1101,7 @@ class PrivateServiceSDK(WorkloadSDK):
         cloud: Optional[str] = None,
         project: Optional[str] = None,
         include_archived: bool = False,
+        tags_filter: Optional[Dict[str, List[str]]] = None,
         # Paging
         max_items: Optional[int] = None,  # Controls total items yielded by iterator
         page_size: Optional[int] = None,  # Controls items fetched per API call
@@ -1096,6 +1142,18 @@ class PrivateServiceSDK(WorkloadSDK):
 
         normalised_states = _normalize_state_filter(state_filter)
 
+        # Convert tags dict into backend tag_filter list[str]
+        backend_tag_filter: Optional[List[str]] = None
+        if tags_filter:
+            flattened: List[str] = []
+            for key, values in tags_filter.items():
+                if not values:
+                    continue
+                for value in values:
+                    if key and value:
+                        flattened.append(f"{key}:{value}")
+            backend_tag_filter = flattened if len(flattened) > 0 else None
+
         def _fetch_page(
             token: Optional[str],
         ) -> DecoratedlistserviceapimodelListResponse:
@@ -1106,6 +1164,7 @@ class PrivateServiceSDK(WorkloadSDK):
                 cloud=cloud,
                 project=project,
                 include_archived=include_archived,
+                tag_filter=backend_tag_filter,
                 count=page_size,
                 paging_token=token,
                 sort_field=sort_field,
@@ -1199,6 +1258,44 @@ class PrivateServiceSDK(WorkloadSDK):
             model.primary_version, head, max_lines
         )
 
+    def add_tags(
+        self,
+        *,
+        id: Optional[str] = None,  # noqa: A002
+        name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        project: Optional[str] = None,
+        tags: Dict[str, str],
+    ) -> None:
+        if id is None:
+            model = self._resolve_to_service_model(
+                name=name, cloud=cloud, project=project
+            )
+            resource_id = model.id
+        assert id is not None
+        self.client.upsert_resource_tags(
+            ResourceTagResourceType.SERVICE, resource_id, tags
+        )
+
+    def remove_tags(
+        self,
+        *,
+        id: Optional[str] = None,  # noqa: A002
+        name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        project: Optional[str] = None,
+        keys: List[str],
+    ) -> None:
+        if id is None:
+            model = self._resolve_to_service_model(
+                name=name, cloud=cloud, project=project
+            )
+            resource_id = model.id
+        assert id is not None
+        self.client.delete_resource_tags(
+            ResourceTagResourceType.SERVICE, resource_id, keys
+        )
+
 
 def _normalize_state_filter(
     states: Optional[Union[List[ServiceState], List[str]]],

anyscale/service/commands.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, Union
+from typing import Dict, List, Optional, Union
 
 from anyscale._private.models.model_base import ResultIterator
 from anyscale._private.sdk import sdk_command
@@ -323,6 +323,77 @@ _CONTROLLER_LOGS_ARG_DOCSTRINGS = {
     "max_lines": "The number of log lines to be fetched. If not provided, 1000 lines will be fetched.",
 }
 
+# SDK docs for Service tag operations
+_TAGS_ADD_EXAMPLE = """
+import anyscale
+
+anyscale.service.add_tags(id="svc_123", tags={"team": "mlops", "env": "prod"})
+"""
+
+_TAGS_ADD_ARG_DOCSTRINGS = {
+    "id": "ID of the service. Provide either id or name.",
+    "name": "Name of the service. Provide either id or name.",
+    "cloud": "Cloud name (used when resolving by name).",
+    "project": "Project name (used when resolving by name).",
+    "tags": "Key/value tags to upsert as a map {key: value}.",
+}
+
+_TAGS_REMOVE_EXAMPLE = """
+import anyscale
+
+anyscale.service.remove_tags(id="svc_123", keys=["team", "env"])
+"""
+
+_TAGS_REMOVE_ARG_DOCSTRINGS = {
+    "id": "ID of the service. Provide either id or name.",
+    "name": "Name of the service. Provide either id or name.",
+    "cloud": "Cloud name (used when resolving by name).",
+    "project": "Project name (used when resolving by name).",
+    "keys": "List of tag keys to remove.",
+}
+
+
+@sdk_command(
+    _SERVICE_SDK_SINGLETON_KEY,
+    PrivateServiceSDK,
+    doc_py_example=_TAGS_ADD_EXAMPLE,
+    arg_docstrings=_TAGS_ADD_ARG_DOCSTRINGS,
+)
+def add_tags(
+    *,
+    id: Optional[str] = None,  # noqa: A002
+    name: Optional[str] = None,
+    cloud: Optional[str] = None,
+    project: Optional[str] = None,
+    tags: Dict[str, str],
+    _private_sdk: Optional[PrivateServiceSDK] = None,
+) -> None:
+    """Upsert (add/update) tag key/value pairs for a service."""
+    return _private_sdk.add_tags(  # type: ignore
+        id=id, name=name, cloud=cloud, project=project, tags=tags
+    )
+
+
+@sdk_command(
+    _SERVICE_SDK_SINGLETON_KEY,
+    PrivateServiceSDK,
+    doc_py_example=_TAGS_REMOVE_EXAMPLE,
+    arg_docstrings=_TAGS_REMOVE_ARG_DOCSTRINGS,
+)
+def remove_tags(
+    *,
+    id: Optional[str] = None,  # noqa: A002
+    name: Optional[str] = None,
+    cloud: Optional[str] = None,
+    project: Optional[str] = None,
+    keys: List[str],
+    _private_sdk: Optional[PrivateServiceSDK] = None,
+) -> None:
+    """Remove tags by key from a service."""
+    return _private_sdk.remove_tags(  # type: ignore
+        id=id, name=name, cloud=cloud, project=project, keys=keys
+    )
+
 
 # This is a private command that is not exposed to the user.
 @sdk_command(
@@ -375,6 +446,7 @@ _LIST_ARG_DOCSTRINGS = {
     "cloud": "Name of the Anyscale Cloud to search in.",
     "project": "Name of the Anyscale Project to search in.",
     "include_archived": "Include archived services (default: False).",
+    "tags_filter": "Filter services by tags as a dict: {key: [values...]}. Values with the same key are ORed; keys are ANDed.",
     # Paging
     "max_items": "Maximum **total** number of items to yield (default: iterate all).",
     "page_size": "Number of items to fetch per API request (default: API default).",
@@ -402,6 +474,7 @@ def list(  # noqa: A001
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     include_archived: bool = False,
+    tags_filter: Optional[Dict[str, List[str]]] = None,
     # Paging
     max_items: Optional[int] = None,
     page_size: Optional[int] = None,
@@ -420,6 +493,7 @@ def list(  # noqa: A001
         cloud=cloud,
         project=project,
         include_archived=include_archived,
+        tags_filter=tags_filter,
         max_items=max_items,
         page_size=page_size,
         sort_field=sort_field,

anyscale/service/models.py

@@ -213,6 +213,9 @@ tracing_config: # (Optional) Configuration options for tracing.
   enabled: true
   exporter_import_path: my_module:custom_tracing_exporter
   sampling_ratio: 1.0
+tags:
+  team: serving
+  cost-center: eng
 """
 
     # Override the `name` field from `WorkloadConfig` so we can document it separately for jobs and services.
@@ -239,8 +242,6 @@ tracing_config: # (Optional) Configuration options for tracing.
         # Allow None if optional; enforce non-empty string if required
         if version_name is None:
             return None
-        if not isinstance(version_name, str) or not version_name.strip():
-            raise ValueError("version_name must be a non-empty string")
         return version_name
 
     def _validate_applications(self, applications: List[Dict[str, Any]]):
@@ -402,6 +403,19 @@ tracing_config: # (Optional) Configuration options for tracing.
 
         return tracing_config
 
+    tags: Optional[Dict[str, str]] = field(
+        default=None, metadata={"docstring": "Tags to associate with the service."},
+    )
+
+    def _validate_tags(self, tags: Optional[Dict[str, str]]):
+        if tags is None:
+            return
+        if not isinstance(tags, dict):
+            raise TypeError("'tags' must be a Dict[str, str].")
+        for k, v in tags.items():
+            if not isinstance(k, str) or not isinstance(v, str):
+                raise TypeError("'tags' must be a Dict[str, str].")
+
 
 class ServiceState(ModelEnum):
     """Current state of a service."""
@@ -499,6 +513,7 @@ id: service2_uz6l8yhy2as5wrer3shzj6kh67
 state: RUNNING
 primary_version:
   id: 601bd56c4b
+  name: v1
   state: RUNNING
   weight: 100
   created_at: 2025-04-18 17:21:28.323174+00:00
@@ -514,6 +529,15 @@ primary_version:
         if not isinstance(id, str):
             raise TypeError("'id' must be a string.")
 
+    name: str = field(
+        metadata={"docstring": "Human-readable name of the service version."}
+    )
+
+    def _validate_name(self, name: str):
+        if name is None:
+            return None
+        return name
+
     state: Union[ServiceVersionState, str] = field(
         metadata={"docstring": "Current state of the service version."}
     )
@@ -700,6 +724,26 @@ primary_version:
         if project is not None and not isinstance(project, str):
             raise TypeError("project must be a string.")
 
+    versions: Optional[List[ServiceVersionStatus]] = field(
+        default=None,
+        repr=False,
+        metadata={
+            "docstring": "All active versions of the service. For multi-version services, this contains all versions with their traffic weights."
+        },
+    )
+
+    def _validate_versions(
+        self, versions: Optional[List[ServiceVersionStatus]]
+    ) -> Optional[List[ServiceVersionStatus]]:
+        if versions is None:
+            return None
+        if not isinstance(versions, list):
+            raise TypeError("'versions' must be a list.")
+        for v in versions:
+            if not isinstance(v, ServiceVersionStatus):
+                raise TypeError("'versions' must be a list of ServiceVersionStatus.")
+        return versions
+
 
 class ServiceSortField(ModelEnum):
     """Fields available for sorting services."""

anyscale/shared_anyscale_utils/latest_ray_version.py

@@ -1,3 +1,3 @@
 # AUTOGENERATED - modify shared_anyscale_util in root directory to make changes
 # RAY_RELEASE_UPDATE: managed by release automation.
-LATEST_RAY_VERSION = "2.50.1"
+LATEST_RAY_VERSION = "2.51.0"

anyscale/version.py

@@ -1 +1 @@
-__version__ = "0.26.70"
+__version__ = "0.26.71"

anyscale/workspace/_private/workspace_sdk.py

@@ -169,6 +169,7 @@ class PrivateWorkspaceSDK(WorkloadSDK):
                 cluster_environment_build_id=build_id,
                 idle_timeout_minutes=config.idle_termination_minutes,
                 cloud_id=cloud_id,
+                tags=getattr(config, "tags", None),
                 skip_start=True,
             )
         )

anyscale/workspace/models.py

@@ -53,6 +53,9 @@ idle_termination_minutes: 220
 env_vars:
     key: value
 requirements: /tmp/requirements.txt
+tags:
+    team: mlops
+    env: prod
 """
 
     name: Optional[str] = field(
@@ -88,6 +91,19 @@ requirements: /tmp/requirements.txt
         if containerfile is not None and not isinstance(containerfile, str):
             raise TypeError("'containerfile' must be a string.")
 
+    tags: Optional[Dict[str, str]] = field(
+        default=None, metadata={"docstring": "Tags to associate with the workspace."},
+    )
+
+    def _validate_tags(self, tags: Optional[Dict[str, str]]):
+        if tags is None:
+            return
+        if not isinstance(tags, dict):
+            raise TypeError("'tags' must be a Dict[str, str].")
+        for k, v in tags.items():
+            if not isinstance(k, str) or not isinstance(v, str):
+                raise TypeError("'tags' must be a Dict[str, str].")
+
     compute_config: Union[ComputeConfigType, Dict, str, None] = field(
         default=None,
         metadata={
@@ -215,6 +231,9 @@ idle_termination_minutes: 220
 env_vars:
     key: value
 requirements: /tmp/requirements.txt
+tags:
+  team: mlops
+  env: prod
 """
 
     project: Optional[str] = field(

{anyscale-0.26.70.dist-info → anyscale-0.26.71.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anyscale
-Version: 0.26.70
+Version: 0.26.71
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License