Skip to content

Commit 9814427

Browse files
authored
Update test_botocore_bedrock.py
1 parent 0a0e7fa commit 9814427

1 file changed

Lines changed: 0 additions & 11 deletions

File tree

instrumentation/opentelemetry-instrumentation-botocore/tests/test_botocore_bedrock.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1487,7 +1487,6 @@ def test_invoke_model_with_content(
14871487
finish_reason = "length"
14881488
else:
14891489
pytest.xfail("model family not handled: {model_family}")
1490-
return
14911490

14921491
assert_message_in_logs(logs[0], "gen_ai.user.message", user_content, span)
14931492

@@ -1628,7 +1627,6 @@ def test_invoke_model_with_content_different_events(
16281627
choice_content = [{"type": "text", "text": "This is a test"}]
16291628
else:
16301629
pytest.xfail("llm_model_value not handled: {llm_model_value}")
1631-
return
16321630

16331631
body = get_invoke_model_body(
16341632
llm_model_value,
@@ -2070,7 +2068,6 @@ def test_invoke_model_with_content_tool_call(
20702068
llm_model_config = AnthropicClaudeModel
20712069
else:
20722070
pytest.xfail("model family not handled: {model_family}")
2073-
return
20742071

20752072
invoke_model_tool_call(
20762073
span_exporter,
@@ -2156,7 +2153,6 @@ def test_invoke_model_no_content(
21562153
finish_reason = "length"
21572154
else:
21582155
pytest.xfail("model family not handled: {model_family}")
2159-
return
21602156

21612157
choice_body = {
21622158
"index": 0,
@@ -2190,7 +2186,6 @@ def test_invoke_model_no_content_different_events(
21902186
finish_reason = "end_turn"
21912187
else:
21922188
pytest.xfail("llm_model_value not handled: {llm_model_value}")
2193-
return
21942189

21952190
body = get_invoke_model_body(
21962191
llm_model_value,
@@ -2246,7 +2241,6 @@ def test_invoke_model_no_content_tool_call(
22462241
llm_model_config = AnthropicClaudeModel
22472242
else:
22482243
pytest.xfail("model family not handled: {model_family}")
2249-
return
22502244

22512245
invoke_model_tool_call(
22522246
span_exporter,
@@ -2401,7 +2395,6 @@ def test_invoke_model_with_response_stream_with_content(
24012395
}
24022396
else:
24032397
pytest.xfail("model family not handled: {model_family}")
2404-
return
24052398

24062399
choice_body = {
24072400
"index": 0,
@@ -2437,7 +2430,6 @@ def test_invoke_model_with_response_stream_with_content_different_events(
24372430
choice_content = [{"text": "This is a test", "type": "text"}]
24382431
else:
24392432
pytest.xfail("llm_model_value not handled: {llm_model_value}")
2440-
return
24412433

24422434
max_tokens = 10
24432435
body = get_invoke_model_body(
@@ -2672,7 +2664,6 @@ def test_invoke_model_with_response_stream_with_content_tool_call(
26722664
llm_model_config = AnthropicClaudeModel
26732665
else:
26742666
pytest.xfail("model family not handled: {model_family}")
2675-
return
26762667

26772668
invoke_model_with_response_stream_tool_call(
26782669
span_exporter,
@@ -2801,7 +2792,6 @@ def test_invoke_model_with_response_stream_no_content_different_events(
28012792
finish_reason = "end_turn"
28022793
else:
28032794
pytest.xfail("llm_model_value not handled: {llm_model_value}")
2804-
return
28052795

28062796
max_tokens = 10
28072797
body = get_invoke_model_body(
@@ -2864,7 +2854,6 @@ def test_invoke_model_with_response_stream_no_content_tool_call(
28642854
llm_model_config = AnthropicClaudeModel
28652855
else:
28662856
pytest.xfail("model family not handled: {model_family}")
2867-
return
28682857

28692858
invoke_model_with_response_stream_tool_call(
28702859
span_exporter,

0 commit comments

Comments
 (0)