azureml-registry-tools 0.1.0a31__tar.gz → 0.1.0a33__tar.gz

{azureml_registry_tools-0.1.0a31/azureml_registry_tools.egg-info → azureml_registry_tools-0.1.0a33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a31
+Version: 0.1.0a33
 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.0a31 → azureml_registry_tools-0.1.0a33}/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",
@@ -495,6 +495,7 @@
         "invisibleUntil": {
           "type": "string",
           "format": "date-time",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
           "description": "Date until which the model is invisible"
         },
         "playgroundRateLimitTier": {
@@ -605,15 +606,37 @@
           },
           "description": "Model capabilities (e.g., agents, assistants)"
         },
-        "playground_rate_limit_tier": {
-          "type": "string",
-          "description": "Rate limit tier for playground",
-          "enum": [
-            "high",
-            "low",
-            "custom",
-            "embeddings"
-          ]
+        "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",
@@ -643,7 +666,8 @@
         },
         "sharedComputeCapacityEnabled": {
           "type": "boolean",
-          "description": "Whether shared compute capacity is enabled"
+          "enum": [true],
+          "description": "Whether shared compute capacity is enabled. Only true values are allowed - remove this field when disabled."
         },
         "freePlayground": {
           "type": "boolean",
@@ -678,7 +702,8 @@
         },
         "isDirectFromAzure": {
           "type": "boolean",
-          "description": "If the model is an Azure Direct model"
+          "enum": [true],
+          "description": "If the model is an Azure Direct model. Only true values are allowed - remove this field when disabled."
         },
         "inferenceEnvironmentAssetId": {
           "type": "string",
@@ -855,14 +880,17 @@
             }
           }
         },
-        "sku": {
+        "SKU": {
           "type": "object",
           "description": "SKU specifications for different operations",
           "additionalProperties": false,
           "properties": {
-            "finetuneRecommendedSku": {
-              "description": "Skus that can be used for finetune maap",
-              "$ref": "#/definitions/skuListDefinition"
+            "finetuneRecommendedSkus": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "Skus that can be used for finetune maap"
             },
             "evaluationMinSkuSpec": {
               "type": "string",
@@ -870,9 +898,12 @@
               "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$",
               "format": "sku-spec"
             },
-            "evaluationRecommendedSku": {
-              "description": "Evaluation recommended sku",
-              "$ref": "#/definitions/skuListDefinition"
+            "evaluationRecommendedSkus": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "Evaluation recommended skus"
             },
             "finetuneMinSkuSpec": {
               "type": "string",
@@ -886,9 +917,12 @@
               "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$",
               "format": "sku-spec"
             },
-            "inferenceRecommendedSku": {
-              "description": "A list of comma separated skus that the client should use to deploy the model",
-              "$ref": "#/definitions/skuListDefinition"
+            "inferenceRecommendedSkus": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "A list of skus that the client should use to deploy the model"
             }
           }
         },
@@ -899,7 +933,8 @@
           "properties": {
             "maasFinetuningTest": {
               "type": "boolean",
-              "description": "Used for testing maas-finetuning in INT"
+              "enum": [true],
+              "description": "Used for testing maas-finetuning in INT. Only true values are allowed - remove this field when disabled."
             },
             "maasFinetuningDeployRegions": {
               "type": "array",
@@ -918,12 +953,23 @@
             },
             "maasFinetuningOffer": {
               "type": "boolean",
-              "description": "Whether MaaS fine-tuning offer is available"
+              "enum": [true],
+              "description": "Whether MaaS fine-tuning offer is available. Only true values are allowed - remove this field when disabled."
             },
-            "maasFinetuningRegions": {
-              "type": "string",
-              "description": "Regions where fine-tuning is available as a comma-separated string",
-              "pattern": "^(westus3|eastus2|eastus|southcentralus|northcentralus|westus)(?:\\s*,\\s*(westus3|eastus2|eastus|southcentralus|northcentralus|westus))*$"
+            "maasFinetuningAzureRegions": {
+              "type": "array",
+              "items": {
+                "type": "string",
+                "enum": [
+                  "westus3",
+                  "eastus2",
+                  "eastus",
+                  "southcentralus",
+                  "northcentralus",
+                  "westus"
+                ]
+              },
+              "description": "Regions where fine-tuning is available"
             }
           }
         },
@@ -934,32 +980,32 @@
           "properties": {
             "finetuneDeprecationDate": {
               "type": "string",
-              "description": "The date that finetune will be deprecated. Date format yyyy-MM-dd. The deprecation will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that finetune will be deprecated. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "finetuneRetirementDate": {
               "type": "string",
-              "description": "The date that finetune will be retired. Date format yyyy-MM-dd. The retirement will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that finetune will be retired. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "inferenceDeprecationDate": {
               "type": "string",
-              "description": "The date that inference will be deprecated. Date format yyyy-MM-dd. The deprecation will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that inference will be deprecated. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "inferenceLegacyDate": {
               "type": "string",
-              "description": "The date that inference will become legacy. Date format yyyy-MM-dd. The change will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that inference will become legacy. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "inferenceRetirementDate": {
               "type": "string",
-              "description": "The date that inference will be retired. Date format yyyy-MM-dd. The retirement will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that inference will be retired. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             }
           }

{azureml_registry_tools-0.1.0a31 → azureml_registry_tools-0.1.0a33}/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.
@@ -236,9 +242,15 @@ def create_or_update_asset(readonly_asset: AssetConfig, registry_name: str, subs
         subscription_id=subscription_id,
         resource_group_name=resource_group,
         registry_name=registry_name,
-        credential=DefaultAzureCredential(),
+        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_tools-0.1.0a31 → azureml_registry_tools-0.1.0a33/azureml_registry_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a31
+Version: 0.1.0a33
 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.0a31 → azureml_registry_tools-0.1.0a33}/azureml_registry_tools.egg-info/requires.txt

@@ -2,4 +2,4 @@ azure-identity<2.0
 ruamel-yaml<0.19,>=0.17.21
 diskcache~=5.6
 azure-ai-ml<2.0,>=1.30.0
-azureml-assets<2.0
+azureml-assets<2.0,>=1.16.99

{azureml_registry_tools-0.1.0a31 → azureml_registry_tools-0.1.0a33}/setup.py

@@ -11,14 +11,14 @@ DEPENDENCIES = [
     "ruamel-yaml>=0.17.21,<0.19",
     "diskcache~=5.6",
     "azure-ai-ml>=1.30.0,<2.0",
-    "azureml-assets<2.0"
+    "azureml-assets>=1.16.99,<2.0"
 ]
 
 exclude_list = ["*.tests"]
 
 setup(
     name='azureml-registry-tools',
-    version="0.1.0a31",
+    version="0.1.0a33",
     description='AzureML Registry tools and CLI',
     author='Microsoft Corp',
     license="https://aka.ms/azureml-sdk-license",