craft-ai-sdk 0.67.1__tar.gz → 0.68.0rc1__tar.gz

{craft_ai_sdk-0.67.1 → craft_ai_sdk-0.68.0rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: craft-ai-sdk
-Version: 0.67.1
+Version: 0.68.0rc1
 Summary: Craft AI MLOps platform SDK
 License: Apache-2.0
 Author: Craft AI

{craft_ai_sdk-0.67.1 → craft_ai_sdk-0.68.0rc1}/craft_ai_sdk/__init__.py

@@ -13,4 +13,4 @@ from .io import (  # noqa: F401
 )
 from .sdk import CraftAiSdk  # noqa: F401
 
-__version__ = "0.67.1"
+__version__ = "0.68.0rc1"

{craft_ai_sdk-0.67.1 → craft_ai_sdk-0.68.0rc1}/craft_ai_sdk/core/data_store.py

@@ -75,7 +75,7 @@ def iter_data_store_objects(
         if continuation_token is None:
             return
         else:
-            params = {"continuation_token": continuation_token, **params}
+            params["continuation_token"] = continuation_token
 
 
 def list_data_store_objects(

{craft_ai_sdk-0.67.1 → craft_ai_sdk-0.68.0rc1}/craft_ai_sdk/core/endpoints.py

@@ -199,15 +199,18 @@ def retrieve_endpoint_results(
         if isinstance(value, dict) and "value" in value:
             outputs[key] = value["value"]
         if isinstance(value, dict) and "url" in value:
-            # If the output is a file, we need to download it
-            url = f"{value['url']}?token={endpoint_token}"
-            file_response = requests.get(url)
-            if file_response.status_code == 200:
-                outputs[key] = file_response.content
+            if not isinstance(value["url"], str):
+                outputs[key] = None
             else:
-                raise ValueError(
-                    f"Failed to download file output {key}: {file_response.text}"
-                )
+                # If the output is a file, we need to download it
+                url = f"{value['url']}?token={endpoint_token}"
+                file_response = requests.get(url)
+                if file_response.status_code == 200:
+                    outputs[key] = file_response.content
+                else:
+                    raise ValueError(
+                        f"Failed to download file output {key}: {file_response.text}"
+                    )
 
     return {
         "outputs": response_data.get("outputs", []),

{craft_ai_sdk-0.67.1 → craft_ai_sdk-0.68.0rc1}/craft_ai_sdk/core/pipelines.py

@@ -84,16 +84,14 @@ def _create_pipeline_from_template(
 
 def _create_pipeline_with_step(
     sdk: BaseCraftAiSdk, pipeline_name: str, step_name: str
-) -> Pipeline:
+) -> None:
     url = f"{sdk.base_environment_api_url}/pipelines"
     body = {
         "pipeline_name": pipeline_name,
         "step_names": [step_name],
     }
 
-    return remove_keys_from_dict(
-        sdk._post(url, json=body), {"is_archived", "pipeline_id"}
-    )
+    sdk._post(url, json=body)
 
 
 @overload
@@ -337,7 +335,8 @@ def create_pipeline(
             FutureWarning,
             stacklevel=2,
         )
-        return _create_pipeline_with_step(sdk, pipeline_name, function_path)
+        _create_pipeline_with_step(sdk, pipeline_name, function_path)
+        return get_pipeline(sdk, pipeline_name, wait_for_completion, timeout_s)
 
     # or with create_pipeline(:pipeline_name:, step_name=:step_name:).
     if "step_name" in kwargs:
@@ -348,7 +347,8 @@ def create_pipeline(
                 "should be provided."
             )
 
-        return _create_pipeline_with_step(sdk, pipeline_name, kwargs["step_name"])
+        _create_pipeline_with_step(sdk, pipeline_name, kwargs["step_name"])
+        return get_pipeline(sdk, pipeline_name, wait_for_completion, timeout_s)
 
     if len(kwargs) != 0:
         raise ValueError(

{craft_ai_sdk-0.67.1 → craft_ai_sdk-0.68.0rc1}/craft_ai_sdk/sdk.py

@@ -141,7 +141,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.67.1"  # Would be better to share it somewhere
+    _version = "0.68.0rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.67.1 → craft_ai_sdk-0.68.0rc1}/pyproject.toml

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