craft-ai-sdk 0.56.0__py3-none-any.whl → 0.57.0__py3-none-any.whl

craft_ai_sdk/__init__.py

@@ -14,4 +14,4 @@ from .io import (  # noqa: F401
 )
 
 
-__version__ = "0.56.0"
+__version__ = "0.57.0"

craft_ai_sdk/core/deployments.py

@@ -529,10 +529,14 @@ def get_deployment(
             deployment_status = deployment.get("status", None)
             elapsed_time = sdk._get_time() - start_time
 
-        if deployment_status not in [
-            DEPLOYMENT_STATUS.UP,
-            DEPLOYMENT_STATUS.CREATION_FAILED,
-        ]:
+        if deployment_status == DEPLOYMENT_STATUS.CREATION_FAILED:
+            raise SdkException(
+                "The deployment creation failed. Please check the logs for more \
+details.",
+                name="CreationFailedException",
+            )
+
+        if deployment_status != DEPLOYMENT_STATUS.UP:
             raise SdkException(
                 'The deployment was not ready in time. It is still being created but \
 this function stopped trying. Please check its status with "get_deployment" with \
@@ -579,6 +583,8 @@ def get_deployment_logs(
 
         * ``"timestamp"`` (:obj:`str`): Timestamp of the log.
         * ``"message"`` (:obj:`str`): Message of the log.
+        * ``"stream"`` (:obj:`str`): Stream of the log. Typically, ``"stdout"`` or
+          ``"stderr"``
         * ``"pod_id"`` (:obj:`str`): ID of the pod.
         * ``"type"`` (:obj:`str`): Type of the log. Can be ``"deployment"``.
     """

craft_ai_sdk/core/pipelines.py

@@ -2,7 +2,6 @@ import warnings
 import os
 import requests
 
-from ..exceptions import SdkException
 from ..sdk import BaseCraftAiSdk
 from ..utils import (
     log_action,
@@ -312,25 +311,20 @@ def get_pipeline(
           * ``"last_execution_id"`` (:obj:`str`): The id of the last execution of
             the pipeline.
     """
-    try:
-        base_url = f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}"
-        if wait_for_completion:
-            pipeline = _wait_create_until_ready(
-                sdk,
-                pipeline_name,
-                lambda sdk, _: sdk._get(
-                    f"{base_url}?wait_for_completion=true", allow_redirects=False
-                ),
-                timeout_s,
-                sdk._get_time(),
-                get_pipeline_logs,
-            )
-        else:
-            pipeline = sdk._get(base_url)
-    except SdkException as error:
-        if error.status_code == 404:
-            return None
-        raise error
+    base_url = f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}"
+    if wait_for_completion:
+        pipeline = _wait_create_until_ready(
+            sdk,
+            pipeline_name,
+            lambda sdk, _: sdk._get(
+                f"{base_url}?wait_for_completion=true", allow_redirects=False
+            ),
+            timeout_s,
+            sdk._get_time(),
+            get_pipeline_logs,
+        )
+    else:
+        pipeline = sdk._get(base_url)
 
     latest_execution = sdk._get(
         f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}/executions/latest"

craft_ai_sdk/sdk.py

@@ -130,7 +130,7 @@ class CraftAiSdk(BaseCraftAiSdk):
         os.environ.get("CRAFT_AI__MULTIPART_PART_SIZE__B", str(38 * 256 * 1024))
     )
     _access_token_margin = timedelta(seconds=30)
-    _version = "0.56.0"  # Would be better to share it somewhere
+    _version = "0.57.0"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.56.0.dist-info → craft_ai_sdk-0.57.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: craft-ai-sdk
-Version: 0.56.0
+Version: 0.57.0
 Summary: Craft AI MLOps platform SDK
 Home-page: https://www.craft.ai/
 License: Apache-2.0

{craft_ai_sdk-0.56.0.dist-info → craft_ai_sdk-0.57.0.dist-info}/RECORD

@@ -1,23 +1,23 @@
-craft_ai_sdk/__init__.py,sha256=YnT2RXICGR9rmJ8DbmrGMsyx_99wdRPApKuckgUlq-s,404
+craft_ai_sdk/__init__.py,sha256=JAsPpy1v4wuMAgpJxCpgwih5gED20EioFytwYg75d-o,404
 craft_ai_sdk/constants.py,sha256=5dR4JfKVz3KhBKRFSdN1nrb06kFkNiYP2mRnacsbzJA,629
 craft_ai_sdk/core/data_store.py,sha256=USONqCEwG_KoTv5a1fgZJmi7FIjbfo4Vi-zq82keo4E,8511
-craft_ai_sdk/core/deployments.py,sha256=fUZBHTgOWHPShMcIvNF7XEHJLAZJwJI8zLzMtdv-Dzc,31427
+craft_ai_sdk/core/deployments.py,sha256=u1xIo7wyaJM84Mt11lgImo1QGBLKTblyEJrcokioQig,31712
 craft_ai_sdk/core/endpoints.py,sha256=ggJ8IU0QVrGa7A0RBWiJY1Lo7Klfi9BbfGnovZD97FQ,4592
 craft_ai_sdk/core/environment_variables.py,sha256=wy0U7y0C7tRXOXOiWGz0vX5ajHSmOb_7J1fXlSahGpI,2045
 craft_ai_sdk/core/pipeline_executions.py,sha256=-ClmW3qtPIA9LIhs_tiECCPL6ykSmG2zMLnrBNDGZiU,18665
 craft_ai_sdk/core/pipeline_metrics.py,sha256=Tt5p6guHAf7nhq1wvJG3t7WqsYKqrNQ7sZcFKqXYmdM,6465
-craft_ai_sdk/core/pipelines.py,sha256=r-wTMDvH6D_4gDMjjWzjGYQdph6vv_DO8uh0wlec9S8,20331
+craft_ai_sdk/core/pipelines.py,sha256=EgqfPuhv95nuy5lymBDmFv9-c51PXrWfTAZ8VcbhHBI,20113
 craft_ai_sdk/core/resource_metrics.py,sha256=yaco_AZ4Wu5sxLRKsA3qt2R3-PMs2e32EGO0gsHtq2Y,2336
 craft_ai_sdk/core/steps.py,sha256=1yTqcGrAKkVQWzslLjz2XQ3I7hILq8Oyc11Z9FeuFk0,20217
 craft_ai_sdk/core/users.py,sha256=Qi3xg7Vzk_0MepccNIZbFzjI0bBqouqkFStTuejUY6s,509
 craft_ai_sdk/exceptions.py,sha256=IC-JfZmmmaTsbMCgirOEByRmWnatQLjKe8BErRkuwM0,1075
 craft_ai_sdk/io.py,sha256=Jh9bFfXRzwCequXvJ3DKO-SIdZ7teYfzO64N940FyDY,11861
-craft_ai_sdk/sdk.py,sha256=lTl_A2TLifv8Qn9tX96pxJfymMdziBtzNIEUM8ycVk8,10068
+craft_ai_sdk/sdk.py,sha256=bhrt9CWTW26_uyH7fcQ2BHaQmB9BonVIxc7TP-8SsSM,10068
 craft_ai_sdk/utils.py,sha256=syOtpNgV8XF9VoeRVnWGkjIdWv0vyS_Z8RN5QfHCfIo,11891
 craft_ai_sdk/warnings.py,sha256=xJXbUR66SVbG_ZO7rxpAZ2hkGxdUtM5JAx6JoWg77Qs,6678
-documentation.pdf,sha256=9NPdcWoQJcO-Fx4CMppLYnN1tDg7FwP6xYTNxDlX-lk,333575
-craft_ai_sdk-0.56.0.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.56.0.dist-info/METADATA,sha256=src4TBy2Y4PuI3rdO281g6Axclvekh2F2THsYipMV-8,1614
-craft_ai_sdk-0.56.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-craft_ai_sdk-0.56.0.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.56.0.dist-info/RECORD,,
+documentation.pdf,sha256=nBaepZVtiRBTCWc3IJnKVdVGU9iPLMz0jkirvyeyTTw,333557
+craft_ai_sdk-0.57.0.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.57.0.dist-info/METADATA,sha256=xFNuWzBiSRO0DPapOOzX1iHifSC8D1kpHjxUu8QPgWo,1614
+craft_ai_sdk-0.57.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+craft_ai_sdk-0.57.0.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
+craft_ai_sdk-0.57.0.dist-info/RECORD,,