feat: Add curatedHub telemetry support

Author: Malav Shastri

src/sagemaker/jumpstart/factory/estimator.py

@@ -432,7 +432,7 @@ def _add_sagemaker_session_to_kwargs(kwargs: JumpStartKwargs) -> JumpStartKwargs
     kwargs.sagemaker_session = (
         kwargs.sagemaker_session
         or get_default_jumpstart_session_with_user_agent_suffix(
-            kwargs.model_id, kwargs.model_version
+            kwargs.model_id, kwargs.model_version, kwargs.hub_arn
         )
     )
     return kwargs

src/sagemaker/jumpstart/factory/model.py

@@ -152,12 +152,14 @@ def _add_sagemaker_session_to_kwargs(
     kwargs: Union[JumpStartModelInitKwargs, JumpStartModelDeployKwargs]
 ) -> JumpStartModelInitKwargs:
     """Sets session in kwargs based on default or override, returns full kwargs."""
+
     kwargs.sagemaker_session = (
         kwargs.sagemaker_session
         or get_default_jumpstart_session_with_user_agent_suffix(
-            kwargs.model_id, kwargs.model_version
+            kwargs.model_id, kwargs.model_version, kwargs.hub_arn
         )
     )
+
     return kwargs
 
 

src/sagemaker/jumpstart/utils.py

@@ -1012,24 +1012,25 @@ def get_jumpstart_configs(
     )
 
 
-def get_jumpstart_user_agent_extra_suffix(model_id: str, model_version: str) -> str:
+def get_jumpstart_user_agent_extra_suffix(model_id: Optional[str], model_version: Optional[str], is_hub_content: Optional[bool]) -> str:
     """Returns the model-specific user agent string to be added to requests."""
     sagemaker_python_sdk_headers = get_user_agent_extra_suffix()
     jumpstart_specific_suffix = f"md/js_model_id#{model_id} md/js_model_ver#{model_version}"
+    hub_specific_suffix = f"md/js_is_hub_content#{is_hub_content}"
     return (
         sagemaker_python_sdk_headers
         if os.getenv(constants.ENV_VARIABLE_DISABLE_JUMPSTART_TELEMETRY, None)
-        else f"{sagemaker_python_sdk_headers} {jumpstart_specific_suffix}"
+        else f"{sagemaker_python_sdk_headers} {jumpstart_specific_suffix} {hub_specific_suffix}"
     )
 
 
 def get_default_jumpstart_session_with_user_agent_suffix(
-    model_id: str, model_version: str
+    model_id: Optional[str] = None, model_version: Optional[str] = None, is_hub_content: Optional[bool] = False
 ) -> Session:
     """Returns default JumpStart SageMaker Session with model-specific user agent suffix."""
     botocore_session = botocore.session.get_session()
     botocore_config = botocore.config.Config(
-        user_agent_extra=get_jumpstart_user_agent_extra_suffix(model_id, model_version),
+        user_agent_extra=get_jumpstart_user_agent_extra_suffix(model_id, model_version, is_hub_content),
     )
     botocore_session.set_default_client_config(botocore_config)
     # shallow copy to not affect default session constant