azureml-registry-tools 0.1.0a20__py3-none-any.whl → 0.1.0a23__py3-none-any.whl

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/_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/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/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.dist-info → azureml_registry_tools-0.1.0a23.dist-info}/METADATA

@@ -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.dist-info → azureml_registry_tools-0.1.0a23.dist-info}/RECORD

@@ -5,15 +5,15 @@ azureml/registry/_cli/registry_syndication_cli.py,sha256=7LyyBDC5OnJu9OMBAl_L_SW
 azureml/registry/_cli/repo2registry_cli.py,sha256=4CDv4tYWLTjVgdXIcoKA9iu_gOv_Z_xn05wSANkdmdg,5378
 azureml/registry/_rest_client/__init__.py,sha256=Eh0vB8AVlwkZlHLO4DCGHyyfRYYgeTeVjiUAX_WujG0,219
 azureml/registry/_rest_client/arm_client.py,sha256=KVSj0V1bN-vCUV3fBbIdZDpWTLKYT6qdWIZtDjx2la4,4516
-azureml/registry/_rest_client/base_rest_client.py,sha256=JcmhckfR-ZdUtEh0EeQIKfe63Zjowa1fLzmVeWpeLno,8173
-azureml/registry/_rest_client/registry_management_client.py,sha256=3-PBwj91lkG1yrlB-s7jaYj0w5j177oSHZHvcoO8icQ,4984
+azureml/registry/_rest_client/base_rest_client.py,sha256=WnjWhOQvigLIBPp2h1eU5IZ75aj0B2ZhCqa-z71bMiY,8280
+azureml/registry/_rest_client/registry_management_client.py,sha256=NsqWRaDdOlPyICNWj74eLVuofVl04RIc6RgIgyrEROA,5113
 azureml/registry/_rest_client/registry_model_client.py,sha256=LqJGTuYQtBnHSeSbOl5KVbiO6vGBkKQ7HD5jc4fvV3k,14466
 azureml/registry/data/__init__.py,sha256=cW3X6HATz6XF-K_7uKdybTbJb9EZSecBN4J27NGdZmU,231
 azureml/registry/data/asset.yaml.template,sha256=WTgfuvKEBp-EVFSQ0JpU0h4z_ULJdULO9kHmB_9Fr1o,96
 azureml/registry/data/description.md.template,sha256=wQLk54U8hoXU1y9235R4irc6FGYPXGO-x9EHUorP15Q,84
 azureml/registry/data/evaluation.md.template,sha256=JaDecIfLV9vZDUrZzVRPzVHHnKD-BQGBgQ-cw1dHavU,277
 azureml/registry/data/model-variant.schema.json,sha256=AT4Dy6cCtp_SFUfSqYIqcER8AldpYm0QIEy1abY3QWE,1699
-azureml/registry/data/model.schema.json,sha256=ZezvItOdGhwPb8drJAxJ6XGKJiOeWQS2W7NWhjQhLu4,28031
+azureml/registry/data/model.schema.json,sha256=j6orgqiVNpQYbvrphBZDLtWxslTXUy0VrTnC6Int200,29192
 azureml/registry/data/model.yaml.template,sha256=h5uqAN22FLaWrbPxIb8yVKH9cGDBrIwooXYYfsKhxDw,245
 azureml/registry/data/notes.md.template,sha256=zSRyOR__9NGL2j0tugY7HgFkwkAdcE2pJyyyGsz1SAk,248
 azureml/registry/data/validate_model_schema.py,sha256=OQp2E01kdxSphvUQYQvelSiD24-qUG6nTFuzW60wX2c,8322
@@ -27,12 +27,12 @@ azureml/registry/mgmt/model_management.py,sha256=STTr_uvdPKV2NaJ5UvS5aMi3yejVF6H
 azureml/registry/mgmt/syndication_manifest.py,sha256=8Sfd49QuCA5en5_mIOLE21kZVpnReUXowx_g0TVRgWg,9025
 azureml/registry/tools/__init__.py,sha256=IAuWWpGfZm__pAkBIxmpJz84QskpkxBr0yDk1TUSnkE,223
 azureml/registry/tools/config.py,sha256=tjPaoBsWtPXBL8Ww1hcJtsr2SuIjPKt79dR8iovcebg,3639
-azureml/registry/tools/create_or_update_assets.py,sha256=Q-_BV7KWn1huQn5JriKT_8xJNoQQ_HK5wCftrq9DepA,15988
+azureml/registry/tools/create_or_update_assets.py,sha256=7LcuBzwU-HNE9ADG2igFXI696aKR028yTYxMuDtjVmA,16095
 azureml/registry/tools/registry_utils.py,sha256=zgYlCiOONtQJ4yZ9wg8tKVoE8dh6rrjB8hYBGhpV9-0,1403
 azureml/registry/tools/repo2registry_config.py,sha256=eXp_tU8Jyi30g8xGf7wbpLgKEPpieohBANKxMSLzq7s,4873
-azureml_registry_tools-0.1.0a20.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
-azureml_registry_tools-0.1.0a20.dist-info/METADATA,sha256=uwEFH4N18_NWol3U6TiBM4JSoWimmf2sVgSKLdfZWUs,522
-azureml_registry_tools-0.1.0a20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-azureml_registry_tools-0.1.0a20.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
-azureml_registry_tools-0.1.0a20.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
-azureml_registry_tools-0.1.0a20.dist-info/RECORD,,
+azureml_registry_tools-0.1.0a23.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
+azureml_registry_tools-0.1.0a23.dist-info/METADATA,sha256=HprALdWRfc7GdfU8IM8xflcYVGWfHauAksUtMqQd4Kk,522
+azureml_registry_tools-0.1.0a23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+azureml_registry_tools-0.1.0a23.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
+azureml_registry_tools-0.1.0a23.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
+azureml_registry_tools-0.1.0a23.dist-info/RECORD,,