lightning-sdk 2025.8.6rc2__py3-none-any.whl → 2025.8.8__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_list_schedule_runs_response.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ListScheduleRunsResponse(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'schedule_runs': 'list[V1ScheduleRun]'
+    }
+
+    attribute_map = {
+        'schedule_runs': 'scheduleRuns'
+    }
+
+    def __init__(self, schedule_runs: 'list[V1ScheduleRun]' =None):  # noqa: E501
+        """V1ListScheduleRunsResponse - a model defined in Swagger"""  # noqa: E501
+        self._schedule_runs = None
+        self.discriminator = None
+        if schedule_runs is not None:
+            self.schedule_runs = schedule_runs
+
+    @property
+    def schedule_runs(self) -> 'list[V1ScheduleRun]':
+        """Gets the schedule_runs of this V1ListScheduleRunsResponse.  # noqa: E501
+
+
+        :return: The schedule_runs of this V1ListScheduleRunsResponse.  # noqa: E501
+        :rtype: list[V1ScheduleRun]
+        """
+        return self._schedule_runs
+
+    @schedule_runs.setter
+    def schedule_runs(self, schedule_runs: 'list[V1ScheduleRun]'):
+        """Sets the schedule_runs of this V1ListScheduleRunsResponse.
+
+
+        :param schedule_runs: The schedule_runs of this V1ListScheduleRunsResponse.  # noqa: E501
+        :type: list[V1ScheduleRun]
+        """
+
+        self._schedule_runs = schedule_runs
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ListScheduleRunsResponse, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ListScheduleRunsResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ListScheduleRunsResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ListScheduleRunsResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py

@@ -44,6 +44,8 @@ class V1NodeMetrics(object):
         'cpu_capacity_cores': 'float',
         'cpu_util': 'float',
         'filesystem': 'dict(str, V1FilesystemMetrics)',
+        'mem_total': 'float',
+        'mem_util': 'float',
         'node_name': 'str',
         'per_gpu_mem_free': 'dict(str, float)',
         'per_gpu_mem_used': 'dict(str, float)',
@@ -57,6 +59,8 @@ class V1NodeMetrics(object):
         'cpu_capacity_cores': 'cpuCapacityCores',
         'cpu_util': 'cpuUtil',
         'filesystem': 'filesystem',
+        'mem_total': 'memTotal',
+        'mem_util': 'memUtil',
         'node_name': 'nodeName',
         'per_gpu_mem_free': 'perGpuMemFree',
         'per_gpu_mem_used': 'perGpuMemUsed',
@@ -66,11 +70,13 @@ class V1NodeMetrics(object):
         'timestamp': 'timestamp'
     }
 
-    def __init__(self, cpu_capacity_cores: 'float' =None, cpu_util: 'float' =None, filesystem: 'dict(str, V1FilesystemMetrics)' =None, node_name: 'str' =None, per_gpu_mem_free: 'dict(str, float)' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_power_usage_watts: 'dict(str, float)' =None, per_gpu_temperature_c: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, timestamp: 'datetime' =None):  # noqa: E501
+    def __init__(self, cpu_capacity_cores: 'float' =None, cpu_util: 'float' =None, filesystem: 'dict(str, V1FilesystemMetrics)' =None, mem_total: 'float' =None, mem_util: 'float' =None, node_name: 'str' =None, per_gpu_mem_free: 'dict(str, float)' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_power_usage_watts: 'dict(str, float)' =None, per_gpu_temperature_c: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, timestamp: 'datetime' =None):  # noqa: E501
         """V1NodeMetrics - a model defined in Swagger"""  # noqa: E501
         self._cpu_capacity_cores = None
         self._cpu_util = None
         self._filesystem = None
+        self._mem_total = None
+        self._mem_util = None
         self._node_name = None
         self._per_gpu_mem_free = None
         self._per_gpu_mem_used = None
@@ -85,6 +91,10 @@ class V1NodeMetrics(object):
             self.cpu_util = cpu_util
         if filesystem is not None:
             self.filesystem = filesystem
+        if mem_total is not None:
+            self.mem_total = mem_total
+        if mem_util is not None:
+            self.mem_util = mem_util
         if node_name is not None:
             self.node_name = node_name
         if per_gpu_mem_free is not None:
@@ -163,6 +173,48 @@ class V1NodeMetrics(object):
 
         self._filesystem = filesystem
 
+    @property
+    def mem_total(self) -> 'float':
+        """Gets the mem_total of this V1NodeMetrics.  # noqa: E501
+
+
+        :return: The mem_total of this V1NodeMetrics.  # noqa: E501
+        :rtype: float
+        """
+        return self._mem_total
+
+    @mem_total.setter
+    def mem_total(self, mem_total: 'float'):
+        """Sets the mem_total of this V1NodeMetrics.
+
+
+        :param mem_total: The mem_total of this V1NodeMetrics.  # noqa: E501
+        :type: float
+        """
+
+        self._mem_total = mem_total
+
+    @property
+    def mem_util(self) -> 'float':
+        """Gets the mem_util of this V1NodeMetrics.  # noqa: E501
+
+
+        :return: The mem_util of this V1NodeMetrics.  # noqa: E501
+        :rtype: float
+        """
+        return self._mem_util
+
+    @mem_util.setter
+    def mem_util(self, mem_util: 'float'):
+        """Sets the mem_util of this V1NodeMetrics.
+
+
+        :param mem_util: The mem_util of this V1NodeMetrics.  # noqa: E501
+        :type: float
+        """
+
+        self._mem_util = mem_util
+
     @property
     def node_name(self) -> 'str':
         """Gets the node_name of this V1NodeMetrics.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_schedule_run.py

@@ -0,0 +1,357 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ScheduleRun(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'action_type': 'V1ScheduleActionType',
+        'command': 'str',
+        'created_at': 'datetime',
+        'id': 'str',
+        'output': 'str',
+        'project_id': 'str',
+        'resource_id': 'str',
+        'schedule_id': 'str',
+        'state': 'str',
+        'stopped_at': 'datetime'
+    }
+
+    attribute_map = {
+        'action_type': 'actionType',
+        'command': 'command',
+        'created_at': 'createdAt',
+        'id': 'id',
+        'output': 'output',
+        'project_id': 'projectId',
+        'resource_id': 'resourceId',
+        'schedule_id': 'scheduleId',
+        'state': 'state',
+        'stopped_at': 'stoppedAt'
+    }
+
+    def __init__(self, action_type: 'V1ScheduleActionType' =None, command: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, output: 'str' =None, project_id: 'str' =None, resource_id: 'str' =None, schedule_id: 'str' =None, state: 'str' =None, stopped_at: 'datetime' =None):  # noqa: E501
+        """V1ScheduleRun - a model defined in Swagger"""  # noqa: E501
+        self._action_type = None
+        self._command = None
+        self._created_at = None
+        self._id = None
+        self._output = None
+        self._project_id = None
+        self._resource_id = None
+        self._schedule_id = None
+        self._state = None
+        self._stopped_at = None
+        self.discriminator = None
+        if action_type is not None:
+            self.action_type = action_type
+        if command is not None:
+            self.command = command
+        if created_at is not None:
+            self.created_at = created_at
+        if id is not None:
+            self.id = id
+        if output is not None:
+            self.output = output
+        if project_id is not None:
+            self.project_id = project_id
+        if resource_id is not None:
+            self.resource_id = resource_id
+        if schedule_id is not None:
+            self.schedule_id = schedule_id
+        if state is not None:
+            self.state = state
+        if stopped_at is not None:
+            self.stopped_at = stopped_at
+
+    @property
+    def action_type(self) -> 'V1ScheduleActionType':
+        """Gets the action_type of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The action_type of this V1ScheduleRun.  # noqa: E501
+        :rtype: V1ScheduleActionType
+        """
+        return self._action_type
+
+    @action_type.setter
+    def action_type(self, action_type: 'V1ScheduleActionType'):
+        """Sets the action_type of this V1ScheduleRun.
+
+
+        :param action_type: The action_type of this V1ScheduleRun.  # noqa: E501
+        :type: V1ScheduleActionType
+        """
+
+        self._action_type = action_type
+
+    @property
+    def command(self) -> 'str':
+        """Gets the command of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The command of this V1ScheduleRun.  # noqa: E501
+        :rtype: str
+        """
+        return self._command
+
+    @command.setter
+    def command(self, command: 'str'):
+        """Sets the command of this V1ScheduleRun.
+
+
+        :param command: The command of this V1ScheduleRun.  # noqa: E501
+        :type: str
+        """
+
+        self._command = command
+
+    @property
+    def created_at(self) -> 'datetime':
+        """Gets the created_at of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The created_at of this V1ScheduleRun.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._created_at
+
+    @created_at.setter
+    def created_at(self, created_at: 'datetime'):
+        """Sets the created_at of this V1ScheduleRun.
+
+
+        :param created_at: The created_at of this V1ScheduleRun.  # noqa: E501
+        :type: datetime
+        """
+
+        self._created_at = created_at
+
+    @property
+    def id(self) -> 'str':
+        """Gets the id of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The id of this V1ScheduleRun.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id: 'str'):
+        """Sets the id of this V1ScheduleRun.
+
+
+        :param id: The id of this V1ScheduleRun.  # noqa: E501
+        :type: str
+        """
+
+        self._id = id
+
+    @property
+    def output(self) -> 'str':
+        """Gets the output of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The output of this V1ScheduleRun.  # noqa: E501
+        :rtype: str
+        """
+        return self._output
+
+    @output.setter
+    def output(self, output: 'str'):
+        """Sets the output of this V1ScheduleRun.
+
+
+        :param output: The output of this V1ScheduleRun.  # noqa: E501
+        :type: str
+        """
+
+        self._output = output
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The project_id of this V1ScheduleRun.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1ScheduleRun.
+
+
+        :param project_id: The project_id of this V1ScheduleRun.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    @property
+    def resource_id(self) -> 'str':
+        """Gets the resource_id of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The resource_id of this V1ScheduleRun.  # noqa: E501
+        :rtype: str
+        """
+        return self._resource_id
+
+    @resource_id.setter
+    def resource_id(self, resource_id: 'str'):
+        """Sets the resource_id of this V1ScheduleRun.
+
+
+        :param resource_id: The resource_id of this V1ScheduleRun.  # noqa: E501
+        :type: str
+        """
+
+        self._resource_id = resource_id
+
+    @property
+    def schedule_id(self) -> 'str':
+        """Gets the schedule_id of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The schedule_id of this V1ScheduleRun.  # noqa: E501
+        :rtype: str
+        """
+        return self._schedule_id
+
+    @schedule_id.setter
+    def schedule_id(self, schedule_id: 'str'):
+        """Sets the schedule_id of this V1ScheduleRun.
+
+
+        :param schedule_id: The schedule_id of this V1ScheduleRun.  # noqa: E501
+        :type: str
+        """
+
+        self._schedule_id = schedule_id
+
+    @property
+    def state(self) -> 'str':
+        """Gets the state of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The state of this V1ScheduleRun.  # noqa: E501
+        :rtype: str
+        """
+        return self._state
+
+    @state.setter
+    def state(self, state: 'str'):
+        """Sets the state of this V1ScheduleRun.
+
+
+        :param state: The state of this V1ScheduleRun.  # noqa: E501
+        :type: str
+        """
+
+        self._state = state
+
+    @property
+    def stopped_at(self) -> 'datetime':
+        """Gets the stopped_at of this V1ScheduleRun.  # noqa: E501
+
+
+        :return: The stopped_at of this V1ScheduleRun.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._stopped_at
+
+    @stopped_at.setter
+    def stopped_at(self, stopped_at: 'datetime'):
+        """Sets the stopped_at of this V1ScheduleRun.
+
+
+        :param stopped_at: The stopped_at of this V1ScheduleRun.  # noqa: E501
+        :type: datetime
+        """
+
+        self._stopped_at = stopped_at
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ScheduleRun, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ScheduleRun') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ScheduleRun):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ScheduleRun') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -76,6 +76,7 @@ class V1UserFeatures(object):
         'f241': 'bool',
         'f242': 'bool',
         'f243': 'bool',
+        'f244': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
         'gcp_overprovisioning': 'bool',
@@ -168,6 +169,7 @@ class V1UserFeatures(object):
         'f241': 'f241',
         'f242': 'f242',
         'f243': 'f243',
+        'f244': 'f244',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
         'gcp_overprovisioning': 'gcpOverprovisioning',
@@ -224,7 +226,7 @@ class V1UserFeatures(object):
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    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, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, model_api_dashboard_clickhouse: 'bool' =None, multicloud_folders: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: '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, 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, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, f244: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, model_api_dashboard_clickhouse: 'bool' =None, multicloud_folders: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: '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, 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._add_data_v2 = None
@@ -261,6 +263,7 @@ class V1UserFeatures(object):
         self._f241 = None
         self._f242 = None
         self._f243 = None
+        self._f244 = None
         self._fair_share = None
         self._featured_studios_admin = None
         self._gcp_overprovisioning = None
@@ -386,6 +389,8 @@ class V1UserFeatures(object):
             self.f242 = f242
         if f243 is not None:
             self.f243 = f243
+        if f244 is not None:
+            self.f244 = f244
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
@@ -1230,6 +1235,27 @@ class V1UserFeatures(object):
 
         self._f243 = f243
 
+    @property
+    def f244(self) -> 'bool':
+        """Gets the f244 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f244 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f244
+
+    @f244.setter
+    def f244(self, f244: 'bool'):
+        """Sets the f244 of this V1UserFeatures.
+
+
+        :param f244: The f244 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f244 = f244
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501

lightning_sdk/llm/llm.py

@@ -138,16 +138,21 @@ class LLM:
         self._cloud_url = LLM._cached_auth_info["cloud_url"]
         self._org = None
 
-    def _parse_model_name(self, name: str) -> Tuple[str, str]:
-        parts = name.split("/")
-        if len(parts) == 1:
-            # a user model or a org model
-            return None, parts[0]
-        if len(parts) == 2:
-            return parts[0].lower(), parts[1]
-        raise ValueError(
-            f"Model name must be in the format `organization/model_name` or `model_name`, but got '{name}'."
-        )
+    @staticmethod
+    def _parse_model_name(name: str) -> Tuple[str, str]:
+        """Parses the model name into provider and model name.
+
+        >>> LLM._parse_model_name("openai/v1/gpt-3.5-turbo")
+        ('openai', 'v1/gpt-3.5-turbo')
+        """
+        if "/" not in name:
+            raise ValueError(
+                f"Invalid model name format: '{name}'. "
+                "Model name must be in the format `provider/model_name`."
+                "(e.g., 'lightning-ai/gpt-oss-20b')"
+            )
+        provider, model_name = name.split("/", maxsplit=1)
+        return provider.lower(), model_name
 
     # returns the assistant ID
     def _get_model_id(self) -> str: