craft-ai-sdk 0.67.1rc1__py3-none-any.whl → 0.68.0rc1__py3-none-any.whl

craft_ai_sdk/__init__.py

@@ -13,4 +13,4 @@ from .io import (  # noqa: F401
 )
 from .sdk import CraftAiSdk  # noqa: F401
 
-__version__ = "0.67.1rc1"
+__version__ = "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/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/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/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.1rc1"  # 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.1rc1.dist-info → craft_ai_sdk-0.68.0rc1.dist-info}/METADATA

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

{craft_ai_sdk-0.67.1rc1.dist-info → craft_ai_sdk-0.68.0rc1.dist-info}/RECORD

@@ -1,20 +1,20 @@
-craft_ai_sdk/__init__.py,sha256=vSigS0OentQ4zqfZyqi1m5cKGQz1nVdRznuZl09gzwM,366
+craft_ai_sdk/__init__.py,sha256=PmZ7jK4ktFe0Vc6ojwajH4aHLHFmR-CBReXEajvn2v4,366
 craft_ai_sdk/constants.py,sha256=rH4JrGlTpbjjjNRrKhk5oScbj5G5INrcVza6Bb6kIzY,980
-craft_ai_sdk/core/data_store.py,sha256=5SYJoU9xXqLBc5540Xb6UnLYOY0V5SQGor31nF3jN8A,9968
+craft_ai_sdk/core/data_store.py,sha256=BPQalGcm6bujgkgu-MtNWWhKiJEg4MiVuCtByNbTbK8,9956
 craft_ai_sdk/core/deployments.py,sha256=Hig0Z6tZXbaLxms9ENtKI4OG3ki7G4Iz3wGk2V2Gmfg,37731
-craft_ai_sdk/core/endpoints.py,sha256=ad8_oHVCwrTpnlrlqvEcO48nXbzmuWTaUbh82JdRX6g,8018
+craft_ai_sdk/core/endpoints.py,sha256=cFf8scftD7XlMvpglLNRew-fesLUC_TcZ2UPyCbmUaA,8158
 craft_ai_sdk/core/environment_variables.py,sha256=CpkyneLf8UF-1QXe36Ay_RZJ8grSfbZZWBvGSQzKlLs,2221
 craft_ai_sdk/core/pipeline_executions.py,sha256=EfhNDzWiKro9-gDUPHJFMFcdPXtznlJo2leFKLloDmA,18709
 craft_ai_sdk/core/pipeline_metrics.py,sha256=AYxF5HpLCfTYK-TydEHswaAHat3N9NiOTHpPd_R9-rY,7190
 craft_ai_sdk/core/pipeline_templates.py,sha256=3WjM4vsURpTNF3dt42h_xk68ysxFkz4kiOxqAb4VyRE,5623
-craft_ai_sdk/core/pipelines.py,sha256=lzDzmnRrwwsoXABb46sGO5tu27QEtBzCisSiuiIFsiU,26521
+craft_ai_sdk/core/pipelines.py,sha256=FD6ai2_YhE1tz6QO57bj_lA-i93TXZzm0Y9PA0mpocg,26587
 craft_ai_sdk/core/resource_metrics.py,sha256=ubCe6QOhud3f-EQr_TXKBsrFFaxZubqTS3fZqfV7kgg,3457
 craft_ai_sdk/core/steps.py,sha256=TwOWXAAVpLPuSZ_gvf-3E5uKEvfGIzR52Y58bqKn_Ds,23487
 craft_ai_sdk/core/users.py,sha256=q5et87q0SOMpRTBOiLX026oaEOTaY1aGqKdbUC51zbA,618
 craft_ai_sdk/core/vector_database.py,sha256=S3h68Ej1FnHYJdc5LdLlNjobYYdmaCloSq2UaylPWko,2334
 craft_ai_sdk/exceptions.py,sha256=IC-JfZmmmaTsbMCgirOEByRmWnatQLjKe8BErRkuwM0,1075
 craft_ai_sdk/io.py,sha256=x1G7ga10W_SeTDLHe-FrhU26_WNwWW32-yprMmCZ5Ck,11836
-craft_ai_sdk/sdk.py,sha256=ulffaxyFBvPJIiZrJaiOtUQ8oDuZ42Yh1NSG7Mo4WaE,11031
+craft_ai_sdk/sdk.py,sha256=Ruica7u91TsK4D8_db149u8-dYAuLaPq2Q4QFao_kQI,11031
 craft_ai_sdk/shared/authentication.py,sha256=5lJS_aNphkgI0-wvHFK4aW316ZRfOq9oywNpv-Oyn2M,1062
 craft_ai_sdk/shared/environments.py,sha256=JOV3s0KIFVl_zskJb--bybzIXEGQJjrf_T4dL-szSlQ,500
 craft_ai_sdk/shared/execution_context.py,sha256=B2Ghq-wiUvq81q5mhsm79Oc59c8c00uQxMIpApFD03o,585
@@ -27,8 +27,8 @@ craft_ai_sdk/utils/__init__.py,sha256=A0sLCXSPD1Z3q2GP1uLDjvif4ivOr__Hzg9RQysEuq
 craft_ai_sdk/utils/datetime_utils.py,sha256=yYP5HVdI879WXxQCajPTnas1pWrwInOxMux-mxqQNQM,734
 craft_ai_sdk/utils/dict_utils.py,sha256=1HQ3A14SN48XPFDmQleujGAgksmkjIs3hyPLpwhwh24,748
 craft_ai_sdk/utils/file_utils.py,sha256=o10-CDt4qzgCJNPykvlNrL6WTouhVLY8C8BVpHJYt18,2795
-craft_ai_sdk-0.67.1rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.67.1rc1.dist-info/METADATA,sha256=7mn1ROp_sqa2QIsKabqTeGYNvxSr-IXP-AdVs0bwDyk,1679
-craft_ai_sdk-0.67.1rc1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-craft_ai_sdk-0.67.1rc1.dist-info/entry_points.txt,sha256=QC96WcXvvUfLMRgFD-l_y7_TgC9SqZybLs9EQ8dsGiQ,417
-craft_ai_sdk-0.67.1rc1.dist-info/RECORD,,
+craft_ai_sdk-0.68.0rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.68.0rc1.dist-info/METADATA,sha256=8ycWBge6n-ZB-gWXnPbERXuPQtMngyJh_QKOkKY5tAc,1679
+craft_ai_sdk-0.68.0rc1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+craft_ai_sdk-0.68.0rc1.dist-info/entry_points.txt,sha256=QC96WcXvvUfLMRgFD-l_y7_TgC9SqZybLs9EQ8dsGiQ,417
+craft_ai_sdk-0.68.0rc1.dist-info/RECORD,,