Bedrock retired the Llama2 model, updated the test to use the closest…

… replacement Llama3 model. (apache#43600)
astronomer · Nov 2, 2024 · c1bd9c5 · c1bd9c5
1 parent eefddcb
commit c1bd9c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/providers/tests/system/amazon/aws/example_bedrock.py b/providers/tests/system/amazon/aws/example_bedrock.py
@@ -63,7 +63,7 @@
 SKIP_PROVISION_THROUGHPUT = environ.get("SKIP_RESTRICTED_SYSTEM_TEST_TASKS", default=True)
 
 
-LLAMA_SHORT_MODEL_ID = "meta.llama2-13b-chat-v1"
+LLAMA_SHORT_MODEL_ID = "meta.llama3-8b-instruct-v1:0"
 TITAN_MODEL_ID = "amazon.titan-text-express-v1:0:8k"
 TITAN_SHORT_MODEL_ID = TITAN_MODEL_ID.split(":")[0]