lightning-sdk 0.2.18__py3-none-any.whl → 0.2.20__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.18"
+__version__ = "0.2.20"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/deployment_api.py

@@ -225,6 +225,7 @@ class DeploymentApi:
         self,
         deployment: V1Deployment,
         from_onboarding: Optional[bool] = None,
+        from_litserve: Optional[bool] = None,
     ) -> V1Deployment:
         return self._client.jobs_service_create_deployment(
             project_id=deployment.project_id,
@@ -238,6 +239,7 @@ class DeploymentApi:
                 spec=deployment.spec,
                 strategy=deployment.strategy,
                 from_onboarding=from_onboarding,
+                from_litserve=from_litserve,
             ),
         )
 

lightning_sdk/api/license_api.py

@@ -1,11 +1,35 @@
+import os
 from typing import Optional
+from urllib.parse import urlencode
 
+from lightning_sdk.lightning_cloud import env
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
+LICENSE_CODE = os.environ.get("LICENSE_CODE", "we843fiji89")
+# https://lightning.ai/home?settings=licenses
+LICENSE_SIGNING_URL = f"{env.LIGHTNING_CLOUD_URL}/ai-hub?settings=licenses"
+
+
+def generate_url_user_settings(redirect_to: str = LICENSE_SIGNING_URL) -> str:
+    params = urlencode({"redirectTo": redirect_to, "mode": "licenses", "okbhrt": LICENSE_CODE})
+    return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
+
 
 class LicenseApi:
-    def __init__(self) -> None:
-        self._client = LightningClient(retry=False, max_tries=0)
+    _client_authenticated: LightningClient = None
+    _client_public: LightningClient = None
+
+    @property
+    def client_public(self) -> LightningClient:
+        if not self._client_public:
+            self._client_public = LightningClient(retry=False, max_tries=0, with_auth=False)
+        return self._client_public
+
+    @property
+    def client_authenticated(self) -> LightningClient:
+        if not self._client_authenticated:
+            self._client_authenticated = LightningClient(retry=True, max_tries=3, with_auth=True)
+        return self._client_authenticated
 
     def valid_license(
         self,
@@ -25,14 +49,12 @@ class LicenseApi:
         Returns:
             True if the license key is valid, False otherwise.
         """
-        response, code, _ = self._client.product_license_service_validate_product_license_with_http_info(
+        response = self.client_public.product_license_service_validate_product_license(
             license_key=license_key,
             product_name=product_name,
             product_version=product_version,
             product_type=product_type,
         )
-        if code != 200:
-            raise ConnectionError(f"Failed to validate license key: {code} - {response}")
         return response.valid
 
     def list_user_licenses(self, user_id: str) -> list:
@@ -44,5 +66,5 @@ class LicenseApi:
         Returns:
             A list of licenses for the user.
         """
-        response = self._client.product_license_service_list_user_licenses(user_id=user_id)
+        response = self.client_authenticated.product_license_service_list_user_licenses(user_id=user_id)
         return response.licenses

lightning_sdk/cli/deploy/_auth.py

@@ -2,8 +2,7 @@ import os
 import time
 from datetime import datetime
 from enum import Enum
-from typing import Any, List, Optional, TypedDict
-from urllib.parse import urlencode
+from typing import List, Optional, TypedDict
 
 from rich.console import Console
 from rich.prompt import Confirm
@@ -11,7 +10,6 @@ from rich.prompt import Confirm
 from lightning_sdk import Teamspace
 from lightning_sdk.api import UserApi
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.lightning_cloud import env
 from lightning_sdk.lightning_cloud.login import Auth, AuthServer
 from lightning_sdk.lightning_cloud.openapi import V1CloudSpace
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
@@ -26,18 +24,7 @@ class _AuthMode(Enum):
     DEPLOY = "deploy"
 
 
-class _AuthServer(AuthServer):
-    def __init__(self, mode: _AuthMode, *args: Any, **kwargs: Any) -> None:
-        self._mode = mode
-        super().__init__(*args, **kwargs)
-
-    def get_auth_url(self, port: int) -> str:
-        redirect_uri = f"http://localhost:{port}/login-complete"
-        params = urlencode({"redirectTo": redirect_uri, "mode": self._mode.value, "okbhrt": LITSERVE_CODE})
-        return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
-
-
-class _Auth(Auth):
+class _AuthLitServe(Auth):
     def __init__(self, mode: _AuthMode, shall_confirm: bool = False) -> None:
         super().__init__()
         self._mode = mode
@@ -51,15 +38,20 @@ class _Auth(Auth):
             if not proceed:
                 raise RuntimeError(
                     "Login cancelled. Please login to Lightning AI to deploy the API. Run `lightning login` to login."
-                ) from None
+                )
         print("Opening browser for authentication...")
         print("Please come back to the terminal after logging in.")
         time.sleep(3)
-        _AuthServer(self._mode).login_with_browser(self)
+        AuthServer({"mode": self._mode, "okbhrt": LITSERVE_CODE}).login_with_browser(self)
 
 
 def authenticate(mode: _AuthMode, shall_confirm: bool = True) -> None:
-    auth = _Auth(mode, shall_confirm)
+    """Authenticate with Lightning AI.
+
+    This will open a browser window for authentication.
+    If `shall_confirm` is True, it will ask for confirmation before proceeding.
+    """
+    auth = _AuthLitServe(mode, shall_confirm)
     auth.authenticate()
 
 
@@ -87,7 +79,7 @@ def poll_verified_status(timeout: int = _POLL_TIMEOUT) -> _UserStatus:
     user_api = UserApi()
     user = _get_authed_user()
     start_time = datetime.now()
-    result = {"onboarded": False, "verified": False}
+    result = _UserStatus(onboarded=False, verified=False)
     while True:
         user_resp = user_api.get_user(name=user.name)
         result["onboarded"] = user_resp.status.completed_project_onboarding

lightning_sdk/cli/entrypoint.py

@@ -1,10 +1,12 @@
 import sys
+import traceback
 from types import TracebackType
 from typing import Type
 
 import click
 from rich.console import Console
 from rich.panel import Panel
+from rich.text import Text
 
 from lightning_sdk import __version__
 from lightning_sdk.api.studio_api import _cloud_url
@@ -26,13 +28,29 @@ from lightning_sdk.cli.start import start
 from lightning_sdk.cli.stop import stop
 from lightning_sdk.cli.switch import switch
 from lightning_sdk.cli.upload import upload
+from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.lightning_cloud.login import Auth
 
 
-def _notify_exception(exception_type: Type[BaseException], value: BaseException, tb: TracebackType) -> None:  # No
+def _notify_exception(exception_type: Type[BaseException], value: BaseException, tb: TracebackType) -> None:
     """CLI won't show tracebacks, just print the exception message."""
+    # if debug mode, print the traceback using rich
     console = Console()
-    console.print(Panel(value))
+    if value.args:
+        message = str(value.args[0]) if value.args[0] else str(value)
+    else:
+        message = str(value) or "An unknown error occurred"
+
+    error_content = Text()
+    error_content.append(f"{exception_type.__name__}: ", style="bold red")
+    error_content.append(message, style="white")
+
+    if _LIGHTNING_DEBUG:
+        error_content.append("\n\nFull traceback:\n", style="bold yellow")
+        tb_lines = traceback.format_exception(exception_type, value, tb)
+        error_content.append("".join(tb_lines), style="dim white")
+
+    console.print(Panel(error_content, title="⚡ Lightning CLI Error", border_style="red"))
 
 
 @click.group(name="lightning", help="Command line interface (CLI) to interact with/manage Lightning AI Studios.")

lightning_sdk/deployment/deployment.py

@@ -129,6 +129,7 @@ class Deployment:
         quantity: Optional[int] = None,
         include_credentials: Optional[bool] = None,
         from_onboarding: Optional[bool] = None,
+        from_litserve: Optional[bool] = None,
     ) -> None:
         """The Lightning AI Deployment.
 
@@ -228,6 +229,7 @@ class Deployment:
                 strategy=to_strategy(release_strategy),
             ),
             from_onboarding=from_onboarding,
+            from_litserve=from_litserve,
         )
 
         # Overrides the name

lightning_sdk/lightning_cloud/login.py

@@ -43,8 +43,8 @@ class Auth:
         for key in Keys:
             setattr(self, key.suffix, os.environ.get(key.value, None))
 
-        self._with_env_var = bool(
-            self.user_id and self.api_key)  # used by authenticate method
+        # used by authenticate method
+        self._with_env_var = bool(self.user_id and self.api_key)
         if self.api_key and not self.user_id:
             raise ValueError(
                 f"{Keys.USER_ID.value} is missing from env variables. "

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -27,6 +27,7 @@ from lightning_sdk.lightning_cloud.openapi.api.analytics_service_api import Anal
 from lightning_sdk.lightning_cloud.openapi.api.assistants_service_api import AssistantsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.auth_service_api import AuthServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import BillingServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.blog_posts_service_api import BlogPostsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_service_api import CloudSpaceEnvironmentTemplateServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi
@@ -68,6 +69,7 @@ from lightning_sdk.lightning_cloud.openapi.configuration import Configuration
 from lightning_sdk.lightning_cloud.openapi.models.affiliatelinks_id_body import AffiliatelinksIdBody
 from lightning_sdk.lightning_cloud.openapi.models.agentmanagedendpoints_id_body import AgentmanagedendpointsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.agents_id_body import AgentsIdBody
+from lightning_sdk.lightning_cloud.openapi.models.alertingevents_id_body import AlertingeventsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.alerts_config_billing import AlertsConfigBilling
 from lightning_sdk.lightning_cloud.openapi.models.alerts_config_studios import AlertsConfigStudios
 from lightning_sdk.lightning_cloud.openapi.models.app_id_works_body import AppIdWorksBody
@@ -79,6 +81,7 @@ from lightning_sdk.lightning_cloud.openapi.models.assistant_id_conversations_bod
 from lightning_sdk.lightning_cloud.openapi.models.billing_checkout_body import BillingCheckoutBody
 from lightning_sdk.lightning_cloud.openapi.models.billing_transfer_body import BillingTransferBody
 from lightning_sdk.lightning_cloud.openapi.models.billing_transfer_body1 import BillingTransferBody1
+from lightning_sdk.lightning_cloud.openapi.models.blogposts_id_body import BlogpostsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.captures_id_body import CapturesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versionpublications_body import CloudSpaceIdVersionpublicationsBody
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versionpublications_body1 import CloudSpaceIdVersionpublicationsBody1
@@ -142,6 +145,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body imp
 from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBody
 from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
+from lightning_sdk.lightning_cloud.openapi.models.id_transfer_body import IdTransferBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
@@ -253,6 +257,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_assistant import V1Assistan
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_knowledge_item_status import V1AssistantKnowledgeItemStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_knowledge_status import V1AssistantKnowledgeStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_model_status import V1AssistantModelStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_author import V1Author
 from lightning_sdk.lightning_cloud.openapi.models.v1_auto_join_domain_validation import V1AutoJoinDomainValidation
 from lightning_sdk.lightning_cloud.openapi.models.v1_auto_join_org_response import V1AutoJoinOrgResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_autoscaling_spec import V1AutoscalingSpec
@@ -263,6 +268,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_billing_feature import V1Bi
 from lightning_sdk.lightning_cloud.openapi.models.v1_billing_period import V1BillingPeriod
 from lightning_sdk.lightning_cloud.openapi.models.v1_billing_subscription import V1BillingSubscription
 from lightning_sdk.lightning_cloud.openapi.models.v1_billing_tier import V1BillingTier
+from lightning_sdk.lightning_cloud.openapi.models.v1_blog_post import V1BlogPost
 from lightning_sdk.lightning_cloud.openapi.models.v1_body import V1Body
 from lightning_sdk.lightning_cloud.openapi.models.v1_build_spec import V1BuildSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_cpu_system_metrics import V1CPUSystemMetrics
@@ -342,6 +348,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_count_metrics_streams_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_agent_multipart_upload_response import V1CreateAgentMultipartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_billing_portal_session_request import V1CreateBillingPortalSessionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_billing_portal_session_response import V1CreateBillingPortalSessionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_blog_post_request import V1CreateBlogPostRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_checkout_session_request import V1CreateCheckoutSessionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_checkout_session_response import V1CreateCheckoutSessionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cloud_space_app_instance_response import V1CreateCloudSpaceAppInstanceResponse
@@ -383,6 +390,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_dataset import V1Dataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1DatasetType
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_affiliate_link_response import V1DeleteAffiliateLinkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_assistant_response import V1DeleteAssistantResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_blog_post_response import V1DeleteBlogPostResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_app_response import V1DeleteCloudSpaceAppResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_environment_template_response import V1DeleteCloudSpaceEnvironmentTemplateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_response import V1DeleteCloudSpaceResponse
@@ -479,7 +487,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
-from lightning_sdk.lightning_cloud.openapi.models.v1_ebs import V1Ebs
 from lightning_sdk.lightning_cloud.openapi.models.v1_efs_config import V1EfsConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
 from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_auth import V1EndpointAuth
@@ -610,6 +617,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_affiliate_links_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_job_artifacts_response import V1ListAgentJobArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_jobs_response import V1ListAgentJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_assistants_response import V1ListAssistantsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_blog_posts_response import V1ListBlogPostsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_apps_response import V1ListCloudSpaceAppsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_cold_start_metrics_response import V1ListCloudSpaceColdStartMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_environment_templates_response import V1ListCloudSpaceEnvironmentTemplatesResponse
@@ -824,7 +832,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_billing_session import V1ReservationBillingSession
 from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
@@ -892,6 +899,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
+from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_project_balance_response import V1TransferProjectBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_user_balance_request import V1TransferUserBalanceRequest

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -8,6 +8,7 @@ from lightning_sdk.lightning_cloud.openapi.api.analytics_service_api import Anal
 from lightning_sdk.lightning_cloud.openapi.api.assistants_service_api import AssistantsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.auth_service_api import AuthServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import BillingServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.blog_posts_service_api import BlogPostsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_service_api import CloudSpaceEnvironmentTemplateServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi