Skip to content

Commit 535eebf

Browse files
committed
add GenAI attribute to verify
1 parent 4387966 commit 535eebf

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

test/contract-tests/tests/test/amazon/awssdk/awssdk_test.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -506,7 +506,8 @@ def test_bedrock_runtime_invoke_model_llama(self):
506506
remote_operation="InvokeModel",
507507
remote_resource_type="AWS::Bedrock::Model",
508508
remote_resource_identifier="meta.llama3-8b-instruct-v1:0",
509-
request_specific_attributes={
509+
request_response_specific_attributes={
510+
_GEN_AI_SYSTEM: "aws_bedrock",
510511
_GEN_AI_REQUEST_MODEL: "meta.llama3-8b-instruct-v1:0",
511512
_GEN_AI_REQUEST_TEMPERATURE: 0.123,
512513
_GEN_AI_REQUEST_TOP_P: 0.456,
@@ -530,7 +531,8 @@ def test_bedrock_runtime_invoke_model_command(self):
530531
remote_operation="InvokeModel",
531532
remote_resource_type="AWS::Bedrock::Model",
532533
remote_resource_identifier="cohere.command-r-v1:0",
533-
request_specific_attributes={
534+
request_response_specific_attributes={
535+
_GEN_AI_SYSTEM: "aws_bedrock",
534536
_GEN_AI_REQUEST_MODEL: "cohere.command-r-v1:0",
535537
_GEN_AI_REQUEST_TEMPERATURE: 0.123,
536538
_GEN_AI_REQUEST_TOP_P: 0.456,
@@ -554,7 +556,8 @@ def test_bedrock_runtime_invoke_model_jamba(self):
554556
remote_operation="InvokeModel",
555557
remote_resource_type="AWS::Bedrock::Model",
556558
remote_resource_identifier="ai21.jamba-1-5-large-v1:0",
557-
request_specific_attributes={
559+
request_response_specific_attributes={
560+
_GEN_AI_SYSTEM: "aws_bedrock",
558561
_GEN_AI_REQUEST_MODEL: "ai21.jamba-1-5-large-v1:0",
559562
_GEN_AI_REQUEST_TEMPERATURE: 0.123,
560563
_GEN_AI_REQUEST_TOP_P: 0.456,
@@ -578,7 +581,8 @@ def test_bedrock_runtime_invoke_model_mistral(self):
578581
remote_operation="InvokeModel",
579582
remote_resource_type="AWS::Bedrock::Model",
580583
remote_resource_identifier="mistral.mistral-7b-instruct-v0:2",
581-
request_specific_attributes={
584+
request_response_specific_attributes={
585+
_GEN_AI_SYSTEM: "aws_bedrock",
582586
_GEN_AI_REQUEST_MODEL: "mistral.mistral-7b-instruct-v0:2",
583587
_GEN_AI_REQUEST_TEMPERATURE: 0.123,
584588
_GEN_AI_REQUEST_TOP_P: 0.456,

0 commit comments

Comments
 (0)