diff --git a/python/sglang/backend/vertexai.py b/python/sglang/backend/vertexai.py
index 07827b76d..871cb4f88 100644
--- a/python/sglang/backend/vertexai.py
+++ b/python/sglang/backend/vertexai.py
@@ -14,15 +14,12 @@ try:
         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):