craft-ai-sdk 0.55.1rc1__tar.gz → 0.55.2__tar.gz

{craft_ai_sdk-0.55.1rc1 → craft_ai_sdk-0.55.2}/PKG-INFO

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

{craft_ai_sdk-0.55.1rc1 → craft_ai_sdk-0.55.2}/craft_ai_sdk/__init__.py

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

{craft_ai_sdk-0.55.1rc1 → craft_ai_sdk-0.55.2}/craft_ai_sdk/core/pipeline_executions.py

@@ -65,8 +65,8 @@ def run_pipeline(
         inputs = {}
     # Retrieve pipeline input types
     pipeline = sdk._get(f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}")
-    pipeline_inputs = pipeline["parameters"]["inputs"]
-    input_types = {input["name"]: input["data_type"] for input in pipeline_inputs}
+    pipeline_inputs = pipeline["inputs"]
+    input_types = {input["input_name"]: input["data_type"] for input in pipeline_inputs}
 
     # Get files to upload and data to send
     files = {}

{craft_ai_sdk-0.55.1rc1 → craft_ai_sdk-0.55.2}/craft_ai_sdk/core/pipelines.py

@@ -124,58 +124,59 @@ def create_pipeline(
         :obj:`dict`: Created pipeline represented as :obj:`dict` with the following
         keys:
 
-        * ``"parameters"`` (:obj:`dict`): Information used to create the pipeline with
-          the following keys:
-
-          * ``"pipeline_name"`` (:obj:`str`): Name of the pipeline.
-          * ``"function_path"`` (:obj:`str`): Path to the file that contains the
-            function.
-          * ``"function_name"`` (:obj:`str`): Name of the function in that file.
-          * ``"description"`` (:obj:`str`): Description.
-          * ``"inputs"`` (:obj:`list` of :obj:`dict`): List of inputs represented
-            as a :obj:`dict` with the following keys:
-
-            * ``"name"`` (:obj:`str`): Input name.
-            * ``"data_type"`` (:obj:`str`): Input data type.
-            * ``"is_required"`` (:obj:`bool`): Whether the input is required.
-            * ``"default_value"`` (:obj:`str`): Input default value.
-
-          * ``"outputs"`` (:obj:`list` of :obj:`dict`): List of outputs
-            represented as a :obj:`dict` with the following keys:
-
-            * ``"name"`` (:obj:`str`): Output name.
-            * ``"data_type"`` (:obj:`str`): Output data type.
-            * ``"description"`` (:obj:`str`): Output description.
-
-          * ``"container_config"`` (:obj:`dict[str, str]`): Some pipeline
-            configuration, with the following optional keys:
-
-            * ``"language"`` (:obj:`str`): Language and version used for the pipeline.
-              The accepted formats are "python:3.X-slim", where "3.X" is a supported
-              version of Python, and "python-cuda:3.X-Y.Z" for GPU environments,
-              where "Y.Z" is a supported version of CUDA. The list of supported
-              versions is available on the official documentation website at
-              https://mlops-platform-documentation.craft.ai.
-            * ``"repository_url"`` (:obj:`str`): Remote repository url.
-            * ``"repository_branch"`` (:obj:`str`): Branch name.
-            * ``"included_folders"`` (:obj:`list[str]`): List of folders and
-              files in the repository required for the pipeline execution.
-            * ``"system_dependencies"`` (:obj:`list[str]`): List of system
-              dependencies.
-            * ``"dockerfile_path"`` (:obj:`str`): Path to the Dockerfile.
-            * ``"requirements_path"`` (:obj:`str`): Path to the requirements.txt
-              file.
-
-        * ``"creation_info"`` (:obj:`dict`): Information about the pipeline creation:
+        * ``"pipeline_name"`` (:obj:`str`): Name of the pipeline.
+        * ``"created_at"`` (:obj:`str`): Pipeline date of creation.
+        * ``"inputs"`` (:obj:`list` of :obj:`dict`): List of open inputs
+          of the pipeline. Each open input is represented as a :obj:`dict` with the
+          following keys:
+
+          * ``"input_name"`` (:obj:`str`): Name of the open input.
+          * ``"step_name"`` (:obj:`str`): Name of the step that provides the open
+            input.
+          * ``"data_type"`` (:obj:`str`): Data type of the open input.
+          * ``"description"`` (:obj:`str`): Description of the open input.
+          * ``"default_value"`` (:obj:`str`): Default value of the open input.
+          * ``"is_required"`` (:obj:`bool`): Whether the open input is required or
+            not.
+
+        * ``"outputs"`` (:obj:`list` of :obj:`dict`): List of open outputs
+          of the pipeline. Each open output is represented as a :obj:`dict` with the
+          following keys:
+
+          * ``"output_name"`` (:obj:`str`): Name of the open output.
+          * ``"step_name"`` (:obj:`str`): Name of the step that provides the open
+            output.
+          * ``"data_type"`` (:obj:`str`): Data type of the open output.
+          * ``"description"`` (:obj:`str`): Description of the open output.
+
+        * ``"container_config"`` (:obj:`dict[str, str]`): Some pipeline
+          configuration, with the following optional keys:
+
+          * ``"language"`` (:obj:`str`): Language and version used for the pipeline.
+            The accepted formats are "python:3.X-slim", where "3.X" is a supported
+            version of Python, and "python-cuda:3.X-Y.Z" for GPU environments,
+            where "Y.Z" is a supported version of CUDA. The list of supported
+            versions is available on the official documentation website at
+            https://mlops-platform-documentation.craft.ai.
+          * ``"repository_branch"`` (:obj:`str`): Branch name.
+          * ``"repository_url"`` (:obj:`str`): Remote repository url.
+          * ``"included_folders"`` (:obj:`list[str]`): List of folders and
+            files in the repository required for the pipeline execution.
+          * ``"system_dependencies"`` (:obj:`list[str]`): List of system
+            dependencies.
+          * ``"dockerfile_path"`` (:obj:`str`): Path to the Dockerfile.
+          * ``"requirements_path"`` (:obj:`str`): Path to the requirements.txt
+            file.
+
+        * ``"creation_info"`` (:obj:`dict`): Information about the step creation (Note
+          that this is only returned if the pipeline was created without a step):
 
-          * ``"created_at"`` (:obj:`str`): The creation date in ISO format.
           * ``"commit_id"`` (:obj:`str`): The commit id on which the pipeline was
             built.
-          * ``"status"`` (:obj:`str`): either ``"creation_pending"`` or ``"ready"``.
-          * ``"origin"`` (:obj:`str`): The origin of the pipeline, can be
+          * ``"status"`` (:obj:`str`): The pipeline status, always ``"ready"`` when
+            this function returns.
+          * ``"origin"`` (:obj:`str`): The origin of the step, can be
             ``"git_repository"`` or ``"local"``.
-          * ``"last_execution_id"`` (:obj:`str`): The id of the last execution of
-            the pipeline.
     """
     # For backward compatibility, a pipeline can be created with a single step:
     args_are_none = all(
@@ -258,59 +259,61 @@ def get_pipeline(
         None if the pipeline does not exist, otherwise pipeline information, with
         the following keys:
 
-        * ``"parameters"`` (:obj:`dict`): Information used to create the pipeline with
-          the following keys:
-
-          * ``"pipeline_name"`` (:obj:`str`): Name of the pipeline.
-          * ``"function_path"`` (:obj:`str`): Path to the file that contains the
-            function.
-          * ``"function_name"`` (:obj:`str`): Name of the function in that file.
-          * ``"description"`` (:obj:`str`): Description.
-          * ``"inputs"`` (:obj:`list` of :obj:`dict`): List of inputs represented
-            as a :obj:`dict` with the following keys:
-
-            * ``"name"`` (:obj:`str`): Input name.
-            * ``"data_type"`` (:obj:`str`): Input data type.
-            * ``"is_required"`` (:obj:`bool`): Whether the input is required.
-            * ``"default_value"`` (:obj:`str`): Input default value.
-
-          * ``"outputs"`` (:obj:`list` of :obj:`dict`): List of outputs
-            represented as a :obj:`dict` with the following keys:
-
-            * ``"name"`` (:obj:`str`): Output name.
-            * ``"data_type"`` (:obj:`str`): Output data type.
-            * ``"description"`` (:obj:`str`): Output description.
-
-          * ``"container_config"`` (:obj:`dict[str, str]`): Some pipeline
-            configuration, with the following optional keys:
-
-            * ``"language"`` (:obj:`str`): Language and version used for the pipeline.
-              The accepted formats are "python:3.X-slim", where "3.X" is a supported
-              version of Python, and "python-cuda:3.X-Y.Z" for GPU environments,
-              where "Y.Z" is a supported version of CUDA. The list of supported
-              versions is available on the official documentation website at
-              https://mlops-platform-documentation.craft.ai.
-            * ``"repository_url"`` (:obj:`str`): Remote repository url.
-            * ``"repository_branch"`` (:obj:`str`): Branch name.
-            * ``"included_folders"`` (:obj:`list[str]`): List of folders and
-              files in the repository required for the pipeline execution.
-            * ``"system_dependencies"`` (:obj:`list[str]`): List of system
-              dependencies.
-            * ``"dockerfile_path"`` (:obj:`str`): Path to the Dockerfile.
-            * ``"requirements_path"`` (:obj:`str`): Path to the requirements.txt
-              file.
-
-        * ``"creation_info"`` (:obj:`dict`): Information about the pipeline creation:
+        * ``"pipeline_name"`` (:obj:`str`): Name of the pipeline.
+        * ``"created_at"`` (:obj:`str`): Pipeline date of creation.
+        * ``"created_by"`` (:obj:`str`): ID of the user who created the deployment.
+        * ``"last_execution_id"`` (:obj:`str`): ID of the last execution of the
+          pipeline.
+        * ``"inputs"`` (:obj:`list` of :obj:`dict`): List of open inputs
+          of the pipeline. Each open input is represented as a :obj:`dict` with the
+          following keys:
+
+          * ``"input_name"`` (:obj:`str`): Name of the open input.
+          * ``"step_name"`` (:obj:`str`): Name of the step that provides the open
+            input.
+          * ``"data_type"`` (:obj:`str`): Data type of the open input.
+          * ``"description"`` (:obj:`str`): Description of the open input.
+          * ``"default_value"`` (:obj:`str`): Default value of the open input.
+          * ``"is_required"`` (:obj:`bool`): Whether the open input is required or
+            not.
+
+        * ``"outputs"`` (:obj:`list` of :obj:`dict`): List of open outputs
+          of the pipeline. Each open output is represented as a :obj:`dict` with the
+          following keys:
+
+          * ``"output_name"`` (:obj:`str`): Name of the open output.
+          * ``"step_name"`` (:obj:`str`): Name of the step that provides the open
+            output.
+          * ``"data_type"`` (:obj:`str`): Data type of the open output.
+          * ``"description"`` (:obj:`str`): Description of the open output.
+
+        * ``"container_config"`` (:obj:`dict[str, str]`): Some pipeline
+          configuration, with the following optional keys:
+
+          * ``"language"`` (:obj:`str`): Language and version used for the pipeline.
+            The accepted formats are "python:3.X-slim", where "3.X" is a supported
+            version of Python, and "python-cuda:3.X-Y.Z" for GPU environments,
+            where "Y.Z" is a supported version of CUDA. The list of supported
+            versions is available on the official documentation website at
+            https://mlops-platform-documentation.craft.ai.
+          * ``"repository_branch"`` (:obj:`str`): Branch name.
+          * ``"repository_url"`` (:obj:`str`): Remote repository url.
+          * ``"included_folders"`` (:obj:`list[str]`): List of folders and
+            files in the repository required for the pipeline execution.
+          * ``"system_dependencies"`` (:obj:`list[str]`): List of system
+            dependencies.
+          * ``"dockerfile_path"`` (:obj:`str`): Path to the Dockerfile.
+          * ``"requirements_path"`` (:obj:`str`): Path to the requirements.txt
+            file.
+
+        * ``"creation_info"`` (:obj:`dict`): Information about the step creation:
 
-          * ``"created_at"`` (:obj:`str`): The creation date in ISO format.
-          * ``"created_by"`` (:obj:`str`): The user who created the pipeline.
           * ``"commit_id"`` (:obj:`str`): The commit id on which the pipeline was
             built.
-          * ``"status"`` (:obj:`str`): either ``"creation_pending"`` or ``"ready"``.
-          * ``"origin"`` (:obj:`str`): The origin of the pipeline, can be
+          * ``"status"`` (:obj:`str`): The pipeline status, always ``"ready"`` when
+            this function returns.
+          * ``"origin"`` (:obj:`str`): The origin of the step, can be
             ``"git_repository"`` or ``"local"``.
-          * ``"last_execution_id"`` (:obj:`str`): The id of the last execution of
-            the pipeline.
     """
     try:
         base_url = f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}"
@@ -335,14 +338,12 @@ def get_pipeline(
     latest_execution = sdk._get(
         f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}/executions/latest"
     )
-    pipeline["creation_info"]["last_execution_id"] = (
+    pipeline["last_execution_id"] = (
         latest_execution.get("execution_id", None)
         if latest_execution is not None
         else None
     )
-    return remove_keys_from_dict(
-        pipeline, {"creation_info.is_archived", "creation_info.id"}
-    )
+    return remove_keys_from_dict(pipeline, {"is_archived", "pipeline_id"})
 
 
 def list_pipelines(sdk: BaseCraftAiSdk):

{craft_ai_sdk-0.55.1rc1 → craft_ai_sdk-0.55.2}/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.55.1rc1"  # Would be better to share it somewhere
+    _version = "0.55.2"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.55.1rc1 → craft_ai_sdk-0.55.2}/pyproject.toml

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