lightning-sdk 0.2.21rc0__py3-none-any.whl → 0.2.22__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py

@@ -42,6 +42,9 @@ class V1UpdateUserRequest(object):
     """
     swagger_types = {
         'agree_to_terms_and_conditions': 'bool',
+        'allow_credits_auto_replenish': 'bool',
+        'auto_replenish_amount': 'float',
+        'auto_replenish_threshold': 'float',
         'complete_sign_up': 'bool',
         'completed_project_onboarding': 'bool',
         'country': 'str',
@@ -69,6 +72,9 @@ class V1UpdateUserRequest(object):
 
     attribute_map = {
         'agree_to_terms_and_conditions': 'agreeToTermsAndConditions',
+        'allow_credits_auto_replenish': 'allowCreditsAutoReplenish',
+        'auto_replenish_amount': 'autoReplenishAmount',
+        'auto_replenish_threshold': 'autoReplenishThreshold',
         'complete_sign_up': 'completeSignUp',
         'completed_project_onboarding': 'completedProjectOnboarding',
         'country': 'country',
@@ -94,9 +100,12 @@ class V1UpdateUserRequest(object):
         'website': 'website'
     }
 
-    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
+    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
         """V1UpdateUserRequest - a model defined in Swagger"""  # noqa: E501
         self._agree_to_terms_and_conditions = None
+        self._allow_credits_auto_replenish = None
+        self._auto_replenish_amount = None
+        self._auto_replenish_threshold = None
         self._complete_sign_up = None
         self._completed_project_onboarding = None
         self._country = None
@@ -123,6 +132,12 @@ class V1UpdateUserRequest(object):
         self.discriminator = None
         if agree_to_terms_and_conditions is not None:
             self.agree_to_terms_and_conditions = agree_to_terms_and_conditions
+        if allow_credits_auto_replenish is not None:
+            self.allow_credits_auto_replenish = allow_credits_auto_replenish
+        if auto_replenish_amount is not None:
+            self.auto_replenish_amount = auto_replenish_amount
+        if auto_replenish_threshold is not None:
+            self.auto_replenish_threshold = auto_replenish_threshold
         if complete_sign_up is not None:
             self.complete_sign_up = complete_sign_up
         if completed_project_onboarding is not None:
@@ -191,6 +206,69 @@ class V1UpdateUserRequest(object):
 
         self._agree_to_terms_and_conditions = agree_to_terms_and_conditions
 
+    @property
+    def allow_credits_auto_replenish(self) -> 'bool':
+        """Gets the allow_credits_auto_replenish of this V1UpdateUserRequest.  # noqa: E501
+
+
+        :return: The allow_credits_auto_replenish of this V1UpdateUserRequest.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_credits_auto_replenish
+
+    @allow_credits_auto_replenish.setter
+    def allow_credits_auto_replenish(self, allow_credits_auto_replenish: 'bool'):
+        """Sets the allow_credits_auto_replenish of this V1UpdateUserRequest.
+
+
+        :param allow_credits_auto_replenish: The allow_credits_auto_replenish of this V1UpdateUserRequest.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_credits_auto_replenish = allow_credits_auto_replenish
+
+    @property
+    def auto_replenish_amount(self) -> 'float':
+        """Gets the auto_replenish_amount of this V1UpdateUserRequest.  # noqa: E501
+
+
+        :return: The auto_replenish_amount of this V1UpdateUserRequest.  # noqa: E501
+        :rtype: float
+        """
+        return self._auto_replenish_amount
+
+    @auto_replenish_amount.setter
+    def auto_replenish_amount(self, auto_replenish_amount: 'float'):
+        """Sets the auto_replenish_amount of this V1UpdateUserRequest.
+
+
+        :param auto_replenish_amount: The auto_replenish_amount of this V1UpdateUserRequest.  # noqa: E501
+        :type: float
+        """
+
+        self._auto_replenish_amount = auto_replenish_amount
+
+    @property
+    def auto_replenish_threshold(self) -> 'float':
+        """Gets the auto_replenish_threshold of this V1UpdateUserRequest.  # noqa: E501
+
+
+        :return: The auto_replenish_threshold of this V1UpdateUserRequest.  # noqa: E501
+        :rtype: float
+        """
+        return self._auto_replenish_threshold
+
+    @auto_replenish_threshold.setter
+    def auto_replenish_threshold(self, auto_replenish_threshold: 'float'):
+        """Sets the auto_replenish_threshold of this V1UpdateUserRequest.
+
+
+        :param auto_replenish_threshold: The auto_replenish_threshold of this V1UpdateUserRequest.  # noqa: E501
+        :type: float
+        """
+
+        self._auto_replenish_threshold = auto_replenish_threshold
+
     @property
     def complete_sign_up(self) -> 'bool':
         """Gets the complete_sign_up of this V1UpdateUserRequest.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -66,7 +66,6 @@ class V1UserFeatures(object):
         'datasets': 'bool',
         'default_one_cluster': 'bool',
         'deployment_persistent_disk': 'bool',
-        'docs_agent': 'bool',
         'drive_v2': 'bool',
         'enterprise_compute_admin': 'bool',
         'fair_share': 'bool',
@@ -111,15 +110,12 @@ class V1UserFeatures(object):
         'restartable_jobs': 'bool',
         'runnable_public_studio_page': 'bool',
         'security_docs': 'bool',
-        'seoul_aws_region': 'bool',
         'show_dev_admin': 'bool',
         'single_wallet': 'bool',
         'slurm': 'bool',
-        'slurm_machine_selector': 'bool',
         'storage_overuse_deletion': 'bool',
         'studio_config': 'bool',
         'studio_version_visibility': 'bool',
-        'teamspace_storage_tab': 'bool',
         'trainium2': 'bool',
         'use_rclone_mounts_only': 'bool',
         'voltage_park': 'bool',
@@ -154,7 +150,6 @@ class V1UserFeatures(object):
         'datasets': 'datasets',
         'default_one_cluster': 'defaultOneCluster',
         'deployment_persistent_disk': 'deploymentPersistentDisk',
-        'docs_agent': 'docsAgent',
         'drive_v2': 'driveV2',
         'enterprise_compute_admin': 'enterpriseComputeAdmin',
         'fair_share': 'fairShare',
@@ -199,15 +194,12 @@ class V1UserFeatures(object):
         'restartable_jobs': 'restartableJobs',
         'runnable_public_studio_page': 'runnablePublicStudioPage',
         'security_docs': 'securityDocs',
-        'seoul_aws_region': 'seoulAwsRegion',
         'show_dev_admin': 'showDevAdmin',
         'single_wallet': 'singleWallet',
         'slurm': 'slurm',
-        'slurm_machine_selector': 'slurmMachineSelector',
         'storage_overuse_deletion': 'storageOveruseDeletion',
         'studio_config': 'studioConfig',
         'studio_version_visibility': 'studioVersionVisibility',
-        'teamspace_storage_tab': 'teamspaceStorageTab',
         'trainium2': 'trainium2',
         'use_rclone_mounts_only': 'useRcloneMountsOnly',
         'voltage_park': 'voltagePark',
@@ -216,7 +208,7 @@ class V1UserFeatures(object):
         'weka': 'weka'
     }
 
-    def __init__(self, accurate_billing: '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, auto_top_up: 'bool' =None, auto_top_up_teamspace: '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, cloud_space_environment_templates: '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, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: '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, 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, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, pipelines: 'bool' =None, plg_control_center: '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, seoul_aws_region: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None):  # noqa: E501
+    def __init__(self, accurate_billing: '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, auto_top_up: 'bool' =None, auto_top_up_teamspace: '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, cloud_space_environment_templates: '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, 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_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: '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, 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, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, pipelines: 'bool' =None, plg_control_center: '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, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._accurate_billing = None
         self._affiliate_links = None
@@ -243,7 +235,6 @@ class V1UserFeatures(object):
         self._datasets = None
         self._default_one_cluster = None
         self._deployment_persistent_disk = None
-        self._docs_agent = None
         self._drive_v2 = None
         self._enterprise_compute_admin = None
         self._fair_share = None
@@ -288,15 +279,12 @@ class V1UserFeatures(object):
         self._restartable_jobs = None
         self._runnable_public_studio_page = None
         self._security_docs = None
-        self._seoul_aws_region = None
         self._show_dev_admin = None
         self._single_wallet = None
         self._slurm = None
-        self._slurm_machine_selector = None
         self._storage_overuse_deletion = None
         self._studio_config = None
         self._studio_version_visibility = None
-        self._teamspace_storage_tab = None
         self._trainium2 = None
         self._use_rclone_mounts_only = None
         self._voltage_park = None
@@ -354,8 +342,6 @@ class V1UserFeatures(object):
             self.default_one_cluster = default_one_cluster
         if deployment_persistent_disk is not None:
             self.deployment_persistent_disk = deployment_persistent_disk
-        if docs_agent is not None:
-            self.docs_agent = docs_agent
         if drive_v2 is not None:
             self.drive_v2 = drive_v2
         if enterprise_compute_admin is not None:
@@ -444,24 +430,18 @@ class V1UserFeatures(object):
             self.runnable_public_studio_page = runnable_public_studio_page
         if security_docs is not None:
             self.security_docs = security_docs
-        if seoul_aws_region is not None:
-            self.seoul_aws_region = seoul_aws_region
         if show_dev_admin is not None:
             self.show_dev_admin = show_dev_admin
         if single_wallet is not None:
             self.single_wallet = single_wallet
         if slurm is not None:
             self.slurm = slurm
-        if slurm_machine_selector is not None:
-            self.slurm_machine_selector = slurm_machine_selector
         if storage_overuse_deletion is not None:
             self.storage_overuse_deletion = storage_overuse_deletion
         if studio_config is not None:
             self.studio_config = studio_config
         if studio_version_visibility is not None:
             self.studio_version_visibility = studio_version_visibility
-        if teamspace_storage_tab is not None:
-            self.teamspace_storage_tab = teamspace_storage_tab
         if trainium2 is not None:
             self.trainium2 = trainium2
         if use_rclone_mounts_only is not None:
@@ -1000,27 +980,6 @@ class V1UserFeatures(object):
 
         self._deployment_persistent_disk = deployment_persistent_disk
 
-    @property
-    def docs_agent(self) -> 'bool':
-        """Gets the docs_agent of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The docs_agent of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._docs_agent
-
-    @docs_agent.setter
-    def docs_agent(self, docs_agent: 'bool'):
-        """Sets the docs_agent of this V1UserFeatures.
-
-
-        :param docs_agent: The docs_agent of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._docs_agent = docs_agent
-
     @property
     def drive_v2(self) -> 'bool':
         """Gets the drive_v2 of this V1UserFeatures.  # noqa: E501
@@ -1945,27 +1904,6 @@ class V1UserFeatures(object):
 
         self._security_docs = security_docs
 
-    @property
-    def seoul_aws_region(self) -> 'bool':
-        """Gets the seoul_aws_region of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The seoul_aws_region of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._seoul_aws_region
-
-    @seoul_aws_region.setter
-    def seoul_aws_region(self, seoul_aws_region: 'bool'):
-        """Sets the seoul_aws_region of this V1UserFeatures.
-
-
-        :param seoul_aws_region: The seoul_aws_region of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._seoul_aws_region = seoul_aws_region
-
     @property
     def show_dev_admin(self) -> 'bool':
         """Gets the show_dev_admin of this V1UserFeatures.  # noqa: E501
@@ -2029,27 +1967,6 @@ class V1UserFeatures(object):
 
         self._slurm = slurm
 
-    @property
-    def slurm_machine_selector(self) -> 'bool':
-        """Gets the slurm_machine_selector of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The slurm_machine_selector of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._slurm_machine_selector
-
-    @slurm_machine_selector.setter
-    def slurm_machine_selector(self, slurm_machine_selector: 'bool'):
-        """Sets the slurm_machine_selector of this V1UserFeatures.
-
-
-        :param slurm_machine_selector: The slurm_machine_selector of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._slurm_machine_selector = slurm_machine_selector
-
     @property
     def storage_overuse_deletion(self) -> 'bool':
         """Gets the storage_overuse_deletion of this V1UserFeatures.  # noqa: E501
@@ -2113,27 +2030,6 @@ class V1UserFeatures(object):
 
         self._studio_version_visibility = studio_version_visibility
 
-    @property
-    def teamspace_storage_tab(self) -> 'bool':
-        """Gets the teamspace_storage_tab of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The teamspace_storage_tab of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._teamspace_storage_tab
-
-    @teamspace_storage_tab.setter
-    def teamspace_storage_tab(self, teamspace_storage_tab: 'bool'):
-        """Sets the teamspace_storage_tab of this V1UserFeatures.
-
-
-        :param teamspace_storage_tab: The teamspace_storage_tab of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._teamspace_storage_tab = teamspace_storage_tab
-
     @property
     def trainium2(self) -> 'bool':
         """Gets the trainium2 of this V1UserFeatures.  # noqa: E501

lightning_sdk/llm/__init__.py

@@ -1,4 +1,3 @@
-from lightning_sdk.llm.asyncllm import AsyncLLM
 from lightning_sdk.llm.llm import LLM
 
-__all__ = ["LLM", "AsyncLLM"]
+__all__ = ["LLM"]

lightning_sdk/llm/llm.py

@@ -1,6 +1,6 @@
 import os
 import warnings
-from typing import Dict, Generator, List, Optional, Set, Tuple, Union
+from typing import AsyncGenerator, Dict, Generator, List, Optional, Set, Tuple, Union
 
 from lightning_sdk.api.llm_api import LLMApi
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
@@ -18,6 +18,7 @@ class LLM:
         self,
         name: str,
         teamspace: Optional[str] = None,
+        enable_async: Optional[bool] = False,
     ) -> None:
         """Initializes the LLM instance with teamspace information, which is required for billing purposes.
 
@@ -30,6 +31,7 @@ class LLM:
             name (str): The name of the model or resource.
             teamspace (Optional[str]): The specified teamspace for billing. If not provided, it will be resolved
                                        through the above methods.
+            enable_async (Optional[bool]): Enable async requests
 
         Raises:
             ValueError: If teamspace information cannot be resolved.
@@ -67,6 +69,7 @@ class LLM:
         self._model_provider, self._model_name = self._parse_model_name(name)
 
         self._llm_api = LLMApi()
+        self._enable_async = enable_async
 
         try:
             # check if it is a org model
@@ -174,6 +177,41 @@ class LLM:
         for line in result:
             yield line.choices[0].delta.content
 
+    async def _async_stream_text(self, output: str) -> AsyncGenerator[str, None]:
+        async for chunk in output:
+            if chunk.choices and chunk.choices[0].delta:
+                yield chunk.choices[0].delta.content
+
+    async def _async_chat(
+        self,
+        prompt: str,
+        system_prompt: Optional[str] = None,
+        max_completion_tokens: Optional[int] = 500,
+        images: Optional[Union[List[str], str]] = None,
+        conversation: Optional[str] = None,
+        metadata: Optional[Dict[str, str]] = None,
+        stream: bool = False,
+        upload_local_images: bool = False,
+    ) -> Union[str, AsyncGenerator[str, None]]:
+        conversation_id = self._conversations.get(conversation) if conversation else None
+        output = await self._llm_api.async_start_conversation(
+            prompt=prompt,
+            system_prompt=system_prompt,
+            max_completion_tokens=max_completion_tokens,
+            images=images,
+            assistant_id=self._model.id,
+            conversation_id=conversation_id,
+            billing_project_id=self._teamspace.id,
+            metadata=metadata,
+            name=conversation,
+            stream=stream,
+        )
+        if not stream:
+            if conversation and not conversation_id:
+                self._conversations[conversation] = output.conversation_id
+            return output.choices[0].delta.content
+        return self._async_stream_text(output)
+
     def chat(
         self,
         prompt: str,
@@ -198,6 +236,19 @@ class LLM:
                     self._teamspace.upload_file(file_path=image, remote_path=f"images/{os.path.basename(image)}")
 
         conversation_id = self._conversations.get(conversation) if conversation else None
+
+        if self._enable_async:
+            return self._async_chat(
+                prompt,
+                system_prompt,
+                max_completion_tokens,
+                images,
+                conversation,
+                metadata,
+                stream,
+                upload_local_images,
+            )
+
         output = self._llm_api.start_conversation(
             prompt=prompt,
             system_prompt=system_prompt,

lightning_sdk/pipeline/__init__.py

@@ -1,4 +1,5 @@
 from lightning_sdk.pipeline.pipeline import Pipeline
+from lightning_sdk.pipeline.schedule import Schedule
 from lightning_sdk.pipeline.types import MMT, Deployment, Job
 
-__all__ = ["Pipeline", "Job", "MMT", "Deployment"]
+__all__ = ["Pipeline", "Job", "MMT", "Deployment", "Schedule"]

lightning_sdk/pipeline/pipeline.py

@@ -1,9 +1,10 @@
-from typing import List, Optional, Union
+from typing import TYPE_CHECKING, List, Optional, Union
 
 from lightning_sdk.api import UserApi
 from lightning_sdk.api.pipeline_api import PipelineApi
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.organization import Organization
+from lightning_sdk.pipeline.printer import PipelinePrinter
 from lightning_sdk.pipeline.types import MMT, Deployment, Job
 from lightning_sdk.pipeline.utils import prepare_steps
 from lightning_sdk.services.utilities import _get_cluster
@@ -11,6 +12,9 @@ from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.user import User
 from lightning_sdk.utils.resolve import _resolve_org, _resolve_teamspace, _resolve_user
 
+if TYPE_CHECKING:
+    from lightning_sdk.pipeline.schedule import Schedule
+
 
 class Pipeline:
     def __init__(
@@ -62,15 +66,16 @@ class Pipeline:
 
         pipeline = None
 
-        if name.startswith("pip_"):
-            pipeline = self._pipeline_api.get_pipeline_by_id(name, self._teamspace.id)
+        pipeline = self._pipeline_api.get_pipeline_by_id(self._teamspace.id, name)
 
         if pipeline:
             self._name = pipeline.name
             self._is_created = True
             self._pipeline = pipeline
+        else:
+            self._pipeline = None
 
-    def run(self, steps: List[Union[Job, Deployment, MMT]]) -> None:
+    def run(self, steps: List[Union[Job, Deployment, MMT]], schedules: Optional[List["Schedule"]] = None) -> None:
         if len(steps) == 0:
             raise ValueError("The provided steps is empty")
 
@@ -83,12 +88,24 @@ class Pipeline:
             for step in steps
         ]
 
+        proto_steps = prepare_steps(steps)
+        schedules = schedules or []
+
+        parent_pipeline_id = None if self._pipeline is None else self._pipeline.id
+
         self._pipeline = self._pipeline_api.create_pipeline(
             self._name,
             self._teamspace.id,
-            prepare_steps(steps),
+            proto_steps,
             self._shared_filesystem or False,
+            schedules,
+            parent_pipeline_id,
+        )
+
+        printer = PipelinePrinter(
+            self._name, parent_pipeline_id is None, self._pipeline, self._teamspace, proto_steps, schedules
         )
+        printer.print_summary()
 
     def stop(self) -> None:
         if self._pipeline is None:
@@ -103,7 +120,7 @@ class Pipeline:
         self._pipeline_api.delete(self._teamspace.id, self._pipeline.id)
 
     @property
-    def name(self) -> str:
+    def name(self) -> Optional[str]:
         if self._pipeline:
             return self._pipeline.name
         return None

lightning_sdk/pipeline/printer.py

@@ -0,0 +1,102 @@
+import os
+from typing import Any, ClassVar, Dict, List
+
+from lightning_sdk.lightning_cloud.openapi.models import V1JobSpec, V1PipelineStepType
+
+
+class PipelinePrinter:
+    """A helper class to print a formatted summary of a pipeline."""
+
+    STEP_TYPE_MAP: ClassVar[Dict[str, str]] = {
+        V1PipelineStepType.DEPLOYMENT: "Deployment",
+        V1PipelineStepType.JOB: "Job",
+        V1PipelineStepType.MMT: "MMT",
+    }
+
+    def __init__(
+        self, name: str, initial: bool, pipeline: Any, teamspace: Any, proto_steps: List[Any], schedules: List[Any]
+    ) -> None:
+        self._name = name
+        self._initial = initial
+        self._pipeline = pipeline
+        self._teamspace = teamspace
+        self._proto_steps = proto_steps
+        self._schedules = schedules
+
+    def print_summary(self) -> None:
+        """Prints the full, formatted summary of the created pipeline."""
+        self._print("\n" + "─" * 60)
+        self._print(f"✅ Pipeline '{self._name}' {'created' if self._initial else 'updated'} successfully!")
+        self._print("─" * 60)
+
+        self._print_steps()
+        self._print_schedules()
+        self._print_cloud_account()
+        self._print_footer()
+
+    def _print(self, value: str) -> None:
+        print(value)
+
+    def _print_steps(self) -> None:
+        """Prints the formatted list of pipeline steps."""
+        self._print("\nWorkflow Steps:")
+        if not self._proto_steps:
+            self._print("  - No steps defined.")
+            return
+
+        for i, step in enumerate(self._proto_steps):
+            step_type = self.STEP_TYPE_MAP.get(step.type, "Unknown Step")
+
+            # Format the 'wait_for' list for cleaner output
+            if not step.wait_for:
+                wait_for_str = "(runs first)"
+            else:
+                # e.g., "'step_a', 'step_b'"
+                wait_for_str = f" waits for {', '.join([f'{w}' for w in step.wait_for])}"
+
+            self._print(f"  ➡️ {i+1}. {step_type} '{step.name}' - {wait_for_str}")
+
+    def _print_schedules(self) -> None:
+        """Prints the formatted list of schedules."""
+        self._print("\n🗓️ Schedules:")
+        if not self._schedules:
+            self._print("  - No schedules defined.")
+            return
+
+        for schedule in self._schedules:
+            self._print(f"  - '{schedule.name}' runs on cron schedule: `{schedule.cron_expression}`")
+
+    def _print_footer(self) -> None:
+        """Prints the final link and closing message."""
+        cloud_url = os.getenv("LIGHTNING_CLOUD_URL", "https://lightning.ai").replace(":443", "")
+
+        # Using properties of assumed objects for a cleaner look
+        owner: str = self._teamspace.owner.name
+        team: str = self._teamspace.name
+        pipeline_name: str = self._name
+
+        pipeline_url = f"{cloud_url}/{owner}/{team}/pipelines/{pipeline_name}?app_id=pipeline&section=Graph"
+
+        self._print("\n" + "─" * 60)
+        self._print(f"🔗 View your pipeline in the browser:\n   {pipeline_url}")
+        self._print("─" * 60 + "\n")
+
+    def _print_cloud_account(self) -> None:
+        if not self._proto_steps:
+            return
+
+        cluster_ids: set[str] = set()
+        for step in self._proto_steps:
+            job_spec = self._get_spec(step)
+            cluster_ids.add(job_spec.cluster_id)
+
+        self._print(f"\nCloud account{'s' if len(cluster_ids) > 1 else ''}:")
+        for cluster_id in sorted(cluster_ids):
+            self._print(f"  - {cluster_id}")
+
+    def _get_spec(self, step: Any) -> V1JobSpec:
+        if step.type == V1PipelineStepType.DEPLOYMENT:
+            return step.deployment.spec
+        if step.type == V1PipelineStepType.MMT:
+            return step.mmt.spec
+        return step.job.spec

lightning_sdk/pipeline/schedule.py

@@ -0,0 +1,7 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class Schedule:
+    name: str
+    cron_expression: str

lightning_sdk/pipeline/types.py

@@ -55,7 +55,7 @@ class Deployment:
         cloud_account: Optional[str] = None,
         custom_domain: Optional[str] = None,
         quantity: Optional[int] = None,
-        wait_for: Union[str, List[str]] = DEFAULT,
+        wait_for: Optional[Union[str, List[str]]] = DEFAULT,
     ) -> None:
         self.name = name
         self.machine = machine
@@ -132,7 +132,7 @@ class Job:
         cloud_account_auth: bool = False,
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
-        wait_for: Union[str, List[str]] = DEFAULT,
+        wait_for: Union[str, List[str], None] = DEFAULT,
     ) -> None:
         self.name = name
         self.machine = machine
@@ -207,7 +207,7 @@ class MMT:
         cloud_account_auth: bool = False,
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
-        wait_for: Union[str, List[str]] = DEFAULT,
+        wait_for: Optional[Union[str, List[str]]] = DEFAULT,
     ) -> None:
         self.machine = machine
         self.num_machines = num_machines