azureml-registry-tools 0.1.0a32__py3-none-any.whl → 0.1.0a34__py3-none-any.whl

azureml/registry/data/model.schema.json

@@ -167,7 +167,7 @@
         "_aml_system_vanity_registry": {
           "description": "AML vanity registry",
           "type": "string",
-          "enum": ["azureml-phi"]
+          "enum": ["azureml-phi", "azureml-oai-oss-ft", "azureml-meta-ft"]
         },
         "author": {
           "description": "Model provider/author name",
@@ -374,7 +374,7 @@
         "outputModalities": {
           "description": "Output modalities supported (e.g., text) as a comma-separated string",
           "type": "string",
-          "pattern": "^(audio|csv|embeddings|image|json|pdf|text|video|3D-image)(?:\\s*,\\s*(audio|csv|embeddings|image|json|pdf|text|video|3D-image))*$"
+          "pattern": "^(audio|csv|embeddings|image|json|pdf|text|video|3D-image|robot-action-chunks)(?:\\s*,\\s*(audio|csv|embeddings|image|json|pdf|text|video|3D-image|robot-action-chunks))*$"
         },
         "playgroundRateLimitTier": {
           "description": "Rate limit tier for playground",
@@ -395,7 +395,7 @@
         "task": {
           "description": "Tasks supported by the model as a comma-separated string",
           "type": "string",
-          "pattern": "^(audio-analysis|audio-classification|audio-generation|automatic-speech-recognition|chat-completion|completions|content-filters|content-safety|conversational-ai|custom-extraction|data-generation|document-analysis|document-ingestion|document-translation|embeddings|face-detection|fill-mask|forecasting|image-analysis|image-classification|image-feature-extraction|image-text-to-text|image-to-image|image-to-text|intelligent-content-processing|intelligent-document-processing|text-pii-extraction|conversation-pii-extraction|document-pii-extraction|detect-language|optical-character-recognition|protein-sequence-generation|protein-structure-prediction|responses|responsible-ai|retrosynthesis-prediction|summarization|text-analysis|text-analytics|text-classification|text-generation|text-to-image|text-to-speech|time-series-forecasting|translation|speech-to-text|speech-translation|video-analysis|video-generation|video-text-to-text|visual-question-answering|zero-shot-classification|zero-shot-image-classification|materials-design|atomistic-modelling|image-to-3D|text-to-3D|3D-generation|task-completion-verification|action-affordance|next-plausible-action-prediction|Structure-Prediction|Genomics|biomolecular-complex-structure-prediction|structure-prediction|protein-folding|web-agent-tasks|gui-grounding|messages)(?:\\s*,\\s*(audio-analysis|audio-classification|audio-generation|automatic-speech-recognition|chat-completion|completions|content-filters|content-safety|conversational-ai|custom-extraction|data-generation|document-analysis|document-ingestion|document-translation|embeddings|face-detection|fill-mask|forecasting|image-analysis|image-classification|image-feature-extraction|image-text-to-text|image-to-image|image-to-text|intelligent-content-processing|intelligent-document-processing|optical-character-recognition|protein-sequence-generation|protein-structure-prediction|responses|responsible-ai|summarization|text-analysis|text-analytics|text-classification|text-generation|text-pii-extraction|conversation-pii-extraction|document-pii-extraction|detect-language|text-to-image|text-to-speech|time-series-forecasting|translation|speech-to-text|speech-translation|video-analysis|video-generation|video-text-to-text|visual-question-answering|zero-shot-classification|zero-shot-image-classification|materials-design|atomistic-modelling|image-to-3D|text-to-3D|3D-generation|task-completion-verification|action-affordance|next-plausible-action-prediction|Structure-Prediction|Genomics|biomolecular-complex-structure-prediction|structure-prediction|protein-folding|web-agent-tasks|gui-grounding|messages))*$"
+          "pattern": "^(audio-analysis|audio-classification|audio-generation|automatic-speech-recognition|chat-completion|completions|content-filters|content-safety|conversational-ai|custom-extraction|data-generation|document-analysis|document-ingestion|document-translation|embeddings|face-detection|fill-mask|forecasting|image-analysis|image-classification|image-feature-extraction|image-text-to-text|image-to-image|image-to-text|intelligent-content-processing|intelligent-document-processing|text-pii-extraction|conversation-pii-extraction|document-pii-extraction|detect-language|optical-character-recognition|protein-sequence-generation|protein-structure-prediction|responses|responsible-ai|retrosynthesis-prediction|summarization|text-analysis|text-analytics|text-classification|text-generation|text-to-image|text-to-speech|time-series-forecasting|translation|speech-to-text|speech-translation|video-analysis|video-generation|video-text-to-text|visual-question-answering|zero-shot-classification|zero-shot-image-classification|materials-design|atomistic-modelling|image-to-3D|text-to-3D|3D-generation|task-completion-verification|action-affordance|next-plausible-action-prediction|Structure-Prediction|Genomics|biomolecular-complex-structure-prediction|structure-prediction|protein-folding|web-agent-tasks|gui-grounding|messages|robot-control)(?:\\s*,\\s*(audio-analysis|audio-classification|audio-generation|automatic-speech-recognition|chat-completion|completions|content-filters|content-safety|conversational-ai|custom-extraction|data-generation|document-analysis|document-ingestion|document-translation|embeddings|face-detection|fill-mask|forecasting|image-analysis|image-classification|image-feature-extraction|image-text-to-text|image-to-image|image-to-text|intelligent-content-processing|intelligent-document-processing|optical-character-recognition|protein-sequence-generation|protein-structure-prediction|responses|responsible-ai|summarization|text-analysis|text-analytics|text-classification|text-generation|text-pii-extraction|conversation-pii-extraction|document-pii-extraction|detect-language|text-to-image|text-to-speech|time-series-forecasting|translation|speech-to-text|speech-translation|video-analysis|video-generation|video-text-to-text|visual-question-answering|zero-shot-classification|zero-shot-image-classification|materials-design|atomistic-modelling|image-to-3D|text-to-3D|3D-generation|task-completion-verification|action-affordance|next-plausible-action-prediction|Structure-Prediction|Genomics|biomolecular-complex-structure-prediction|structure-prediction|protein-folding|web-agent-tasks|gui-grounding|messages|robot-control))*$"
         },
         "textContextWindow": {
           "description": "Context window size",
@@ -606,6 +606,38 @@
           },
           "description": "Model capabilities (e.g., agents, assistants)"
         },
+        "toolsSupported": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "web_search",
+              "file_search",
+              "image_generation",
+              "code_interpreter",
+              "computer_use",
+              "mcp"
+            ]
+          },
+          "description": "Specific tools that the model supports (e.g., web_search, file_search, code_interpreter, computer_use)"
+        },
+        "featuresSupported": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "streaming",
+              "function_calling",
+              "structured_outputs",
+              "fine_tuning",
+              "distillation",
+              "predicted_outputs",
+              "inpainting",
+              "image_input"
+            ]
+          },
+          "description": "Model features and capabilities supported (e.g., streaming, function_calling, structured_outputs)"
+        },
         "azureOffers": {
           "type": "array",
           "items": {

azureml/registry/mgmt/asset_management.py

@@ -20,13 +20,16 @@ from azureml.registry.data.validate_model_variant_schema import validate_model_v
 from azureml.registry._rest_client.registry_management_client import RegistryManagementClient
 from azureml.registry.mgmt.util import resolve_from_file_for_asset
 
-# Windows compatibility patch - must be applied before importing azureml.assets
+# Cross-platform compatibility patch - must be applied before importing azureml.assets
+import os
 from subprocess import run
 
 
 def patched_run_command(cmd: List[str]):
-    """Run command with shell=True for Windows compatibility."""
-    result = run(cmd, capture_output=True, encoding=sys.stdout.encoding, errors="ignore", shell=True)
+    """Run command with OS-appropriate shell setting for cross-platform compatibility."""
+    # Use shell=True on Windows, shell=False on Unix-like systems
+    use_shell = os.name == 'nt'  # 'nt' is Windows
+    result = run(cmd, capture_output=True, encoding=sys.stdout.encoding, errors="ignore", shell=use_shell)
     return result
 
 
@@ -40,6 +43,9 @@ from azureml.assets.config import AssetConfig, AssetType  # noqa: E402
 from azureml.assets.publish_utils import create_asset  # noqa: E402
 from azureml.assets.validate_assets import validate_assets  # noqa: E402
 
+# Set azcopy job plan location to avoid permission issues
+os.environ.setdefault("AZCOPY_JOB_PLAN_LOCATION", tempfile.gettempdir())
+
 
 def validate_model(asset_path: Path, allow_additional_properties: bool = False) -> bool:
     """Validate model.
@@ -239,6 +245,12 @@ def create_or_update_asset(readonly_asset: AssetConfig, registry_name: str, subs
         credential=DefaultAzureCredential(),  # CodeQL [SM05139] DefaultAzureCredential should only be used for local development and testing purposes.
     )
 
+    if not shutil.which("azcopy"):
+        raise RuntimeError(
+            "azcopy is not available in PATH. "
+            "Please install azcopy: https://learn.microsoft.com/en-us/azure/storage/common/storage-use-azcopy-v10"
+        )
+
     with tempfile.TemporaryDirectory() as mutable_asset_dir:
         mutable_asset, system_metadata_payload = build_mutable_asset(base_asset=readonly_asset, mutable_asset_dir=mutable_asset_dir, override_storage=override_storage)
         try:

azureml/registry/mgmt/create_manifest.py

@@ -20,14 +20,6 @@ ALLOWED_ASSET_TYPES = {
     'environments', 'models', 'deployment-templates', 'components', 'datasets'
 }
 
-ASSET_TYPE_MAP = {
-    'models': 'Models',
-    'environments': 'Environments',
-    'deployment-templates': 'DeploymentTemplates',
-    'components': 'Components',
-    'datasets': 'Datasets',
-}
-
 
 def validate_manifest_data(data: dict, file_path: str) -> None:
     """Validate required root-level fields and allowed values for nested fields under 'assets'.

azureml/registry/mgmt/syndication_manifest.py

@@ -7,6 +7,18 @@ from dataclasses import dataclass, field
 from typing import List
 from uuid import UUID
 
+# Asset type mapping from YAML keys to JSON/API keys
+ASSET_TYPE_MAP = {
+    'models': 'Models',
+    'environments': 'Environments',
+    'deployment-templates': 'DeploymentTemplates',
+    'components': 'Components',
+    'datasets': 'Datasets',
+}
+
+# Reverse mapping for deserializing API responses back to YAML keys
+ASSET_TYPE_REVERSE_MAP = {v: k for k, v in ASSET_TYPE_MAP.items()}
+
 
 def _norm_key(k):
     return k.replace('_', '').lower()
@@ -70,7 +82,13 @@ class SourceRegistry:
         asset_dict = {}
         for asset_type, asset_list in self.assets.items():
             if asset_list:
-                asset_dict[asset_type] = [
+                # Convert YAML asset type keys for API
+                if asset_type not in ASSET_TYPE_MAP:
+                    raise ValueError(
+                        f"Invalid asset type '{asset_type}'. Allowed types: {', '.join(ASSET_TYPE_MAP.keys())}"
+                    )
+                api_asset_type = ASSET_TYPE_MAP[asset_type]
+                asset_dict[api_asset_type] = [
                     {"Name": a.name} if a.name == ".*" else {"Name": a.name, "Version": a.version}
                     for a in asset_list
                 ]
@@ -86,7 +104,9 @@ class SourceRegistry:
         assets = {}
         assets_dict = _get_key(d, "Assets", "assets", normalize_keys=normalize_keys)
         for asset_type, asset_list in assets_dict.items():
-            assets[asset_type] = [Asset.from_dict(a, normalize_keys=normalize_keys) for a in asset_list]
+            # Normalize API keys back to YAML keys
+            normalized_asset_type = ASSET_TYPE_REVERSE_MAP.get(asset_type, asset_type)
+            assets[normalized_asset_type] = [Asset.from_dict(a, normalize_keys=normalize_keys) for a in asset_list]
         return SourceRegistry(
             registry_name=_get_key(d, "RegistryName", "registry_name", normalize_keys=normalize_keys),
             tenant_id=UUID(str(_get_key(d, "TenantId", "tenant_id", normalize_keys=normalize_keys))),

{azureml_registry_tools-0.1.0a32.dist-info → azureml_registry_tools-0.1.0a34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a32
+Version: 0.1.0a34
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license
@@ -10,7 +10,7 @@ Requires-Dist: azure-identity<2.0
 Requires-Dist: ruamel-yaml<0.19,>=0.17.21
 Requires-Dist: diskcache~=5.6
 Requires-Dist: azure-ai-ml<2.0,>=1.30.0
-Requires-Dist: azureml-assets<2.0
+Requires-Dist: azureml-assets<2.0,>=1.16.99
 Dynamic: author
 Dynamic: license
 Dynamic: license-file

{azureml_registry_tools-0.1.0a32.dist-info → azureml_registry_tools-0.1.0a34.dist-info}/RECORD

@@ -13,27 +13,27 @@ azureml/registry/data/asset.yaml.template,sha256=WTgfuvKEBp-EVFSQ0JpU0h4z_ULJdUL
 azureml/registry/data/description.md.template,sha256=DiVAQEXGXoKmhV4LPqE3NupxVtcsuDZ1pn2UA5Fzd6U,2821
 azureml/registry/data/evaluation.md.template,sha256=FC9U8EI_1Dg9Vz18ftAFDDmTqvpwELDkIDlYqB8C9Dk,1031
 azureml/registry/data/model-variant.schema.json,sha256=AT4Dy6cCtp_SFUfSqYIqcER8AldpYm0QIEy1abY3QWE,1699
-azureml/registry/data/model.schema.json,sha256=yP5YejvH9VBrWrh8mXjBLcRotjwd5kMqs4hSNAqc9aU,47605
+azureml/registry/data/model.schema.json,sha256=OoxFRCKCEnNjB5kPPTDUlzq_LAZBP_AbcQYqNNbdnvY,48726
 azureml/registry/data/model.yaml.template,sha256=h5uqAN22FLaWrbPxIb8yVKH9cGDBrIwooXYYfsKhxDw,245
 azureml/registry/data/notes.md.template,sha256=rgGGHQaxfVg6COIzZU8EVBa48sRPnNezVkCMGNyRRNo,1528
 azureml/registry/data/validate_model_schema.py,sha256=4Exd9K6ry93URbx21gn2wSJ6xPacYfO7tmZ96cr8ViM,8376
 azureml/registry/data/validate_model_variant_schema.py,sha256=0nC1zWzvtYAwowY1MG4lIhFDtBBhezwpya4z-ItafWo,3665
 azureml/registry/mgmt/__init__.py,sha256=LMhqcEC8ItmmpKZljElGXH-6olHlT3SLl0dJU01OvuM,226
-azureml/registry/mgmt/asset_management.py,sha256=pfQsaXzt_BllEtFHXclnHEG2ShXJmHp1H2QrU_kKj-8,14736
+azureml/registry/mgmt/asset_management.py,sha256=zMAgrwMEQTDh8YjL5B1_p7_Wi0bR-vKx2qyuc3FudsE,15269
 azureml/registry/mgmt/create_asset_template.py,sha256=ejwLuIsmzJOoUePoxbM-eGMg2E3QHfdX-nPMBzYUVMQ,3525
-azureml/registry/mgmt/create_manifest.py,sha256=N9wRmjAKO09A3utN_lCUsM_Ufpj7PL0SJz-XHPHWuyM,9528
+azureml/registry/mgmt/create_manifest.py,sha256=WuL1UNCi-vLOgI71FHNSIP3CAIYJ-jwPvTbEtJPCuTw,9335
 azureml/registry/mgmt/create_model_spec.py,sha256=1PdAcUf-LomvljoT8wKQihXMTLd7DoTgN0qDX4Lol1A,10473
 azureml/registry/mgmt/model_management.py,sha256=STTr_uvdPKV2NaJ5UvS5aMi3yejVF6Hkj9DjofJLQik,7453
-azureml/registry/mgmt/syndication_manifest.py,sha256=8Sfd49QuCA5en5_mIOLE21kZVpnReUXowx_g0TVRgWg,9025
+azureml/registry/mgmt/syndication_manifest.py,sha256=kSL6SbbpsraIBoo2XdCv6lbg1XxHs3YqPc0ozLX1v54,9900
 azureml/registry/mgmt/util.py,sha256=1q90sLr70w8FrXYqyHUwszjXa7OdT2MRSI_gJf-aVdY,2403
 azureml/registry/tools/__init__.py,sha256=IAuWWpGfZm__pAkBIxmpJz84QskpkxBr0yDk1TUSnkE,223
 azureml/registry/tools/config.py,sha256=eZ8F6vKvKX-rHkZrnLmjX-h_1ccqSSTkUK0s3ADoD9o,3662
 azureml/registry/tools/create_or_update_assets.py,sha256=xKGmcc3GVQ0qoZav68NENL86dYN1jdTPXCHK1SfK_vM,16207
 azureml/registry/tools/registry_utils.py,sha256=zgYlCiOONtQJ4yZ9wg8tKVoE8dh6rrjB8hYBGhpV9-0,1403
 azureml/registry/tools/repo2registry_config.py,sha256=YxSpLl16m-Cs9RkrXjpWgg6la2zukl-y1i12pV4UIco,4891
-azureml_registry_tools-0.1.0a32.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
-azureml_registry_tools-0.1.0a32.dist-info/METADATA,sha256=KG3mK4xG_yuwbfjmCRAAckAJxDOzAglwCz581KMH298,531
-azureml_registry_tools-0.1.0a32.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-azureml_registry_tools-0.1.0a32.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
-azureml_registry_tools-0.1.0a32.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
-azureml_registry_tools-0.1.0a32.dist-info/RECORD,,
+azureml_registry_tools-0.1.0a34.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
+azureml_registry_tools-0.1.0a34.dist-info/METADATA,sha256=PfpeXbIue5E66DwBnydtrt9LbWYm8qr3Q3b87IZLPks,541
+azureml_registry_tools-0.1.0a34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+azureml_registry_tools-0.1.0a34.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
+azureml_registry_tools-0.1.0a34.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
+azureml_registry_tools-0.1.0a34.dist-info/RECORD,,