craft-ai-sdk 0.54.0rc1__py3-none-any.whl → 0.54.1__py3-none-any.whl

craft_ai_sdk/__init__.py

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

craft_ai_sdk/core/data_store.py

@@ -148,7 +148,7 @@ def _upload_multipart_data_store_object(
             "chunk_size": sdk._MULTIPART_PART_SIZE,
             "part_number": part_idx,
         }
-        if len < sdk._MULTIPART_PART_SIZE:
+        if part["lastChunk"]:
             data["size"] = (part_idx - 1) * sdk._MULTIPART_PART_SIZE + len
         multipart_part_result = sdk._post(url=f"{multipart_base_url}/part", json=data)
         resp = requests.put(
@@ -156,10 +156,12 @@ def _upload_multipart_data_store_object(
             data=chunk,
             headers=multipart_part_result["headers"],
         )
-        part = {"number": part_idx}
+        partData = {"number": part_idx}
         if "ETag" in resp.headers:
-            part["metadata"] = resp.headers["ETag"]
-        parts.append(part)
+            partData["metadata"] = resp.headers["ETag"]
+        parts.append(partData)
+        if part["lastChunk"]:
+            break
 
     sdk._post(
         url=f"{multipart_base_url}/end",

craft_ai_sdk/io.py

@@ -237,7 +237,7 @@ class OutputDestination:
             )
         if pipeline_output_name is None and step_output_name is None:
             raise ValueError('missing "pipeline_output_name" argument.')
-        if pipeline_output_name is not None:
+        if step_output_name is not None:
             warnings.warn(
                 "Providing the 'step_output_name' argument is deprecated and will "
                 "be removed in a future version. Please use the 'pipeline_output_name' keyword argument instead.",  # noqa: E501

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

craft_ai_sdk/utils.py

@@ -278,13 +278,25 @@ def map_container_config_step_parameter(container_config):
 def chunk_buffer(buffer: IOBase, size: int) -> Iterable[Union[BytesIO, StringIO]]:
     size_int = int(size)
     b = buffer.read(size_int)
+    next_data = None
     while b:
-        yield (
-            {"chunk": StringIO(b), "len": len(b)}
-            if isinstance(b, str)
-            else {"chunk": BytesIO(b), "len": len(b)}
-        )
-        b = buffer.read(size_int)
+        chunk = StringIO() if isinstance(b, str) else BytesIO()
+        previous_data = next_data
+        if previous_data:
+            chunk.write(next_data)
+        chunk.write(b)
+        chunk.seek(0)
+
+        next_data = buffer.read(1)
+
+        data = {
+            "chunk": chunk,
+            "len": len(b) + (len(previous_data) if previous_data else 0),
+            "lastChunk": len(next_data) == 0,
+        }
+        yield data
+        chunk.close()
+        b = buffer.read(size_int - 1)
 
 
 def convert_size(size_in_bytes):

{craft_ai_sdk-0.54.0rc1.dist-info → craft_ai_sdk-0.54.1.dist-info}/METADATA

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

{craft_ai_sdk-0.54.0rc1.dist-info → craft_ai_sdk-0.54.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
-craft_ai_sdk/__init__.py,sha256=aFYFR9NYnekNpnJGtyEfoZkYbaPxEMg-fBLVUMSavz8,407
+craft_ai_sdk/__init__.py,sha256=Tg8k-Ecy9qqYytNNAbLFFZ83yVZ9bRIblwS_bFnFn3I,404
 craft_ai_sdk/constants.py,sha256=5dR4JfKVz3KhBKRFSdN1nrb06kFkNiYP2mRnacsbzJA,629
-craft_ai_sdk/core/data_store.py,sha256=kI0nUIXfkz-9ENVEeFW70_PGhBNB_S-nntJJkKhGRoQ,8464
+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/endpoints.py,sha256=ggJ8IU0QVrGa7A0RBWiJY1Lo7Klfi9BbfGnovZD97FQ,4592
 craft_ai_sdk/core/environment_variables.py,sha256=wy0U7y0C7tRXOXOiWGz0vX5ajHSmOb_7J1fXlSahGpI,2045
@@ -11,13 +11,13 @@ craft_ai_sdk/core/resource_metrics.py,sha256=yaco_AZ4Wu5sxLRKsA3qt2R3-PMs2e32EGO
 craft_ai_sdk/core/steps.py,sha256=fodbswiMRzHBarBDUCcH1J3O8N8iKu9X66-wZ_Qfnb0,20375
 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=y9lg5unGIwQICAPuKUdUPDAJCsxe2lbseIlmiWka2Ec,11865
-craft_ai_sdk/sdk.py,sha256=qajssLGx7mP6EHpj9JCQ6eRWprW0_09bB7StCSnmPSo,10071
-craft_ai_sdk/utils.py,sha256=w-Y4_xxTHnl9_zLBQVhDg-KB7wqJEPoTArcksf4Fu-c,11573
+craft_ai_sdk/io.py,sha256=Jh9bFfXRzwCequXvJ3DKO-SIdZ7teYfzO64N940FyDY,11861
+craft_ai_sdk/sdk.py,sha256=WN7zGsbAmI3-NMkCErVSvOJaSZEP44wR-GxsekfW5oY,10068
+craft_ai_sdk/utils.py,sha256=syOtpNgV8XF9VoeRVnWGkjIdWv0vyS_Z8RN5QfHCfIo,11891
 craft_ai_sdk/warnings.py,sha256=xJXbUR66SVbG_ZO7rxpAZ2hkGxdUtM5JAx6JoWg77Qs,6678
-documentation.pdf,sha256=45J8Q29ptcM0X52UEfe_JLCNOsWH3SgU-0MhzmGkQko,334272
-craft_ai_sdk-0.54.0rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.54.0rc1.dist-info/METADATA,sha256=DVdVqmvB6kYeFJEp_wnLJ4lgHr59Z6RtnbduUE6272A,1617
-craft_ai_sdk-0.54.0rc1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-craft_ai_sdk-0.54.0rc1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.54.0rc1.dist-info/RECORD,,
+documentation.pdf,sha256=90a4ViFZAGOUzlrKiZjrt7Tx1Um__CKHwa4_m-6zT6s,333705
+craft_ai_sdk-0.54.1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.54.1.dist-info/METADATA,sha256=8TbrLS-64Q7xvj1o3zsFfOs08qPURECb5RbE8XdCPoA,1614
+craft_ai_sdk-0.54.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+craft_ai_sdk-0.54.1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
+craft_ai_sdk-0.54.1.dist-info/RECORD,,