phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) vertex ai project/location

Browse source
This commit is contained in:
ishaan-jaff 2024-02-28 08:13:13 -08:00
parent a40590d849
commit 9078707672
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/main.py
View file

@ -1467,12 +1467,12 @@ def completion(
response = model_response
elif custom_llm_provider == "vertex_ai":
vertex_ai_project = (
optional_params.pop("vertex_ai_project", None)
optional_params.pop("vertex_project", None)
or litellm.vertex_project
or get_secret("VERTEXAI_PROJECT")
)
vertex_ai_location = (
optional_params.pop("vertex_ai_location", None)
optional_params.pop("vertex_location", None)
or litellm.vertex_location
or get_secret("VERTEXAI_LOCATION")
)
@ -2566,12 +2566,12 @@ def embedding(
)
elif custom_llm_provider == "vertex_ai":
vertex_ai_project = (
optional_params.pop("vertex_ai_project", None)
optional_params.pop("vertex_project", None)
or litellm.vertex_project
or get_secret("VERTEXAI_PROJECT")
)
vertex_ai_location = (
optional_params.pop("vertex_ai_location", None)
optional_params.pop("vertex_location", None)
or litellm.vertex_location
or get_secret("VERTEXAI_LOCATION")
)