Skip to content

Commit fe5d1e1

Browse files
remove if condition in test
1 parent 9cf5643 commit fe5d1e1

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

tests/integrations/openai_agents/test_openai_agents.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2063,11 +2063,10 @@ def test_openai_agents_message_truncation(sentry_init, capture_events):
20632063
with start_span(op="gen_ai.chat") as span:
20642064
scope = sentry_sdk.get_current_scope()
20652065
_set_input_data(span, get_response_kwargs)
2066-
if hasattr(scope, "_gen_ai_original_message_count"):
2067-
truncated_count = scope._gen_ai_original_message_count.get(span.span_id)
2068-
assert truncated_count == 5, (
2069-
f"Expected 5 original messages, got {truncated_count}"
2070-
)
2066+
truncated_count = scope._gen_ai_original_message_count.get(span.span_id)
2067+
assert truncated_count == 5, (
2068+
f"Expected 5 original messages, got {truncated_count}"
2069+
)
20712070

20722071
assert SPANDATA.GEN_AI_REQUEST_MESSAGES in span._data
20732072
messages_data = span._data[SPANDATA.GEN_AI_REQUEST_MESSAGES]

0 commit comments

Comments
 (0)