diff --git a/python/sglang/backend/vertexai.py b/python/sglang/backend/vertexai.py
index 07827b76dbfe..871cb4f882c2 100644
--- a/python/sglang/backend/vertexai.py
+++ b/python/sglang/backend/vertexai.py
@@ -14,15 +14,12 @@
         GenerativeModel,
         Image,
     )
-    from vertexai.generative_models._generative_models import SafetySettingsType
 except ImportError as e:
     GenerativeModel = e
 
 
 class VertexAI(BaseBackend):
-    def __init__(
-        self, model_name, safety_settings: Optional[SafetySettingsType] = None
-    ):
+    def __init__(self, model_name, safety_settings=None):
         super().__init__()
 
         if isinstance(GenerativeModel, Exception):