craft-ai-sdk 0.66.0rc1__py3-none-any.whl → 0.66.1rc1__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.66.0rc1"
+__version__ = "0.66.1rc1"

craft_ai_sdk/core/endpoints.py

@@ -1,4 +1,5 @@
 import io
+import json
 from typing import Any, Literal, TypedDict, Union, overload
 from urllib.parse import urlencode
 
@@ -90,13 +91,20 @@ def trigger_endpoint(
     if inputs is None:
         inputs = {}
 
-    body = {}
-    files = {}
+    parameters = {}
+    artifacts = {}
     for input_name, input_value in inputs.items():
         if isinstance(input_value, io.IOBase) and input_value.readable():
-            files[input_name] = input_value
+            artifacts[input_name] = input_value
         else:
-            body[input_name] = input_value
+            parameters[input_name] = input_value
+
+    data = None
+    # When using both files and parameters, parameters should
+    # be passed through `data` instead
+    if len(parameters) > 0 and len(artifacts) > 0:
+        data = {key: json.dumps(value) for key, value in parameters.items()}
+        parameters = None
 
     if endpoint_token is None:
         url = f"{sdk.base_environment_api_url}/deployments/{endpoint_name}/executions"
@@ -107,8 +115,9 @@ def trigger_endpoint(
             sdk,
             url,
             allow_redirects=False,
-            json=body,
-            files=files,
+            json=parameters,
+            files=artifacts,
+            data=data,
         )
 
     else:
@@ -121,8 +130,9 @@ def trigger_endpoint(
                 "craft-ai-client": f"craft-ai-sdk@{sdk._version}",
             },
             allow_redirects=False,
-            json=body,
-            files=files,
+            json=parameters,
+            files=artifacts,
+            data=data,
         )
     response = handle_http_response(post_result)
     execution_id = response.get("execution_id", "")

craft_ai_sdk/sdk.py

@@ -23,6 +23,7 @@ class BaseCraftAiSdk(ABC):
     _MULTIPART_THRESHOLD: int
     _MULTIPART_PART_SIZE: int
     _version: str
+    _session: requests.Session
     warn_on_metric_outside_of_step: bool
 
     @abstractmethod
@@ -138,7 +139,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.66.0rc1"  # Would be better to share it somewhere
+    _version = "0.66.1rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,

craft_ai_sdk/shared/authentication.py

@@ -3,9 +3,19 @@ from datetime import datetime
 from typing import Callable
 
 from requests import Response
+from typing_extensions import Concatenate, ParamSpec
 
+import craft_ai_sdk.sdk as sdk
 
-def use_authentication(action_func: Callable[..., Response]) -> Callable[..., Response]:
+Args = ParamSpec("Args")
+
+
+def use_authentication(
+    action_func: Callable[
+        Concatenate["sdk.BaseCraftAiSdk", Args],
+        Response,
+    ],
+) -> Callable[..., Response]:
     @functools.wraps(action_func)
     def wrapper(sdk, *args, headers=None, **kwargs):
         actual_headers = None

{craft_ai_sdk-0.66.0rc1.dist-info → craft_ai_sdk-0.66.1rc1.dist-info}/METADATA

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

{craft_ai_sdk-0.66.0rc1.dist-info → craft_ai_sdk-0.66.1rc1.dist-info}/RECORD

@@ -1,8 +1,8 @@
-craft_ai_sdk/__init__.py,sha256=lzXeiTGkDxqC05y57VzYQsSYXxMTyBcTegOpafelEGE,366
+craft_ai_sdk/__init__.py,sha256=3gNWDWcydXNO3rAKuyXKYD5-uvKD2f800or0kSVKeQA,366
 craft_ai_sdk/constants.py,sha256=rH4JrGlTpbjjjNRrKhk5oScbj5G5INrcVza6Bb6kIzY,980
 craft_ai_sdk/core/data_store.py,sha256=dlVZajXGwcI_4mzqFctHpzKO-uySIP5lvQaJy6CKwY0,9131
 craft_ai_sdk/core/deployments.py,sha256=O4JjH8CjfA4Dy-Y7H_ABVEPS3TbMhO9j66F2mKf22K4,37617
-craft_ai_sdk/core/endpoints.py,sha256=smNLSDG4T6ZMNyu494Ip52SURBOqmmx3oSWtz8GzCCc,7648
+craft_ai_sdk/core/endpoints.py,sha256=ad8_oHVCwrTpnlrlqvEcO48nXbzmuWTaUbh82JdRX6g,8018
 craft_ai_sdk/core/environment_variables.py,sha256=CpkyneLf8UF-1QXe36Ay_RZJ8grSfbZZWBvGSQzKlLs,2221
 craft_ai_sdk/core/pipeline_executions.py,sha256=p_vjsOMR92_6UUG5S-pgyen3V7yYHTjHBJ7pNfANe5I,18661
 craft_ai_sdk/core/pipeline_metrics.py,sha256=IJFzAptwqjanb2Jl-XQCh0FtashEd4I2Pb51ky51cyY,7166
@@ -13,8 +13,8 @@ craft_ai_sdk/core/users.py,sha256=q5et87q0SOMpRTBOiLX026oaEOTaY1aGqKdbUC51zbA,61
 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=KxInD0Bvj6wKb0WeKa9yLXXMOsW0nLrlV6H4CTI40e8,10815
-craft_ai_sdk/shared/authentication.py,sha256=OdwtAH47tOUS-u_HhxlI8JdjZT5REr5B5cGSrX7sz00,885
+craft_ai_sdk/sdk.py,sha256=jD6OREimT6yyQ3ykeaoYZ9Q6eGBLrtTo4OMpAGgjCb8,10846
+craft_ai_sdk/shared/authentication.py,sha256=5lJS_aNphkgI0-wvHFK4aW316ZRfOq9oywNpv-Oyn2M,1062
 craft_ai_sdk/shared/environments.py,sha256=LbpRK-ACpwFfN7WTuo0nrtbbi2NAXhuMRKf7YkktEYI,510
 craft_ai_sdk/shared/execution_context.py,sha256=B2Ghq-wiUvq81q5mhsm79Oc59c8c00uQxMIpApFD03o,585
 craft_ai_sdk/shared/helpers.py,sha256=5ZBiHlS-k1OgKejZgMlDJiYwIPYk7yg-lzeJI67poCc,1070
@@ -26,8 +26,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.66.0rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.66.0rc1.dist-info/METADATA,sha256=Z_g7Xyu1sJriOF8BC2yaVkgIceK7t1makhmWJwNKRUM,1679
-craft_ai_sdk-0.66.0rc1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-craft_ai_sdk-0.66.0rc1.dist-info/entry_points.txt,sha256=QC96WcXvvUfLMRgFD-l_y7_TgC9SqZybLs9EQ8dsGiQ,417
-craft_ai_sdk-0.66.0rc1.dist-info/RECORD,,
+craft_ai_sdk-0.66.1rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.66.1rc1.dist-info/METADATA,sha256=b7TO-yVdsfSGvFCUonaBQPEwT4qVJBbgtKQgaZEsTpo,1679
+craft_ai_sdk-0.66.1rc1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+craft_ai_sdk-0.66.1rc1.dist-info/entry_points.txt,sha256=QC96WcXvvUfLMRgFD-l_y7_TgC9SqZybLs9EQ8dsGiQ,417
+craft_ai_sdk-0.66.1rc1.dist-info/RECORD,,