phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #1199 from neubig/add_safety_settings_default

Browse source 
Add a default for safety settings in vertex AI
This commit is contained in:
Krish Dholakia 2023-12-21 09:07:23 +05:30 • committed by GitHub
commit 5edc987209
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/llms/vertex_ai.py
View file

@ -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):