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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a20
+Version: 0.1.0a23
 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.0a23}/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.0a23}/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.0a23}/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",
@@ -265,7 +275,7 @@
         "keywords": {
           "description": "List of key words as a comma-separated string. Max 3 keywords",
           "type": "string",
-          "pattern": "^(Agents|Audio|Coding|Conversation|Instruction|Large context|Low latency|Multilingual|Multimodal|Multipurpose|RAG|Reasoning|Summarization|Translation|Understanding|Vision|Atomistic generation)(?:\\s*,\\s*(Agents|Audio|Coding|Conversation|Instruction|Large context|Low latency|Multilingual|Multimodal|Multipurpose|RAG|Reasoning|Summarization|Translation|Understanding|Vision|Atomistic generation)){0,2}$"
+          "pattern": "^(Agents|Audio|Coding|Conversation|Instruction|Large context|Low latency|Multilingual|Multimodal|Multipurpose|RAG|Reasoning|Summarization|Translation|Understanding|Vision|Atomistic generation|Chat|Hybrid Reasoning|Vision Language Model)(?:\\s*,\\s*(Agents|Audio|Coding|Conversation|Instruction|Large context|Low latency|Multilingual|Multimodal|Multipurpose|RAG|Reasoning|Summarization|Translation|Understanding|Vision|Atomistic generation|Chat|Hybrid Reasoning|Vision Language Model)){0,2}$"
         },
         "languages": {
           "description": "Comma-separated list of supported languages (ISO 639-1 language codes)",
@@ -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|explanation-and-chain-of-thought-reasoning|video-analytics-and-spatial-temporal-awareness)(?:\\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|explanation-and-chain-of-thought-reasoning|video-analytics-and-spatial-temporal-awareness))*$"
         },
         "textContextWindow": {
           "description": "Context window size",
@@ -558,7 +568,8 @@
         "speech-translation",
         "video-analysis",
         "video-generation",
-        "materials-design"
+        "materials-design",
+        "atomistic-modelling"
       ]
     },
     "finetuningTasksListDefinition": {
@@ -619,7 +630,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.0a23}/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.0a23/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.0a23
 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.0a23}/setup.py

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