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

fix(main.py): allow vertex ai project and location to be set in completion() call

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-25 15:00:51 -08:00
parent 06506f65f0
commit fbab810189
3 changed files with 16 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -3347,6 +3347,10 @@ def get_optional_params(
custom_llm_provider != "bedrock" and custom_llm_provider != "sagemaker"
): # allow dynamically setting boto3 init logic
continue
elif (
k.startswith("vertex_") and custom_llm_provider != "vertex_ai"
): # allow dynamically setting vertex ai init logic
continue
passed_params[k] = v
default_params = {
"functions": None,