Skip to content

Commit da8813b

Browse files
committed
fix test
1 parent 6ac119b commit da8813b

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tests/integrations/litellm/test_litellm.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -637,10 +637,8 @@ def test_litellm_single_large_message_preservation(sentry_init, capture_events):
637637

638638
assert SPANDATA.GEN_AI_REQUEST_MESSAGES in span["data"]
639639
messages_data = span["data"][SPANDATA.GEN_AI_REQUEST_MESSAGES]
640-
assert isinstance(messages_data, str)
640+
assert isinstance(messages_data, list)
641641

642-
parsed_messages = json.loads(messages_data)
643-
assert isinstance(parsed_messages, list)
644-
assert len(parsed_messages) == 1
645-
assert parsed_messages[0]["role"] == "user"
646-
assert len(parsed_messages[0]["content"]) < len(huge_content)
642+
assert len(messages_data) == 1
643+
assert messages_data[0]["role"] == "user"
644+
assert len(messages_data[0]["content"]) < len(huge_content)

0 commit comments

Comments
 (0)