azureml-registry-tools 0.1.0a26__tar.gz → 0.1.0a28__tar.gz

{azureml_registry_tools-0.1.0a26/azureml_registry_tools.egg-info → azureml_registry_tools-0.1.0a28}/PKG-INFO

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

{azureml_registry_tools-0.1.0a26 → azureml_registry_tools-0.1.0a28}/azureml/registry/data/model.schema.json

@@ -272,6 +272,11 @@
           "type": "string",
           "enum": ["true"]
         },
+        "isCustomerInfoRequired": {
+          "description": "This will trigger a customer info form to be shown to the user before they can access the model",
+          "type": "string",
+          "enum": ["true", "false"]
+        },
         "keywords": {
           "description": "List of key words as a comma-separated string. Max 3 keywords",
           "type": "string",
@@ -299,6 +304,11 @@
           "type": "string",
           "format": "long-string"
         },
+        "fineTuningMethods": {
+          "description": "Fine-tuning strategy applied to customize the model behavior",
+          "type": "string",
+          "pattern": "^(Supervised|Direct Preference Optimization|Reinforcement)(?:\\s*,\\s*(Supervised|Direct Preference Optimization|Reinforcement))*$"
+        },
         "maas-finetuning": {
           "description": "Whether MaaS fine-tuning is enabled. True means enabled, if disabled remove the key",
           "type": "string",
@@ -351,7 +361,7 @@
         "modelCapabilities": {
           "description": "Model capabilities (e.g., agents, assistants) as a comma-separated string",
           "type": "string",
-          "pattern": "^(agents|assistants|routing|reasoning|streaming|tool-calling)(?:\\s*,\\s*(agents|assistants|routing|reasoning|streaming|tool-calling))*$"
+          "pattern": "^(agents|agentsV2|assistants|routing|reasoning|streaming|tool-calling)(?:\\s*,\\s*(agents|agentsV2|assistants|routing|reasoning|streaming|tool-calling))*$"
         },
         "modelHash": {
           "description": "Hash of the model",
@@ -385,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)(?:\\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))*$"
+          "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)(?:\\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))*$"
         },
         "textContextWindow": {
           "description": "Context window size",
@@ -583,6 +593,7 @@
             "type": "string",
             "enum": [
               "agents",
+              "agentsV2",
               "assistants",
               "routing",
               "reasoning",

{azureml_registry_tools-0.1.0a26 → azureml_registry_tools-0.1.0a28}/azureml/registry/mgmt/asset_management.py

@@ -18,6 +18,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
+from azureml.registry.mgmt.util import resolve_from_file_for_asset
 
 # Windows compatibility patch - must be applied before importing azureml.assets
 from subprocess import run
@@ -98,6 +99,9 @@ def put_system_metadata(ml_client: MLClient, asset: AssetConfig, registry: str,
         registry (str): Name of registry.
         system_metadata (dict): System metadata payload.
     """
+    # First, transform system metadata such that any files are read and their content used
+    system_metadata = {k: resolve_from_file_for_asset(asset, v) for k, v in system_metadata.items()}
+
     # Use RegistryManagementClient for discovery
     registry_mgmt_client = RegistryManagementClient(registry_name=registry)
     discovery = registry_mgmt_client.discovery()
@@ -120,7 +124,7 @@ def put_system_metadata(ml_client: MLClient, asset: AssetConfig, registry: str,
     }
 
     print("Attempting to PUT system metadata")
-    print(f"Payload: {system_metadata}")
+    print(f"System Metadata Payload: {system_metadata}")
 
     response = requests.put(url, headers=headers, json=system_metadata)
 

azureml_registry_tools-0.1.0a28/azureml/registry/mgmt/util.py

@@ -0,0 +1,75 @@
+"""File resolution utilities for asset management."""
+
+import os
+from pathlib import Path, PurePath
+from typing import Tuple, Union, Any
+
+import azureml.assets as assets  # noqa: E402
+
+
+def is_file_relative_to_asset_path(asset: assets.AssetConfig, value: Any) -> bool:
+    """Check if the value is a file with respect to the asset path.
+
+    Args:
+        asset (AssetConfig): the asset to try and resolve the value for
+        value: value to check
+
+    Returns:
+        bool: True if value represents a file relative to asset path, False otherwise
+    """
+    if not isinstance(value, str) and not isinstance(value, PurePath):
+        return False
+
+    path_value = value if isinstance(value, Path) else Path(value)
+
+    if not path_value.is_relative_to(asset.file_path):
+        path_value = asset._append_to_file_path(path_value)
+
+    return os.path.isfile(path_value)
+
+
+def resolve_from_file_for_asset(asset: assets.AssetConfig, value: Any) -> Any:
+    """Resolve the value from a file for an asset if it is a file, otherwise returns the value.
+
+    Args:
+        asset (AssetConfig): the asset to try and resolve the value for
+        value: value to try and resolve
+
+    Returns:
+        Any: File content if value is a file path relative to asset, otherwise the original value
+    """
+    if not is_file_relative_to_asset_path(asset, value):
+        return value
+
+    path_value = value if isinstance(value, Path) else Path(value)
+
+    if not path_value.is_relative_to(asset.file_path):
+        path_value = asset._append_to_file_path(path_value)
+
+    is_resolved_from_file, resolved_value = _resolve_from_file(path_value)
+
+    if is_resolved_from_file:
+        return resolved_value
+    else:
+        return value
+
+
+def _resolve_from_file(value: Union[str, Path]) -> Tuple[bool, Union[str, None]]:
+    """Resolve file content (internal helper).
+
+    Args:
+        value: File path to resolve
+
+    Returns:
+        Tuple[bool, Union[str, None]]: (success, content) where success indicates
+                                       if file was read and content is the file content
+    """
+    if os.path.isfile(value):
+        try:
+            with open(value, 'r') as f:
+                content = f.read()
+                return (True, content)
+        except Exception as e:
+            raise Exception(f"Failed to read file {value}: {e}")
+    else:
+        return (False, None)

{azureml_registry_tools-0.1.0a26 → azureml_registry_tools-0.1.0a28/azureml_registry_tools.egg-info}/PKG-INFO

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

{azureml_registry_tools-0.1.0a26 → azureml_registry_tools-0.1.0a28}/azureml_registry_tools.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ azureml/registry/mgmt/create_manifest.py
 azureml/registry/mgmt/create_model_spec.py
 azureml/registry/mgmt/model_management.py
 azureml/registry/mgmt/syndication_manifest.py
+azureml/registry/mgmt/util.py
 azureml/registry/tools/__init__.py
 azureml/registry/tools/config.py
 azureml/registry/tools/create_or_update_assets.py

{azureml_registry_tools-0.1.0a26 → azureml_registry_tools-0.1.0a28}/setup.py

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