craft-ai-sdk 0.66.0rc1__tar.gz → 0.66.1rc1__tar.gz

{craft_ai_sdk-0.66.0rc1 → craft_ai_sdk-0.66.1rc1}/PKG-INFO

@@ -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 → craft_ai_sdk-0.66.1rc1}/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-0.66.0rc1 → craft_ai_sdk-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-0.66.0rc1 → craft_ai_sdk-0.66.1rc1}/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-0.66.0rc1 → craft_ai_sdk-0.66.1rc1}/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 → craft_ai_sdk-0.66.1rc1}/pyproject.toml

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