From b85e49ad1919bcc609a657bfe0fcb9304847019c Mon Sep 17 00:00:00 2001
From: "open-swe[bot]" <open-swe@users.noreply.github.com>
Date: Mon, 11 Aug 2025 20:54:39 +0000
Subject: [PATCH] Apply patch [skip ci]

---
 libs/partners/openai/tests/unit_tests/test_imports.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/libs/partners/openai/tests/unit_tests/test_imports.py b/libs/partners/openai/tests/unit_tests/test_imports.py
index 144a394c2ff..23f7f814b98 100644
--- a/libs/partners/openai/tests/unit_tests/test_imports.py
+++ b/libs/partners/openai/tests/unit_tests/test_imports.py
@@ -8,8 +8,11 @@ EXPECTED_ALL = [
     "AzureChatOpenAI",
     "AzureOpenAIEmbeddings",
     "custom_tool",
+    "BatchError",
+    "BatchStatus",
 ]
 
 
 def test_all_imports() -> None:
     assert sorted(EXPECTED_ALL) == sorted(__all__)
+