Add a default for safety settings in vertex AI

2023-12-20 13:12:50 -05:00 · 2023-12-20 13:12:50 -05:00 · 482b3b5bc3
commit 482b3b5bc3
parent 04bbd0649f
1 changed files with 1 additions and 0 deletions
--- 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):