chore: address PR comments

Author: evakravi

src/sagemaker/jumpstart/cache.py

@@ -293,8 +293,8 @@ def _model_id_retrieval_function(
             raise KeyError(error_msg)
 
         error_msg = f"Unable to find model manifest for '{model_id}' with version '{version}'. "
-        error_msg += f"Visit {MODEL_ID_LIST_WEB_URL} for updated list of models, "
-        error_msg += "or try another AWS region. "
+        error_msg += "Specify a different model ID or try a different AWS Region. "
+        error_msg += f"For a list of available models, see {MODEL_ID_LIST_WEB_URL}. "
 
         other_model_id_version = None
         if model_type == JumpStartModelType.OPEN_WEIGHTS:

src/sagemaker/jumpstart/exceptions.py

@@ -32,8 +32,8 @@
 )
 
 INVALID_MODEL_ID_ERROR_MSG = (
-    "Invalid model ID: '{model_id}'. Please visit "
-    f"{MODEL_ID_LIST_WEB_URL} for a list of valid model IDs, or try another AWS region. "
+    "Invalid model ID: '{model_id}'. Specify a different model ID or try a different AWS Region. "
+    f"For a list of available models, see {MODEL_ID_LIST_WEB_URL}. "
     "The module `sagemaker.jumpstart.notebook_utils` contains utilities for "
     "fetching model IDs. We recommend upgrading to the latest version of sagemaker "
     "to get access to the most models."

tests/unit/sagemaker/jumpstart/test_cache.py

@@ -205,8 +205,10 @@ def test_jumpstart_cache_get_header():
         )
     assert (
         "Unable to find model manifest for 'pytorch-ic-imagenet-inception-v3-classification-4' with "
-        "version '3.*'. Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html "
-        "for updated list of models, or try another AWS region. Consider using model ID "
+        "version '3.*'. Specify a different model ID or try a different AWS Region. "
+        "For a list of available models, see "
+        "https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html. "
+        "Consider using model ID "
         "'pytorch-ic-imagenet-inception-v3-"
         "classification-4' with version '2.0.0'."
     ) in str(e.value)
@@ -215,17 +217,19 @@ def test_jumpstart_cache_get_header():
         cache.get_header(model_id="pytorch-ic-", semantic_version_str="*")
     assert (
         "Unable to find model manifest for 'pytorch-ic-' with version '*'. "
-        "Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html "
-        "for updated list of models, or try another AWS region. "
+        "Specify a different model ID or try a different AWS Region. "
+        "For a list of available models, see "
+        "https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html. "
         "Did you mean to use model ID 'pytorch-ic-imagenet-inception-v3-classification-4'?"
     ) in str(e.value)
 
     with pytest.raises(KeyError) as e:
         cache.get_header(model_id="tensorflow-ic-", semantic_version_str="*")
     assert (
         "Unable to find model manifest for 'tensorflow-ic-' with version '*'. "
-        "Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html "
-        "for updated list of models, or try another AWS region. "
+        "Specify a different model ID or try a different AWS Region. For a list "
+        "of available models, see "
+        "https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html. "
         "Did you mean to use model ID 'tensorflow-ic-imagenet-inception-"
         "v3-classification-4'?"
     ) in str(e.value)
@@ -238,8 +242,9 @@ def test_jumpstart_cache_get_header():
         )
     assert (
         "Unable to find model manifest for 'ai21-summarize' with version '1.1.003'. "
-        "Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html "
-        "for updated list of models, or try another AWS region. "
+        "Specify a different model ID or try a different AWS Region. "
+        "For a list of available models, see "
+        "https://sagemaker.readthedocs.io/en/stable/doc_utils/pretrainedmodels.html. "
         "Did you mean to use model ID 'ai21-summarization'?"
     ) in str(e.value)