Skip to content

Commit 0b00cca

Browse files
authored
chore: Update bedrock_ranker_example.py (#1740)
1 parent f837c94 commit 0b00cca

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

integrations/amazon_bedrock/examples/bedrock_ranker_example.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,14 @@
33
from haystack import Document
44
from haystack.utils import Secret
55

6-
from haystack_integrations.components.rankers.amazon_bedrock import BedrockRanker
6+
from haystack_integrations.components.rankers.amazon_bedrock import AmazonBedrockRanker
77

88
# Set up AWS credentials
99
# You can also set these as environment variables
1010
aws_profile_name = os.environ.get("AWS_PROFILE") or "default"
1111
aws_region_name = os.environ.get("AWS_DEFAULT_REGION") or "eu-central-1"
12-
# Initialize the BedrockRanker with AWS credentials
13-
ranker = BedrockRanker(
12+
# Initialize the AmazonBedrockRanker with AWS credentials
13+
ranker = AmazonBedrockRanker(
1414
model="cohere.rerank-v3-5:0",
1515
top_k=2,
1616
aws_profile_name=Secret.from_token(aws_profile_name),

0 commit comments

Comments
 (0)