From 88d578a6de353e78de3dbc6de1cf14f581bc14fd Mon Sep 17 00:00:00 2001
From: fbpo23 <77107111+fbpo23@users.noreply.github.com>
Date: Wed, 2 Oct 2024 17:15:14 +0100
Subject: [PATCH] Rremoving safety settings from generation config for Vertex
 AI models (#16337)

---
 .../llama-index-llms-vertex/llama_index/llms/vertex/base.py     | 1 -
 .../llms/llama-index-llms-vertex/pyproject.toml                 | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/base.py b/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/base.py
index be02ac3e52608..23db7a6ffc4ab 100644
--- a/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/base.py
+++ b/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/base.py
@@ -187,7 +187,6 @@ def _model_kwargs(self) -> Dict[str, Any]:
         base_kwargs = {
             "temperature": self.temperature,
             "max_output_tokens": self.max_tokens,
-            "safety_settings": self._safety_settings,
         }
         return {
             **base_kwargs,
diff --git a/llama-index-integrations/llms/llama-index-llms-vertex/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-vertex/pyproject.toml
index 0468d2e1c1e4c..cf188c2b3deec 100644
--- a/llama-index-integrations/llms/llama-index-llms-vertex/pyproject.toml
+++ b/llama-index-integrations/llms/llama-index-llms-vertex/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-llms-vertex"
 readme = "README.md"
-version = "0.3.6"
+version = "0.3.7"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"