From 433a271613c0e7909c358598b4c4160cfaab7b1c Mon Sep 17 00:00:00 2001
From: Eric Schmidt <erschmid@google.com>
Date: Sun, 17 Nov 2024 18:33:01 -0800
Subject: [PATCH] feat: expose HarmBlockMethod to external callers

This enum is required for the SafetySettings() class to allow users to toggle harmful content blocking
---
 vertexai/generative_models/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

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",