craft-ai-sdk 0.64.0__py3-none-any.whl → 0.64.1__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.64.0"
+__version__ = "0.64.1"

craft_ai_sdk/core/endpoints.py

@@ -8,6 +8,7 @@ from ..sdk import BaseCraftAiSdk
 from ..shared.logger import log_func_result
 from ..shared.request_response_handler import handle_http_response
 from .deployments import get_deployment
+from ..shared.authentication import use_authentication
 
 
 def _get_endpoint_url_path(sdk: BaseCraftAiSdk, endpoint_name: str):
@@ -55,7 +56,7 @@ def trigger_endpoint(
 def trigger_endpoint(
     sdk: BaseCraftAiSdk,
     endpoint_name: str,
-    endpoint_token: str,
+    endpoint_token: Union[str, None] = None,
     inputs: Union[dict[str, Any], None] = None,
     wait_for_completion=True,
 ) -> Union[EndpointTriggerWithOutputs, EndpointTriggerBase]:
@@ -63,7 +64,8 @@ def trigger_endpoint(
 
     Args:
         endpoint_name (:obj:`str`): Name of the endpoint.
-        endpoint_token (:obj:`str`): Token to access endpoint.
+        endpoint_token (:obj:`str`, optional): Token to access endpoint. If not set,
+            the SDK token will be used.
         inputs (:obj:`dict`, optional): Dictionary of inputs to pass to the endpoint
             with input names as keys and corresponding values as values.
             For files, the value should be an instance of io.IOBase.
@@ -85,7 +87,6 @@ def trigger_endpoint(
     """
     if inputs is None:
         inputs = {}
-    endpoint_url_path = _get_endpoint_url_path(sdk, endpoint_name)
 
     body = {}
     files = {}
@@ -95,19 +96,36 @@ def trigger_endpoint(
         else:
             body[input_name] = input_value
 
-    url = f"{sdk.base_environment_url}/endpoints/{endpoint_url_path}"
-    post_result = requests.post(
-        url,
-        headers={
-            "Authorization": f"EndpointToken {endpoint_token}",
-            "craft-ai-client": f"craft-ai-sdk@{sdk._version}",
-        },
-        allow_redirects=False,
-        json=body,
-        files=files,
-    )
-    handle_http_response(post_result)
-    execution_id = post_result.headers.get("Craft-Ai-Execution-Id", "")
+    if endpoint_token is None:
+        url = (
+            f"{sdk.base_environment_api_url}" f"/deployments/{endpoint_name}/executions"
+        )
+        do_post = use_authentication(
+            lambda sdk, *args, **kwargs: sdk._session.post(*args, **kwargs)
+        )
+        post_result = do_post(
+            sdk,
+            url,
+            allow_redirects=False,
+            json=body,
+            files=files,
+        )
+
+    else:
+        endpoint_url_path = _get_endpoint_url_path(sdk, endpoint_name)
+        url = f"{sdk.base_environment_url}/endpoints/{endpoint_url_path}"
+        post_result = requests.post(
+            url,
+            headers={
+                "Authorization": f"EndpointToken {endpoint_token}",
+                "craft-ai-client": f"craft-ai-sdk@{sdk._version}",
+            },
+            allow_redirects=False,
+            json=body,
+            files=files,
+        )
+    response = handle_http_response(post_result)
+    execution_id = response.get("execution_id", "")
     if wait_for_completion and 200 <= post_result.status_code < 400:
         return retrieve_endpoint_results(
             sdk,
@@ -123,7 +141,7 @@ def retrieve_endpoint_results(
     sdk: BaseCraftAiSdk,
     endpoint_name: str,
     execution_id: str,
-    endpoint_token: str,
+    endpoint_token: Union[str, None] = None,
 ) -> EndpointTriggerWithOutputs:
     """Get the results of an endpoint execution.
 
@@ -131,7 +149,8 @@ def retrieve_endpoint_results(
         endpoint_name (:obj:`str`): Name of the endpoint.
         execution_id (:obj:`str`): ID of the execution returned by
             `trigger_endpoint`.
-        endpoint_token (:obj:`str`): Token to access endpoint.
+        endpoint_token (:obj:`str`, optional): Token to access endpoint. If not set,
+            the SDK token will be used.
 
     Returns:
         :obj:`dict`: Created pipeline execution represented as :obj:`dict` with the
@@ -140,6 +159,10 @@ def retrieve_endpoint_results(
         * ``"outputs"`` (:obj:`dict`): Dictionary of outputs of the pipeline with
           output names as keys and corresponding values as values.
     """
+
+    if endpoint_token is None:
+        return sdk.retrieve_pipeline_execution_outputs(execution_id)
+
     endpoint_url_path = _get_endpoint_url_path(sdk, endpoint_name)
 
     url = (
@@ -149,11 +172,13 @@ def retrieve_endpoint_results(
     query = urlencode({"token": endpoint_token})
     response = requests.get(f"{url}?{query}")
 
+    handled_response = handle_http_response(response)
+
     # 500 is returned if the pipeline failed too. In that case, it is not a
     # standard API error
     if response.status_code == 500:
         try:
-            return handle_http_response(response)
+            return handled_response
         except KeyError:
             return response.json()
 
@@ -161,11 +186,15 @@ def retrieve_endpoint_results(
         content_disposition = response.headers.get("Content-Disposition", "")
         output_name = content_disposition.split(f"_{execution_id}_")[1]
         return {
-            "outputs": {output_name: handle_http_response(response)},
+            "outputs": {output_name: handled_response},
             "execution_id": execution_id,
         }
     else:
-        return {**handle_http_response(response), "execution_id": execution_id}
+        response_data = handle_http_response(response)
+        return {
+            "outputs": response_data.get("outputs", []),
+            "execution_id": execution_id,
+        }
 
 
 def generate_new_endpoint_token(

craft_ai_sdk/core/pipeline_executions.py

@@ -106,12 +106,26 @@ Its execution ID is "{execution_id}".',
 
     # Wait for pipeline execution to finish
     if wait_for_completion:
-        return _retrieve_pipeline_execution_outputs(sdk, execution_id)
+        return retrieve_pipeline_execution_outputs(sdk, execution_id)
     return {"execution_id": execution_id}
 
 
 @log_func_result("Pipeline execution results retrieval")
-def _retrieve_pipeline_execution_outputs(sdk: BaseCraftAiSdk, execution_id):
+def retrieve_pipeline_execution_outputs(sdk: BaseCraftAiSdk, execution_id):
+    """Get the results of a pipeline execution.
+
+    Args:
+        execution_id (:obj:`str`): Name of the pipeline execution.
+
+    Returns:
+        :obj:`dict`: Created pipeline execution represented as :obj:`dict` with the
+        following keys:
+
+        * ``execution_id`` (str): Name of the pipeline execution.
+        * ``"outputs"`` (:obj:`dict`): Dictionary of outputs of the pipeline with
+          output names as keys and corresponding values as values.
+    """
+
     url = (
         f"{sdk.base_environment_api_url}"
         f"/executions/{execution_id}/outputs?wait_for_completion=true"

craft_ai_sdk/sdk.py

@@ -93,6 +93,7 @@ class CraftAiSdk(BaseCraftAiSdk):
         get_pipeline_execution_output,
         list_pipeline_executions,
         run_pipeline,
+        retrieve_pipeline_execution_outputs,
     )
     from .core.pipeline_metrics import (
         get_list_metrics,
@@ -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.64.0"  # Would be better to share it somewhere
+    _version = "0.64.1"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.64.0.dist-info → craft_ai_sdk-0.64.1.dist-info}/METADATA

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

{craft_ai_sdk-0.64.0.dist-info → craft_ai_sdk-0.64.1.dist-info}/RECORD

@@ -1,10 +1,10 @@
-craft_ai_sdk/__init__.py,sha256=yN6aX8menUCFkdEBd3ZeT_OnhFUKP8ntg2ZlXDojrME,363
+craft_ai_sdk/__init__.py,sha256=1B7fdloJQsWnPA90TMoCRarnHriveRHWvgHMlU8CCfw,363
 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=-XMPQJ9ZMmd3i7gEAQB7kONfScg-8JqVAD9VI1fQNEQ,37174
-craft_ai_sdk/core/endpoints.py,sha256=pLIxSYzp3omz7D3Lq6lz9ZbCnHxAfxVnWV4PBlBkAGI,6107
+craft_ai_sdk/core/endpoints.py,sha256=cB4MgEr5L0XVS_QtE1ewsFqq586NZoxqeI-ZINSS9R8,7034
 craft_ai_sdk/core/environment_variables.py,sha256=FYxwWqz8XdmxWuJLI3MP9QOObWwly3P4k6lon-KHUmU,2224
-craft_ai_sdk/core/pipeline_executions.py,sha256=w4dKFlFt9XgQC77H0S4AoXnURRCo9ZRxkCM4PiZ4VYU,18153
+craft_ai_sdk/core/pipeline_executions.py,sha256=uasCxQewvD4pYXRSPhYQnrApFVzL5covsN1tbJkxtT8,18624
 craft_ai_sdk/core/pipeline_metrics.py,sha256=cyDuhQCGVRMruo-ZGGd9vfQB0rbNu8Fs1sQxnfk6Aow,7174
 craft_ai_sdk/core/pipelines.py,sha256=QrpCH9Yn6SA92xyDfF0Te-iumtkhlNRLtQAiztqcl6o,21574
 craft_ai_sdk/core/resource_metrics.py,sha256=jwxDZJIhj28rpmFJKopt_1O0X58TGjUThRpAeMvVz1Y,3455
@@ -13,7 +13,7 @@ craft_ai_sdk/core/users.py,sha256=OBwOrjE2hrqQde28ZItGxNDj52dhKY-mxabkbLaMnGQ,61
 craft_ai_sdk/core/vector_database.py,sha256=JQsz7Rj8HL9tmB8y4lSsAd3IPnuVMX2LpAK1z8yrz9M,2323
 craft_ai_sdk/exceptions.py,sha256=IC-JfZmmmaTsbMCgirOEByRmWnatQLjKe8BErRkuwM0,1075
 craft_ai_sdk/io.py,sha256=4BlUV2Rf3Y8YXMC2mRdUqbI8EzvbgkrLB-cbcrv2gco,13468
-craft_ai_sdk/sdk.py,sha256=G_cdOzzEKOF9rWrkqHbUef6Bd-E0nepWQQ5w0rGwxDo,10736
+craft_ai_sdk/sdk.py,sha256=3E--Ta-02FKIE1S5dAq0h_6SFA7wENn5vlb72BZplfY,10781
 craft_ai_sdk/shared/authentication.py,sha256=OdwtAH47tOUS-u_HhxlI8JdjZT5REr5B5cGSrX7sz00,885
 craft_ai_sdk/shared/environments.py,sha256=NI7DgQa4uj1PnffJUZ1Q4Car4r9OoFdXpYod26xgOIo,547
 craft_ai_sdk/shared/execution_context.py,sha256=B2Ghq-wiUvq81q5mhsm79Oc59c8c00uQxMIpApFD03o,585
@@ -26,8 +26,8 @@ craft_ai_sdk/utils/__init__.py,sha256=A0sLCXSPD1Z3q2GP1uLDjvif4ivOr__Hzg9RQysEuq
 craft_ai_sdk/utils/datetime_utils.py,sha256=ziWQ2Xa_ypKIm50F_1R5OtubXSNlM4JWbFUeh7qjwpE,672
 craft_ai_sdk/utils/dict_utils.py,sha256=1HQ3A14SN48XPFDmQleujGAgksmkjIs3hyPLpwhwh24,748
 craft_ai_sdk/utils/file_utils.py,sha256=dKn8NzrCMc5nn0ptUL6VuUOvLo474NJFWrVBe7ccYI8,2256
-craft_ai_sdk-0.64.0.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.64.0.dist-info/METADATA,sha256=QwetLvIeXSBGQSb5RTmzKn7I8xx6ST5H_-NE7aH5Pcw,1676
-craft_ai_sdk-0.64.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-craft_ai_sdk-0.64.0.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.64.0.dist-info/RECORD,,
+craft_ai_sdk-0.64.1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.64.1.dist-info/METADATA,sha256=A2IgdNLdxMtj3u5u9BssLEXTYfQ9JWapuu9o7VM_nOo,1676
+craft_ai_sdk-0.64.1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+craft_ai_sdk-0.64.1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
+craft_ai_sdk-0.64.1.dist-info/RECORD,,