anyscale 0.24.46__tar.gz → 0.24.47__tar.gz

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

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

{anyscale-0.24.46 → anyscale-0.24.47}/anyscale/__init__.py

@@ -18,7 +18,7 @@ path.append(os.path.join(anyscale_dir, "client"))
 path.append(os.path.join(anyscale_dir, "sdk"))
 
 import anyscale
-from anyscale import compute_config, image, integrations, job, service
+from anyscale import compute_config, image, integrations, job, llm, schedule, service
 from anyscale._private.anyscale_client import AnyscaleClient, AnyscaleClientInterface
 from anyscale._private.sdk.base_sdk import Timer
 from anyscale.authenticate import AuthenticationBlock
@@ -28,6 +28,7 @@ from anyscale.cluster_compute import get_cluster_compute_from_name
 from anyscale.compute_config import ComputeConfigSDK
 from anyscale.connect import ClientBuilder
 from anyscale.job import JobSDK
+from anyscale.schedule import ScheduleSDK
 from anyscale.sdk.anyscale_client.sdk import AnyscaleSDK
 from anyscale.service import ServiceSDK
 
@@ -88,6 +89,7 @@ class Anyscale:
         self._job_sdk = JobSDK(client=self._anyscale_client)
         self._service_sdk = ServiceSDK(client=self._anyscale_client)
         self._compute_config_sdk = ComputeConfigSDK(client=self._anyscale_client)
+        self._schedule_sdk = ScheduleSDK(client=self._anyscale_client)
 
     @classmethod
     def _init_private(
@@ -104,6 +106,9 @@ class Anyscale:
         obj._compute_config_sdk = ComputeConfigSDK(  # noqa: SLF001
             client=client, logger=logger, timer=timer
         )
+        obj._schedule_sdk = ScheduleSDK(  # noqa: SLF001
+            client=client, logger=logger, timer=timer,
+        )
         return obj
 
     @property
@@ -118,6 +123,10 @@ class Anyscale:
     def compute_config(self) -> ComputeConfigSDK:  # noqa: F811
         return self._compute_config_sdk
 
+    @property
+    def schedule(self) -> ScheduleSDK:  # noqa: F811
+        return self._schedule_sdk
+
 
 def _validate_headers(headers: Optional[Dict[str, str]]):
     if not headers:

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

@@ -7,6 +7,7 @@ import time
 from typing import Any, Callable, Dict, Generator, IO, List, Optional, Tuple
 from urllib.parse import urlparse
 
+from openapi_client.exceptions import ApiException
 import requests
 
 from anyscale._private.anyscale_client.common import (
@@ -35,6 +36,8 @@ from anyscale.client.openapi_client.models import (
     DecoratedComputeTemplate,
     InternalProductionJob,
 )
+from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
+from anyscale.client.openapi_client.models.decorated_schedule import DecoratedSchedule
 from anyscale.client.openapi_client.models.production_job import ProductionJob
 from anyscale.client.openapi_client.rest import ApiException as InternalApiException
 from anyscale.cluster_compute import parse_cluster_compute_name_version
@@ -1019,3 +1022,20 @@ class AnyscaleClient(AnyscaleClientInterface):
 
         logs = self._read_log_lines(all_log_chunk_urls, head, bearer_token, max_lines)
         return logs
+
+    def apply_schedule(self, model: CreateSchedule) -> DecoratedSchedule:
+        try:
+            return self._internal_api_client.create_or_update_job_api_v2_experimental_cron_jobs_put(
+                model
+            ).result
+        except ApiException as e:
+            if e.status == 400 and (
+                "Please specify a valid cron expression" in e.body
+                or "Please specify a valid tz database timezone" in e.body
+            ):
+                import json
+
+                body_dict = json.loads(e.body)
+                msg = body_dict["error"]["detail"]
+                raise ValueError(msg) from None
+            raise e from None

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

@@ -10,6 +10,8 @@ from anyscale.client.openapi_client.models import (
     InternalProductionJob,
     Project,
 )
+from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
+from anyscale.client.openapi_client.models.decorated_schedule import DecoratedSchedule
 from anyscale.client.openapi_client.models.production_job import ProductionJob
 from anyscale.sdk.anyscale_client.models import (
     ApplyServiceModel,
@@ -366,3 +368,11 @@ class AnyscaleClientInterface(ABC):
         The job_run_id is the backend id for the job run.
         """
         raise NotImplementedError
+
+    @abstractmethod
+    def apply_schedule(self, model: CreateSchedule) -> DecoratedSchedule:
+        """Creates or applies the schedule.
+
+        Returns the DecoratedSchedule created.
+        """
+        raise NotImplementedError

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

@@ -22,6 +22,8 @@ from anyscale.client.openapi_client.models import (
     ProductionJobStateTransition,
     Project,
 )
+from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
+from anyscale.client.openapi_client.models.decorated_schedule import DecoratedSchedule
 from anyscale.cluster_compute import parse_cluster_compute_name_version
 from anyscale.sdk.anyscale_client.configuration import Configuration
 from anyscale.sdk.anyscale_client.models import (
@@ -123,6 +125,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._submitted_job: Optional[CreateInternalProductionJob] = None
         self._env_vars: Optional[Dict[str, str]] = None
         self._job_run_logs: Dict[str, str] = {}
+        self._schedules: Dict[str, DecoratedSchedule] = {}
 
         # Cloud ID -> Cloud.
         self._clouds: Dict[str, Cloud] = {
@@ -726,3 +729,22 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             return "\n".join(log_lines[:max_lines])
         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 update_schedule(self, model: DecoratedSchedule):
+        self._schedules[model.id] = model
+
+    def apply_schedule(self, model: CreateSchedule) -> DecoratedSchedule:
+        schedule = DecoratedSchedule(
+            id=f"sched-{uuid.uuid4()!s}",
+            name=model.name,
+            project_id=model.project_id,
+            config=model.config,
+            schedule=model.schedule,
+            local_vars_configuration=OPENAPI_NO_VALIDATION,
+        )
+
+        self.update_schedule(schedule)
+        return schedule

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

@@ -85,6 +85,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**archive_compute_template_api_v2_compute_templates_compute_template_id_archive_post**](docs/DefaultApi.md#archive_compute_template_api_v2_compute_templates_compute_template_id_archive_post) | **POST** /api/v2/compute_templates/{compute_template_id}/archive | Archive Compute Template
 *DefaultApi* | [**archive_job_api_v2_decorated_ha_jobs_production_job_id_archive_post**](docs/DefaultApi.md#archive_job_api_v2_decorated_ha_jobs_production_job_id_archive_post) | **POST** /api/v2/decorated_ha_jobs/{production_job_id}/archive | Archive Job
 *DefaultApi* | [**attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post**](docs/DefaultApi.md#attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post) | **POST** /api/v2/machine_pools/attach | Attach Machine Pool To Cloud
+*DefaultApi* | [**aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post**](docs/DefaultApi.md#aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post) | **POST** /api/v2/organization_billing/aws_marketplace_registration | Aws Marketplace Registration
 *DefaultApi* | [**ban_organization_api_v2_organizations_organization_id_ban_put**](docs/DefaultApi.md#ban_organization_api_v2_organizations_organization_id_ban_put) | **PUT** /api/v2/organizations/{organization_id}/ban | Ban Organization
 *DefaultApi* | [**batch_create_invitations_api_v2_organization_invitations_batch_create_post**](docs/DefaultApi.md#batch_create_invitations_api_v2_organization_invitations_batch_create_post) | **POST** /api/v2/organization_invitations/batch_create | Batch Create Invitations
 *DefaultApi* | [**batch_get_job_api_v2_decorated_ha_jobs_batch_get_post**](docs/DefaultApi.md#batch_get_job_api_v2_decorated_ha_jobs_batch_get_post) | **POST** /api/v2/decorated_ha_jobs/batch_get | Batch Get Job
@@ -109,6 +110,8 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**create_cloud_collaborator_api_v2_projects_project_id_collaborators_cloud_post**](docs/DefaultApi.md#create_cloud_collaborator_api_v2_projects_project_id_collaborators_cloud_post) | **POST** /api/v2/projects/{project_id}/collaborators/cloud | Create Cloud Collaborator
 *DefaultApi* | [**create_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_post**](docs/DefaultApi.md#create_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_post) | **POST** /api/v2/clouds_with_cloud_resource_router/ | Create Cloud With Cloud Resource
 *DefaultApi* | [**create_compute_template_api_v2_compute_templates_post**](docs/DefaultApi.md#create_compute_template_api_v2_compute_templates_post) | **POST** /api/v2/compute_templates/ | Create Compute Template
+*DefaultApi* | [**create_configuration_api_v2_organization_configurations_post**](docs/DefaultApi.md#create_configuration_api_v2_organization_configurations_post) | **POST** /api/v2/organization_configurations/ | Create Configuration
+*DefaultApi* | [**create_configuration_script_api_v2_organization_configurations_create_helper_get**](docs/DefaultApi.md#create_configuration_script_api_v2_organization_configurations_create_helper_get) | **GET** /api/v2/organization_configurations/create_helper | Create Configuration Script
 *DefaultApi* | [**create_connect_session_api_v2_sessions_create_connect_session_post**](docs/DefaultApi.md#create_connect_session_api_v2_sessions_create_connect_session_post) | **POST** /api/v2/sessions/create_connect_session | Create Connect Session
 *DefaultApi* | [**create_endpoint_user_for_org_api_v2_organizations_organization_id_create_endpoint_user_post**](docs/DefaultApi.md#create_endpoint_user_for_org_api_v2_organizations_organization_id_create_endpoint_user_post) | **POST** /api/v2/organizations/{organization_id}/create_endpoint_user | Create Endpoint User For Org
 *DefaultApi* | [**create_fine_tuning_job_api_v2_fine_tuning_jobs_create_post**](docs/DefaultApi.md#create_fine_tuning_job_api_v2_fine_tuning_jobs_create_post) | **POST** /api/v2/fine_tuning/jobs/create | Create Fine Tuning Job
@@ -276,6 +279,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_application_templates_api_v2_application_templates_get**](docs/DefaultApi.md#list_application_templates_api_v2_application_templates_get) | **GET** /api/v2/application_templates/ | List Application Templates
 *DefaultApi* | [**list_builds_api_v2_builds_get**](docs/DefaultApi.md#list_builds_api_v2_builds_get) | **GET** /api/v2/builds/ | List Builds
 *DefaultApi* | [**list_clouds_api_v2_clouds_get**](docs/DefaultApi.md#list_clouds_api_v2_clouds_get) | **GET** /api/v2/clouds/ | List Clouds
+*DefaultApi* | [**list_configurations_api_v2_organization_configurations_get**](docs/DefaultApi.md#list_configurations_api_v2_organization_configurations_get) | **GET** /api/v2/organization_configurations/ | List Configurations
 *DefaultApi* | [**list_cron_jobs_api_v2_experimental_cron_jobs_get**](docs/DefaultApi.md#list_cron_jobs_api_v2_experimental_cron_jobs_get) | **GET** /api/v2/experimental_cron_jobs/ | List Cron Jobs
 *DefaultApi* | [**list_decorated_clusters_api_v2_decorated_sessions_get**](docs/DefaultApi.md#list_decorated_clusters_api_v2_decorated_sessions_get) | **GET** /api/v2/decorated_sessions/ | List Decorated Clusters
 *DefaultApi* | [**list_decorated_interactive_sessions_api_v2_decorated_interactive_sessions_get**](docs/DefaultApi.md#list_decorated_interactive_sessions_api_v2_decorated_interactive_sessions_get) | **GET** /api/v2/decorated_interactive_sessions/ | List Decorated Interactive Sessions
@@ -441,6 +445,7 @@ Class | Method | HTTP request | Description
  - [BillingInformation](docs/BillingInformation.md)
  - [BillingVersionCode](docs/BillingVersionCode.md)
  - [BlockDeviceMapping](docs/BlockDeviceMapping.md)
+ - [BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost](docs/BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost.md)
  - [BodyUploadFileApiV2FilesPost](docs/BodyUploadFileApiV2FilesPost.md)
  - [BufferRegistration](docs/BufferRegistration.md)
  - [Build](docs/Build.md)
@@ -548,6 +553,7 @@ Class | Method | HTTP request | Description
  - [CreateMachineResponse](docs/CreateMachineResponse.md)
  - [CreateMetronomeWebhookNotification](docs/CreateMetronomeWebhookNotification.md)
  - [CreateOTPReturnApiModel](docs/CreateOTPReturnApiModel.md)
+ - [CreateOrganizationConfiguration](docs/CreateOrganizationConfiguration.md)
  - [CreateOrganizationInvitation](docs/CreateOrganizationInvitation.md)
  - [CreateProductionJobConfig](docs/CreateProductionJobConfig.md)
  - [CreateSchedule](docs/CreateSchedule.md)
@@ -566,6 +572,7 @@ Class | Method | HTTP request | Description
  - [CreatesessionresponseResponse](docs/CreatesessionresponseResponse.md)
  - [CreditCardInformation](docs/CreditCardInformation.md)
  - [CustomerAlertStatus](docs/CustomerAlertStatus.md)
+ - [CustomerBillingType](docs/CustomerBillingType.md)
  - [DataplaneServices](docs/DataplaneServices.md)
  - [DecoratedApplicationTemplate](docs/DecoratedApplicationTemplate.md)
  - [DecoratedBuild](docs/DecoratedBuild.md)
@@ -754,6 +761,8 @@ Class | Method | HTTP request | Description
  - [Organization](docs/Organization.md)
  - [OrganizationAvailability](docs/OrganizationAvailability.md)
  - [OrganizationCollaborator](docs/OrganizationCollaborator.md)
+ - [OrganizationConfiguration](docs/OrganizationConfiguration.md)
+ - [OrganizationConfigurationResponse](docs/OrganizationConfigurationResponse.md)
  - [OrganizationInvitation](docs/OrganizationInvitation.md)
  - [OrganizationInvitationBase](docs/OrganizationInvitationBase.md)
  - [OrganizationMarketingQuestions](docs/OrganizationMarketingQuestions.md)
@@ -767,6 +776,8 @@ Class | Method | HTTP request | Description
  - [OrganizationUsageAlertSeverity](docs/OrganizationUsageAlertSeverity.md)
  - [OrganizationavailabilityResponse](docs/OrganizationavailabilityResponse.md)
  - [OrganizationcollaboratorListResponse](docs/OrganizationcollaboratorListResponse.md)
+ - [OrganizationconfigurationListResponse](docs/OrganizationconfigurationListResponse.md)
+ - [OrganizationconfigurationresponseResponse](docs/OrganizationconfigurationresponseResponse.md)
  - [OrganizationinvitationListResponse](docs/OrganizationinvitationListResponse.md)
  - [OrganizationinvitationResponse](docs/OrganizationinvitationResponse.md)
  - [OrganizationinvitationbaseResponse](docs/OrganizationinvitationbaseResponse.md)

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

@@ -86,6 +86,7 @@ from openapi_client.models.batched_result_organization_invitation_base import Ba
 from openapi_client.models.billing_information import BillingInformation
 from openapi_client.models.billing_version_code import BillingVersionCode
 from openapi_client.models.block_device_mapping import BlockDeviceMapping
+from openapi_client.models.body_aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post import BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost
 from openapi_client.models.body_upload_file_api_v2_files_post import BodyUploadFileApiV2FilesPost
 from openapi_client.models.buffer_registration import BufferRegistration
 from openapi_client.models.build import Build
@@ -193,6 +194,7 @@ from openapi_client.models.create_machine_request import CreateMachineRequest
 from openapi_client.models.create_machine_response import CreateMachineResponse
 from openapi_client.models.create_metronome_webhook_notification import CreateMetronomeWebhookNotification
 from openapi_client.models.create_otp_return_api_model import CreateOTPReturnApiModel
+from openapi_client.models.create_organization_configuration import CreateOrganizationConfiguration
 from openapi_client.models.create_organization_invitation import CreateOrganizationInvitation
 from openapi_client.models.create_production_job_config import CreateProductionJobConfig
 from openapi_client.models.create_schedule import CreateSchedule
@@ -211,6 +213,7 @@ from openapi_client.models.createotpreturnapimodel_response import Createotpretu
 from openapi_client.models.createsessionresponse_response import CreatesessionresponseResponse
 from openapi_client.models.credit_card_information import CreditCardInformation
 from openapi_client.models.customer_alert_status import CustomerAlertStatus
+from openapi_client.models.customer_billing_type import CustomerBillingType
 from openapi_client.models.dataplane_services import DataplaneServices
 from openapi_client.models.decorated_application_template import DecoratedApplicationTemplate
 from openapi_client.models.decorated_build import DecoratedBuild
@@ -399,6 +402,8 @@ from openapi_client.models.onboarding_user_cards_query import OnboardingUserCard
 from openapi_client.models.organization import Organization
 from openapi_client.models.organization_availability import OrganizationAvailability
 from openapi_client.models.organization_collaborator import OrganizationCollaborator
+from openapi_client.models.organization_configuration import OrganizationConfiguration
+from openapi_client.models.organization_configuration_response import OrganizationConfigurationResponse
 from openapi_client.models.organization_invitation import OrganizationInvitation
 from openapi_client.models.organization_invitation_base import OrganizationInvitationBase
 from openapi_client.models.organization_marketing_questions import OrganizationMarketingQuestions
@@ -412,6 +417,8 @@ from openapi_client.models.organization_usage_alert import OrganizationUsageAler
 from openapi_client.models.organization_usage_alert_severity import OrganizationUsageAlertSeverity
 from openapi_client.models.organizationavailability_response import OrganizationavailabilityResponse
 from openapi_client.models.organizationcollaborator_list_response import OrganizationcollaboratorListResponse
+from openapi_client.models.organizationconfiguration_list_response import OrganizationconfigurationListResponse
+from openapi_client.models.organizationconfigurationresponse_response import OrganizationconfigurationresponseResponse
 from openapi_client.models.organizationinvitation_list_response import OrganizationinvitationListResponse
 from openapi_client.models.organizationinvitation_response import OrganizationinvitationResponse
 from openapi_client.models.organizationinvitationbase_response import OrganizationinvitationbaseResponse