lightning-sdk 0.2.24rc1__py3-none-any.whl → 2025.7.9__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.24.rc1"
+__version__ = "2025.07.09"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/cluster_api.py

@@ -4,6 +4,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     Externalv1Cluster,
     V1CloudProvider,
     V1ClusterType,
+    V1ExternalCluster,
     V1ListClusterAcceleratorsResponse,
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
@@ -28,6 +29,20 @@ class ClusterApi:
             raise ValueError(f"Cluster {cluster_id} does not exist")
         return res
 
+    def list_clusters(self, project_id: str) -> List[V1ExternalCluster]:
+        """Lists the clusters for a given project.
+
+        Args:
+            project_id: The project to list clusters for
+
+        Returns:
+            A list of clusters
+        """
+        res = self._client.cluster_service_list_project_clusters(
+            project_id=project_id,
+        )
+        return res.clusters
+
     def list_cluster_accelerators(self, cluster_id: str, org_id: str) -> V1ListClusterAcceleratorsResponse:
         """Lists the accelerators for a given cluster.
 

lightning_sdk/api/pipeline_api.py

@@ -55,7 +55,7 @@ class PipelineApi:
         body = ProjectIdPipelinesBody(
             name=name,
             steps=steps,
-            shared_filesystem=self._prepare_shared_filesytem(shared_filesystem, steps, teamspace),
+            shared_filesystem=self._prepare_shared_filesystem(shared_filesystem, steps, teamspace),
             parent_pipeline_id=parent_pipeline_id or "",
         )
 
@@ -89,7 +89,7 @@ class PipelineApi:
     def delete(self, project_id: str, pipeline_id: str) -> V1DeletePipelineResponse:
         return self._client.pipelines_service_delete_pipeline(project_id, pipeline_id)
 
-    def _prepare_shared_filesytem(
+    def _prepare_shared_filesystem(
         self, shared_filesystem: Union[bool, V1SharedFilesystem], steps: List["V1PipelineStep"], teamspace: Teamspace
     ) -> V1SharedFilesystem:
         if not shared_filesystem:
@@ -97,14 +97,22 @@ class PipelineApi:
 
         from lightning_sdk.pipeline.utils import _get_cloud_account
 
-        cluster = self._cluster_api.get_cluster(
-            cluster_id=_get_cloud_account(steps), project_id=teamspace.id, org_id=teamspace.owner.id
-        )
+        clusters = self._cluster_api.list_clusters(project_id=teamspace.id)
+
+        selected_cluster = None
+        selected_cluster_id = _get_cloud_account(steps)
+        for cluster in clusters:
+            if cluster.id == selected_cluster_id:
+                selected_cluster = cluster
+                break
+
+        if selected_cluster is None:
+            raise ValueError(f"Cloud Account {selected_cluster_id} not found")
 
-        if cluster.spec.aws_v1:
+        if selected_cluster.spec.aws_v1:
             return V1SharedFilesystem(enabled=True, s3_folder=True)
 
-        if cluster.spec.google_cloud_v1:
+        if selected_cluster.spec.google_cloud_v1:
             return V1SharedFilesystem(enabled=True, gcs_folder=True)
 
         raise NotImplementedError("This cluster isn't support yet")

lightning_sdk/api/studio_api.py

@@ -2,7 +2,6 @@ import json
 import os
 import tempfile
 import time
-import warnings
 import zipfile
 from threading import Event, Thread
 from typing import Any, Dict, Generator, Mapping, Optional, Tuple, Union
@@ -172,14 +171,6 @@ class StudioApi:
         self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: False
     ) -> None:
         """Start an existing Studio."""
-        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
-            warnings.warn(
-                f"{Machine.CPU_SMALL} is not a valid machine for starting a Studio. "
-                "It is reserved for running jobs only. "
-                "The Studio will be started with a CPU machine instead."
-            )
-            machine = Machine.CPU
-
         self._client.cloud_space_service_start_cloud_space_instance(
             IdStartBody(
                 compute_config=V1UserRequestedComputeConfig(name=_machine_to_compute_name(machine), spot=interruptible)
@@ -229,14 +220,6 @@ class StudioApi:
         self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
     ) -> None:
         """Switches given Studio to a new machine type."""
-        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
-            warnings.warn(
-                f"{Machine.CPU_SMALL} is not a valid machine for switching a Studio. "
-                "It is reserved for running jobs only. "
-                "The Studio will be switched to a CPU machine instead."
-            )
-            machine = Machine.CPU
-
         compute_name = _machine_to_compute_name(machine)
         # TODO: UI sends disk size here, maybe we need to also?
         body = IdCodeconfigBody(compute_config=V1UserRequestedComputeConfig(name=compute_name, spot=interruptible))

lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py

@@ -53,6 +53,7 @@ class AssistantIdConversationsBody(object):
         'parent_conversation_id': 'str',
         'parent_message_id': 'str',
         'reasoning_effort': 'str',
+        'sent_at': 'datetime',
         'store': 'bool',
         'stream': 'bool',
         'system_prompt': 'str'
@@ -71,12 +72,13 @@ class AssistantIdConversationsBody(object):
         'parent_conversation_id': 'parentConversationId',
         'parent_message_id': 'parentMessageId',
         'reasoning_effort': 'reasoningEffort',
+        'sent_at': 'sentAt',
         'store': 'store',
         'stream': 'stream',
         'system_prompt': 'systemPrompt'
     }
 
-    def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None):  # noqa: E501
+    def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, sent_at: 'datetime' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None):  # noqa: E501
         """AssistantIdConversationsBody - a model defined in Swagger"""  # noqa: E501
         self._auto_name = None
         self._billing_project_id = None
@@ -90,6 +92,7 @@ class AssistantIdConversationsBody(object):
         self._parent_conversation_id = None
         self._parent_message_id = None
         self._reasoning_effort = None
+        self._sent_at = None
         self._store = None
         self._stream = None
         self._system_prompt = None
@@ -118,6 +121,8 @@ class AssistantIdConversationsBody(object):
             self.parent_message_id = parent_message_id
         if reasoning_effort is not None:
             self.reasoning_effort = reasoning_effort
+        if sent_at is not None:
+            self.sent_at = sent_at
         if store is not None:
             self.store = store
         if stream is not None:
@@ -377,6 +382,27 @@ class AssistantIdConversationsBody(object):
 
         self._reasoning_effort = reasoning_effort
 
+    @property
+    def sent_at(self) -> 'datetime':
+        """Gets the sent_at of this AssistantIdConversationsBody.  # noqa: E501
+
+
+        :return: The sent_at of this AssistantIdConversationsBody.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._sent_at
+
+    @sent_at.setter
+    def sent_at(self, sent_at: 'datetime'):
+        """Sets the sent_at of this AssistantIdConversationsBody.
+
+
+        :param sent_at: The sent_at of this AssistantIdConversationsBody.  # noqa: E501
+        :type: datetime
+        """
+
+        self._sent_at = sent_at
+
     @property
     def store(self) -> 'bool':
         """Gets the store of this AssistantIdConversationsBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py

@@ -46,6 +46,7 @@ class V1ConversationResponseChunk(object):
         'executable': 'bool',
         'id': 'str',
         'object': 'str',
+        'stats': 'dict(str, str)',
         'throughput': 'float',
         'usage': 'V1TokenUsage'
     }
@@ -56,17 +57,19 @@ class V1ConversationResponseChunk(object):
         'executable': 'executable',
         'id': 'id',
         'object': 'object',
+        'stats': 'stats',
         'throughput': 'throughput',
         'usage': 'usage'
     }
 
-    def __init__(self, choices: 'list[V1ResponseChoice]' =None, conversation_id: 'str' =None, executable: 'bool' =None, id: 'str' =None, object: 'str' =None, throughput: 'float' =None, usage: 'V1TokenUsage' =None):  # noqa: E501
+    def __init__(self, choices: 'list[V1ResponseChoice]' =None, conversation_id: 'str' =None, executable: 'bool' =None, id: 'str' =None, object: 'str' =None, stats: 'dict(str, str)' =None, throughput: 'float' =None, usage: 'V1TokenUsage' =None):  # noqa: E501
         """V1ConversationResponseChunk - a model defined in Swagger"""  # noqa: E501
         self._choices = None
         self._conversation_id = None
         self._executable = None
         self._id = None
         self._object = None
+        self._stats = None
         self._throughput = None
         self._usage = None
         self.discriminator = None
@@ -80,6 +83,8 @@ class V1ConversationResponseChunk(object):
             self.id = id
         if object is not None:
             self.object = object
+        if stats is not None:
+            self.stats = stats
         if throughput is not None:
             self.throughput = throughput
         if usage is not None:
@@ -190,6 +195,27 @@ class V1ConversationResponseChunk(object):
 
         self._object = object
 
+    @property
+    def stats(self) -> 'dict(str, str)':
+        """Gets the stats of this V1ConversationResponseChunk.  # noqa: E501
+
+
+        :return: The stats of this V1ConversationResponseChunk.  # noqa: E501
+        :rtype: dict(str, str)
+        """
+        return self._stats
+
+    @stats.setter
+    def stats(self, stats: 'dict(str, str)'):
+        """Sets the stats of this V1ConversationResponseChunk.
+
+
+        :param stats: The stats of this V1ConversationResponseChunk.  # noqa: E501
+        :type: dict(str, str)
+        """
+
+        self._stats = stats
+
     @property
     def throughput(self) -> 'float':
         """Gets the throughput of this V1ConversationResponseChunk.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -81,12 +81,12 @@ class V1UserFeatures(object):
         'lambda_labs_studios': 'bool',
         'landing_studios': 'bool',
         'lit_logger': 'bool',
-        'litcr_byoc_gcp': 'bool',
         'machine_selector_v2': 'bool',
         'marketplace': 'bool',
         'mmt_fault_tolerance': 'bool',
         'mmt_strategy_selector': 'bool',
         'model_api_dashboard': 'bool',
+        'model_api_dashboard_clickhouse': 'bool',
         'multicloud_folders': 'bool',
         'multicloud_saas': 'bool',
         'multiple_studio_versions': 'bool',
@@ -175,12 +175,12 @@ class V1UserFeatures(object):
         'lambda_labs_studios': 'lambdaLabsStudios',
         'landing_studios': 'landingStudios',
         'lit_logger': 'litLogger',
-        'litcr_byoc_gcp': 'litcrByocGcp',
         'machine_selector_v2': 'machineSelectorV2',
         'marketplace': 'marketplace',
         'mmt_fault_tolerance': 'mmtFaultTolerance',
         'mmt_strategy_selector': 'mmtStrategySelector',
         'model_api_dashboard': 'modelApiDashboard',
+        'model_api_dashboard_clickhouse': 'modelApiDashboardClickhouse',
         'multicloud_folders': 'multicloudFolders',
         'multicloud_saas': 'multicloudSaas',
         'multiple_studio_versions': 'multipleStudioVersions',
@@ -228,7 +228,7 @@ class V1UserFeatures(object):
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, academic_tier: 'bool' =None, accurate_billing: 'bool' =None, add_data_v2: '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, 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, cloudspace_schedules: '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, data_connection_flushing_v2: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, gcs_fuse: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_clusters: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, litcr_byoc_gcp: 'bool' =None, machine_selector_v2: '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, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, persistent_disk: '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, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_internal_data_connection_mounts: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, academic_tier: 'bool' =None, accurate_billing: 'bool' =None, add_data_v2: '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, 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, cloudspace_schedules: '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, data_connection_flushing_v2: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, gcs_fuse: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_clusters: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, machine_selector_v2: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, model_api_dashboard_clickhouse: '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, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, persistent_disk: '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, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_internal_data_connection_mounts: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._academic_tier = None
         self._accurate_billing = None
@@ -270,12 +270,12 @@ class V1UserFeatures(object):
         self._lambda_labs_studios = None
         self._landing_studios = None
         self._lit_logger = None
-        self._litcr_byoc_gcp = None
         self._machine_selector_v2 = None
         self._marketplace = None
         self._mmt_fault_tolerance = None
         self._mmt_strategy_selector = None
         self._model_api_dashboard = None
+        self._model_api_dashboard_clickhouse = None
         self._multicloud_folders = None
         self._multicloud_saas = None
         self._multiple_studio_versions = None
@@ -402,8 +402,6 @@ class V1UserFeatures(object):
             self.landing_studios = landing_studios
         if lit_logger is not None:
             self.lit_logger = lit_logger
-        if litcr_byoc_gcp is not None:
-            self.litcr_byoc_gcp = litcr_byoc_gcp
         if machine_selector_v2 is not None:
             self.machine_selector_v2 = machine_selector_v2
         if marketplace is not None:
@@ -414,6 +412,8 @@ class V1UserFeatures(object):
             self.mmt_strategy_selector = mmt_strategy_selector
         if model_api_dashboard is not None:
             self.model_api_dashboard = model_api_dashboard
+        if model_api_dashboard_clickhouse is not None:
+            self.model_api_dashboard_clickhouse = model_api_dashboard_clickhouse
         if multicloud_folders is not None:
             self.multicloud_folders = multicloud_folders
         if multicloud_saas is not None:
@@ -1345,27 +1345,6 @@ class V1UserFeatures(object):
 
         self._lit_logger = lit_logger
 
-    @property
-    def litcr_byoc_gcp(self) -> 'bool':
-        """Gets the litcr_byoc_gcp of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The litcr_byoc_gcp of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._litcr_byoc_gcp
-
-    @litcr_byoc_gcp.setter
-    def litcr_byoc_gcp(self, litcr_byoc_gcp: 'bool'):
-        """Sets the litcr_byoc_gcp of this V1UserFeatures.
-
-
-        :param litcr_byoc_gcp: The litcr_byoc_gcp of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._litcr_byoc_gcp = litcr_byoc_gcp
-
     @property
     def machine_selector_v2(self) -> 'bool':
         """Gets the machine_selector_v2 of this V1UserFeatures.  # noqa: E501
@@ -1471,6 +1450,27 @@ class V1UserFeatures(object):
 
         self._model_api_dashboard = model_api_dashboard
 
+    @property
+    def model_api_dashboard_clickhouse(self) -> 'bool':
+        """Gets the model_api_dashboard_clickhouse of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The model_api_dashboard_clickhouse of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._model_api_dashboard_clickhouse
+
+    @model_api_dashboard_clickhouse.setter
+    def model_api_dashboard_clickhouse(self, model_api_dashboard_clickhouse: 'bool'):
+        """Sets the model_api_dashboard_clickhouse of this V1UserFeatures.
+
+
+        :param model_api_dashboard_clickhouse: The model_api_dashboard_clickhouse of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._model_api_dashboard_clickhouse = model_api_dashboard_clickhouse
+
     @property
     def multicloud_folders(self) -> 'bool':
         """Gets the multicloud_folders of this V1UserFeatures.  # noqa: E501

lightning_sdk/machine.py

@@ -5,7 +5,6 @@ from typing import Any, ClassVar, Optional, Tuple
 @dataclass(frozen=True)
 class Machine:
     # Default Machines
-    CPU_SMALL: ClassVar["Machine"]
     CPU: ClassVar["Machine"]
     DATA_PREP: ClassVar["Machine"]
     DATA_PREP_MAX: ClassVar["Machine"]
@@ -45,8 +44,7 @@ class Machine:
     def is_cpu(self) -> bool:
         """Whether the machine is a CPU."""
         return (
-            self == Machine.CPU_SMALL
-            or self == Machine.CPU
+            self == Machine.CPU
             or self == Machine.DATA_PREP
             or self == Machine.DATA_PREP_MAX
             or self == Machine.DATA_PREP_ULTRA
@@ -67,7 +65,6 @@ class Machine:
         return cls(machine, machine)
 
 
-Machine.CPU_SMALL = Machine(name="CPU_SMALL", instance_type="m3.medium")
 Machine.CPU = Machine(name="CPU", instance_type="cpu-4")
 Machine.DATA_PREP = Machine(name="DATA_PREP", instance_type="data-large")
 Machine.DATA_PREP_MAX = Machine(name="DATA_PREP_MAX", instance_type="data-max")

lightning_sdk/pipeline/pipeline.py

@@ -1,9 +1,7 @@
 import os
 from typing import TYPE_CHECKING, List, Optional, Union
 
-from lightning_sdk.api import UserApi
 from lightning_sdk.api.pipeline_api import PipelineApi
-from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.organization import Organization
 from lightning_sdk.pipeline.printer import PipelinePrinter
 from lightning_sdk.pipeline.steps import DeploymentStep, JobStep, MMTStep, _get_studio
@@ -12,7 +10,7 @@ from lightning_sdk.services.utilities import _get_cluster
 from lightning_sdk.studio import Studio
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.user import User
-from lightning_sdk.utils.resolve import _resolve_org, _resolve_teamspace, _resolve_user
+from lightning_sdk.utils.resolve import _resolve_teamspace
 
 if TYPE_CHECKING:
     from lightning_sdk.pipeline.schedule import Schedule
@@ -40,19 +38,7 @@ class Pipeline:
             shared_filesystem: Whether the pipeline should use a shared filesystem across all nodes.
                 Note: This forces the pipeline steps to be in the cloud_account and same region
         """
-        self._auth = Auth()
-        self._user = None
-
-        try:
-            self._auth.authenticate()
-            if user is None:
-                self._user = User(name=UserApi()._get_user_by_id(self._auth.user_id).username)
-        except ConnectionError as e:
-            raise e
-
         self._name = name
-        self._org = _resolve_org(org)
-        self._user = _resolve_user(self._user or user)
 
         self._teamspace = _resolve_teamspace(
             teamspace=teamspace,

{lightning_sdk-0.2.24rc1.dist-info → lightning_sdk-2025.7.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.2.24rc1
+Version: 2025.7.9
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License

{lightning_sdk-0.2.24rc1.dist-info → lightning_sdk-2025.7.9.dist-info}/RECORD

@@ -1,12 +1,12 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=eOQXMuFcIp5MYfSKoBHS3BG9KbBoanr7fy0bbHkd0Qo,1109
+lightning_sdk/__init__.py,sha256=9BSQ7hp-GqfVziz3SrmjcYVugWAR5Po4PmfDEhFBVQI,1109
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
 lightning_sdk/base_studio.py,sha256=_Pwwl37R9GRd7t-f2kO5aQXiLNrP4sUtUNht2ZkP8LE,3678
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
 lightning_sdk/helpers.py,sha256=KWMWnORHItIIA3PGn71YPs-7RjzGi8IXa2kQ5Qo4U8M,2459
 lightning_sdk/lit_container.py,sha256=8ys49TXI9MO89jLTA7MwDrKrssTRARAIF9OVmolDLq0,5273
-lightning_sdk/machine.py,sha256=EMr-ulyYEYhIkKFnGBOofnFf4asndTSeoOQxlyd3xW4,3632
+lightning_sdk/machine.py,sha256=WNzUUQzThSY34tnHPUXZ_FeG4fDamPAz6LnC4yswqFI,3483
 lightning_sdk/models.py,sha256=tPFiiDQiDsFhfStcVPH5_PcujbBEfKS2IB5FS9NfwZk,7838
 lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
 lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
@@ -21,7 +21,7 @@ lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
 lightning_sdk/api/ai_hub_api.py,sha256=azqDZ-PzasVAcoQHno7k7OO_xFOHQ4NDozxF8jEh83Y,7864
 lightning_sdk/api/base_studio_api.py,sha256=3R8tucZX2e9yKHBcY2rWFRP4dxqLrC6H75vdBDkH0ck,3617
-lightning_sdk/api/cluster_api.py,sha256=mfHGKHy-RNpI7nKlhw9Zc7Jv6f4DCFq-qRIwOR3VptY,3972
+lightning_sdk/api/cluster_api.py,sha256=kAZeJbcNYG9rsYE7if5k1HlcND1Z_6cKivZ7lY4S20U,4392
 lightning_sdk/api/deployment_api.py,sha256=KvBwGaG896q7WS1OPMOIjFUh05W2g7S1djS2XJMVieQ,24215
 lightning_sdk/api/job_api.py,sha256=_mMAI_BG_48i-BLwCP_U72zgmM5zYa2KUZ7u66HWkIc,13568
 lightning_sdk/api/license_api.py,sha256=XV3RhefyPQDYjwY9AaBZe4rByZTEAnsvLDxcdm9q0Wo,2438
@@ -29,8 +29,8 @@ lightning_sdk/api/lit_container_api.py,sha256=jCJVwd-3MNjejL9FyvH89pzt-SeG3G8qCJ
 lightning_sdk/api/llm_api.py,sha256=G8WKTslOIS8cTYFoOlQNRgtIvzEpqIr2Ez0rxcMAh4k,9100
 lightning_sdk/api/mmt_api.py,sha256=-v7ATab-ThAM-HRClS92Ehxuu9MlBfdKWWFCGvVUHiM,8962
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
-lightning_sdk/api/pipeline_api.py,sha256=OUt_YSKFd7O5mQJKhWJbzvk2TDXicodC0PVGTjn-kkU,4180
-lightning_sdk/api/studio_api.py,sha256=K_1tMM__wR0l4CFAiS6UWrFwNoMnMz8F_MDfmbV7hYI,31488
+lightning_sdk/api/pipeline_api.py,sha256=oS0WIBB748SLD7bqkEQ8rnkixrsAML_M01OUunmLFSQ,4470
+lightning_sdk/api/studio_api.py,sha256=J-kp6IoXlQaVz1S4BtaK93VrLAbMOolQYDGJyrkPDRo,30702
 lightning_sdk/api/teamspace_api.py,sha256=CsaaxmaLmTRIRwu37wtQ3quGYql62HJT3BZJ3Q-1d9c,16854
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
 lightning_sdk/api/utils.py,sha256=K0Uc6D1KxxjFElqjsuCXg85SYreQi3P7jJ1VzP0MpEQ,23966
@@ -138,7 +138,7 @@ lightning_sdk/lightning_cloud/openapi/models/appinstances_id_body.py,sha256=DfwY
 lightning_sdk/lightning_cloud/openapi/models/approveautojoindomain_domain_body.py,sha256=gC2puM75rrgB36ekVmgF9rnzWE4mAp6tpaGUrlhARRU,3771
 lightning_sdk/lightning_cloud/openapi/models/apps_id_body.py,sha256=tjIpEmU6Y8nhLDT0c3JAdywxEjI-NzvgLdEhf2UXWf4,17912
 lightning_sdk/lightning_cloud/openapi/models/apps_id_body1.py,sha256=Ja_1s0PjD1hIjAvyqyp8U4WXKv2GoUGoQ-HHm8hvRbc,8874
-lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py,sha256=RmDAeuHhyqiMvZoZNP3w64p6n1HDHrI2PWA0rfgk8gg,15551
+lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py,sha256=XRAgtHlFxqIhML1z_4kGNvlrOCQQph3kjm9JWIr-ijQ,16305
 lightning_sdk/lightning_cloud/openapi/models/billing_checkout_body.py,sha256=bMABc4XkRzK-t2RzWg1bd2YZSwZkqYZoUex9iWO1lPI,5379
 lightning_sdk/lightning_cloud/openapi/models/billing_transfer_body.py,sha256=MaovoJiJDzOXNhCskZAN5UNwnFEFP7ONF83DPRgfhgk,4510
 lightning_sdk/lightning_cloud/openapi/models/billing_transfer_body1.py,sha256=6zrWBAHmmKHJgqJtxahcM1mDrO3Z55WxaGK0RY10oOw,5275
@@ -411,7 +411,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_compute_config.py,sha256=YGkx66e
 lightning_sdk/lightning_cloud/openapi/models/v1_contact_assistant_owner_reason.py,sha256=9J2PuNGXlDXMcsmINbocUYGXVKhRlvqDdRwgtAH_qPU,3121
 lightning_sdk/lightning_cloud/openapi/models/v1_contact_assistant_owner_response.py,sha256=_FXW0hslMncLSPPxDmWEbhePRz-4tFc98KpqMf0M_6s,3076
 lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py,sha256=xINDLIVGuMEXSog0ryP3rQW1y_9PmEIMsL7P7LHzb_M,9220
-lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py,sha256=kMBvhXdYj4yE_ccxKfcSeL1zvSfyl5b8u4rvzBwmRAo,8402
+lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py,sha256=7JovH85aN8mwC_gUVdCBDyqUCwwNlsEAqkBGKMTmwMo,9149
 lightning_sdk/lightning_cloud/openapi/models/v1_count_metrics_streams_response.py,sha256=ovqawh81CI7PvAqcElhGYw0-NVwSzwbm4s2PFErdxHA,3751
 lightning_sdk/lightning_cloud/openapi/models/v1_cpu_system_metrics.py,sha256=Mgybx_EGelKpTVzfW9WXp5HiiVMLPoes8vdVq8We1TU,10120
 lightning_sdk/lightning_cloud/openapi/models/v1_create_agent_multipart_upload_response.py,sha256=Ov9pBANdLtAbfz0ed_N-UGBeUwZozYKuqbst0F8u2Cw,3900
@@ -1027,7 +1027,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_upstream_open_ai.py,sha256=jt1qQ
 lightning_sdk/lightning_cloud/openapi/models/v1_usage.py,sha256=ozMzoGD9mfZGYy4J5j50Dps19Y6o8cn-aYW_oRMZMy8,16865
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py,sha256=U7qC698Xj5tb3D93ZskG6sDf3lTXE13UTlGeDTvtRU4,14062
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py,sha256=k9pDp9UIaOEEWz6bTNWF_KMfcNCOp-F67N-IZ9MO2Rs,8304
-lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=aTKyd0pXaiXsKEvaigU6tPNhqGQyzkJKZIDI8gklIs4,82298
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=5qLtkUfv5Vyf9xZaOmcDJWMnDKo7Jkjl5q2sNQq6HYU,82617
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py,sha256=0fnZpdhxNRXGNyILHtfm3rVztfHpSF4kXGc5kTt4zls,13960
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py,sha256=3WpZ-lf7xPwuYyQDMdP7Uc6-dh3vf5TaaUlcMfesfMk,5208
 lightning_sdk/lightning_cloud/openapi/models/v1_user_slurm_job_action_response.py,sha256=BdNzXH8Vsf5PHjl9Rd-TVkpAgx1YC9rf8LD0js-ba20,3058
@@ -1078,7 +1078,7 @@ lightning_sdk/mmt/mmt.py,sha256=swdGP1DOM42a_QmmY1vg3--6ZBDiC4zToAzU9Eycv4U,1344
 lightning_sdk/mmt/v1.py,sha256=TxLtL0ssDoP8eyleDaFyYr4evkOKbLJcckLVIfalOno,8429
 lightning_sdk/mmt/v2.py,sha256=Em1XBoqViqUMKm-sshzdMcSH5UTtZZwbJcsgqY6-mw0,9625
 lightning_sdk/pipeline/__init__.py,sha256=Sja_0NJ8vgh-2ThSVP3WDI9a9qghrWd21LkaQp4Zsp8,378
-lightning_sdk/pipeline/pipeline.py,sha256=xctjiZj4jIXr9klkOvnzYWYjcmLgN9nwKQKqPVOOiVM,5926
+lightning_sdk/pipeline/pipeline.py,sha256=yF8ebZlhOa0pWxLUpmKqW5oy4VEMT7NFSXpUhdpWbak,5425
 lightning_sdk/pipeline/printer.py,sha256=uNi3JD3RjQMVflJxik9o4CDfifROP70aV_vt6mms5zg,4671
 lightning_sdk/pipeline/schedule.py,sha256=r8L4M_NH5zoXoCf13jpn2icNIJ3hUgUFZrrq-ArnTWA,147
 lightning_sdk/pipeline/steps.py,sha256=5IM7BNiH42Rxq0sJNuwIN64qs-NCa81_XeP8pg_dCL8,12294
@@ -1092,9 +1092,9 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/resolve.py,sha256=6qlBUkOmcFTjhQx_CAGfnvWBbMYp6XrCV_sX_IqplLE,6748
-lightning_sdk-0.2.24rc1.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.2.24rc1.dist-info/METADATA,sha256=YXSCn2tDME1-uvxH6B5o4i8Ay2jPNxTRjQPz1GhcU1s,3995
-lightning_sdk-0.2.24rc1.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-0.2.24rc1.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.2.24rc1.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.2.24rc1.dist-info/RECORD,,
+lightning_sdk-2025.7.9.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.7.9.dist-info/METADATA,sha256=RNn6JYhfsIfEL1Cq8yizfZ5drW82S9vN8Eai0hytyvI,3994
+lightning_sdk-2025.7.9.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.7.9.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.7.9.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.7.9.dist-info/RECORD,,