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

craft_ai_sdk/__init__.py

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

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["inputs"]
-    input_types = {input["input_name"]: input["data_type"] for input in pipeline_inputs}
+    pipeline_inputs = pipeline["parameters"]["inputs"]
+    input_types = {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,59 +124,58 @@ def create_pipeline(
         :obj:`dict`: Created pipeline represented as :obj:`dict` with the following
         keys:
 
-        * ``"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):
+        * ``"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:
+
+          * ``"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`): The pipeline status, always ``"ready"`` when
-            this function returns.
-          * ``"origin"`` (:obj:`str`): The origin of the step, can be
+          * ``"status"`` (:obj:`str`): either ``"creation_pending"`` or ``"ready"``.
+          * ``"origin"`` (:obj:`str`): The origin of the pipeline, 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(
@@ -259,61 +258,59 @@ def get_pipeline(
         None if the pipeline does not exist, otherwise pipeline information, with
         the following keys:
 
-        * ``"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:
+        * ``"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:
 
+          * ``"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`): The pipeline status, always ``"ready"`` when
-            this function returns.
-          * ``"origin"`` (:obj:`str`): The origin of the step, can be
+          * ``"status"`` (:obj:`str`): either ``"creation_pending"`` or ``"ready"``.
+          * ``"origin"`` (:obj:`str`): The origin of the pipeline, 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}"
@@ -338,12 +335,14 @@ def get_pipeline(
     latest_execution = sdk._get(
         f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}/executions/latest"
     )
-    pipeline["last_execution_id"] = (
+    pipeline["creation_info"]["last_execution_id"] = (
         latest_execution.get("execution_id", None)
         if latest_execution is not None
         else None
     )
-    return remove_keys_from_dict(pipeline, {"is_archived", "pipeline_id"})
+    return remove_keys_from_dict(
+        pipeline, {"creation_info.is_archived", "creation_info.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.0"  # Would be better to share it somewhere
+    _version = "0.55.1rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.55.0.dist-info → craft_ai_sdk-0.55.1rc1.dist-info}/METADATA

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

{craft_ai_sdk-0.55.0.dist-info → craft_ai_sdk-0.55.1rc1.dist-info}/RECORD

@@ -1,23 +1,23 @@
-craft_ai_sdk/__init__.py,sha256=sscg1NNCdglApUnNt3Nz9NbRVlu3aEt90eeHwNQHFec,404
+craft_ai_sdk/__init__.py,sha256=fZkpdM_nSiAXaajh7y1mKYzmu-zs9H8yMW7nLh3EiJQ,407
 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=caFN7-4tyVoxuAVNlldcRi2PmnvUYcazIZwmTOBBqik,18657
+craft_ai_sdk/core/pipeline_executions.py,sha256=-ClmW3qtPIA9LIhs_tiECCPL6ykSmG2zMLnrBNDGZiU,18665
 craft_ai_sdk/core/pipeline_metrics.py,sha256=Tt5p6guHAf7nhq1wvJG3t7WqsYKqrNQ7sZcFKqXYmdM,6465
-craft_ai_sdk/core/pipelines.py,sha256=J1zcuvOjCVAsau2YxZ_VLV1ctM4LnPUsADq3kFEJQ0U,20440
+craft_ai_sdk/core/pipelines.py,sha256=r-wTMDvH6D_4gDMjjWzjGYQdph6vv_DO8uh0wlec9S8,20331
 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=ZTRF_JrWo9E2tbtFC539u6GDs8fY3fx0ob862tjiwwE,10068
+craft_ai_sdk/sdk.py,sha256=BEqJVl-vCG0PSecYfqS3aR8uEQHjLgo6Ay82C-Ku2t8,10071
 craft_ai_sdk/utils.py,sha256=syOtpNgV8XF9VoeRVnWGkjIdWv0vyS_Z8RN5QfHCfIo,11891
 craft_ai_sdk/warnings.py,sha256=xJXbUR66SVbG_ZO7rxpAZ2hkGxdUtM5JAx6JoWg77Qs,6678
-documentation.pdf,sha256=fb3cG2vUBMSSQlIHgDNlyCJeI3EmhQ7aMmbQ5fQ1rgQ,333430
-craft_ai_sdk-0.55.0.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.55.0.dist-info/METADATA,sha256=4ysYCscHZF6Y4gshbiP0RcioeTSIxKVh5sbU_ZPwgWg,1614
-craft_ai_sdk-0.55.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-craft_ai_sdk-0.55.0.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.55.0.dist-info/RECORD,,
+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,,