lightning-sdk 2025.8.7__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_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/public_assistants.py

@@ -1,11 +1,32 @@
 PUBLIC_MODELS = {
-    "openai/gpt-4o": {"id": "ast_01jdjds71fs8gt47jexzed4czs", "context_length": 128000},
-    "openai/gpt-4": {"id": "ast_01jd38ze6tjbrcd4942nhz41zn", "context_length": 8192},
-    "openai/o3-mini": {"id": "ast_01jz3t13fhnjhh11t1k8b5gyp1", "context_length": 128000},
-    "anthropic/claude-3-5-sonnet-20240620": {"id": "ast_01jd3923a6p98rqwh3dpj686pq", "context_length": 200000},
-    "google/gemini-2.5-pro": {"id": "ast_01jz3tdb1fhey798k95pv61v57", "context_length": 1048576},
-    "google/gemini-2.5-flash": {"id": "ast_01jz3thxskg4fcdk4xhkjkym5a", "context_length": 8000},
-    "google/gemini-2.5-flash-lite-preview-06-17": {"id": "ast_01jz3thxskg4fcdk4xhkjkym5b", "context_length": 8000},
+    "openai/gpt-4o": {
+        "id": "ast_01jdjds71fs8gt47jexzed4czs",
+        "context_length": 128000
+    },
+    "openai/gpt-4": {
+        "id": "ast_01jd38ze6tjbrcd4942nhz41zn",
+        "context_length": 8192
+    },
+    "openai/o3-mini": {
+        "id": "ast_01jz3t13fhnjhh11t1k8b5gyp1",
+        "context_length": 128000
+    },
+    "anthropic/claude-3-5-sonnet-20240620": {
+        "id": "ast_01jd3923a6p98rqwh3dpj686pq",
+        "context_length": 200000
+    },
+    "google/gemini-2.5-pro": {
+        "id": "ast_01jz3tdb1fhey798k95pv61v57",
+        "context_length": 1048576
+    },
+    "google/gemini-2.5-flash": {
+        "id": "ast_01jz3thxskg4fcdk4xhkjkym5a",
+        "context_length": 8000
+    },
+    "google/gemini-2.5-flash-lite-preview-06-17": {
+        "id": "ast_01jz3thxskg4fcdk4xhkjkym5b",
+        "context_length": 8000
+    },
     "lightning-ai/gpt-oss-20b": {
         "id": "ast_01k1y2ywfh9zzexjzrva6y96ms",
         "context_length": 128000,
@@ -14,4 +35,16 @@ PUBLIC_MODELS = {
         "id": "ast_01k1z7ajxbz7wq62xj5bc0mrcs",
         "context_length": 128000,
     },
+    "openai/gpt-5-2025-08-07": {
+        "id": "ast_01jdjds71fs8gt47jexzed4cza",
+        "context_length": 400000
+    },
+    "openai/gpt-5-mini-2025-08-07": {
+        "id": "ast_01jdjds71fs8gt47jexzed4czb",
+        "context_length": 400000
+    },
+    "openai/gpt-5-nano-2025-08-07": {
+        "id": "ast_01jdjds71fs8gt47jexzed4czc",
+        "context_length": 400000
+    },
 }

{lightning_sdk-2025.8.7.dist-info → lightning_sdk-2025.8.8.dist-info}/METADATA

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