fix AmazonBedrockConfigurationError import

deepset-ai · Dec 29, 2023 · 56fc6ed · 56fc6ed
1 parent 1ae9236
commit 56fc6ed
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/integrations/amazon_bedrock/src/amazon_bedrock_haystack/generators/amazon_bedrock.py b/integrations/amazon_bedrock/src/amazon_bedrock_haystack/generators/amazon_bedrock.py
@@ -6,12 +6,12 @@
 import boto3
 from botocore.exceptions import BotoCoreError, ClientError
 from haystack import component
-from haystack.errors import (
+
+from amazon_bedrock_haystack.errors import (
     AmazonBedrockConfigurationError,
     AmazonBedrockInferenceError,
     AWSConfigurationError,
 )
-
 from amazon_bedrock_haystack.generators.amazon_bedrock_adapters import (
     AI21LabsJurassic2Adapter,
     AmazonTitanAdapter,

diff --git a/integrations/amazon_bedrock/tests/test_amazon_bedrock.py b/integrations/amazon_bedrock/tests/test_amazon_bedrock.py
@@ -3,8 +3,8 @@
 
 import pytest
 from botocore.exceptions import BotoCoreError
-from haystack.errors import AmazonBedrockConfigurationError
 
+from amazon_bedrock_haystack.errors import AmazonBedrockConfigurationError
 from amazon_bedrock_haystack.generators.amazon_bedrock import AmazonBedrockGenerator
 from amazon_bedrock_haystack.generators.amazon_bedrock_adapters import (
     AI21LabsJurassic2Adapter,