From 482b3b5bc3dc6acfd38f741de7ccf338216bad6c Mon Sep 17 00:00:00 2001 From: Graham Neubig Date: Wed, 20 Dec 2023 13:12:50 -0500 Subject: [PATCH] Add a default for safety settings in vertex AI --- litellm/llms/vertex_ai.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/llms/vertex_ai.py b/litellm/llms/vertex_ai.py index f55575227..7cfc91701 100644 --- a/litellm/llms/vertex_ai.py +++ b/litellm/llms/vertex_ai.py @@ -195,6 +195,7 @@ def completion( optional_params[k] = v ## Process safety settings into format expected by vertex AI + safety_settings = None if "safety_settings" in optional_params: safety_settings = optional_params.pop("safety_settings") if not isinstance(safety_settings, list):