From 7119d188f628360539784760a98632cfd4be1b46 Mon Sep 17 00:00:00 2001
From: Yuhao Yang <47235274+yhyang201@users.noreply.github.com>
Date: Tue, 18 Nov 2025 10:32:59 +0800
Subject: [PATCH] [CI] re-enable test_vision_openai_server_a ci (#13444)

---
 test/srt/run_suite.py                   |  4 ++--
 test/srt/test_vision_openai_server_a.py | 24 ++++++++++++++++--------
 2 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py
index 504515273..a83a0df20 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -124,7 +124,7 @@ suites = {
         TestFile("test_triton_sliding_window.py", 100),
         TestFile("test_utils_update_weights.py", 48),
         TestFile("test_vision_chunked_prefill.py", 170),
-        # TestFile("test_vision_openai_server_a.py", 900),
+        TestFile("test_vision_openai_server_a.py", 900),
         TestFile("test_vlm_input_format.py", 300),
         TestFile("test_modelopt_loader.py", 30),
         TestFile("test_modelopt_export.py", 30),
@@ -365,7 +365,7 @@ suites = {
         TestFile("test_verl_engine_4_gpu.py"),
         TestFile("test_verl_engine_server.py"),
         TestFile("test_vertex_endpoint.py"),
-        TestFile("test_vision_openai_server_a.py"),  # TODO: Fix timeout
+        # TestFile("test_vision_openai_server_a.py"),  # TODO: Fix timeout
         TestFile("test_vision_openai_server_b.py"),
         TestFile("test_vision_openai_server_common.py"),
         TestFile("test_vlm_accuracy.py"),
diff --git a/test/srt/test_vision_openai_server_a.py b/test/srt/test_vision_openai_server_a.py
index 7f8cc87b5..5068dcda1 100644
--- a/test/srt/test_vision_openai_server_a.py
+++ b/test/srt/test_vision_openai_server_a.py
@@ -6,7 +6,16 @@ python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_multi_
 
 import unittest
 
+import openai
 from test_vision_openai_server_common import *
+from test_vision_openai_server_common import (  # DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,; DEFAULT_URL_FOR_TEST,; IMAGE_MAN_IRONING_URL,; popen_launch_server,
+    AudioOpenAITestMixin,
+    CustomTestCase,
+    ImageOpenAITestMixin,
+    OmniOpenAITestMixin,
+    TestOpenAIMLLMServerBase,
+    VideoOpenAITestMixin,
+)
 
 
 class TestLlavaServer(ImageOpenAITestMixin):
@@ -207,12 +216,11 @@ class TestDeepseekOCRServer(TestOpenAIMLLMServerBase):
 
 
 if __name__ == "__main__":
-    # Note: Cannot delete mixin classes imported via * since they're not in local scope
-    # del (
-    #     TestOpenAIMLLMServerBase,
-    #     ImageOpenAITestMixin,
-    #     VideoOpenAITestMixin,
-    #     AudioOpenAITestMixin,
-    #     OmniOpenAITestMixin,
-    # )
+    del (
+        TestOpenAIMLLMServerBase,
+        ImageOpenAITestMixin,
+        VideoOpenAITestMixin,
+        AudioOpenAITestMixin,
+        OmniOpenAITestMixin,
+    )
     unittest.main()