craft-ai-sdk 0.65.1rc1__tar.gz → 0.65.2rc1__tar.gz

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/PKG-INFO

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

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/__init__.py

@@ -13,4 +13,4 @@ from .io import (  # noqa: F401
 )
 from .sdk import CraftAiSdk  # noqa: F401
 
-__version__ = "0.65.1rc1"
+__version__ = "0.65.2rc1"

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/deployments.py

@@ -1,8 +1,10 @@
 from datetime import datetime
 from typing import Literal, TypedDict, Union
+
 from typing_extensions import NotRequired
 
 from craft_ai_sdk.shared.types import Log
+
 from ..constants import DEPLOYMENT_EXECUTION_RULES, DEPLOYMENT_MODES, DEPLOYMENT_STATUS
 from ..exceptions import SdkException
 from ..io import (

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/endpoints.py

@@ -5,10 +5,10 @@ from urllib.parse import urlencode
 import requests
 
 from ..sdk import BaseCraftAiSdk
+from ..shared.authentication import use_authentication
 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):
@@ -36,9 +36,9 @@ class EndpointNewToken(TypedDict):
 def trigger_endpoint(
     sdk: BaseCraftAiSdk,
     endpoint_name: str,
-    endpoint_token: str,
-    inputs: dict[str, Any],
-    wait_for_completion: Literal[True],
+    endpoint_token: Union[str, None] = None,
+    inputs: Union[dict[str, Any], None] = None,
+    wait_for_completion: Literal[True] = True,
 ) -> EndpointTriggerWithOutputs: ...
 
 
@@ -46,9 +46,9 @@ def trigger_endpoint(
 def trigger_endpoint(
     sdk: BaseCraftAiSdk,
     endpoint_name: str,
-    endpoint_token: str,
-    inputs: dict[str, Any],
-    wait_for_completion: Literal[False],
+    endpoint_token: Union[str, None] = None,
+    inputs: Union[dict[str, Any], None] = None,
+    wait_for_completion: Literal[False] = False,
 ) -> EndpointTriggerBase: ...
 
 
@@ -97,9 +97,7 @@ def trigger_endpoint(
             body[input_name] = input_value
 
     if endpoint_token is None:
-        url = (
-            f"{sdk.base_environment_api_url}" f"/deployments/{endpoint_name}/executions"
-        )
+        url = f"{sdk.base_environment_api_url}/deployments/{endpoint_name}/executions"
         do_post = use_authentication(
             lambda sdk, *args, **kwargs: sdk._session.post(*args, **kwargs)
         )
@@ -167,7 +165,7 @@ def retrieve_endpoint_results(
 
     url = (
         f"{sdk.base_environment_url}"
-        f"/endpoints/{endpoint_url_path}/executions/{execution_id}"
+        f"/endpoints/v1/{endpoint_url_path}/executions/{execution_id}"
     )
     query = urlencode({"token": endpoint_token})
     response = requests.get(f"{url}?{query}")
@@ -182,19 +180,27 @@ def retrieve_endpoint_results(
         except KeyError:
             return response.json()
 
-    if "application/octet-stream" in response.headers.get("Content-Type", ""):
-        content_disposition = response.headers.get("Content-Disposition", "")
-        output_name = content_disposition.split(f"_{execution_id}_")[1]
-        return {
-            "outputs": {output_name: handled_response},
-            "execution_id": execution_id,
-        }
-    else:
-        response_data = handle_http_response(response)
-        return {
-            "outputs": response_data.get("outputs", []),
-            "execution_id": execution_id,
-        }
+    response_data = handle_http_response(response)
+    outputs = response_data.get("outputs", {})
+
+    for key, value in outputs.items():
+        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
+            else:
+                raise ValueError(
+                    f"Failed to download file output {key}: {file_response.text}"
+                )
+
+    return {
+        "outputs": response_data.get("outputs", []),
+        "execution_id": execution_id,
+    }
 
 
 def generate_new_endpoint_token(
@@ -211,8 +217,5 @@ def generate_new_endpoint_token(
 
         * ``"endpoint_token"`` (:obj:`str`): New endpoint token.
     """
-    url = (
-        f"{sdk.base_environment_api_url}"
-        f"/endpoints/{endpoint_name}/generate-new-token"
-    )
+    url = f"{sdk.base_environment_api_url}/endpoints/{endpoint_name}/generate-new-token"
     return sdk._post(url)

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/environment_variables.py

@@ -1,4 +1,5 @@
 from typing import TypedDict
+
 from ..sdk import BaseCraftAiSdk
 from ..shared.logger import log_func_result
 
@@ -24,7 +25,7 @@ def create_or_update_environment_variable(
     Returns:
         None
     """
-    url = f"{sdk.base_environment_api_url}" f"/environment-variables"
+    url = f"{sdk.base_environment_api_url}/environment-variables"
     data = {
         "environment_variables": {
             environment_variable_name: environment_variable_value

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/pipeline_executions.py

@@ -177,8 +177,7 @@ def _retrieve_pipeline_execution_input_value(
     sdk: BaseCraftAiSdk, execution_id, input_name
 ):
     url = (
-        f"{sdk.base_environment_api_url}"
-        f"/executions/{execution_id}/inputs/{input_name}"
+        f"{sdk.base_environment_api_url}/executions/{execution_id}/inputs/{input_name}"
     )
     response = sdk._get(url)
     return response

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/pipeline_metrics.py

@@ -1,5 +1,5 @@
-from typing import TypedDict, Union
 import warnings
+from typing import TypedDict, Union
 
 from ..sdk import BaseCraftAiSdk
 from ..shared.execution_context import get_execution_id
@@ -46,7 +46,7 @@ been sent",
                 stacklevel=2,
             )
         return False
-    url = f"{sdk.base_environment_api_url}" f"/metrics/single-value/{name}"
+    url = f"{sdk.base_environment_api_url}/metrics/single-value/{name}"
     data = {"value": value, "execution_id": get_execution_id()}
     sdk._put(url, json=data)
     return True
@@ -81,7 +81,7 @@ been sent",
 
     BATCH_SIZE = 10000
     for i in range(0, len(values), BATCH_SIZE):
-        url = f"{sdk.base_environment_api_url}" f"/metrics/list-values/{name}"
+        url = f"{sdk.base_environment_api_url}/metrics/list-values/{name}"
         data = {
             "values": values[i : i + BATCH_SIZE],
             "execution_id": get_execution_id(),

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/resource_metrics.py

@@ -1,6 +1,8 @@
 from datetime import datetime
 from typing import Literal, TypedDict, Union, overload
+
 from typing_extensions import NotRequired
+
 from ..sdk import BaseCraftAiSdk
 from ..utils import datetime_to_timestamp_in_ms
 

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/steps.py

@@ -1,11 +1,11 @@
-from datetime import datetime
 import io
 import os
 import tarfile
-from typing import TypeVar, TypedDict, Union, cast
-from typing_extensions import NotRequired
+from datetime import datetime
+from typing import TypedDict, TypeVar, Union, cast
 
 import requests
+from typing_extensions import NotRequired
 
 from craft_ai_sdk.shared.types import Log
 
@@ -97,11 +97,11 @@ def _validate_create_step_parameters(
         raise ValueError("The timeout must be greater than 0 or None.")
 
     if inputs is not None:
-        if any([not isinstance(input_, Input) for input_ in inputs]):
+        if any(not isinstance(input_, Input) for input_ in inputs):
             raise ValueError("'inputs' must be a list of instances of Input.")
 
     if outputs is not None:
-        if any([not isinstance(output_, Output) for output_ in outputs]):
+        if any(not isinstance(output_, Output) for output_ in outputs):
             raise ValueError("'outputs' must be a list of instances of Output.")
 
 
@@ -191,12 +191,12 @@ def _add_inputs_outputs_in_message(message, inputs, outputs):
         message += "\n  Inputs: "
         for inp in inputs:
             required_str = ", required" if inp.get("is_required", False) else ""
-            message += f'\n    - {inp.get("name", inp.get("input_name"))} ({inp["data_type"]}{required_str})'  # noqa: E501
+            message += f"\n    - {inp.get('name', inp.get('input_name'))} ({inp['data_type']}{required_str})"  # noqa: E501
 
     if outputs:
         message += "\n  Outputs: "
         for output in outputs:
-            message += f'\n    - {output.get("name", output.get("output_name"))} ({output["data_type"]})'  # noqa: E501
+            message += f"\n    - {output.get('name', output.get('output_name'))} ({output['data_type']})"  # noqa: E501
     return message
 
 

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/users.py

@@ -1,4 +1,5 @@
 from typing import TypedDict
+
 from ..sdk import BaseCraftAiSdk
 
 

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/vector_database.py

@@ -1,4 +1,5 @@
 from typing import TypedDict
+
 from craft_ai_sdk.shared.environments import get_environment_id
 
 from ..sdk import BaseCraftAiSdk
@@ -41,7 +42,7 @@ def get_weaviate_client(sdk: BaseCraftAiSdk):
         raise ModuleNotFoundError(
             "The 'weaviate' package is required to use the vector database. "
             "You can install it with 'pip install weaviate-client'."
-        )
+        ) from None
     credentials = get_vector_database_credentials(sdk)
 
     is_secure = credentials["vector_database_url"].startswith("https://")

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/sdk.py

@@ -1,10 +1,10 @@
 import os
 import sys
 import time
-from typing import Any
 import warnings
 from abc import ABC, abstractmethod
 from datetime import timedelta
+from typing import Any
 
 import jwt
 import requests
@@ -92,8 +92,8 @@ class CraftAiSdk(BaseCraftAiSdk):
         get_pipeline_execution_logs,
         get_pipeline_execution_output,
         list_pipeline_executions,
-        run_pipeline,
         retrieve_pipeline_execution_outputs,
+        run_pipeline,
     )
     from .core.pipeline_metrics import (
         get_list_metrics,
@@ -119,7 +119,6 @@ class CraftAiSdk(BaseCraftAiSdk):
         list_steps,
     )
     from .core.users import get_user
-
     from .core.vector_database import (
         get_vector_database_credentials,
         get_weaviate_client,
@@ -139,7 +138,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.65.1rc1"  # Would be better to share it somewhere
+    _version = "0.65.2rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,
@@ -216,7 +215,9 @@ class CraftAiSdk(BaseCraftAiSdk):
             verbose_log = (
                 True
                 if env_verbose_log == "true"
-                else False if env_verbose_log == "false" else hasattr(sys, "ps1")
+                else False
+                if env_verbose_log == "false"
+                else hasattr(sys, "ps1")
             )
         self.verbose_log = verbose_log
 

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/shared/logger.py

@@ -7,7 +7,7 @@ from craft_ai_sdk.exceptions import SdkException
 
 def log_action(sdk, message: str, should_log: Union[bool, Callable[[], bool]] = True):
     if sdk.verbose_log and (should_log() if callable(should_log) else should_log):
-        print(message, file=sys.stderr)
+        print(message, file=sys.stderr)  # noqa: T201
 
 
 def log_func_result(message: str, should_log: Union[bool, Callable[[], bool]] = True):

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/shared/request_response_handler.py

@@ -1,6 +1,6 @@
-from typing import Any, Callable
 import xml.etree.ElementTree as ET
 from json import JSONDecodeError
+from typing import Any, Callable
 
 from requests import RequestException, Response
 
@@ -20,6 +20,7 @@ def handle_data_store_response(response: Response):
 
     Returns:
         :obj:`str`: Content of the response.
+
     """
     if 200 <= response.status_code < 300:
         return response.content
@@ -36,12 +37,12 @@ def handle_data_store_response(response: Response):
             name=error_code,
             additional_data=error_infos,
         )
-    except ET.ParseError:
+    except ET.ParseError as error:
         raise SdkException(
             "Unable to decode response from the data store: "
             f"Content being:\n'{response.text}'",
             status_code=response.status_code,
-        )
+        ) from error
 
 
 def _parse_json_response(response: Response):

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/shared/warnings.py

@@ -13,8 +13,7 @@ MULTIPLE_EXPERIMENTAL_WARNING_JOINER = "\n            * "
 def _documentation_function_warning_decorator(
     func_or_message, default_message, warning_category
 ):
-    """
-    A general-purpose decorator for issuing function warnings. This function is
+    """A general-purpose decorator for issuing function warnings. This function is
         designed to be used within other decorators to add warning functionality.
 
     Args:
@@ -26,6 +25,7 @@ def _documentation_function_warning_decorator(
         warning_category (class): The category of warning to be used.
             This should be a class derived from the Warning class,
             such as DeprecationWarning, FutureWarning, etc.
+
     """
 
     def _get_wrapper(func, warning_message):
@@ -161,7 +161,6 @@ def deprecated(func_or_message):
             (i.e. `@deprecated`), a default deprecation warning message will be used.
 
     """
-
     DEFAULT_DEPRECATION_WARNING_MESSAGE = (
         "This function is deprecated and its usage is not supported."
     )

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/utils/datetime_utils.py

@@ -9,7 +9,7 @@ def datetime_to_timestamp_in_ms(dt: datetime) -> int:
 
 
 def parse_isodate(date_string: str):
-    """_summary_
+    """Parse a date string in ISO 8601 format and return a `datetime` object.
 
     Args:
         date_string (str): date in ISO 8601 format potentially ending with
@@ -17,6 +17,7 @@ def parse_isodate(date_string: str):
 
     Returns:
         :obj:`datetime.datetime`: A `datetime` corresponding to `date_string`.
+
     """
     if date_string[-1] == "Z":
         date_string = date_string.rstrip("Z")

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/utils/file_utils.py

@@ -39,9 +39,7 @@ def chunk_buffer(buffer: IOBase, size: int) -> Iterable[ChunkedIO]:
 
 
 def convert_size(size_in_bytes: Union[int, float]):
-    """
-    Convert a size in bytes to a human readable string.
-    """
+    """Convert a size in bytes to a human readable string."""
     units = ["B", "KB", "MB", "GB", "TB"]
     for unit in units:  # noqa: B007
         if size_in_bytes < 1024.0:
@@ -53,9 +51,9 @@ def convert_size(size_in_bytes: Union[int, float]):
 # Adapted from
 # https://gist.github.com/kazqvaizer/4cebebe5db654a414132809f9f88067b#file-multipartify-py-L13-L33
 def multipartify(data, parent_key: Union[str, None] = None) -> dict:
-    """
-    Convert a nested dictionary or list into a format suitable for multipart/form-data.
-    This is useful for triggering endpoints that require multipart/form-data payloads.
+    """Convert a nested dictionary or list into a format suitable for
+    multipart/form-data. This is useful for triggering endpoints that require
+    multipart/form-data payloads.
 
     Args:
         data (:obj:`any`):
@@ -65,6 +63,7 @@ def multipartify(data, parent_key: Union[str, None] = None) -> dict:
 
     Returns:
         :obj:`dict`: A dictionary where keys are formatted for multipart/form-data.
+
     """
 
     def formatter(v):

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "craft-ai-sdk"
-version = "0.65.1rc1"
+version = "0.65.2rc1"
 description = "Craft AI MLOps platform SDK"
 license = "Apache-2.0"
 authors = ["Craft AI <contact@craft.ai>"]
@@ -11,6 +11,7 @@ include = ["documentation.pdf"]
 
 [tool.poetry.scripts]
 lint = "scripts.scripts:lint"
+lintfix = "scripts.scripts:lintfix"
 format = "scripts.scripts:format"
 reformat = "scripts.scripts:reformat"
 test = "scripts.scripts:test"
@@ -34,22 +35,29 @@ weaviate-client = "^4.10.4"
 [tool.poetry.group.dev.dependencies]
 ipython = "^8.18.1"
 ipykernel = "^6.29.5"
-black = "^25.1.0"
-flake8 = "^7.1.2"
-flake8-bugbear = "^24.12.12"
-pylint = "^3.3.4"
+pylint = "^3.3.7"
 python-dotenv = "^1.0.1"
 pytest = "^8.3.4"
 requests-mock = { extras = ["fixture"], version = "^1.9.3" }
 pytest-watch = "^4.2.0"
 pyarrow = "^19.0.1"
 pandas = "^2.2.3"
+ruff = "^0.12.2"
 
 [tool.poetry.group.docs.dependencies]
 doc8 = "^0.10.1"
 Sphinx = "^5.0.1"
 sphinxcontrib-restbuilder = "^0.3"
 
+[tool.ruff.lint]
+extend-select = ["B", "C", "E", "F", "I", "T", "W"]
+extend-ignore = ["C401", "C901"]
+
+[tool.ruff.lint.per-file-ignores]
+"**/docs/**/*.py" = ["T"]
+"**/tests/**/*.py" = ["C", "T"]
+"**/scripts/**/*.py" = ["T"]
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/core/pipelines.py

@@ -1,9 +1,9 @@
 import os
-from typing import TypedDict, Union, cast
-from typing_extensions import NotRequired
 import warnings
+from typing import TypedDict, Union, cast
 
 import requests
+from typing_extensions import NotRequired
 
 from craft_ai_sdk.io import Input, Output
 from craft_ai_sdk.shared.types import Log

{craft_ai_sdk-0.65.1rc1 → craft_ai_sdk-0.65.2rc1}/craft_ai_sdk/io.py

@@ -1,8 +1,8 @@
-from typing import Any, TypedDict, cast
-from typing_extensions import NotRequired
 import warnings
+from typing import Any, TypedDict, cast
 
 from strenum import LowercaseStrEnum
+from typing_extensions import NotRequired
 
 from craft_ai_sdk.utils import remove_none_values