craft-ai-sdk 0.55.2rc1__tar.gz → 0.56.0rc1__tar.gz

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

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

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

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

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

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

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