From 9e8de8f1d36010cc930702ccf9f3e56b1dcd7533 Mon Sep 17 00:00:00 2001 From: agnieszka-m Date: Mon, 5 Aug 2024 13:56:27 +0200 Subject: [PATCH] Fix formatting --- .../generators/amazon_bedrock/chat/chat_generator.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py b/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py index 24eb7579c..c6ee14b12 100644 --- a/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py +++ b/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py @@ -55,9 +55,7 @@ class AmazonBedrockChatGenerator: If the AWS environment is not configured, set `aws_access_key_id`, `aws_secret_access_key`, and `aws_region_name` as environment variables or pass them as [Secret](https://docs.haystack.deepset.ai/v2.0/docs/secret-management) arguments. Make sure the region you set - supports Amazon Bedrock. - - + supports Amazon Bedrock. """ SUPPORTED_MODEL_PATTERNS: ClassVar[Dict[str, Type[BedrockModelChatAdapter]]] = { @@ -98,11 +96,13 @@ def __init__( :param aws_region_name: AWS region name. Make sure the region you set supports Amazon Bedrock. :param aws_profile_name: AWS profile name. :param generation_kwargs: Keyword arguments sent to the model. These - parameters are specific to a model. You can find them in the model's documentation. For example, you can find the + parameters are specific to a model. You can find them in the model's documentation. + For example, you can find the Anthropic Claude generation parameters in [Anthropic documentation](https://docs.anthropic.com/claude/reference/complete_post). :param stop_words: A list of stop words that stop the model from generating more text when encountered. You can provide them using - this parameter or using the model's `generation_kwargs` under a model's specific key for stop words. For example, you can provide + this parameter or using the model's `generation_kwargs` under a model's specific key for stop words. + For example, you can provide stop words for Anthropic Claude in the `stop_sequences` key. :param streaming_callback: A callback function called when a new token is received from the stream. By default, the model is not set up for streaming. To enable streaming, set this parameter to a callback