diff --git a/vertexai/generative_models/__init__.py b/vertexai/generative_models/__init__.py
index f460b9e408..976160b20f 100644
--- a/vertexai/generative_models/__init__.py
+++ b/vertexai/generative_models/__init__.py
@@ -27,6 +27,7 @@
     GenerativeModel,
     GenerationResponse,
     HarmCategory,
+    HarmBlockMethod,
     HarmBlockThreshold,
     Image,
     Part,
@@ -48,6 +49,7 @@
     "FunctionCall",
     "FunctionDeclaration",
     "HarmCategory",
+    "HarmBlockMethod",
     "HarmBlockThreshold",
     "Image",
     "Part",