two version of R2R are here HEAD master

8 files changed, 283 insertions, 0 deletions

diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/__init__.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/__init__.py
new file mode 100644
index 00000000..29a4fcd3
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/__init__.py
@@ -0,0 +1,5 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/base_environment_source.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/base_environment_source.py
new file mode 100644
index 00000000..09e0a56c
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/base_environment_source.py
@@ -0,0 +1,23 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+import logging
+from marshmallow import fields, post_load
+from azure.ai.ml._schema.core.schema import PathAwareSchema
+
+
+module_logger = logging.getLogger(__name__)
+
+
+class BaseEnvironmentSourceSchema(PathAwareSchema):
+    type = fields.Str()
+    resource_id = fields.Str()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities import BaseEnvironment
+
+        return BaseEnvironment(**data)
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/inference_server.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/inference_server.py
new file mode 100644
index 00000000..c6e38331
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/inference_server.py
@@ -0,0 +1,51 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument,no-else-return
+
+import logging
+
+from marshmallow import post_load
+from azure.ai.ml._schema._deployment.code_configuration_schema import CodeConfigurationSchema
+from azure.ai.ml._schema.core.fields import StringTransformedEnum, NestedField
+from azure.ai.ml._schema.core.schema import PathAwareSchema
+from azure.ai.ml.constants._common import InferenceServerType
+from .online_inference_configuration import OnlineInferenceConfigurationSchema
+
+
+module_logger = logging.getLogger(__name__)
+
+
+class InferenceServerSchema(PathAwareSchema):
+    type = StringTransformedEnum(
+        allowed_values=[
+            InferenceServerType.AZUREML_ONLINE,
+            InferenceServerType.AZUREML_BATCH,
+            InferenceServerType.CUSTOM,
+            InferenceServerType.TRITON,
+        ],
+        required=True,
+    )
+    code_configuration = NestedField(CodeConfigurationSchema)  # required for batch and online
+    inference_configuration = NestedField(OnlineInferenceConfigurationSchema)  # required for custom and Triton
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities import (
+            AzureMLOnlineInferencingServer,
+            AzureMLBatchInferencingServer,
+            CustomInferencingServer,
+            TritonInferencingServer,
+        )
+
+        if data["type"] == InferenceServerType.AZUREML_ONLINE:
+            return AzureMLOnlineInferencingServer(**data)
+        elif data["type"] == InferenceServerType.AZUREML_BATCH:
+            return AzureMLBatchInferencingServer(**data)
+        elif data["type"] == InferenceServerType.CUSTOM:
+            return CustomInferencingServer(**data)
+        elif data["type"] == InferenceServerType.TRITON:
+            return TritonInferencingServer(**data)
+        else:
+            return None
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_configuration.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_configuration.py
new file mode 100644
index 00000000..0e5a54a5
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_configuration.py
@@ -0,0 +1,30 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+import logging
+
+from marshmallow import fields, post_load
+from azure.ai.ml._schema.core.schema import PathAwareSchema
+from azure.ai.ml._schema.core.fields import StringTransformedEnum
+
+
+module_logger = logging.getLogger(__name__)
+
+
+class ModelConfigurationSchema(PathAwareSchema):
+    mode = StringTransformedEnum(
+        allowed_values=[
+            "copy",
+            "download",
+        ]
+    )
+    mount_path = fields.Str()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities import ModelConfiguration
+
+        return ModelConfiguration(**data)
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_package.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_package.py
new file mode 100644
index 00000000..142c85c8
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_package.py
@@ -0,0 +1,41 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+import logging
+
+from marshmallow import fields, post_load
+
+from azure.ai.ml._schema.core.schema import PathAwareSchema
+from azure.ai.ml._schema.core.fields import UnionField, NestedField, StringTransformedEnum
+from .inference_server import InferenceServerSchema
+from .model_configuration import ModelConfigurationSchema
+from .model_package_input import ModelPackageInputSchema
+from .base_environment_source import BaseEnvironmentSourceSchema
+
+module_logger = logging.getLogger(__name__)
+
+
+class ModelPackageSchema(PathAwareSchema):
+    target_environment = UnionField(
+        union_fields=[
+            fields.Dict(keys=StringTransformedEnum(allowed_values=["name"]), values=fields.Str()),
+            fields.Str(required=True),
+        ]
+    )
+    base_environment_source = NestedField(BaseEnvironmentSourceSchema)
+    inferencing_server = NestedField(InferenceServerSchema)
+    model_configuration = NestedField(ModelConfigurationSchema)
+    inputs = fields.List(NestedField(ModelPackageInputSchema))
+    tags = fields.Dict()
+    environment_variables = fields.Dict(
+        metadata={"description": "Environment variables configuration for the model package."}
+    )
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities import ModelPackage
+
+        return ModelPackage(**data)
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_package_input.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_package_input.py
new file mode 100644
index 00000000..a1a1dd8b
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/model_package_input.py
@@ -0,0 +1,81 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+import logging
+
+from marshmallow import fields, post_load
+from azure.ai.ml._utils.utils import camel_to_snake
+from azure.ai.ml._schema.core.schema import PathAwareSchema
+from azure.ai.ml._schema.core.fields import StringTransformedEnum, UnionField, NestedField
+
+module_logger = logging.getLogger(__name__)
+
+
+class PathBaseSchema(PathAwareSchema):
+    input_path_type = StringTransformedEnum(
+        allowed_values=[
+            "path_id",
+            "url",
+            "path_version",
+        ],
+        casing_transform=camel_to_snake,
+    )
+
+
+class PackageInputPathIdSchema(PathBaseSchema):
+    resource_id = fields.Str()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities._assets._artifacts._package.model_package import PackageInputPathId
+
+        return PackageInputPathId(**data)
+
+
+class PackageInputPathUrlSchema(PathBaseSchema):
+    url = fields.Str()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities._assets._artifacts._package.model_package import PackageInputPathUrl
+
+        return PackageInputPathUrl(**data)
+
+
+class PackageInputPathSchema(PathBaseSchema):
+    resource_name = fields.Str()
+    resource_version = fields.Str()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities._assets._artifacts._package.model_package import PackageInputPathVersion
+
+        return PackageInputPathVersion(**data)
+
+
+class ModelPackageInputSchema(PathAwareSchema):
+    type = StringTransformedEnum(allowed_values=["uri_file", "uri_folder"], casing_transform=camel_to_snake)
+    mode = StringTransformedEnum(
+        allowed_values=[
+            "read_only_mount",
+            "download",
+        ],
+        casing_transform=camel_to_snake,
+    )
+    path = UnionField(
+        [
+            NestedField(PackageInputPathIdSchema),
+            NestedField(PackageInputPathUrlSchema),
+            NestedField(PackageInputPathSchema),
+        ]
+    )
+    mount_path = fields.Str()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities._assets._artifacts._package.model_package import ModelPackageInput
+
+        return ModelPackageInput(**data)
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/online_inference_configuration.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/online_inference_configuration.py
new file mode 100644
index 00000000..b5c313ed
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/online_inference_configuration.py
@@ -0,0 +1,30 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+import logging
+from marshmallow import fields, post_load
+from azure.ai.ml._schema.core.fields import NestedField
+from azure.ai.ml._schema.core.schema import PathAwareSchema
+from .route import RouteSchema
+
+
+module_logger = logging.getLogger(__name__)
+
+
+class OnlineInferenceConfigurationSchema(PathAwareSchema):
+    liveness_route = NestedField(RouteSchema)
+    readiness_route = NestedField(RouteSchema)
+    scoring_route = NestedField(RouteSchema)
+    entry_script = fields.Str()
+    configuration = fields.Dict()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities._assets._artifacts._package.inferencing_server import (
+            OnlineInferenceConfiguration,
+        )
+
+        return OnlineInferenceConfiguration(**data)
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/route.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/route.py
new file mode 100644
index 00000000..86f37e06
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/package/route.py
@@ -0,0 +1,22 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument,bad-mcs-method-argument
+
+import logging
+from marshmallow import fields, post_load
+from azure.ai.ml._schema.core.schema_meta import PatchedSchemaMeta
+
+module_logger = logging.getLogger(__name__)
+
+
+class RouteSchema(PatchedSchemaMeta):
+    port = fields.Str()
+    path = fields.Str()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities._assets._artifacts._package.inferencing_server import Route
+
+        return Route(**data)