fix: Fixing tag generation

Author: chrstfu

src/sagemaker/jumpstart/hub/interfaces.py

@@ -457,6 +457,7 @@ class HubModelDocument(HubDataHolderType):
         "url",
         "min_sdk_version",
         "training_supported",
+        "capabilities",
         "incremental_training_supported",
         "dynamic_container_deployment_supported",
         "hosting_ecr_uri",
@@ -560,6 +561,7 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:
             JumpStartEnvironmentVariable(env_variable, is_hub_content=True)
             for env_variable in json_obj.get("InferenceEnvironmentVariables", [])
         ]
+        self.capabilities: Optional[List[str]] = json_obj.get("Capabilities")
         self.training_supported: bool = bool(json_obj.get("TrainingSupported"))
         self.incremental_training_supported: bool = bool(json_obj.get("IncrementalTrainingSupported"))
         self.dynamic_container_deployment_supported: Optional[bool] = (

src/sagemaker/jumpstart/hub/parsers.py

@@ -137,6 +137,7 @@ def make_model_specs_from_describe_hub_content_response(
     hub_model_document: HubModelDocument = response.hub_content_document
     specs["url"] = hub_model_document.url
     specs["min_sdk_version"] = hub_model_document.min_sdk_version
+    specs["capabilities"] = hub_model_document.capabilities
     specs["training_supported"] = bool(hub_model_document.training_supported)
     specs["incremental_training_supported"] = bool(
         hub_model_document.incremental_training_supported