azureml-registry-tools 0.1.0a20__tar.gz → 0.1.0a25__tar.gz

{azureml_registry_tools-0.1.0a20/azureml_registry_tools.egg-info → azureml_registry_tools-0.1.0a25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a20
+Version: 0.1.0a25
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license

{azureml_registry_tools-0.1.0a20 → azureml_registry_tools-0.1.0a25}/azureml/registry/_rest_client/base_rest_client.py

@@ -178,7 +178,7 @@ class BaseAzureRestClient(BaseRestClient):
         super().__init__(base_url, api_key=api_key, max_retries=max_retries, backoff_factor=backoff_factor)
         # Only after super().__init__ is self.session available
         if api_key is None:
-            self._credential = DefaultAzureCredential()
+            self._credential = DefaultAzureCredential()  # CodeQL [SM05139] DefaultAzureCredential should only be used for local development and testing purposes.
             self._refresh_api_key_if_needed()
             # Ensure self.api_key is set for future use
             api_key = self.api_key

{azureml_registry_tools-0.1.0a20 → azureml_registry_tools-0.1.0a25}/azureml/registry/_rest_client/registry_management_client.py

@@ -24,7 +24,10 @@ class RegistryManagementClient(BaseAzureRestClient):
         if primary_region is None:
             # Resolve the primary region if not provided
             primary_region = self.resolve_registry_primary_region(registry_name)
-        base_url = f"https://{primary_region}.api.azureml.ms"
+        if primary_region.lower() == "centraluseuap":
+            base_url = "https://int.api.azureml-test.ms"
+        else:
+            base_url = f"https://{primary_region}.api.azureml.ms"
         super().__init__(base_url, api_key=api_key, max_retries=max_retries, backoff_factor=backoff_factor)
         self.registry_name = registry_name
 

{azureml_registry_tools-0.1.0a20 → azureml_registry_tools-0.1.0a25}/azureml/registry/data/model.schema.json

@@ -176,7 +176,17 @@
         "azureOffers": {
           "description": "Supported azure offer types as a comma-separated string",
           "type": "string",
-          "pattern": "^(standard-paygo|RestrictedAccess|PTU|batch-paygo|VM)(?:\\s*,\\s*(standard-paygo|RestrictedAccess|PTU|batch-paygo|VM))*$"
+          "pattern": "^(standard-paygo|RestrictedAccess|PTU|batch-paygo|VM|VM-withSurcharge)(?:\\s*,\\s*(standard-paygo|RestrictedAccess|PTU|batch-paygo|VM|VM-withSurcharge))*$"
+        },
+        "maap-surcharge": {
+          "description": "MaaP Surcharge",
+          "type": "string",
+          "enum": ["true", "false"]
+        },
+        "maap-surcharge-type": {
+          "description": "MaaP Surcharge Type",
+          "type": "string",
+          "pattern": "^(model)(?:\\s*,\\s*(model))*$"
         },
         "benchmark": {
           "description": "Benchmark type",
@@ -270,7 +280,7 @@
         "languages": {
           "description": "Comma-separated list of supported languages (ISO 639-1 language codes)",
           "type": "string",
-          "pattern": "^(aa|ab|ae|af|ak|am|an|ar|as|av|ay|az|ba|be|bg|bh|bi|bm|bn|bo|br|bs|ca|ce|ch|co|cr|cs|cu|cv|cy|da|de|dv|dz|ee|el|en|eo|es|et|eu|fa|ff|fi|fj|fo|fr|fy|ga|gd|gl|gn|gu|gv|ha|he|hi|ho|hr|ht|hu|hy|hz|ia|id|ie|ig|ii|ik|io|is|it|iu|ja|jv|ka|kg|ki|kj|kk|kl|km|kn|ko|kr|ks|ku|kv|kw|ky|la|lb|lg|li|ln|lo|lt|lu|lv|mg|mh|mi|mk|ml|mn|mr|ms|mt|my|na|nb|nd|ne|ng|nl|nn|\"no\"|nr|nv|ny|oc|oj|om|or|os|pa|pi|pl|ps|pt|pt-br|qu|rm|rn|ro|ru|rw|sa|sc|sd|se|sg|si|sk|sl|sm|sn|so|sq|sr|ss|st|su|sv|sw|ta|te|tg|th|ti|tk|tl|tn|to|tr|ts|tt|tw|ty|ug|uk|ur|uz|ve|vi|vo|wa|wo|xh|yi|yo|za|zh|zh-cn|zu)(?:\\s*,\\s*(aa|ab|ae|af|ak|am|an|ar|as|av|ay|az|ba|be|bg|bh|bi|bm|bn|bo|br|bs|ca|ce|ch|co|cr|cs|cu|cv|cy|da|de|dv|dz|ee|el|en|eo|es|et|eu|fa|ff|fi|fj|fo|fr|fy|ga|gd|gl|gn|gu|gv|ha|he|hi|ho|hr|ht|hu|hy|hz|ia|id|ie|ig|ii|ik|io|is|it|iu|ja|jv|ka|kg|ki|kj|kk|kl|km|kn|ko|kr|ks|ku|kv|kw|ky|la|lb|lg|li|ln|lo|lt|lu|lv|mg|mh|mi|mk|ml|mn|mr|ms|mt|my|na|nb|nd|ne|ng|nl|nn|\"no\"|nr|nv|ny|oc|oj|om|or|os|pa|pi|pl|ps|pt|pt-br|qu|rm|rn|ro|ru|rw|sa|sc|sd|se|sg|si|sk|sl|sm|sn|so|sq|sr|ss|st|su|sv|sw|ta|te|tg|th|ti|tk|tl|tn|to|tr|ts|tt|tw|ty|ug|uk|ur|uz|ve|vi|vo|wa|wo|xh|yi|yo|za|zh|zh-cn|zu))*$"
+          "pattern": "^(aa|ab|ae|af|ak|am|an|ar|as|av|ay|az|ba|be|bg|bh|bi|bm|bn|bo|br|bs|ca|ce|ch|co|cr|cs|cu|cv|cy|da|de|dv|dz|ee|el|en|eo|es|et|eu|fa|ff|fi|fj|fo|fr|fy|ga|gd|gl|gn|gu|gv|ha|he|hi|ho|hr|ht|hu|hy|hz|ia|id|ie|ig|ii|ik|io|is|it|iu|ja|jv|ka|kg|ki|kj|kk|kl|km|kn|ko|kr|ks|ku|kv|kw|ky|la|lb|lg|li|ln|lo|lt|lu|lv|mg|mh|mi|mk|ml|mn|mr|ms|mt|my|na|nb|nd|ne|ng|nl|nn|no|nr|nv|ny|oc|oj|om|or|os|pa|pi|pl|ps|pt|pt-br|qu|rm|rn|ro|ru|rw|sa|sc|sd|se|sg|si|sk|sl|sm|sn|so|sq|sr|ss|st|su|sv|sw|ta|te|tg|th|ti|tk|tl|tn|to|tr|ts|tt|tw|ty|ug|uk|ur|uz|ve|vi|vo|wa|wo|xh|yi|yo|za|zh|zh-cn|zu)(?:\\s*,\\s*(aa|ab|ae|af|ak|am|an|ar|as|av|ay|az|ba|be|bg|bh|bi|bm|bn|bo|br|bs|ca|ce|ch|co|cr|cs|cu|cv|cy|da|de|dv|dz|ee|el|en|eo|es|et|eu|fa|ff|fi|fj|fo|fr|fy|ga|gd|gl|gn|gu|gv|ha|he|hi|ho|hr|ht|hu|hy|hz|ia|id|ie|ig|ii|ik|io|is|it|iu|ja|jv|ka|kg|ki|kj|kk|kl|km|kn|ko|kr|ks|ku|kv|kw|ky|la|lb|lg|li|ln|lo|lt|lu|lv|mg|mh|mi|mk|ml|mn|mr|ms|mt|my|na|nb|nd|ne|ng|nl|nn|no|nr|nv|ny|oc|oj|om|or|os|pa|pi|pl|ps|pt|pt-br|qu|rm|rn|ro|ru|rw|sa|sc|sd|se|sg|si|sk|sl|sm|sn|so|sq|sr|ss|st|su|sv|sw|ta|te|tg|th|ti|tk|tl|tn|to|tr|ts|tt|tw|ty|ug|uk|ur|uz|ve|vi|vo|wa|wo|xh|yi|yo|za|zh|zh-cn|zu))*$"
         },
         "license": {
           "description": "License type",
@@ -354,7 +364,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)(?:\\s*,\\s*(audio|csv|embeddings|image|json|pdf|text|video))*$"
+          "pattern": "^(audio|csv|embeddings|image|json|pdf|text|video|3D-image)(?:\\s*,\\s*(audio|csv|embeddings|image|json|pdf|text|video|3D-image))*$"
         },
         "playgroundRateLimitTier": {
           "description": "Rate limit tier for playground",
@@ -375,7 +385,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|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)(?:\\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-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))*$"
+          "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)(?:\\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))*$"
         },
         "textContextWindow": {
           "description": "Context window size",
@@ -442,12 +452,488 @@
         }
       }
     },
+    "system_metadata": {
+      "type": "object",
+      "description": "Model system metadata",
+      "additionalProperties": false,
+      "properties": {
+        "invisibleUntil": {
+          "type": "string",
+          "format": "date-time",
+          "description": "Date until which the model is invisible"
+        },
+        "playgroundRateLimitTier": {
+          "type": "string",
+          "description": "Rate limit tier for playground",
+          "enum": [
+            "high",
+            "low",
+            "custom",
+            "embeddings"
+          ]
+        },
+        "publisher": {
+          "type": "string",
+          "description": "Model provider/author name"
+        },
+        "license": {
+          "type": "string",
+          "description": "License type",
+          "enum": [
+            "custom",
+            "other",
+            "llama2",
+            "apache-2.0",
+            "Llama 3.1 Community Licensed",
+            "MIT"
+          ]
+        },
+        "licenseDescription": {
+          "type": "string",
+          "description": "Description of the license terms",
+          "format": "long-string"
+        },
+        "summary": {
+          "type": "string",
+          "description": "Brief description of the model",
+          "format": "long-string"
+        },
+        "displayName": {
+          "type": "string",
+          "description": "Display name for marketing materials"
+        },
+        "keywords": {
+          "type": "array",
+          "items": {
+            "$ref": "#/definitions/keywordsEnum"
+          },
+          "description": "List of key words. Max 3 keywords"
+        },
+        "languages": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "pattern": "^(aa|ab|ae|af|ak|am|an|ar|as|av|ay|az|ba|be|bg|bh|bi|bm|bn|bo|br|bs|ca|ce|ch|co|cr|cs|cu|cv|cy|da|de|dv|dz|ee|el|en|eo|es|et|eu|fa|ff|fi|fj|fo|fr|fy|ga|gd|gl|gn|gu|gv|ha|he|hi|ho|hr|ht|hu|hy|hz|ia|id|ie|ig|ii|ik|io|is|it|iu|ja|jv|ka|kg|ki|kj|kk|kl|km|kn|ko|kr|ks|ku|kv|kw|ky|la|lb|lg|li|ln|lo|lt|lu|lv|mg|mh|mi|mk|ml|mn|mr|ms|mt|my|na|nb|nd|ne|ng|nl|nn|no|nr|nv|ny|oc|oj|om|or|os|pa|pi|pl|ps|pt|pt-br|qu|rm|rn|ro|ru|rw|sa|sc|sd|se|sg|si|sk|sl|sm|sn|so|sq|sr|ss|st|su|sv|sw|ta|te|tg|th|ti|tk|tl|tn|to|tr|ts|tt|tw|ty|ug|uk|ur|uz|ve|vi|vo|wa|wo|xh|yi|yo|za|zh|zh-cn|zu)$"
+          },
+          "description": "Supported languages (ISO 639-1 language codes)"
+        },
+        "textContextWindow": {
+          "type": "integer",
+          "minimum": 1,
+          "description": "Context window size"
+        },
+        "maxOutputTokens": {
+          "type": "integer",
+          "minimum": 1,
+          "description": "Maximum number of output tokens"
+        },
+        "inputModalities": {
+          "type": "array",
+          "items": {
+            "$ref": "#/definitions/modalitiesEnum"
+          },
+          "description": "Input modalities supported (e.g., text, image, audio)"
+        },
+        "outputModalities": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "audio",
+              "csv",
+              "embeddings",
+              "image",
+              "json",
+              "pdf",
+              "text",
+              "video",
+              "3D-image"
+            ]
+          },
+          "description": "Output modalities supported (e.g., text)"
+        },
+        "modelCapabilities": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "agents",
+              "assistants",
+              "routing",
+              "reasoning",
+              "streaming",
+              "tool-calling"
+            ]
+          },
+          "description": "Model capabilities (e.g., agents, assistants)"
+        },
+        "playground_rate_limit_tier": {
+          "type": "string",
+          "description": "Rate limit tier for playground",
+          "enum": [
+            "high",
+            "low",
+            "custom",
+            "embeddings"
+          ]
+        },
+        "azureOffers": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "standard-paygo",
+              "RestrictedAccess",
+              "PTU",
+              "batch-paygo",
+              "VM",
+              "VM-withSurcharge"
+            ]
+          },
+          "description": "Supported azure offer types"
+        },
+        "baseModelWeightsVersion": {
+          "oneOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "number",
+              "minimum": 1.0
+            }
+          ]
+        },
+        "sharedComputeCapacityEnabled": {
+          "type": "boolean",
+          "description": "Whether shared compute capacity is enabled"
+        },
+        "freePlayground": {
+          "type": "boolean",
+          "description": "Whether the model is available in free playground"
+        },
+        "deploymentOptions": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "ServerlessMaaS",
+              "UnifiedEndpointMaaS",
+              "MaaP",
+              "AOAI",
+              "Restricted",
+              "MonetizedMaaP"
+            ]
+          },
+          "description": "Which types of deployment are supported in the UI"
+        },
+        "inferenceSupportedEnvs": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "ds_mii",
+              "hf",
+              "vllm"
+            ]
+          },
+          "description": "Supported inference environments"
+        },
+        "isDirectFromAzure": {
+          "type": "boolean",
+          "description": "If the model is an Azure Direct model"
+        },
+        "inferenceEnvironmentAssetId": {
+          "type": "string",
+          "pattern": "azureml://registries/([^/]+)/environments/([^/]+)/versions/([^/]+)"
+        },
+        "modelArchitecture": {
+          "type": "string",
+          "description": "Model architecture type"
+        },
+        "componentVersion": {
+          "type": "string",
+          "description": "Version of the model component"
+        },
+        "modelPath": {
+          "type": "string",
+          "description": "Path to the model files or artifacts"
+        },
+        "aotManifest": {
+          "type": "string",
+          "description": "Ahead-of-time compilation manifest for the model"
+        },
+        "hasDeltaWeights": {
+          "type": "boolean",
+          "description": "Whether the model uses delta weights"
+        },
+        "featured": {
+          "type": "boolean",
+          "description": "Whether the model is featured"
+        },
+        "preview": {
+          "type": "boolean",
+          "description": "Whether the model is in preview"
+        },
+        "benchmark": {
+          "type": "string",
+          "description": "Benchmark type",
+          "enum": [
+            "quality",
+            "embeddings"
+          ]
+        },
+        "datasets": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "Publicly available sources",
+              "bookcorpus",
+              "wikipedia"
+            ]
+          },
+          "description": "Where data was sourced for training"
+        },
+        "industry": {
+          "type": "string",
+          "description": "The industry for the model",
+          "enum": [
+            "consumer-goods",
+            "health-and-life-sciences",
+            "manufacturing",
+            "mobility",
+            "financial-services"
+          ]
+        },
+        "trainingDataDate": {
+          "type": "string",
+          "description": "Training cutoff date",
+          "pattern": "^(January|February|March|April|May|June|July|August|September|October|November|December)\\s+\\d{4}$",
+          "format": "month-year"
+        },
+        "evaluation": {
+          "type": "string",
+          "description": "Reference to evaluation documentation"
+        },
+        "notes": {
+          "type": "string",
+          "description": "Reference to notes documentation"
+        },
+        "featureFlags": {
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": [
+              "OpenAI.EAGatingTier",
+              "OpenAI.EnterpriseTier",
+              "OpenAI.1PGatingTier",
+              "OpenAI.OneVetGatingTier",
+              "OpenAI.1000CUGatingTier",
+              "OpenAI.computer-use-preview",
+              "OpenAI.Preview.202409",
+              "OpenAITest",
+              "AIServices.grok-4-code",
+              "AIServices.grok-4"
+            ]
+          },
+          "description": "List of feature flags for the models to gate usage"
+        },
+        "tasks": {
+          "type": "object",
+          "description": "Tasks that the model can perform",
+          "additionalProperties": false,
+          "properties": {
+            "inferenceTasks": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/taskEnum"
+              },
+              "description": "Tasks supported by the model"
+            },
+            "fineTuningTasks": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/finetuningTasksEnum"
+              },
+              "description": "Tasks for which the model can be fine-tuned"
+            },
+            "fineTuningMethods": {
+              "type": "array",
+              "description": "Methods for finetuning the model",
+              "items": {
+                "type": "string"
+              }
+            }
+          }
+        },
+        "platformCapabilities": {
+          "type": "object",
+          "description": "Platform-specific capabilities and settings",
+          "additionalProperties": false,
+          "properties": {
+            "enableMaap": {
+              "type": "boolean",
+              "description": "Whether MAAP is enabled (inverse of disable-maap)"
+            },
+            "enableBatch": {
+              "type": "boolean",
+              "description": "Whether batch processing is enabled (inverse of disable-batch)"
+            },
+            "maasInference": {
+              "type": "boolean",
+              "description": "Whether MaaS inference is enabled"
+            },
+            "maasFineTuning": {
+              "type": "boolean",
+              "description": "Whether MaaS fine-tuning is enabled"
+            }
+          }
+        },
+        "supportedComputes": {
+          "type": "object",
+          "description": "Compute types supported for different operations",
+          "additionalProperties": false,
+          "properties": {
+            "evaluationComputes": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "List of compute SKUs allowed for evaluation"
+            },
+            "finetuneComputes": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "List of compute SKUs allowed for finetune with maap"
+            },
+            "inferenceComputes": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "List of compute SKUs allowed for inference"
+            }
+          }
+        },
+        "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"
+            },
+            "evaluationMinSkuSpec": {
+              "type": "string",
+              "description": "Minimum SKU specification for evaluation in the format of 'cores|GPUs|RAM(GB)|VRAM(GB)'",
+              "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$",
+              "format": "sku-spec"
+            },
+            "evaluationRecommendedSku": {
+              "description": "Evaluation recommended sku",
+              "$ref": "#/definitions/skuListDefinition"
+            },
+            "finetuneMinSkuSpec": {
+              "type": "string",
+              "description": "Minimum SKU specification for finetune in the format of 'cores|GPUs|RAM(GB)|VRAM(GB)'",
+              "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$",
+              "format": "sku-spec"
+            },
+            "inferenceMinSkuSpec": {
+              "type": "string",
+              "description": "Minimum SKU specification for inference in the format of 'cores|GPUs|RAM(GB)|VRAM(GB)'",
+              "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"
+            }
+          }
+        },
+        "maasFinetuning": {
+          "type": "object",
+          "description": "Model-as-a-Service fine-tuning configuration",
+          "additionalProperties": false,
+          "properties": {
+            "maasFinetuningTest": {
+              "type": "boolean",
+              "description": "Used for testing maas-finetuning in INT"
+            },
+            "maasFinetuningDeployRegions": {
+              "type": "array",
+              "items": {
+                "type": "string",
+                "enum": [
+                  "eastus2",
+                  "eastus",
+                  "northcentralus",
+                  "westus3",
+                  "westus",
+                  "southcentralus"
+                ]
+              },
+              "description": "Regions where fine-tuned models can be deployed"
+            },
+            "maasFinetuningOffer": {
+              "type": "boolean",
+              "description": "Whether MaaS fine-tuning offer is available"
+            },
+            "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))*$"
+            }
+          }
+        },
+        "deprecation": {
+          "type": "object",
+          "description": "Deprecation and retirement schedule information",
+          "additionalProperties": false,
+          "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}$",
+              "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}$",
+              "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}$",
+              "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}$",
+              "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}$",
+              "format": "date-time"
+            }
+          }
+        }
+      }
+    },
     "version": {
       "description": "Model version",
       "oneOf": [{ "type": "string" }, { "type": "integer", "minimum": 1 }]
     }
   },
-  "required": ["$schema", "name", "path", "tags", "version"],
+  "required": ["$schema", "name", "path", "version"],
   "definitions": {
     "skuListDefinition": {
       "type": "string",
@@ -558,7 +1044,8 @@
         "speech-translation",
         "video-analysis",
         "video-generation",
-        "materials-design"
+        "materials-design",
+        "atomistic-modelling"
       ]
     },
     "finetuningTasksListDefinition": {
@@ -619,7 +1106,7 @@
     "featureFlagsListDefinition": {
       "type": "string",
       "description": "A comma-separated list of valid feature flags with no duplicates",
-      "pattern": "^(OpenAI\\.EAGatingTier|OpenAI\\.EnterpriseTier|OpenAI\\.1PGatingTier|OpenAI\\.OneVetGatingTier|OpenAI\\.1000CUGatingTier|OpenAI\\.computer-use-preview|OpenAI\\.Preview\\.202409|OpenAITest|AIServices\\.grok-4-code)(?:\\s*,\\s*(OpenAI\\.EAGatingTier|OpenAI\\.EnterpriseTier|OpenAI\\.1PGatingTier|OpenAI\\.OneVetGatingTier|OpenAI\\.1000CUGatingTier|OpenAI\\.computer-use-preview|OpenAI\\.Preview\\.202409|OpenAITest|AIServices\\.grok-4-code))*$"
+      "pattern": "^(OpenAI\\.EAGatingTier|OpenAI\\.EnterpriseTier|OpenAI\\.1PGatingTier|OpenAI\\.OneVetGatingTier|OpenAI\\.1000CUGatingTier|OpenAI\\.computer-use-preview|OpenAI\\.Preview\\.202409|OpenAITest|AIServices\\.grok-4-code|AIServices\\.grok-4)(?:\\s*,\\s*(OpenAI\\.EAGatingTier|OpenAI\\.EnterpriseTier|OpenAI\\.1PGatingTier|OpenAI\\.OneVetGatingTier|OpenAI\\.1000CUGatingTier|OpenAI\\.computer-use-preview|OpenAI\\.Preview\\.202409|OpenAITest|AIServices\\.grok-4-code|AIServices\\.grok-4))*$"
     }
   }
 }

{azureml_registry_tools-0.1.0a20 → azureml_registry_tools-0.1.0a25}/azureml/registry/mgmt/asset_management.py

@@ -3,6 +3,7 @@
 # ---------------------------------------------------------
 """Asset management commands for registry-mgmt CLI."""
 
+import requests
 import sys
 import shutil
 import tempfile
@@ -16,6 +17,7 @@ from azure.identity import DefaultAzureCredential
 
 from azureml.registry.data.validate_model_schema import validate_model_schema
 from azureml.registry.data.validate_model_variant_schema import validate_model_variant_schema
+from azureml.registry._rest_client.registry_management_client import RegistryManagementClient
 
 # Windows compatibility patch - must be applied before importing azureml.assets
 from subprocess import run
@@ -87,6 +89,52 @@ def validate_model(asset_path: Path, allow_additional_properties: bool = False)
     return True
 
 
+def put_system_metadata(ml_client: MLClient, asset: AssetConfig, registry: str, system_metadata: dict):
+    """PUT system metadata for an asset.
+
+    Args:
+        ml_client (MLClient): ML client.
+        asset (AssetConfig): Asset config.
+        registry (str): Name of registry.
+        system_metadata (dict): System metadata payload.
+    """
+    # Use RegistryManagementClient for discovery
+    registry_mgmt_client = RegistryManagementClient(registry_name=registry)
+    discovery = registry_mgmt_client.discovery()
+
+    # Extract the needed components
+    base_url = discovery.get('registryFqdns', {}).get(discovery.get('primaryRegion', '').lower(), {}).get('uri')
+    subscription_id = discovery.get('subscriptionId')
+    resource_group = discovery.get('resourceGroup')
+
+    # Build the URL
+    mms_id = f"{asset.name}:{asset.version}"
+    url = (f"{base_url}/modelregistry/v1.0/subscriptions/{subscription_id}/resourceGroups/{resource_group}/"
+           f"providers/Microsoft.MachineLearningServices/registries/{registry}/models/{mms_id}/systemMetadata")
+
+    # Get token and make request
+    access_token = ml_client._credential.get_token("https://ml.azure.com").token
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {access_token}"
+    }
+
+    print("Attempting to PUT system metadata")
+    print(f"Payload: {system_metadata}")
+
+    response = requests.put(url, headers=headers, json=system_metadata)
+
+    if response.status_code in [200, 201, 202]:
+        print("PUT request successful")
+        return response.json()
+    else:
+        print(f"PUT request failed: {url}")
+        print(f"Status Code: {response.status_code}")
+        print(f"Response: {response.reason}")
+        print(f"Content: {response.text}")
+        raise Exception(response.reason)
+
+
 def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, override_storage: bool = False) -> AssetConfig:
     """Build a mutable copy of the asset in a temporary directory.
 
@@ -97,6 +145,7 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, overrid
 
     Returns:
         AssetConfig: Mutable asset configuration object
+        system_metadata_payload (dict): Extracted system metadata from asset spec
     """
     common_dir, _ = util.find_common_directory(base_asset.release_paths)
 
@@ -114,20 +163,19 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, overrid
     spec_config_file = mutable_asset_dir / base_spec_file.relative_to(common_dir)
     model_config_file = mutable_asset_dir / base_model_file.relative_to(common_dir)
 
+    with open(spec_config_file, "r") as f:
+        spec_config = yaml.safe_load(f)
+
     # Override storage info for model card preview
     if override_storage:
         print("Model card preview - Default setting version and storage info")
 
         # Autoincrement version for mutable asset
-        with open(spec_config_file, "r") as f:
-            spec_config = yaml.safe_load(f)
-            spec_config["version"] = datetime.now().strftime("%Y%m%d%H%M%S")
+        spec_config["version"] = datetime.now().strftime("%Y%m%d%H%M%S")
 
         # Remove intellectualPropertyPublisher if it exists in properties
-        spec_config["properties"].pop("intellectualPropertyPublisher", None)
-
-        with open(spec_config_file, "w") as f:
-            yaml.dump(spec_config, f)
+        if "properties" in spec_config:
+            spec_config["properties"].pop("intellectualPropertyPublisher", None)
 
         # Create dummy file to upload to storage
         with open(mutable_asset_dir / "dummy.txt", "w") as f:
@@ -147,9 +195,23 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, overrid
         with open(model_config_file, "w") as f:
             yaml.dump(new_model_config, f)
 
+    # Extract system_metadata from model spec
+    # PUT request will be made after asset creation while the SDK update/PATCH logic is being updated
+    system_metadata_payload = spec_config.pop("system_metadata", {})
+    if system_metadata_payload:
+        print(f"system_metadata found in model spec. Extracted system_metadata and saved "
+              f"to system_metadata: {system_metadata_payload}")
+    else:
+        print(f"No system_metadata found in model spec. Using system metadata payload {system_metadata_payload}")
+
+    # Write updated spec config back to the file
+    if override_storage or system_metadata_payload:
+        with open(spec_config_file, "w") as f:
+            yaml.dump(spec_config, f)
+
     mutable_asset = AssetConfig(asset_config_file)
 
-    return mutable_asset
+    return mutable_asset, system_metadata_payload
 
 
 def create_or_update_asset(readonly_asset: AssetConfig, registry_name: str, subscription_id: str, resource_group: str, override_storage: bool = False):
@@ -174,7 +236,7 @@ def create_or_update_asset(readonly_asset: AssetConfig, registry_name: str, subs
     )
 
     with tempfile.TemporaryDirectory() as mutable_asset_dir:
-        mutable_asset = build_mutable_asset(base_asset=readonly_asset, mutable_asset_dir=mutable_asset_dir, override_storage=override_storage)
+        mutable_asset, system_metadata_payload = build_mutable_asset(base_asset=readonly_asset, mutable_asset_dir=mutable_asset_dir, override_storage=override_storage)
         try:
             success = create_asset(mutable_asset, registry_name, ml_client)
             if not success:
@@ -183,6 +245,14 @@ def create_or_update_asset(readonly_asset: AssetConfig, registry_name: str, subs
             print(f"Failed to create/update asset: {e}")
             raise
 
+        # PUT system metadata after model creation
+        if success:
+            try:
+                put_system_metadata(ml_client, mutable_asset, registry_name, system_metadata_payload)
+            except Exception as e:
+                print(f"Failed to update {mutable_asset.name} with system metadata: {e}")
+                raise
+
         print("\n[VALIDATE YOUR ASSET IN THE UI HERE]")
         print(f" - Model Catalog link: https://ai.azure.com/explore/models/{mutable_asset.name}/version/{mutable_asset.version}/registry/{registry_name}")
         print(f" - Azure Portal link: https://ml.azure.com/registries/{registry_name}/models/{mutable_asset.name}/version/{mutable_asset.version}")

{azureml_registry_tools-0.1.0a20 → azureml_registry_tools-0.1.0a25}/azureml/registry/tools/create_or_update_assets.py

@@ -275,7 +275,7 @@ def create_or_update_assets(input_dir: Path,
         subscription_id=repo2registry_config.subscription_id,
         resource_group_name=repo2registry_config.resource_group,
         registry_name=repo2registry_config.registry_name,
-        credential=DefaultAzureCredential(),
+        credential=DefaultAzureCredential(),  # CodeQL [SM05139] DefaultAzureCredential should only be used for local development and testing purposes.
     )
 
     results_dict = {}

{azureml_registry_tools-0.1.0a20 → azureml_registry_tools-0.1.0a25/azureml_registry_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a20
+Version: 0.1.0a25
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license

{azureml_registry_tools-0.1.0a20 → azureml_registry_tools-0.1.0a25}/setup.py

@@ -18,7 +18,7 @@ exclude_list = ["*.tests"]
 
 setup(
     name='azureml-registry-tools',
-    version="0.1.0a20",
+    version="0.1.0a25",
     description='AzureML Registry tools and CLI',
     author='Microsoft Corp',
     license="https://aka.ms/azureml-sdk-license",