craft-ai-sdk 0.55.1rc1__py3-none-any.whl → 0.55.2__py3-none-any.whl

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/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/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/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.dist-info → craft_ai_sdk-0.55.2.dist-info}/METADATA

@@ -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.dist-info → craft_ai_sdk-0.55.2.dist-info}/RECORD

@@ -1,23 +1,23 @@
-craft_ai_sdk/__init__.py,sha256=fZkpdM_nSiAXaajh7y1mKYzmu-zs9H8yMW7nLh3EiJQ,407
+craft_ai_sdk/__init__.py,sha256=5UXdjh1QLhhuFskMfUwDDbjmI0Q2b3J4EcrRTcfMU6c,404
 craft_ai_sdk/constants.py,sha256=5dR4JfKVz3KhBKRFSdN1nrb06kFkNiYP2mRnacsbzJA,629
 craft_ai_sdk/core/data_store.py,sha256=USONqCEwG_KoTv5a1fgZJmi7FIjbfo4Vi-zq82keo4E,8511
 craft_ai_sdk/core/deployments.py,sha256=fUZBHTgOWHPShMcIvNF7XEHJLAZJwJI8zLzMtdv-Dzc,31427
 craft_ai_sdk/core/endpoints.py,sha256=ggJ8IU0QVrGa7A0RBWiJY1Lo7Klfi9BbfGnovZD97FQ,4592
 craft_ai_sdk/core/environment_variables.py,sha256=wy0U7y0C7tRXOXOiWGz0vX5ajHSmOb_7J1fXlSahGpI,2045
-craft_ai_sdk/core/pipeline_executions.py,sha256=-ClmW3qtPIA9LIhs_tiECCPL6ykSmG2zMLnrBNDGZiU,18665
+craft_ai_sdk/core/pipeline_executions.py,sha256=caFN7-4tyVoxuAVNlldcRi2PmnvUYcazIZwmTOBBqik,18657
 craft_ai_sdk/core/pipeline_metrics.py,sha256=Tt5p6guHAf7nhq1wvJG3t7WqsYKqrNQ7sZcFKqXYmdM,6465
-craft_ai_sdk/core/pipelines.py,sha256=r-wTMDvH6D_4gDMjjWzjGYQdph6vv_DO8uh0wlec9S8,20331
+craft_ai_sdk/core/pipelines.py,sha256=J1zcuvOjCVAsau2YxZ_VLV1ctM4LnPUsADq3kFEJQ0U,20440
 craft_ai_sdk/core/resource_metrics.py,sha256=yaco_AZ4Wu5sxLRKsA3qt2R3-PMs2e32EGO0gsHtq2Y,2336
 craft_ai_sdk/core/steps.py,sha256=1yTqcGrAKkVQWzslLjz2XQ3I7hILq8Oyc11Z9FeuFk0,20217
 craft_ai_sdk/core/users.py,sha256=Qi3xg7Vzk_0MepccNIZbFzjI0bBqouqkFStTuejUY6s,509
 craft_ai_sdk/exceptions.py,sha256=IC-JfZmmmaTsbMCgirOEByRmWnatQLjKe8BErRkuwM0,1075
 craft_ai_sdk/io.py,sha256=Jh9bFfXRzwCequXvJ3DKO-SIdZ7teYfzO64N940FyDY,11861
-craft_ai_sdk/sdk.py,sha256=BEqJVl-vCG0PSecYfqS3aR8uEQHjLgo6Ay82C-Ku2t8,10071
+craft_ai_sdk/sdk.py,sha256=ehS0Jf9kaQxcbuHYxYFDeL_xZKrHfgg7FUmWT1zoYGg,10068
 craft_ai_sdk/utils.py,sha256=syOtpNgV8XF9VoeRVnWGkjIdWv0vyS_Z8RN5QfHCfIo,11891
 craft_ai_sdk/warnings.py,sha256=xJXbUR66SVbG_ZO7rxpAZ2hkGxdUtM5JAx6JoWg77Qs,6678
-documentation.pdf,sha256=eSLh9i3QDQnRe1zz0upLX_J4AU7qy_Eo8XDoEZhthh8,334078
-craft_ai_sdk-0.55.1rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.55.1rc1.dist-info/METADATA,sha256=svsgM22xoweX0sBLicpRKw6gCLTqsfCYlqzwRNKwlUo,1617
-craft_ai_sdk-0.55.1rc1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-craft_ai_sdk-0.55.1rc1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.55.1rc1.dist-info/RECORD,,
+documentation.pdf,sha256=pZdOoPnLklDUVAhdldYvgESbsC4-3NJRqMN_p6k2Nl8,333529
+craft_ai_sdk-0.55.2.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.55.2.dist-info/METADATA,sha256=TMpKA6COOTsh1btq-S_Wh-LCmvjSToLLddCtp3vkDoA,1614
+craft_ai_sdk-0.55.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+craft_ai_sdk-0.55.2.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
+craft_ai_sdk-0.55.2.dist-info/RECORD,,