From e1fa482f9790eb4c3669d926d66b058a29283cab Mon Sep 17 00:00:00 2001 From: "David S. Batista" Date: Fri, 16 Feb 2024 18:03:59 +0100 Subject: [PATCH] attending PR comments --- .../components/generators/amazon_sagemaker/sagemaker.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/integrations/amazon_sagemaker/src/haystack_integrations/components/generators/amazon_sagemaker/sagemaker.py b/integrations/amazon_sagemaker/src/haystack_integrations/components/generators/amazon_sagemaker/sagemaker.py index 39e347f86..4eacf33a4 100644 --- a/integrations/amazon_sagemaker/src/haystack_integrations/components/generators/amazon_sagemaker/sagemaker.py +++ b/integrations/amazon_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.