Add secret_key to endpoint mode

Author: Bryannah Hernandez

src/sagemaker/serve/mode/sagemaker_endpoint_mode.py

@@ -110,6 +110,7 @@ def prepare(
                 model_path=model_path,
                 sagemaker_session=sagemaker_session,
                 s3_model_data_url=s3_model_data_url,
+                secret_key=secret_key,
                 image=image,
             )
 

src/sagemaker/serve/model_server/multi_model_server/server.py

@@ -92,6 +92,7 @@ class SageMakerMultiModelServer:
     def _upload_server_artifacts(
         self,
         model_path: str,
+        secret_key: str,
         sagemaker_session: Session,
         s3_model_data_url: str = None,
         image: str = None,
@@ -132,6 +133,7 @@ def _upload_server_artifacts(
         env_vars = {
             "SAGEMAKER_SUBMIT_DIRECTORY": "/opt/ml/model/code",
             "SAGEMAKER_PROGRAM": "inference.py",
+            "SAGEMAKER_SERVE_SECRET_KEY": secret_key,
             "SAGEMAKER_REGION": sagemaker_session.boto_region_name,
             "SAGEMAKER_CONTAINER_LOG_LEVEL": "10",
             "LOCAL_PYTHON": platform.python_version(),