@@ -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