chore: improve vulnerable/deprecated error message

Author: evakravi

src/sagemaker/jumpstart/exceptions.py

@@ -39,6 +39,12 @@
     "Note that models may have different input/output signatures after a major version upgrade."
 )
 
+_VULNERABLE_DEPRECATED_ERROR_RECOMMENDATION = (
+    "We recommend that you specify a more recent "
+    "model version or choose a different model. To access the latest models "
+    "and model versions, be sure to upgrade to the latest version of the SageMaker Python SDK."
+)
+
 
 def get_wildcard_model_version_msg(
     model_id: str, wildcard_model_version: str, full_model_version: str
@@ -115,16 +121,15 @@ def __init__(
                 self.message = (
                     f"Version '{version}' of JumpStart model '{model_id}' "  # type: ignore
                     "has at least 1 vulnerable dependency in the inference script. "
-                    "Please try targeting a higher version of the model or using a "
-                    "different model. List of vulnerabilities: "
+                    f"{_VULNERABLE_DEPRECATED_ERROR_RECOMMENDATION} "
+                    "List of vulnerabilities: "
                     f"{', '.join(vulnerabilities)}"  # type: ignore
                 )
             elif scope == JumpStartScriptScope.TRAINING:
                 self.message = (
                     f"Version '{version}' of JumpStart model '{model_id}' "  # type: ignore
                     "has at least 1 vulnerable dependency in the training script. "
-                    "Please try targeting a higher version of the model, upgrading "
-                    "SageMaker Python SDK version or using a different model. "
+                    f"{_VULNERABLE_DEPRECATED_ERROR_RECOMMENDATION} "
                     "List of vulnerabilities: "
                     f"{', '.join(vulnerabilities)}"  # type: ignore
                 )
@@ -160,8 +165,7 @@ def __init__(
                 raise RuntimeError("Must specify `model_id` and `version` arguments.")
             self.message = (
                 f"Version '{version}' of JumpStart model '{model_id}' is deprecated. "
-                "Please try targeting a higher version of the model, upgrading "
-                "SageMaker Python SDK version or using a different model."
+                f"{_VULNERABLE_DEPRECATED_ERROR_RECOMMENDATION}"
             )
 
         super().__init__(self.message)

tests/unit/sagemaker/jumpstart/test_utils.py

@@ -1052,7 +1052,10 @@ def make_vulnerable_inference_spec(*largs, **kwargs):
     assert (
         "Version '*' of JumpStart model 'pytorch-eqa-bert-base-cased' has at least 1 "
         "vulnerable dependency in the inference script. "
-        "Please try targeting a higher version of the model or using a different model. "
+        "We recommend that you specify a more recent model version or "
+        "choose a different model. To access the "
+        "latest models and model versions, be sure to upgrade "
+        "to the latest version of the SageMaker Python SDK. "
         "List of vulnerabilities: some, vulnerability"
     ) == str(e.value.message)
 
@@ -1074,8 +1077,10 @@ def make_vulnerable_training_spec(*largs, **kwargs):
     assert (
         "Version '*' of JumpStart model 'pytorch-eqa-bert-base-cased' has at least 1 "
         "vulnerable dependency in the training script. "
-        "Please try targeting a higher version of the model, "
-        "upgrading SageMaker Python SDK version or using a different model. "
+        "We recommend that you specify a more recent model version or "
+        "choose a different model. To access the "
+        "latest models and model versions, be sure to upgrade "
+        "to the latest version of the SageMaker Python SDK. "
         "List of vulnerabilities: some, vulnerability"
     ) == str(e.value.message)