@@ -153,7 +153,7 @@ def test_start_model_invoke_span(mock_tracer):
153153
154154 mock_tracer .start_span .assert_called_once ()
155155 assert mock_tracer .start_span .call_args [1 ]["name" ] == "chat"
156- assert mock_tracer .start_span .call_args [1 ]["kind" ] == SpanKind .CLIENT
156+ assert mock_tracer .start_span .call_args [1 ]["kind" ] == SpanKind .INTERNAL
157157 mock_span .set_attribute .assert_any_call ("gen_ai.system" , "strands-agents" )
158158 mock_span .set_attribute .assert_any_call ("gen_ai.operation.name" , "chat" )
159159 mock_span .set_attribute .assert_any_call ("gen_ai.request.model" , model_id )
@@ -188,7 +188,7 @@ def test_start_model_invoke_span_latest_conventions(mock_tracer):
188188
189189 mock_tracer .start_span .assert_called_once ()
190190 assert mock_tracer .start_span .call_args [1 ]["name" ] == "chat"
191- assert mock_tracer .start_span .call_args [1 ]["kind" ] == SpanKind .CLIENT
191+ assert mock_tracer .start_span .call_args [1 ]["kind" ] == SpanKind .INTERNAL
192192 mock_span .set_attribute .assert_any_call ("gen_ai.provider.name" , "strands-agents" )
193193 mock_span .set_attribute .assert_any_call ("gen_ai.operation.name" , "chat" )
194194 mock_span .set_attribute .assert_any_call ("gen_ai.request.model" , model_id )
@@ -670,6 +670,7 @@ def test_start_agent_span(mock_tracer):
670670
671671 mock_tracer .start_span .assert_called_once ()
672672 assert mock_tracer .start_span .call_args [1 ]["name" ] == "invoke_agent WeatherAgent"
673+ assert mock_tracer .start_span .call_args [1 ]["kind" ] == SpanKind .INTERNAL
673674 mock_span .set_attribute .assert_any_call ("gen_ai.system" , "strands-agents" )
674675 mock_span .set_attribute .assert_any_call ("gen_ai.agent.name" , "WeatherAgent" )
675676 mock_span .set_attribute .assert_any_call ("gen_ai.request.model" , model_id )
0 commit comments