nit

Author: abrahamy

integrations/amazon_bedrock/tests/test_chat_generator.py

@@ -13,6 +13,7 @@
 
 from haystack_integrations.common.amazon_bedrock.errors import AmazonBedrockInferenceError
 from haystack_integrations.components.generators.amazon_bedrock import AmazonBedrockChatGenerator
+from haystack_integrations.components.generators.amazon_bedrock.chat.chat_generator import MAX_RETRIES
 
 CLASS_TYPE = "haystack_integrations.components.generators.amazon_bedrock.chat.chat_generator.AmazonBedrockChatGenerator"
 MODELS_TO_TEST = [
@@ -1267,8 +1268,6 @@ def test_run_raises_on_non_throttling_client_error(self, mock_boto3_session, set
 
     def test_run_raises_after_max_retries_exhausted(self, mock_boto3_session, set_env_variables):
         """After MAX_RETRIES throttling errors the error is propagated."""
-        from haystack_integrations.components.generators.amazon_bedrock.chat.chat_generator import MAX_RETRIES
-
         generator = AmazonBedrockChatGenerator(model="anthropic.claude-3-5-sonnet-20240620-v1:0")
         mock_client = mock_boto3_session.return_value.client.return_value
         mock_client.converse.side_effect = _make_client_error("ThrottlingException")
@@ -1351,8 +1350,6 @@ async def test_run_async_raises_on_non_throttling_client_error(self, mock_boto3_
     @pytest.mark.asyncio
     async def test_run_async_raises_after_max_retries_exhausted(self, mock_boto3_session, mock_aioboto3_session):
         """After MAX_RETRIES throttling errors the error is propagated in async mode."""
-        from haystack_integrations.components.generators.amazon_bedrock.chat.chat_generator import MAX_RETRIES
-
         generator = AmazonBedrockChatGenerator(model="anthropic.claude-3-5-sonnet-20240620-v1:0")
 
         mock_async_client = AsyncMock()

integrations/amazon_bedrock/tests/test_generator.py

@@ -20,6 +20,7 @@
     MetaLlamaAdapter,
     MistralAdapter,
 )
+from haystack_integrations.components.generators.amazon_bedrock.generator import MAX_RETRIES
 
 
 @pytest.mark.parametrize("boto3_config", [None, {"read_timeout": 1000}])
@@ -1822,8 +1823,6 @@ def test_run_raises_on_non_throttling_client_error(self, mock_boto3_session):
 
     def test_run_raises_after_max_retries_exhausted(self, mock_boto3_session):
         """After MAX_RETRIES throttling errors the error is propagated."""
-        from haystack_integrations.components.generators.amazon_bedrock.generator import MAX_RETRIES
-
         generator = AmazonBedrockGenerator(model="anthropic.claude-v2")
         mock_client = mock_boto3_session.return_value.client.return_value
         mock_client.invoke_model.side_effect = _make_client_error("ThrottlingException")