File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -15,7 +15,9 @@ def test_flex_service_tier():
1515 client = boto3 .client ("bedrock-runtime" , region_name = REGION )
1616 response = client .converse (
1717 modelId = MODEL_ID ,
18- messages = [{"role" : "user" , "content" : [{"text" : "What is 2+2? Answer in one word." }]}],
18+ messages = [
19+ {"role" : "user" , "content" : [{"text" : "What is 2+2? Answer in one word." }]}
20+ ],
1921 inferenceConfig = {"maxTokens" : 10 },
2022 serviceTier = {"type" : "flex" },
2123 )
@@ -89,7 +91,9 @@ def test_no_service_tier():
8991 print ("✅ ALL TESTS PASSED" )
9092 print ("=" * 60 )
9193 print ("\n Verification:" )
92- print ("- serviceTier parameter correctly formatted as {'type': 'flex|priority|default'}" )
94+ print (
95+ "- serviceTier parameter correctly formatted as {'type': 'flex|priority|default'}"
96+ )
9397 print ("- All service tiers work with Nova 2 Lite model" )
9498 print ("- Backward compatibility maintained (no serviceTier works)" )
9599
You can’t perform that action at this time.
0 commit comments