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

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.24.rc1"
+__version__ = "2025.07.10"
 _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/llm_api.py

@@ -1,5 +1,6 @@
 import asyncio
 import base64
+import datetime
 import json
 import os
 import threading
@@ -46,6 +47,8 @@ class LLMApi:
                 executable=result_data.get("executable"),
                 id=result_data.get("id"),
                 throughput=result_data.get("throughput"),
+                stats=result_data.get("stats"),
+                usage=result_data.get("usage"),
             )
         except json.JSONDecodeError:
             warnings.warn("Error decoding JSON:", decoded_line)
@@ -158,6 +161,7 @@ class LLMApi:
             "ephemeral": ephemeral,
             "parent_conversation_id": kwargs.get("parent_conversation_id", ""),
             "parent_message_id": kwargs.get("parent_message_id", ""),
+            "sent_at": datetime.datetime.now(datetime.timezone.utc).isoformat(timespec="microseconds"),
         }
         if images:
             for image in images:

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/job/base.py

@@ -296,6 +296,27 @@ class _BaseJob(ABC):
 
             time.sleep(interval)
 
+    async def async_wait(self, interval: float = 5.0, timeout: Optional[float] = None) -> None:
+        """Waits for the job to be either completed, manually stopped or failed.
+
+        Args:
+            interval: The number of seconds to spend in-between status checks.
+            timeout: The maximum number of seconds to wait before raising an error. If None, waits forever.
+        """
+        import asyncio
+
+        from lightning_sdk.status import Status
+
+        start = asyncio.get_event_loop().time()
+        while True:
+            if self.status in (Status.Completed, Status.Stopped, Status.Failed):
+                break
+
+            if timeout is not None and asyncio.get_event_loop().time() - start > timeout:
+                raise TimeoutError("Job didn't finish within the provided timeout.")
+
+            await asyncio.sleep(interval)
+
     @property
     @abstractmethod
     def status(self) -> "Status":

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

@@ -42,7 +42,6 @@ class V1UserFeatures(object):
     """
     swagger_types = {
         'academic_tier': 'bool',
-        'accurate_billing': 'bool',
         'add_data_v2': 'bool',
         'affiliate_links': 'bool',
         'agents_v2': 'bool',
@@ -59,7 +58,6 @@ class V1UserFeatures(object):
         'cloudy_vibe_code': 'bool',
         'code_tab': 'bool',
         'collab_screen_sharing': 'bool',
-        'concurrent_gpu_limit': 'bool',
         'control_center_monitoring': 'bool',
         'cost_attribution_settings': 'bool',
         'custom_app_domain': 'bool',
@@ -76,17 +74,18 @@ class V1UserFeatures(object):
         'gcs_fuse': 'bool',
         'instant_capacity_reservation': 'bool',
         'job_artifacts_v2': 'bool',
+        'kubernetes_cluster_ui': 'bool',
         'kubernetes_clusters': 'bool',
         'lambda_labs': 'bool',
         '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',
@@ -136,7 +135,6 @@ class V1UserFeatures(object):
 
     attribute_map = {
         'academic_tier': 'academicTier',
-        'accurate_billing': 'accurateBilling',
         'add_data_v2': 'addDataV2',
         'affiliate_links': 'affiliateLinks',
         'agents_v2': 'agentsV2',
@@ -153,7 +151,6 @@ class V1UserFeatures(object):
         'cloudy_vibe_code': 'cloudyVibeCode',
         'code_tab': 'codeTab',
         'collab_screen_sharing': 'collabScreenSharing',
-        'concurrent_gpu_limit': 'concurrentGpuLimit',
         'control_center_monitoring': 'controlCenterMonitoring',
         'cost_attribution_settings': 'costAttributionSettings',
         'custom_app_domain': 'customAppDomain',
@@ -170,17 +167,18 @@ class V1UserFeatures(object):
         'gcs_fuse': 'gcsFuse',
         'instant_capacity_reservation': 'instantCapacityReservation',
         'job_artifacts_v2': 'jobArtifactsV2',
+        'kubernetes_cluster_ui': 'kubernetesClusterUi',
         'kubernetes_clusters': 'kubernetesClusters',
         'lambda_labs': 'lambdaLabs',
         '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,10 +226,9 @@ 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, 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, 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_cluster_ui: '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
         self._add_data_v2 = None
         self._affiliate_links = None
         self._agents_v2 = None
@@ -248,7 +245,6 @@ class V1UserFeatures(object):
         self._cloudy_vibe_code = None
         self._code_tab = None
         self._collab_screen_sharing = None
-        self._concurrent_gpu_limit = None
         self._control_center_monitoring = None
         self._cost_attribution_settings = None
         self._custom_app_domain = None
@@ -265,17 +261,18 @@ class V1UserFeatures(object):
         self._gcs_fuse = None
         self._instant_capacity_reservation = None
         self._job_artifacts_v2 = None
+        self._kubernetes_cluster_ui = None
         self._kubernetes_clusters = None
         self._lambda_labs = None
         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
@@ -324,8 +321,6 @@ class V1UserFeatures(object):
         self.discriminator = None
         if academic_tier is not None:
             self.academic_tier = academic_tier
-        if accurate_billing is not None:
-            self.accurate_billing = accurate_billing
         if add_data_v2 is not None:
             self.add_data_v2 = add_data_v2
         if affiliate_links is not None:
@@ -358,8 +353,6 @@ class V1UserFeatures(object):
             self.code_tab = code_tab
         if collab_screen_sharing is not None:
             self.collab_screen_sharing = collab_screen_sharing
-        if concurrent_gpu_limit is not None:
-            self.concurrent_gpu_limit = concurrent_gpu_limit
         if control_center_monitoring is not None:
             self.control_center_monitoring = control_center_monitoring
         if cost_attribution_settings is not None:
@@ -392,6 +385,8 @@ class V1UserFeatures(object):
             self.instant_capacity_reservation = instant_capacity_reservation
         if job_artifacts_v2 is not None:
             self.job_artifacts_v2 = job_artifacts_v2
+        if kubernetes_cluster_ui is not None:
+            self.kubernetes_cluster_ui = kubernetes_cluster_ui
         if kubernetes_clusters is not None:
             self.kubernetes_clusters = kubernetes_clusters
         if lambda_labs is not None:
@@ -402,8 +397,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 +407,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:
@@ -526,27 +521,6 @@ class V1UserFeatures(object):
 
         self._academic_tier = academic_tier
 
-    @property
-    def accurate_billing(self) -> 'bool':
-        """Gets the accurate_billing of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The accurate_billing of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._accurate_billing
-
-    @accurate_billing.setter
-    def accurate_billing(self, accurate_billing: 'bool'):
-        """Sets the accurate_billing of this V1UserFeatures.
-
-
-        :param accurate_billing: The accurate_billing of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._accurate_billing = accurate_billing
-
     @property
     def add_data_v2(self) -> 'bool':
         """Gets the add_data_v2 of this V1UserFeatures.  # noqa: E501
@@ -883,27 +857,6 @@ class V1UserFeatures(object):
 
         self._collab_screen_sharing = collab_screen_sharing
 
-    @property
-    def concurrent_gpu_limit(self) -> 'bool':
-        """Gets the concurrent_gpu_limit of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The concurrent_gpu_limit of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._concurrent_gpu_limit
-
-    @concurrent_gpu_limit.setter
-    def concurrent_gpu_limit(self, concurrent_gpu_limit: 'bool'):
-        """Sets the concurrent_gpu_limit of this V1UserFeatures.
-
-
-        :param concurrent_gpu_limit: The concurrent_gpu_limit of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._concurrent_gpu_limit = concurrent_gpu_limit
-
     @property
     def control_center_monitoring(self) -> 'bool':
         """Gets the control_center_monitoring of this V1UserFeatures.  # noqa: E501
@@ -1240,6 +1193,27 @@ class V1UserFeatures(object):
 
         self._job_artifacts_v2 = job_artifacts_v2
 
+    @property
+    def kubernetes_cluster_ui(self) -> 'bool':
+        """Gets the kubernetes_cluster_ui of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The kubernetes_cluster_ui of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._kubernetes_cluster_ui
+
+    @kubernetes_cluster_ui.setter
+    def kubernetes_cluster_ui(self, kubernetes_cluster_ui: 'bool'):
+        """Sets the kubernetes_cluster_ui of this V1UserFeatures.
+
+
+        :param kubernetes_cluster_ui: The kubernetes_cluster_ui of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._kubernetes_cluster_ui = kubernetes_cluster_ui
+
     @property
     def kubernetes_clusters(self) -> 'bool':
         """Gets the kubernetes_clusters of this V1UserFeatures.  # noqa: E501
@@ -1345,27 +1319,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 +1424,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/lightning_cloud/utils/data_connection.py

@@ -1,10 +1,9 @@
 import os
 from time import sleep, time
 from lightning_sdk.lightning_cloud import rest_client
-from lightning_sdk.lightning_cloud.openapi import Create, V1AwsDataConnection, V1S3FolderDataConnection, V1EfsConfig
+from lightning_sdk.lightning_cloud.openapi import Create, V1AwsDataConnection, V1S3FolderDataConnection, V1EfsConfig, V1GcpDataConnection
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 import urllib3
-import os
 
 def add_s3_connection(bucket_name: str, region: str = "us-east-1", create_timeout: int = 15) -> None:
     """Utility to add a data connection."""
@@ -217,6 +216,62 @@ def add_efs_connection(name: str, filesystem_id: str, region: str = "us-east-1",
 
     return
 
+def add_gcs_connection(connection_name: str, bucket_name: str, create_timeout: int = 15) -> None:
+    """
+    Utility function to add a GCS data connection.
+
+    Parameters:
+        1. connection_name (str): The name of the data connection.
+        2. bucket_name (str): The name of the bucket to attach.
+        3. create_timeout (int): The timeout for the data connectio creation.
+    Returns:
+        None
+    """
+
+    client = rest_client.LightningClient(retry=False)
+
+    project_id = os.getenv("LIGHTNING_CLOUD_PROJECT_ID")
+    cluster_id = os.getenv("LIGHTNING_CLUSTER_ID")
+
+    # Get existing data connections and ensure there is no an existing one with the same name and type
+    data_connections = client.data_connection_service_list_data_connections(project_id).data_connections
+
+    for connection in data_connections:
+        existing_connection_name = getattr(connection, 'name', None)
+        isGCSConnection = getattr(connection, 'gcp', None) is not None
+
+        # Same name and type already exists
+        if existing_connection_name == connection_name and isGCSConnection:
+            return
+
+    body = Create(
+        name=connection_name,
+        create_index=True,
+        cluster_id=cluster_id,
+        access_cluster_ids=[cluster_id],
+        gcp=V1GcpDataConnection(
+            source=f"gs://{bucket_name}",
+        ))
+    try:
+        client.data_connection_service_create_data_connection(body, project_id)
+    except (ApiException, urllib3.exceptions.HTTPError) as ex:
+        # Note: This function can be called in a distributed way.
+        # There is a race condition where one machine might create the entry before another machine
+        # and this request would fail with duplicated key
+        # In this case, it is fine not to raise
+        if isinstance(ex, ApiException) and 'duplicate key value violates unique constraint' in str(ex.body):
+            pass
+        else:
+            raise ex
+
+    # Wait for the filesystem picks up the newly added GCS data connection
+    start = time()
+
+    while not os.path.isdir(f"/teamspace/gcs_connections/{bucket_name}") and (time() - start) < create_timeout:
+        sleep(1)
+
+    return
+
 def delete_data_connection(name: str):
     """Utility to delete a data connection
 
@@ -249,4 +304,3 @@ def delete_data_connection(name: str):
         #  for now it's best to actually stop the studio and all other things where the connection
         #  is mounted before trying to delete it
         raise e from None
-

lightning_sdk/llm/llm.py

@@ -169,21 +169,33 @@ class LLM:
                 self._conversations[conversation.name] = conversation.id
 
     def _stream_chat_response(
-        self, result: Generator[V1ConversationResponseChunk, None, None], conversation: Optional[str] = None
+        self,
+        result: Generator[V1ConversationResponseChunk, None, None],
+        conversation: Optional[str] = None,
+        full_response: bool = False,
     ) -> Generator[str, None, None]:
         first_line = next(result, None)
         if first_line:
             if conversation and first_line.conversation_id:
                 self._conversations[conversation] = first_line.conversation_id
-            yield first_line.choices[0].delta.content
+            if full_response:
+                yield first_line
+            else:
+                yield first_line.choices[0].delta.content
 
         for line in result:
-            yield line.choices[0].delta.content
+            if full_response:
+                yield line
+            else:
+                yield line.choices[0].delta.content
 
-    async def _async_stream_text(self, output: str) -> AsyncGenerator[str, None]:
+    async def _async_stream_text(self, output: str, full_response: bool = False) -> AsyncGenerator[str, None]:
         async for chunk in output:
             if chunk.choices and chunk.choices[0].delta:
-                yield chunk.choices[0].delta.content
+                if full_response:
+                    yield chunk
+                else:
+                    yield chunk.choices[0].delta.content
 
     async def _async_chat(
         self,
@@ -194,6 +206,7 @@ class LLM:
         conversation: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        full_response: bool = False,
         **kwargs: Any,
     ) -> Union[str, AsyncGenerator[str, None]]:
         conversation_id = self._conversations.get(conversation) if conversation else None
@@ -213,8 +226,10 @@ class LLM:
         if not stream:
             if conversation and not conversation_id:
                 self._conversations[conversation] = output.conversation_id
+            if full_response:
+                return output
             return output.choices[0].delta.content
-        return self._async_stream_text(output)
+        return self._async_stream_text(output, full_response)
 
     def chat(
         self,
@@ -225,8 +240,11 @@ class LLM:
         conversation: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        full_response: bool = False,
         **kwargs: Any,
-    ) -> Union[str, Generator[str, None, None]]:
+    ) -> Union[
+        V1ConversationResponseChunk, Generator[V1ConversationResponseChunk, None, None], str, Generator[str, None, None]
+    ]:
         if conversation and conversation not in self._conversations:
             self._get_conversations()
 
@@ -248,6 +266,7 @@ class LLM:
                 conversation,
                 metadata,
                 stream,
+                full_response,
                 **kwargs,
             )
 
@@ -267,8 +286,10 @@ class LLM:
         if not stream:
             if conversation and not conversation_id:
                 self._conversations[conversation] = output.conversation_id
+            if full_response:
+                return output
             return output.choices[0].delta.content
-        return self._stream_chat_response(output, conversation=conversation)
+        return self._stream_chat_response(output, conversation=conversation, full_response=full_response)
 
     def list_conversations(self) -> List[Dict]:
         self._get_conversations()

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.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.2.24rc1
+Version: 2025.7.10
 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.10.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=fG3HKc1WBcAYah6nhvTOh7OxcQrKdc47OVufxrxpbbE,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,16 +21,16 @@ 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
 lightning_sdk/api/lit_container_api.py,sha256=jCJVwd-3MNjejL9FyvH89pzt-SeG3G8qCJD16iTMJAQ,11454
-lightning_sdk/api/llm_api.py,sha256=G8WKTslOIS8cTYFoOlQNRgtIvzEpqIr2Ez0rxcMAh4k,9100
+lightning_sdk/api/llm_api.py,sha256=ngumhZrGk_XRjgN-JXHN2ksXJMwEXfadcg9p6re4Efk,9316
 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
@@ -67,7 +67,7 @@ lightning_sdk/cli/deploy/serve.py,sha256=qqml7oYf4IPcoiL13P2ly247_PQ8IGv-8hM2NBW
 lightning_sdk/deployment/__init__.py,sha256=dXsa4psDzFYFklsq3JC-2V_L4FQjGZnQAf-ZiVlqG9c,545
 lightning_sdk/deployment/deployment.py,sha256=kD3P3uonV7Vr33BVnyD-HkxdUuQs0F2nt0GYoWb3Wx4,21057
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
-lightning_sdk/job/base.py,sha256=TS5KavtfBAFbLbNqqumEizY9edjO1joSmtUdcO5CThQ,17748
+lightning_sdk/job/base.py,sha256=u6d_Q56zzM_nzKGPvjVqR2MKEYzFawEGoaSsD747ZHQ,18603
 lightning_sdk/job/job.py,sha256=1Xf0ne4wwXpkb_GJgWD8mfueJZoKR8G4lC5T6OXijlw,13075
 lightning_sdk/job/v1.py,sha256=zgdZFnlyhzOBlbwdwx6Ob9YKBrzMnJflzLXy4qyb4So,9070
 lightning_sdk/job/v2.py,sha256=9O31CwIQjwm6qZSEZtFBWbZyBuCSmyFMsHCRbQvJdBU,10283
@@ -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=Bm5XX7YRf4skZnQVnKSMfaHGGnErXYcXNwhgFMAlMDg,81783
 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
@@ -1065,12 +1065,12 @@ lightning_sdk/lightning_cloud/source_code/logs_socket_api.py,sha256=fXuRQfnRLNkY
 lightning_sdk/lightning_cloud/source_code/tar.py,sha256=mSKRWEX1SIAbXFcESNrQAqWSVULv4G0jgcvR_o2KJ5g,5813
 lightning_sdk/lightning_cloud/source_code/uploader.py,sha256=CQrWV6JNDiglyRfhc42ht4GWFtgRz_fBxw4tOCERzco,3207
 lightning_sdk/lightning_cloud/utils/__init__.py,sha256=kSNKoybELDJHs9WeCf279V5JtzOf1VqfKODlKYr9uRo,115
-lightning_sdk/lightning_cloud/utils/data_connection.py,sha256=esKg01F9vdb05fLUf0DUPBIa1F3MsY2WE63wWPCnclk,9268
+lightning_sdk/lightning_cloud/utils/data_connection.py,sha256=WeC7HWpQL2HxSEuX-7iB0mjoQN_YNr5dgfSkPtCGlds,11448
 lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgydm78kJzhOIx3C9qKls,2011
 lightning_sdk/lightning_cloud/utils/name_generator.py,sha256=MkciuA10332V0mcE2PxLIiwWomWE0Fm_gNGK01vwRr4,58046
 lightning_sdk/lightning_cloud/utils/network.py,sha256=axPgl8rhyPcPjxiztDxyksfxax3VNg2OXL5F5Uc81b4,406
 lightning_sdk/llm/__init__.py,sha256=ErZva0HqN2iPtK_6hI6GN7A_HPGNrHo3wYh7vyFdO3Q,57
-lightning_sdk/llm/llm.py,sha256=9lYhOlJvRQD7pcO-mD35r5vUcnA83zJXWGvgrqbD7Ck,12335
+lightning_sdk/llm/llm.py,sha256=M43OQli7cRqTtHUB484-OOdWs-jiwJv8j1LV-jFUolY,13038
 lightning_sdk/llm/public_assistants.json,sha256=PFbA1p2J6XFRafuW0zfaz8tu_Ac4iI2hNeh94_PNoUQ,374
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
 lightning_sdk/mmt/base.py,sha256=B3HC-c82bPHprEZh1mhLCPCrCE8BOKqwIhY7xCF9CXg,15152
@@ -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.10.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.7.10.dist-info/METADATA,sha256=TH70L4MhS7ZbK7lPcmMB6QftgtEp4tGo5wUlOuwAS78,3995
+lightning_sdk-2025.7.10.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.7.10.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.7.10.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.7.10.dist-info/RECORD,,