From 56fc6edf4fa1d6523a82d239cc17b94189a0a06c Mon Sep 17 00:00:00 2001
From: Julian Risch <julian.risch@deepset.ai>
Date: Fri, 29 Dec 2023 12:02:12 +0100
Subject: [PATCH] fix AmazonBedrockConfigurationError import

---
 .../src/amazon_bedrock_haystack/generators/amazon_bedrock.py  | 4 ++--
 integrations/amazon_bedrock/tests/test_amazon_bedrock.py      | 2 +-
 2 files changed, 3 insertions(+), 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
index fc96b1b03..c90a20cd5 100644
--- 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
index cecb28578..1dab863e5 100644
--- 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,