craft-ai-sdk 0.56.0rc1__tar.gz → 0.57.0__tar.gz

{craft_ai_sdk-0.56.0rc1 → craft_ai_sdk-0.57.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: craft-ai-sdk
-Version: 0.56.0rc1
+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.0rc1 → craft_ai_sdk-0.57.0}/craft_ai_sdk/__init__.py

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

{craft_ai_sdk-0.56.0rc1 → craft_ai_sdk-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-0.56.0rc1 → craft_ai_sdk-0.57.0}/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-0.56.0rc1 → craft_ai_sdk-0.57.0}/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.0rc1"  # 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.0rc1 → craft_ai_sdk-0.57.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "craft-ai-sdk"
-version = "0.56.0rc1"
+version = "0.57.0"
 description = "Craft AI MLOps platform SDK"
 license = "Apache-2.0"
 authors = ["Craft AI <contact@craft.ai>"]