lightning-sdk 2025.7.9__py3-none-any.whl → 2025.7.17__py3-none-any.whl

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',
@@ -50,6 +49,7 @@ class V1UserFeatures(object):
         'auto_fast_load': 'bool',
         'auto_join_orgs': 'bool',
         'b2c_experience': 'bool',
+        'byo_machine_type': 'bool',
         'cap_add': 'list[str]',
         'cap_drop': 'list[str]',
         'capacity_reservation_byoc': 'bool',
@@ -59,7 +59,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',
@@ -69,13 +68,16 @@ class V1UserFeatures(object):
         'deployment_persistent_disk': 'bool',
         'drive_v2': 'bool',
         'enterprise_compute_admin': 'bool',
+        'f236': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
+        'gcp_overprovisioning': 'bool',
         'gcs_connections_optimized': 'bool',
         'gcs_folders': 'bool',
         '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',
@@ -112,6 +114,7 @@ class V1UserFeatures(object):
         'project_selector': 'bool',
         'publish_pipelines': 'bool',
         'r2_data_connections': 'bool',
+        'r2_uploads': 'bool',
         'reserved_machines_tab': 'bool',
         'restartable_jobs': 'bool',
         'runnable_public_studio_page': 'bool',
@@ -136,7 +139,6 @@ class V1UserFeatures(object):
 
     attribute_map = {
         'academic_tier': 'academicTier',
-        'accurate_billing': 'accurateBilling',
         'add_data_v2': 'addDataV2',
         'affiliate_links': 'affiliateLinks',
         'agents_v2': 'agentsV2',
@@ -144,6 +146,7 @@ class V1UserFeatures(object):
         'auto_fast_load': 'autoFastLoad',
         'auto_join_orgs': 'autoJoinOrgs',
         'b2c_experience': 'b2cExperience',
+        'byo_machine_type': 'byoMachineType',
         'cap_add': 'capAdd',
         'cap_drop': 'capDrop',
         'capacity_reservation_byoc': 'capacityReservationByoc',
@@ -153,7 +156,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',
@@ -163,13 +165,16 @@ class V1UserFeatures(object):
         'deployment_persistent_disk': 'deploymentPersistentDisk',
         'drive_v2': 'driveV2',
         'enterprise_compute_admin': 'enterpriseComputeAdmin',
+        'f236': 'f236',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
+        'gcp_overprovisioning': 'gcpOverprovisioning',
         'gcs_connections_optimized': 'gcsConnectionsOptimized',
         'gcs_folders': 'gcsFolders',
         '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',
@@ -206,6 +211,7 @@ class V1UserFeatures(object):
         'project_selector': 'projectSelector',
         'publish_pipelines': 'publishPipelines',
         'r2_data_connections': 'r2DataConnections',
+        'r2_uploads': 'r2Uploads',
         'reserved_machines_tab': 'reservedMachinesTab',
         'restartable_jobs': 'restartableJobs',
         'runnable_public_studio_page': 'runnablePublicStudioPage',
@@ -228,10 +234,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, 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
+    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, byo_machine_type: '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, f236: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: '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, r2_uploads: '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
@@ -239,6 +244,7 @@ class V1UserFeatures(object):
         self._auto_fast_load = None
         self._auto_join_orgs = None
         self._b2c_experience = None
+        self._byo_machine_type = None
         self._cap_add = None
         self._cap_drop = None
         self._capacity_reservation_byoc = None
@@ -248,7 +254,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
@@ -258,13 +263,16 @@ class V1UserFeatures(object):
         self._deployment_persistent_disk = None
         self._drive_v2 = None
         self._enterprise_compute_admin = None
+        self._f236 = None
         self._fair_share = None
         self._featured_studios_admin = None
+        self._gcp_overprovisioning = None
         self._gcs_connections_optimized = None
         self._gcs_folders = None
         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
@@ -301,6 +309,7 @@ class V1UserFeatures(object):
         self._project_selector = None
         self._publish_pipelines = None
         self._r2_data_connections = None
+        self._r2_uploads = None
         self._reserved_machines_tab = None
         self._restartable_jobs = None
         self._runnable_public_studio_page = None
@@ -324,8 +333,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:
@@ -340,6 +347,8 @@ class V1UserFeatures(object):
             self.auto_join_orgs = auto_join_orgs
         if b2c_experience is not None:
             self.b2c_experience = b2c_experience
+        if byo_machine_type is not None:
+            self.byo_machine_type = byo_machine_type
         if cap_add is not None:
             self.cap_add = cap_add
         if cap_drop is not None:
@@ -358,8 +367,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:
@@ -378,10 +385,14 @@ class V1UserFeatures(object):
             self.drive_v2 = drive_v2
         if enterprise_compute_admin is not None:
             self.enterprise_compute_admin = enterprise_compute_admin
+        if f236 is not None:
+            self.f236 = f236
         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 gcp_overprovisioning is not None:
+            self.gcp_overprovisioning = gcp_overprovisioning
         if gcs_connections_optimized is not None:
             self.gcs_connections_optimized = gcs_connections_optimized
         if gcs_folders is not None:
@@ -392,6 +403,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:
@@ -464,6 +477,8 @@ class V1UserFeatures(object):
             self.publish_pipelines = publish_pipelines
         if r2_data_connections is not None:
             self.r2_data_connections = r2_data_connections
+        if r2_uploads is not None:
+            self.r2_uploads = r2_uploads
         if reserved_machines_tab is not None:
             self.reserved_machines_tab = reserved_machines_tab
         if restartable_jobs is not None:
@@ -526,27 +541,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
@@ -694,6 +688,27 @@ class V1UserFeatures(object):
 
         self._b2c_experience = b2c_experience
 
+    @property
+    def byo_machine_type(self) -> 'bool':
+        """Gets the byo_machine_type of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The byo_machine_type of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._byo_machine_type
+
+    @byo_machine_type.setter
+    def byo_machine_type(self, byo_machine_type: 'bool'):
+        """Sets the byo_machine_type of this V1UserFeatures.
+
+
+        :param byo_machine_type: The byo_machine_type of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._byo_machine_type = byo_machine_type
+
     @property
     def cap_add(self) -> 'list[str]':
         """Gets the cap_add of this V1UserFeatures.  # noqa: E501
@@ -883,27 +898,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
@@ -1093,6 +1087,27 @@ class V1UserFeatures(object):
 
         self._enterprise_compute_admin = enterprise_compute_admin
 
+    @property
+    def f236(self) -> 'bool':
+        """Gets the f236 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f236 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f236
+
+    @f236.setter
+    def f236(self, f236: 'bool'):
+        """Sets the f236 of this V1UserFeatures.
+
+
+        :param f236: The f236 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f236 = f236
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501
@@ -1135,6 +1150,27 @@ class V1UserFeatures(object):
 
         self._featured_studios_admin = featured_studios_admin
 
+    @property
+    def gcp_overprovisioning(self) -> 'bool':
+        """Gets the gcp_overprovisioning of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The gcp_overprovisioning of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._gcp_overprovisioning
+
+    @gcp_overprovisioning.setter
+    def gcp_overprovisioning(self, gcp_overprovisioning: 'bool'):
+        """Sets the gcp_overprovisioning of this V1UserFeatures.
+
+
+        :param gcp_overprovisioning: The gcp_overprovisioning of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._gcp_overprovisioning = gcp_overprovisioning
+
     @property
     def gcs_connections_optimized(self) -> 'bool':
         """Gets the gcs_connections_optimized of this V1UserFeatures.  # noqa: E501
@@ -1240,6 +1276,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
@@ -1996,6 +2053,27 @@ class V1UserFeatures(object):
 
         self._r2_data_connections = r2_data_connections
 
+    @property
+    def r2_uploads(self) -> 'bool':
+        """Gets the r2_uploads of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The r2_uploads of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._r2_uploads
+
+    @r2_uploads.setter
+    def r2_uploads(self, r2_uploads: 'bool'):
+        """Sets the r2_uploads of this V1UserFeatures.
+
+
+        :param r2_uploads: The r2_uploads of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._r2_uploads = r2_uploads
+
     @property
     def reserved_machines_tab(self) -> 'bool':
         """Gets the reserved_machines_tab of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_volume.py

@@ -57,6 +57,7 @@ class V1Volume(object):
         'server_id': 'str',
         'size_gb': 'str',
         'state': 'V1VolumeState',
+        'subnet_id': 'str',
         'throughput': 'str',
         'type': 'str',
         'updated_at': 'datetime',
@@ -80,13 +81,14 @@ class V1Volume(object):
         'server_id': 'serverId',
         'size_gb': 'sizeGb',
         'state': 'state',
+        'subnet_id': 'subnetId',
         'throughput': 'throughput',
         'type': 'type',
         'updated_at': 'updatedAt',
         'user_id': 'userId'
     }
 
-    def __init__(self, attached_at: 'datetime' =None, availability_zone: 'str' =None, detached_at: 'datetime' =None, encrypted: 'bool' =None, iops: 'str' =None, metadata: 'V1Metadata' =None, path: 'str' =None, provider: 'str' =None, provider_id: 'str' =None, region: 'str' =None, resource_id: 'str' =None, resource_type: 'str' =None, retention_period: 'str' =None, server_id: 'str' =None, size_gb: 'str' =None, state: 'V1VolumeState' =None, throughput: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, attached_at: 'datetime' =None, availability_zone: 'str' =None, detached_at: 'datetime' =None, encrypted: 'bool' =None, iops: 'str' =None, metadata: 'V1Metadata' =None, path: 'str' =None, provider: 'str' =None, provider_id: 'str' =None, region: 'str' =None, resource_id: 'str' =None, resource_type: 'str' =None, retention_period: 'str' =None, server_id: 'str' =None, size_gb: 'str' =None, state: 'V1VolumeState' =None, subnet_id: 'str' =None, throughput: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Volume - a model defined in Swagger"""  # noqa: E501
         self._attached_at = None
         self._availability_zone = None
@@ -104,6 +106,7 @@ class V1Volume(object):
         self._server_id = None
         self._size_gb = None
         self._state = None
+        self._subnet_id = None
         self._throughput = None
         self._type = None
         self._updated_at = None
@@ -141,6 +144,8 @@ class V1Volume(object):
             self.size_gb = size_gb
         if state is not None:
             self.state = state
+        if subnet_id is not None:
+            self.subnet_id = subnet_id
         if throughput is not None:
             self.throughput = throughput
         if type is not None:
@@ -486,6 +491,27 @@ class V1Volume(object):
 
         self._state = state
 
+    @property
+    def subnet_id(self) -> 'str':
+        """Gets the subnet_id of this V1Volume.  # noqa: E501
+
+
+        :return: The subnet_id of this V1Volume.  # noqa: E501
+        :rtype: str
+        """
+        return self._subnet_id
+
+    @subnet_id.setter
+    def subnet_id(self, subnet_id: 'str'):
+        """Sets the subnet_id of this V1Volume.
+
+
+        :param subnet_id: The subnet_id of this V1Volume.  # noqa: E501
+        :type: str
+        """
+
+        self._subnet_id = subnet_id
+
     @property
     def throughput(self) -> 'str':
         """Gets the throughput of this V1Volume.  # 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

@@ -9,6 +9,7 @@ PUBLIC_MODEL_PROVIDERS: Dict[str, str] = {
     "openai": "OpenAI",
     "anthropic": "Anthropic",
     "google": "Google",
+    "lightning-ai": "lightning-ai",
 }
 
 
@@ -169,21 +170,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 +207,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 +227,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 +241,12 @@ class LLM:
         conversation: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        full_response: bool = False,
+        tools: Optional[List[Dict[str, Any]]] = None,
         **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 +268,7 @@ class LLM:
                 conversation,
                 metadata,
                 stream,
+                full_response,
                 **kwargs,
             )
 
@@ -262,13 +283,16 @@ class LLM:
             metadata=metadata,
             name=conversation,
             stream=stream,
+            tools=tools,
             **kwargs,
         )
         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/llm/public_assistants.json

@@ -4,5 +4,7 @@
     "openai/o3-mini": "ast_01jz3t13fhnjhh11t1k8b5gyp1",
     "anthropic/claude-3-5-sonnet-20240620": "ast_01jd3923a6p98rqwh3dpj686pq",
     "google/gemini-2.5-pro": "ast_01jz3tdb1fhey798k95pv61v57",
-    "google/gemini-2.5-flash": "ast_01jz3thxskg4fcdk4xhkjkym5a"
+    "google/gemini-2.5-flash": "ast_01jz3thxskg4fcdk4xhkjkym5a",
+    "google/gemini-2.5-flash-lite-preview-06-17": "ast_01jz3thxskg4fcdk4xhkjkym5b",
+    "lightning-ai/llama4-maverick": "ast_01k073vsqs66tenpns02cf5jnq"
 }