lightning-sdk 0.2.20__py3-none-any.whl → 0.2.21rc0__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -50,8 +50,6 @@ class V1UserFeatures(object):
         'auto_top_up': 'bool',
         'auto_top_up_teamspace': 'bool',
         'b2c_experience': 'bool',
-        'blog_route': 'bool',
-        'byoc_litcr': 'bool',
         'cap_add': 'list[str]',
         'cap_drop': 'list[str]',
         'capacity_reservation_byoc': 'bool',
@@ -67,35 +65,29 @@ class V1UserFeatures(object):
         'custom_app_domain': 'bool',
         'datasets': 'bool',
         'default_one_cluster': 'bool',
-        'deployment_alerts': 'bool',
         'deployment_persistent_disk': 'bool',
-        'deployment_requests_tab': 'bool',
-        'dgx_cloud': 'bool',
         'docs_agent': 'bool',
         'drive_v2': 'bool',
-        'enable_storage_limits': 'bool',
         'enterprise_compute_admin': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
-        'filestore': 'bool',
         'gcs_folders': 'bool',
         'instant_capacity_reservation': 'bool',
         'job_artifacts_v2': 'bool',
         'lambda_labs': 'bool',
         'lambda_labs_studios': 'bool',
         'landing_studios': 'bool',
-        'lightning_cloud': 'bool',
         'lit_logger': 'bool',
         'marketplace': 'bool',
         'mmt_fault_tolerance': 'bool',
         'mmt_strategy_selector': 'bool',
+        'model_api_dashboard': 'bool',
         'multicloud_folders': 'bool',
         'multicloud_saas': 'bool',
         'multiple_studio_versions': 'bool',
         'nebius': 'bool',
         'nebius_cpu_studios': 'bool',
         'nebius_gpu_studios': 'bool',
-        'neocloud_studios': 'bool',
         'nerf_fs_nonpaying': 'bool',
         'org_level_member_permissions': 'bool',
         'org_usage_limits': 'bool',
@@ -124,13 +116,11 @@ class V1UserFeatures(object):
         'single_wallet': 'bool',
         'slurm': 'bool',
         'slurm_machine_selector': 'bool',
-        'stop_ide_container_on_shutdown': 'bool',
         'storage_overuse_deletion': 'bool',
         'studio_config': 'bool',
         'studio_version_visibility': 'bool',
         'teamspace_storage_tab': 'bool',
         'trainium2': 'bool',
-        'transfer_studios': 'bool',
         'use_rclone_mounts_only': 'bool',
         'voltage_park': 'bool',
         'voltage_park_studios': 'bool',
@@ -148,8 +138,6 @@ class V1UserFeatures(object):
         'auto_top_up': 'autoTopUp',
         'auto_top_up_teamspace': 'autoTopUpTeamspace',
         'b2c_experience': 'b2cExperience',
-        'blog_route': 'blogRoute',
-        'byoc_litcr': 'byocLitcr',
         'cap_add': 'capAdd',
         'cap_drop': 'capDrop',
         'capacity_reservation_byoc': 'capacityReservationByoc',
@@ -165,35 +153,29 @@ class V1UserFeatures(object):
         'custom_app_domain': 'customAppDomain',
         'datasets': 'datasets',
         'default_one_cluster': 'defaultOneCluster',
-        'deployment_alerts': 'deploymentAlerts',
         'deployment_persistent_disk': 'deploymentPersistentDisk',
-        'deployment_requests_tab': 'deploymentRequestsTab',
-        'dgx_cloud': 'dgxCloud',
         'docs_agent': 'docsAgent',
         'drive_v2': 'driveV2',
-        'enable_storage_limits': 'enableStorageLimits',
         'enterprise_compute_admin': 'enterpriseComputeAdmin',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
-        'filestore': 'filestore',
         'gcs_folders': 'gcsFolders',
         'instant_capacity_reservation': 'instantCapacityReservation',
         'job_artifacts_v2': 'jobArtifactsV2',
         'lambda_labs': 'lambdaLabs',
         'lambda_labs_studios': 'lambdaLabsStudios',
         'landing_studios': 'landingStudios',
-        'lightning_cloud': 'lightningCloud',
         'lit_logger': 'litLogger',
         'marketplace': 'marketplace',
         'mmt_fault_tolerance': 'mmtFaultTolerance',
         'mmt_strategy_selector': 'mmtStrategySelector',
+        'model_api_dashboard': 'modelApiDashboard',
         'multicloud_folders': 'multicloudFolders',
         'multicloud_saas': 'multicloudSaas',
         'multiple_studio_versions': 'multipleStudioVersions',
         'nebius': 'nebius',
         'nebius_cpu_studios': 'nebiusCpuStudios',
         'nebius_gpu_studios': 'nebiusGpuStudios',
-        'neocloud_studios': 'neocloudStudios',
         'nerf_fs_nonpaying': 'nerfFsNonpaying',
         'org_level_member_permissions': 'orgLevelMemberPermissions',
         'org_usage_limits': 'orgUsageLimits',
@@ -222,13 +204,11 @@ class V1UserFeatures(object):
         'single_wallet': 'singleWallet',
         'slurm': 'slurm',
         'slurm_machine_selector': 'slurmMachineSelector',
-        'stop_ide_container_on_shutdown': 'stopIdeContainerOnShutdown',
         'storage_overuse_deletion': 'storageOveruseDeletion',
         'studio_config': 'studioConfig',
         'studio_version_visibility': 'studioVersionVisibility',
         'teamspace_storage_tab': 'teamspaceStorageTab',
         'trainium2': 'trainium2',
-        'transfer_studios': 'transferStudios',
         'use_rclone_mounts_only': 'useRcloneMountsOnly',
         'voltage_park': 'voltagePark',
         'voltage_park_studios': 'voltageParkStudios',
@@ -236,7 +216,7 @@ class V1UserFeatures(object):
         'weka': 'weka'
     }
 
-    def __init__(self, accurate_billing: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, auto_top_up: 'bool' =None, auto_top_up_teamspace: 'bool' =None, b2c_experience: 'bool' =None, blog_route: 'bool' =None, byoc_litcr: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloud_space_environment_templates: 'bool' =None, cloudy_vibe_code: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, concurrent_gpu_limit: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_alerts: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_requests_tab: 'bool' =None, dgx_cloud: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_storage_limits: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, filestore: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lightning_cloud: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multicloud_folders: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, nebius: 'bool' =None, nebius_cpu_studios: 'bool' =None, nebius_gpu_studios: 'bool' =None, neocloud_studios: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, pipelines: 'bool' =None, plg_control_center: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, r2_data_connections: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, seoul_aws_region: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, stop_ide_container_on_shutdown: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, transfer_studios: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None):  # noqa: E501
+    def __init__(self, accurate_billing: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, auto_top_up: 'bool' =None, auto_top_up_teamspace: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloud_space_environment_templates: 'bool' =None, cloudy_vibe_code: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, concurrent_gpu_limit: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multicloud_folders: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, nebius: 'bool' =None, nebius_cpu_studios: 'bool' =None, nebius_gpu_studios: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, pipelines: 'bool' =None, plg_control_center: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, r2_data_connections: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, seoul_aws_region: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._accurate_billing = None
         self._affiliate_links = None
@@ -247,8 +227,6 @@ class V1UserFeatures(object):
         self._auto_top_up = None
         self._auto_top_up_teamspace = None
         self._b2c_experience = None
-        self._blog_route = None
-        self._byoc_litcr = None
         self._cap_add = None
         self._cap_drop = None
         self._capacity_reservation_byoc = None
@@ -264,35 +242,29 @@ class V1UserFeatures(object):
         self._custom_app_domain = None
         self._datasets = None
         self._default_one_cluster = None
-        self._deployment_alerts = None
         self._deployment_persistent_disk = None
-        self._deployment_requests_tab = None
-        self._dgx_cloud = None
         self._docs_agent = None
         self._drive_v2 = None
-        self._enable_storage_limits = None
         self._enterprise_compute_admin = None
         self._fair_share = None
         self._featured_studios_admin = None
-        self._filestore = None
         self._gcs_folders = None
         self._instant_capacity_reservation = None
         self._job_artifacts_v2 = None
         self._lambda_labs = None
         self._lambda_labs_studios = None
         self._landing_studios = None
-        self._lightning_cloud = None
         self._lit_logger = None
         self._marketplace = None
         self._mmt_fault_tolerance = None
         self._mmt_strategy_selector = None
+        self._model_api_dashboard = None
         self._multicloud_folders = None
         self._multicloud_saas = None
         self._multiple_studio_versions = None
         self._nebius = None
         self._nebius_cpu_studios = None
         self._nebius_gpu_studios = None
-        self._neocloud_studios = None
         self._nerf_fs_nonpaying = None
         self._org_level_member_permissions = None
         self._org_usage_limits = None
@@ -321,13 +293,11 @@ class V1UserFeatures(object):
         self._single_wallet = None
         self._slurm = None
         self._slurm_machine_selector = None
-        self._stop_ide_container_on_shutdown = None
         self._storage_overuse_deletion = None
         self._studio_config = None
         self._studio_version_visibility = None
         self._teamspace_storage_tab = None
         self._trainium2 = None
-        self._transfer_studios = None
         self._use_rclone_mounts_only = None
         self._voltage_park = None
         self._voltage_park_studios = None
@@ -352,10 +322,6 @@ class V1UserFeatures(object):
             self.auto_top_up_teamspace = auto_top_up_teamspace
         if b2c_experience is not None:
             self.b2c_experience = b2c_experience
-        if blog_route is not None:
-            self.blog_route = blog_route
-        if byoc_litcr is not None:
-            self.byoc_litcr = byoc_litcr
         if cap_add is not None:
             self.cap_add = cap_add
         if cap_drop is not None:
@@ -386,28 +352,18 @@ class V1UserFeatures(object):
             self.datasets = datasets
         if default_one_cluster is not None:
             self.default_one_cluster = default_one_cluster
-        if deployment_alerts is not None:
-            self.deployment_alerts = deployment_alerts
         if deployment_persistent_disk is not None:
             self.deployment_persistent_disk = deployment_persistent_disk
-        if deployment_requests_tab is not None:
-            self.deployment_requests_tab = deployment_requests_tab
-        if dgx_cloud is not None:
-            self.dgx_cloud = dgx_cloud
         if docs_agent is not None:
             self.docs_agent = docs_agent
         if drive_v2 is not None:
             self.drive_v2 = drive_v2
-        if enable_storage_limits is not None:
-            self.enable_storage_limits = enable_storage_limits
         if enterprise_compute_admin is not None:
             self.enterprise_compute_admin = enterprise_compute_admin
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
             self.featured_studios_admin = featured_studios_admin
-        if filestore is not None:
-            self.filestore = filestore
         if gcs_folders is not None:
             self.gcs_folders = gcs_folders
         if instant_capacity_reservation is not None:
@@ -420,8 +376,6 @@ class V1UserFeatures(object):
             self.lambda_labs_studios = lambda_labs_studios
         if landing_studios is not None:
             self.landing_studios = landing_studios
-        if lightning_cloud is not None:
-            self.lightning_cloud = lightning_cloud
         if lit_logger is not None:
             self.lit_logger = lit_logger
         if marketplace is not None:
@@ -430,6 +384,8 @@ class V1UserFeatures(object):
             self.mmt_fault_tolerance = mmt_fault_tolerance
         if mmt_strategy_selector is not None:
             self.mmt_strategy_selector = mmt_strategy_selector
+        if model_api_dashboard is not None:
+            self.model_api_dashboard = model_api_dashboard
         if multicloud_folders is not None:
             self.multicloud_folders = multicloud_folders
         if multicloud_saas is not None:
@@ -442,8 +398,6 @@ class V1UserFeatures(object):
             self.nebius_cpu_studios = nebius_cpu_studios
         if nebius_gpu_studios is not None:
             self.nebius_gpu_studios = nebius_gpu_studios
-        if neocloud_studios is not None:
-            self.neocloud_studios = neocloud_studios
         if nerf_fs_nonpaying is not None:
             self.nerf_fs_nonpaying = nerf_fs_nonpaying
         if org_level_member_permissions is not None:
@@ -500,8 +454,6 @@ class V1UserFeatures(object):
             self.slurm = slurm
         if slurm_machine_selector is not None:
             self.slurm_machine_selector = slurm_machine_selector
-        if stop_ide_container_on_shutdown is not None:
-            self.stop_ide_container_on_shutdown = stop_ide_container_on_shutdown
         if storage_overuse_deletion is not None:
             self.storage_overuse_deletion = storage_overuse_deletion
         if studio_config is not None:
@@ -512,8 +464,6 @@ class V1UserFeatures(object):
             self.teamspace_storage_tab = teamspace_storage_tab
         if trainium2 is not None:
             self.trainium2 = trainium2
-        if transfer_studios is not None:
-            self.transfer_studios = transfer_studios
         if use_rclone_mounts_only is not None:
             self.use_rclone_mounts_only = use_rclone_mounts_only
         if voltage_park is not None:
@@ -714,48 +664,6 @@ class V1UserFeatures(object):
 
         self._b2c_experience = b2c_experience
 
-    @property
-    def blog_route(self) -> 'bool':
-        """Gets the blog_route of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The blog_route of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._blog_route
-
-    @blog_route.setter
-    def blog_route(self, blog_route: 'bool'):
-        """Sets the blog_route of this V1UserFeatures.
-
-
-        :param blog_route: The blog_route of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._blog_route = blog_route
-
-    @property
-    def byoc_litcr(self) -> 'bool':
-        """Gets the byoc_litcr of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The byoc_litcr of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._byoc_litcr
-
-    @byoc_litcr.setter
-    def byoc_litcr(self, byoc_litcr: 'bool'):
-        """Sets the byoc_litcr of this V1UserFeatures.
-
-
-        :param byoc_litcr: The byoc_litcr of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._byoc_litcr = byoc_litcr
-
     @property
     def cap_add(self) -> 'list[str]':
         """Gets the cap_add of this V1UserFeatures.  # noqa: E501
@@ -1071,27 +979,6 @@ class V1UserFeatures(object):
 
         self._default_one_cluster = default_one_cluster
 
-    @property
-    def deployment_alerts(self) -> 'bool':
-        """Gets the deployment_alerts of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The deployment_alerts of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._deployment_alerts
-
-    @deployment_alerts.setter
-    def deployment_alerts(self, deployment_alerts: 'bool'):
-        """Sets the deployment_alerts of this V1UserFeatures.
-
-
-        :param deployment_alerts: The deployment_alerts of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._deployment_alerts = deployment_alerts
-
     @property
     def deployment_persistent_disk(self) -> 'bool':
         """Gets the deployment_persistent_disk of this V1UserFeatures.  # noqa: E501
@@ -1113,48 +1000,6 @@ class V1UserFeatures(object):
 
         self._deployment_persistent_disk = deployment_persistent_disk
 
-    @property
-    def deployment_requests_tab(self) -> 'bool':
-        """Gets the deployment_requests_tab of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The deployment_requests_tab of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._deployment_requests_tab
-
-    @deployment_requests_tab.setter
-    def deployment_requests_tab(self, deployment_requests_tab: 'bool'):
-        """Sets the deployment_requests_tab of this V1UserFeatures.
-
-
-        :param deployment_requests_tab: The deployment_requests_tab of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._deployment_requests_tab = deployment_requests_tab
-
-    @property
-    def dgx_cloud(self) -> 'bool':
-        """Gets the dgx_cloud of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The dgx_cloud of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._dgx_cloud
-
-    @dgx_cloud.setter
-    def dgx_cloud(self, dgx_cloud: 'bool'):
-        """Sets the dgx_cloud of this V1UserFeatures.
-
-
-        :param dgx_cloud: The dgx_cloud of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._dgx_cloud = dgx_cloud
-
     @property
     def docs_agent(self) -> 'bool':
         """Gets the docs_agent of this V1UserFeatures.  # noqa: E501
@@ -1197,27 +1042,6 @@ class V1UserFeatures(object):
 
         self._drive_v2 = drive_v2
 
-    @property
-    def enable_storage_limits(self) -> 'bool':
-        """Gets the enable_storage_limits of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The enable_storage_limits of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._enable_storage_limits
-
-    @enable_storage_limits.setter
-    def enable_storage_limits(self, enable_storage_limits: 'bool'):
-        """Sets the enable_storage_limits of this V1UserFeatures.
-
-
-        :param enable_storage_limits: The enable_storage_limits of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._enable_storage_limits = enable_storage_limits
-
     @property
     def enterprise_compute_admin(self) -> 'bool':
         """Gets the enterprise_compute_admin of this V1UserFeatures.  # noqa: E501
@@ -1281,27 +1105,6 @@ class V1UserFeatures(object):
 
         self._featured_studios_admin = featured_studios_admin
 
-    @property
-    def filestore(self) -> 'bool':
-        """Gets the filestore of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The filestore of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._filestore
-
-    @filestore.setter
-    def filestore(self, filestore: 'bool'):
-        """Sets the filestore of this V1UserFeatures.
-
-
-        :param filestore: The filestore of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._filestore = filestore
-
     @property
     def gcs_folders(self) -> 'bool':
         """Gets the gcs_folders of this V1UserFeatures.  # noqa: E501
@@ -1428,27 +1231,6 @@ class V1UserFeatures(object):
 
         self._landing_studios = landing_studios
 
-    @property
-    def lightning_cloud(self) -> 'bool':
-        """Gets the lightning_cloud of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The lightning_cloud of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._lightning_cloud
-
-    @lightning_cloud.setter
-    def lightning_cloud(self, lightning_cloud: 'bool'):
-        """Sets the lightning_cloud of this V1UserFeatures.
-
-
-        :param lightning_cloud: The lightning_cloud of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._lightning_cloud = lightning_cloud
-
     @property
     def lit_logger(self) -> 'bool':
         """Gets the lit_logger of this V1UserFeatures.  # noqa: E501
@@ -1533,6 +1315,27 @@ class V1UserFeatures(object):
 
         self._mmt_strategy_selector = mmt_strategy_selector
 
+    @property
+    def model_api_dashboard(self) -> 'bool':
+        """Gets the model_api_dashboard of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The model_api_dashboard of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._model_api_dashboard
+
+    @model_api_dashboard.setter
+    def model_api_dashboard(self, model_api_dashboard: 'bool'):
+        """Sets the model_api_dashboard of this V1UserFeatures.
+
+
+        :param model_api_dashboard: The model_api_dashboard of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._model_api_dashboard = model_api_dashboard
+
     @property
     def multicloud_folders(self) -> 'bool':
         """Gets the multicloud_folders of this V1UserFeatures.  # noqa: E501
@@ -1659,27 +1462,6 @@ class V1UserFeatures(object):
 
         self._nebius_gpu_studios = nebius_gpu_studios
 
-    @property
-    def neocloud_studios(self) -> 'bool':
-        """Gets the neocloud_studios of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The neocloud_studios of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._neocloud_studios
-
-    @neocloud_studios.setter
-    def neocloud_studios(self, neocloud_studios: 'bool'):
-        """Sets the neocloud_studios of this V1UserFeatures.
-
-
-        :param neocloud_studios: The neocloud_studios of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._neocloud_studios = neocloud_studios
-
     @property
     def nerf_fs_nonpaying(self) -> 'bool':
         """Gets the nerf_fs_nonpaying of this V1UserFeatures.  # noqa: E501
@@ -2268,27 +2050,6 @@ class V1UserFeatures(object):
 
         self._slurm_machine_selector = slurm_machine_selector
 
-    @property
-    def stop_ide_container_on_shutdown(self) -> 'bool':
-        """Gets the stop_ide_container_on_shutdown of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The stop_ide_container_on_shutdown of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._stop_ide_container_on_shutdown
-
-    @stop_ide_container_on_shutdown.setter
-    def stop_ide_container_on_shutdown(self, stop_ide_container_on_shutdown: 'bool'):
-        """Sets the stop_ide_container_on_shutdown of this V1UserFeatures.
-
-
-        :param stop_ide_container_on_shutdown: The stop_ide_container_on_shutdown of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._stop_ide_container_on_shutdown = stop_ide_container_on_shutdown
-
     @property
     def storage_overuse_deletion(self) -> 'bool':
         """Gets the storage_overuse_deletion of this V1UserFeatures.  # noqa: E501
@@ -2394,27 +2155,6 @@ class V1UserFeatures(object):
 
         self._trainium2 = trainium2
 
-    @property
-    def transfer_studios(self) -> 'bool':
-        """Gets the transfer_studios of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The transfer_studios of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._transfer_studios
-
-    @transfer_studios.setter
-    def transfer_studios(self, transfer_studios: 'bool'):
-        """Sets the transfer_studios of this V1UserFeatures.
-
-
-        :param transfer_studios: The transfer_studios of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._transfer_studios = transfer_studios
-
     @property
     def use_rclone_mounts_only(self) -> 'bool':
         """Gets the use_rclone_mounts_only of this V1UserFeatures.  # noqa: E501

lightning_sdk/llm/__init__.py

@@ -1,3 +1,4 @@
+from lightning_sdk.llm.asyncllm import AsyncLLM
 from lightning_sdk.llm.llm import LLM
 
-__all__ = ["LLM"]
+__all__ = ["LLM", "AsyncLLM"]

lightning_sdk/llm/asyncllm.py

@@ -0,0 +1,48 @@
+import os
+from typing import AsyncGenerator, Dict, List, Optional, Union
+
+from lightning_sdk.llm.llm import LLM
+
+
+class AsyncLLM(LLM):
+    async def chat(
+        self,
+        prompt: str,
+        system_prompt: Optional[str] = None,
+        max_completion_tokens: Optional[int] = 500,
+        images: Optional[Union[List[str], str]] = None,
+        conversation: Optional[str] = None,
+        metadata: Optional[Dict[str, str]] = None,
+        stream: bool = False,
+        upload_local_images: bool = False,
+    ) -> Union[str, AsyncGenerator[str, None]]:
+        if conversation and conversation not in self._conversations:
+            self._get_conversations()
+
+        if images:
+            if isinstance(images, str):
+                images = [images]
+            for image in images:
+                if not isinstance(image, str):
+                    raise NotImplementedError(f"Image type {type(image)} are not supported yet.")
+                if not image.startswith("http") and upload_local_images:
+                    self._teamspace.upload_file(file_path=image, remote_path=f"images/{os.path.basename(image)}")
+
+        conversation_id = self._conversations.get(conversation) if conversation else None
+        output = await self._llm_api.async_start_conversation(
+            prompt=prompt,
+            system_prompt=system_prompt,
+            max_completion_tokens=max_completion_tokens,
+            images=images,
+            assistant_id=self._model.id,
+            conversation_id=conversation_id,
+            billing_project_id=self._teamspace.id,
+            metadata=metadata,
+            name=conversation,
+            stream=stream,
+        )
+        if not stream:
+            if conversation and not conversation_id:
+                self._conversations[conversation] = output.conversation_id
+            return output.choices[0].delta.content
+        return self._stream_chat_response(output, conversation=conversation)

lightning_sdk/services/utilities.py

@@ -35,7 +35,9 @@ def _get_project(client: LightningClient, project_name: Optional[str] = None) ->
     raise ValueError("No valid projects found. Please reach out to lightning.ai team to create a project")
 
 
-def _get_cluster(client: LightningClient, project_id: str, cluster_id: Optional[str] = None) -> V1ProjectClusterBinding:
+def _get_cluster(
+    client: LightningClient, project_id: str, cluster_id: Optional[str] = None, allow_neoclouds: bool = False
+) -> V1ProjectClusterBinding:
     """Get a project membership for the user from the backend."""
     clusters = client.projects_service_list_project_cluster_bindings(project_id=project_id)
     if cluster_id:
@@ -47,6 +49,18 @@ def _get_cluster(client: LightningClient, project_id: str, cluster_id: Optional[
             f"Found {[c.cluster_id for c in clusters.clusters]}."
         )
 
+    # filter neoclouds out
+    if not allow_neoclouds:
+        cluster_objs = client.cluster_service_list_clusters(project_id=project_id)
+        # filter for aws or gcp cluster
+        valid_clusters = filter(
+            lambda c: c.spec.aws_v1 is not None or c.spec.google_cloud_v1 is not None, cluster_objs.clusters
+        )
+        valid_clusters = {c.id for c in valid_clusters}
+
+        if len(valid_clusters):
+            clusters.clusters = list(filter(lambda c: c.cluster_id in valid_clusters, clusters.clusters))
+
     clusters = sorted(clusters.clusters, key=lambda x: x.created_at)
     if len(clusters):
         return clusters[0]