From f3c863447f1d98a6c72b64918bf7b5244580b183 Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Mon, 4 Aug 2025 23:24:47 -0400 Subject: [PATCH] fix: core imports tests --- libs/core/tests/unit_tests/messages/test_imports.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/libs/core/tests/unit_tests/messages/test_imports.py b/libs/core/tests/unit_tests/messages/test_imports.py index 9fda5493244..750f2f49f06 100644 --- a/libs/core/tests/unit_tests/messages/test_imports.py +++ b/libs/core/tests/unit_tests/messages/test_imports.py @@ -43,6 +43,10 @@ EXPECTED_ALL = [ "convert_to_messages", "get_buffer_string", "is_data_content_block", + "is_reasoning_block", + "is_text_block", + "is_tool_call_block", + "is_tool_call_chunk", "merge_content", "message_chunk_to_message", "message_to_dict",