anyscale 0.24.47__tar.gz → 0.24.49__tar.gz

{anyscale-0.24.47 → anyscale-0.24.49}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.47
+Version: 0.24.49
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.24.47 → anyscale-0.24.49}/anyscale/_private/anyscale_client/anyscale_client.py

@@ -1039,3 +1039,53 @@ class AnyscaleClient(AnyscaleClientInterface):
                 msg = body_dict["error"]["detail"]
                 raise ValueError(msg) from None
             raise e from None
+
+    def get_schedule(
+        self,
+        *,
+        name: Optional[str],
+        id: Optional[str],  # noqa: A002
+        cloud: Optional[str],
+        project: Optional[str],
+    ) -> Optional[DecoratedSchedule]:
+        if id is not None:
+            try:
+                return self._internal_api_client.get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get(
+                    id
+                ).result
+            except ExternalApiException as e:
+                if e.status == 404:
+                    return None
+                raise e from None
+        else:
+            paging_token = None
+            cloud_id = self.get_cloud_id(cloud_name=cloud)
+            project_id = self.get_project_id(parent_cloud_id=cloud_id, name=project)
+            result: Optional[DecoratedSchedule] = None
+            while True:
+                resp = self._internal_api_client.list_cron_jobs_api_v2_experimental_cron_jobs_get(
+                    project_id=project_id,
+                    name=name,
+                    count=self.LIST_ENDPOINT_COUNT,
+                    paging_token=paging_token,
+                )
+                for schedule in resp.results:
+                    if schedule is not None and schedule.name == name:
+                        result = schedule
+                        break
+
+                paging_token = resp.metadata.next_paging_token
+                if paging_token is None:
+                    break
+
+            return result
+
+    def set_schedule_state(self, id: str, is_paused: bool):  # noqa: A002
+        self._internal_api_client.pause_cron_job_api_v2_experimental_cron_jobs_cron_job_id_pause_post(
+            id, {"is_paused": is_paused}
+        ).result
+
+    def trigger_schedule(self, id: str):  # noqa: A002
+        self._internal_api_client.trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post(
+            id
+        )

{anyscale-0.24.47 → anyscale-0.24.49}/anyscale/_private/anyscale_client/common.py

@@ -376,3 +376,30 @@ class AnyscaleClientInterface(ABC):
         Returns the DecoratedSchedule created.
         """
         raise NotImplementedError
+
+    @abstractmethod
+    def get_schedule(
+        self,
+        *,
+        name: Optional[str],
+        id: Optional[str],  # noqa: A002
+        cloud: Optional[str],
+        project: Optional[str],
+    ) -> Optional[DecoratedSchedule]:
+        """Get a schedule by either name or id. Filter by cloud and project.
+
+        Returns None if not found.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def set_schedule_state(self, id: str, is_paused: bool):  # noqa: A002
+        """Set the state of a schedule with id.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def trigger_schedule(self, id: str):  # noqa: A002
+        """Trigger a schedule with id.
+        """
+        raise NotImplementedError

{anyscale-0.24.47 → anyscale-0.24.49}/anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -70,6 +70,8 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     WORKSPACE_CLUSTER_COMPUTE_ID = "fake-workspace-cluster-compute-id"
     WORKSPACE_CLUSTER_ENV_BUILD_ID = "fake-workspace-cluster-env-build-id"
 
+    SCHEDULE_NEXT_TRIGGER_AT_TIME = datetime.utcnow()
+
     def __init__(self):
         self._builds: Dict[str, ClusterEnvironmentBuild] = {
             self.DEFAULT_CLUSTER_ENV_BUILD_ID: ClusterEnvironmentBuild(
@@ -126,6 +128,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._env_vars: Optional[Dict[str, str]] = None
         self._job_run_logs: Dict[str, str] = {}
         self._schedules: Dict[str, DecoratedSchedule] = {}
+        self._schedule_trigger_counts: Dict[str, int] = defaultdict(int)
 
         # Cloud ID -> Cloud.
         self._clouds: Dict[str, Cloud] = {
@@ -730,8 +733,33 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         else:
             return "\n".join(log_lines[-1 * max_lines :])
 
-    def get_schedule(self, schedule_id: str) -> Optional[DecoratedSchedule]:
-        return self._schedules.get(schedule_id, None)
+    def get_schedule(
+        self,
+        *,
+        name: Optional[str],
+        id: Optional[str],  # noqa: A002
+        cloud: Optional[str],
+        project: Optional[str],
+    ) -> Optional[DecoratedSchedule]:
+        if id is not None:
+            return self._schedules.get(id, None)
+        else:
+            cloud_id = self.get_cloud_id(cloud_name=cloud)
+            cloud_project_dict = self._project_to_id.get(cloud_id, None)
+            project_id = (
+                cloud_project_dict.get(project, None) if cloud_project_dict else None
+            )
+            result: DecoratedSchedule = None
+            for schedule in self._schedules.values():
+                if (
+                    schedule is not None
+                    and schedule.name == name
+                    and (project_id is None or schedule.project_id == project_id)
+                ):
+                    result = schedule
+                    break
+
+        return result
 
     def update_schedule(self, model: DecoratedSchedule):
         self._schedules[model.id] = model
@@ -743,8 +771,25 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             project_id=model.project_id,
             config=model.config,
             schedule=model.schedule,
+            # Fill in dummy time to represent schedule is enabled.
+            next_trigger_at=self.SCHEDULE_NEXT_TRIGGER_AT_TIME,
             local_vars_configuration=OPENAPI_NO_VALIDATION,
         )
 
         self.update_schedule(schedule)
         return schedule
+
+    def set_schedule_state(self, id: str, is_paused: bool):  # noqa: A002
+        if is_paused:
+            self._schedules[id].next_trigger_at = None
+        else:
+            self._schedules[id].next_trigger_at = self.SCHEDULE_NEXT_TRIGGER_AT_TIME
+
+    def schedule_is_enabled(self, id: str) -> bool:  # noqa: A002
+        return self._schedules[id].next_trigger_at is not None
+
+    def trigger_schedule(self, id: str):  # noqa: A002
+        self._schedule_trigger_counts[id] += 1
+
+    def trigger_counts(self, id: str):  # noqa: A002
+        return self._schedule_trigger_counts[id]

{anyscale-0.24.47 → anyscale-0.24.49}/anyscale/client/README.md

@@ -424,6 +424,7 @@ Class | Method | HTTP request | Description
  - [AppConfigConfigSchema](docs/AppConfigConfigSchema.md)
  - [AppconfigListResponse](docs/AppconfigListResponse.md)
  - [AppconfigResponse](docs/AppconfigResponse.md)
+ - [ApplicationType](docs/ApplicationType.md)
  - [AppliedSnapshot](docs/AppliedSnapshot.md)
  - [ApplyProductionServiceV2Model](docs/ApplyProductionServiceV2Model.md)
  - [ArchiveStatus](docs/ArchiveStatus.md)
@@ -518,6 +519,7 @@ Class | Method | HTTP request | Description
  - [ClusterfeaturesResponse](docs/ClusterfeaturesResponse.md)
  - [CompanySize](docs/CompanySize.md)
  - [ComputeNodeType](docs/ComputeNodeType.md)
+ - [ComputeStack](docs/ComputeStack.md)
  - [ComputeTemplate](docs/ComputeTemplate.md)
  - [ComputeTemplateConfig](docs/ComputeTemplateConfig.md)
  - [ComputeTemplateQuery](docs/ComputeTemplateQuery.md)

{anyscale-0.24.47 → anyscale-0.24.49}/anyscale/client/openapi_client/__init__.py

@@ -65,6 +65,7 @@ from openapi_client.models.app_config import AppConfig
 from openapi_client.models.app_config_config_schema import AppConfigConfigSchema
 from openapi_client.models.appconfig_list_response import AppconfigListResponse
 from openapi_client.models.appconfig_response import AppconfigResponse
+from openapi_client.models.application_type import ApplicationType
 from openapi_client.models.applied_snapshot import AppliedSnapshot
 from openapi_client.models.apply_production_service_v2_model import ApplyProductionServiceV2Model
 from openapi_client.models.archive_status import ArchiveStatus
@@ -159,6 +160,7 @@ from openapi_client.models.clustereventsoutput_response import Clustereventsoutp
 from openapi_client.models.clusterfeatures_response import ClusterfeaturesResponse
 from openapi_client.models.company_size import CompanySize
 from openapi_client.models.compute_node_type import ComputeNodeType
+from openapi_client.models.compute_stack import ComputeStack
 from openapi_client.models.compute_template import ComputeTemplate
 from openapi_client.models.compute_template_config import ComputeTemplateConfig
 from openapi_client.models.compute_template_query import ComputeTemplateQuery

{anyscale-0.24.47 → anyscale-0.24.49}/anyscale/client/openapi_client/api/default_api.py

@@ -26798,6 +26798,7 @@ class DefaultApi(object):
         :param list[ServiceEventCurrentState] state_filter: A list of Service states to filter by
         :param str creator_id: creator_id to filter by
         :param str cloud_id: cloud_id to filter by
+        :param list[ApplicationType] application_type: Application type to filter by
         :param ServiceSortField sort_field: If absent, the default sorting order is 1. status (active first).2. Last updated at (desc). 3. Name (asc).
         :param SortOrder sort_order: If sort_field is absent, this field is ignored.If absent, this field defaults to ascending.
         :param str paging_token:
@@ -26830,6 +26831,7 @@ class DefaultApi(object):
         :param list[ServiceEventCurrentState] state_filter: A list of Service states to filter by
         :param str creator_id: creator_id to filter by
         :param str cloud_id: cloud_id to filter by
+        :param list[ApplicationType] application_type: Application type to filter by
         :param ServiceSortField sort_field: If absent, the default sorting order is 1. status (active first).2. Last updated at (desc). 3. Name (asc).
         :param SortOrder sort_order: If sort_field is absent, this field is ignored.If absent, this field defaults to ascending.
         :param str paging_token:
@@ -26856,6 +26858,7 @@ class DefaultApi(object):
             'state_filter',
             'creator_id',
             'cloud_id',
+            'application_type',
             'sort_field',
             'sort_order',
             'paging_token',
@@ -26899,6 +26902,9 @@ class DefaultApi(object):
             query_params.append(('creator_id', local_var_params['creator_id']))  # noqa: E501
         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 'application_type' in local_var_params and local_var_params['application_type'] is not None:  # noqa: E501
+            query_params.append(('application_type', local_var_params['application_type']))  # noqa: E501
+            collection_formats['application_type'] = 'multi'  # noqa: E501
         if 'sort_field' in local_var_params and local_var_params['sort_field'] is not None:  # noqa: E501
             query_params.append(('sort_field', local_var_params['sort_field']))  # noqa: E501
         if 'sort_order' in local_var_params and local_var_params['sort_order'] is not None:  # noqa: E501

{anyscale-0.24.47 → anyscale-0.24.49}/anyscale/client/openapi_client/models/__init__.py

@@ -51,6 +51,7 @@ from openapi_client.models.app_config import AppConfig
 from openapi_client.models.app_config_config_schema import AppConfigConfigSchema
 from openapi_client.models.appconfig_list_response import AppconfigListResponse
 from openapi_client.models.appconfig_response import AppconfigResponse
+from openapi_client.models.application_type import ApplicationType
 from openapi_client.models.applied_snapshot import AppliedSnapshot
 from openapi_client.models.apply_production_service_v2_model import ApplyProductionServiceV2Model
 from openapi_client.models.archive_status import ArchiveStatus
@@ -145,6 +146,7 @@ from openapi_client.models.clustereventsoutput_response import Clustereventsoutp
 from openapi_client.models.clusterfeatures_response import ClusterfeaturesResponse
 from openapi_client.models.company_size import CompanySize
 from openapi_client.models.compute_node_type import ComputeNodeType
+from openapi_client.models.compute_stack import ComputeStack
 from openapi_client.models.compute_template import ComputeTemplate
 from openapi_client.models.compute_template_config import ComputeTemplateConfig
 from openapi_client.models.compute_template_query import ComputeTemplateQuery

anyscale-0.24.49/anyscale/client/openapi_client/models/application_type.py

@@ -0,0 +1,99 @@
+# coding: utf-8
+
+"""
+    Managed Ray API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by: https://openapi-generator.tech
+"""
+
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+from openapi_client.configuration import Configuration
+
+
+class ApplicationType(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    RAYLLM = "RAYLLM"
+
+    allowable_values = [RAYLLM]  # noqa: E501
+
+    """
+    Attributes:
+      openapi_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    openapi_types = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self, local_vars_configuration=None):  # noqa: E501
+        """ApplicationType - a model defined in OpenAPI"""  # noqa: E501
+        if local_vars_configuration is None:
+            local_vars_configuration = Configuration()
+        self.local_vars_configuration = local_vars_configuration
+        self.discriminator = None
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.openapi_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+
+        return result
+
+    def to_str(self):
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self):
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other):
+        """Returns true if both objects are equal"""
+        if not isinstance(other, ApplicationType):
+            return False
+
+        return self.to_dict() == other.to_dict()
+
+    def __ne__(self, other):
+        """Returns true if both objects are not equal"""
+        if not isinstance(other, ApplicationType):
+            return True
+
+        return self.to_dict() != other.to_dict()