From 14ab13ed83595f09cd2317d44f2297bb3e6bb9e3 Mon Sep 17 00:00:00 2001 From: Erick Friis Date: Wed, 24 Apr 2024 10:48:45 -0700 Subject: [PATCH 1/2] fix integration test region --- .../integration_tests/embeddings/test_bedrock_embeddings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py b/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py index c35084ee..3046883f 100644 --- a/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py +++ b/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py @@ -8,7 +8,7 @@ @pytest.fixture def bedrock_embeddings() -> BedrockEmbeddings: return BedrockEmbeddings( - model_id="amazon.titan-embed-text-v1", region_name="us-west-2" + model_id="amazon.titan-embed-text-v1" ) From f627a85e0594dd9d887f8a7047ef2d8e89224fb0 Mon Sep 17 00:00:00 2001 From: Erick Friis Date: Wed, 24 Apr 2024 10:49:44 -0700 Subject: [PATCH 2/2] x --- .../integration_tests/embeddings/test_bedrock_embeddings.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py b/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py index 3046883f..927204aa 100644 --- a/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py +++ b/libs/aws/tests/integration_tests/embeddings/test_bedrock_embeddings.py @@ -7,9 +7,7 @@ @pytest.fixture def bedrock_embeddings() -> BedrockEmbeddings: - return BedrockEmbeddings( - model_id="amazon.titan-embed-text-v1" - ) + return BedrockEmbeddings(model_id="amazon.titan-embed-text-v1") @pytest.mark.scheduled