From 13a77289b9c641cb18d564ac4dcf5b9bf2a96b43 Mon Sep 17 00:00:00 2001 From: Ari Webb Date: Thu, 6 Nov 2025 14:07:06 -0800 Subject: [PATCH] fix: send message test for gpt 4o (#6012) fix Co-authored-by: Ari Webb --- tests/integration_test_send_message.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/integration_test_send_message.py b/tests/integration_test_send_message.py index d91ce8de..d85a4c65 100644 --- a/tests/integration_test_send_message.py +++ b/tests/integration_test_send_message.py @@ -464,13 +464,6 @@ def assert_tool_call_response( ): return - try: - assert len(messages) == expected_message_count, messages - except: - if "claude-3-7-sonnet" not in llm_config.model: - raise - assert len(messages) == expected_message_count - 1, messages - # OpenAI gpt-4o-mini can sometimes omit the final AssistantMessage in streaming, # yielding the shorter sequence: # Reasoning -> ToolCall -> ToolReturn -> Reasoning -> StopReason -> Usage @@ -489,6 +482,13 @@ def assert_tool_call_response( ): return + try: + assert len(messages) == expected_message_count, messages + except: + if "claude-3-7-sonnet" not in llm_config.model: + raise + assert len(messages) == expected_message_count - 1, messages + index = 0 if from_db: assert isinstance(messages[index], UserMessage)