doc update

Author: Captainia

src/sagemaker/jumpstart/cache.py

@@ -319,7 +319,7 @@ def _get_open_weight_manifest_key_from_model_id(
         key: JumpStartVersionedModelId,
         value: Optional[JumpStartVersionedModelId],  # pylint: disable=W0613
     ) -> JumpStartVersionedModelId:
-        """For open weights models, retrieve model manifest key for open source model.
+        """For open weights models, retrieve model manifest key for open weight model.
 
         Filters models list by supported versions.
         """

src/sagemaker/jumpstart/enums.py

@@ -37,8 +37,10 @@ class ModelFramework(str, Enum):
 class JumpStartModelType(str, Enum):
     """Enum class for JumpStart model type.
 
-    Open source model refers to JumpStart owned community models.
-    Proprietary model refers to external provider owned Marketplace models.
+    OPEN_WEIGHTS: Publicly available models have open weights
+    and are onboarded and maintained by JumpStart.
+    PROPRIETARY: Proprietary models from third-party providers do not have open weights.
+    You must subscribe to proprietary models in AWS Marketplace before use.
     """
 
     OPEN_WEIGHTS = "open_weights"

src/sagemaker/jumpstart/exceptions.py

@@ -71,7 +71,7 @@ def get_proprietary_model_subscription_msg(
 
     return (
         f"INFO: Using proprietary model '{model_id}'. "
-        f"Please make sure to subscribe to the model on {subscription_link}"
+        f"To subscribe to this model in AWS Marketplace, see {subscription_link}"
     )
 
 
@@ -215,7 +215,7 @@ class MarketplaceModelSubscriptionError(ValueError):
     """Exception raised when trying to deploy a JumpStart Marketplace model.
 
     A caller is required to subscribe to the Marketplace product in order to deploy.
-    This exception is raised when a caller tries to deploy a JumpStart Marketplace
+    This exception is raised when a caller tries to deploy a JumpStart Marketplace model
     but the caller is not subscribed to the model.
     """
 
@@ -227,8 +227,11 @@ def __init__(
         if message:
             self.message = message
         else:
-            self.message = "You have not subscribed to this Marketplace model. "
+            self.message = (
+                "To use a proprietary JumpStart model, "
+                "you must first subscribe to the model in AWS Marketplace. "
+            )
             if model_subscription_link:
-                self.message += f"Please subscribe following this link {model_subscription_link}"
+                self.message += f"To subscribe to this model, see {model_subscription_link}"
 
         super().__init__(self.message)