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

Merge pull request #4461 from t968914/litellm-fix-vertexaibeta

Browse source 
fix: Include vertex_ai_beta in vertex_ai param mapping/Do not use google auth project_id
This commit is contained in:
Krish Dholakia 2024-07-04 15:27:20 -07:00 • committed by GitHub
commit 18d8edc145
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/vertex_httpx.py
View file

@ -748,10 +748,12 @@ class VertexLLM(BaseLLM):
if project_id is None:
project_id = creds.project_id
else:
creds, project_id = google_auth.default(
creds, creds_project_id = google_auth.default(
quota_project_id=project_id,
scopes=["https://www.googleapis.com/auth/cloud-platform"],
)
if project_id is None:
project_id = creds_project_id
creds.refresh(Request())
@ -974,7 +976,7 @@ class VertexLLM(BaseLLM):
api_base: Optional[str] = None,
) -> Union[ModelResponse, CustomStreamWrapper]:
stream: Optional[bool] = optional_params.pop("stream", None) # type: ignore
auth_header, url = self._get_token_and_url(
model=model,
gemini_api_key=gemini_api_key,