attending PR comments

deepset-ai · Feb 16, 2024 · e1fa482 · e1fa482
1 parent 0a6b741
commit e1fa482
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...n_sagemaker/src/haystack_integrations/components/generators/amazon_sagemaker/sagemaker.py b/...n_sagemaker/src/haystack_integrations/components/generators/amazon_sagemaker/sagemaker.py
@@ -63,6 +63,11 @@ def __init__(
         """
         Instantiates the session with SageMaker.
 
+        :param aws_access_key_id: The `Secret` for AWS access key ID.
+        :param aws_secret_access_key: The `Secret` for AWS secret access key.
+        :param aws_session_token: The `Secret` for AWS session token.
+        :param aws_region_name: The `Secret` for AWS region name. If not provided, the default region will be used.
+        :param aws_profile_name: The `Secret` for AWS profile name. If not provided, the default profile will be used.
         :param model: The name for SageMaker Model Endpoint.
         :param aws_custom_attributes: Custom attributes to be passed to SageMaker, for example `{"accept_eula": True}`
             in case of Llama-2 models.