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

fix vertex use async func to set auth creds

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-10 16:12:18 -07:00
parent 26ae86e59b
commit 1c6f8b1be2
8 changed files with 420 additions and 230 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/llms/fine_tuning_apis/vertex_ai.py
View file

@ -185,8 +185,14 @@ class VertexFineTuningAPI(VertexLLM):
"creating fine tuning job, args= %s", create_fine_tuning_job_data
)
_auth_header, vertex_project = self._ensure_access_token(
credentials=vertex_credentials,
project_id=vertex_project,
)
auth_header, _ = self._get_token_and_url(
model="",
auth_header=_auth_header,
gemini_api_key=None,
vertex_credentials=vertex_credentials,
vertex_project=vertex_project,
@ -251,8 +257,14 @@ class VertexFineTuningAPI(VertexLLM):
vertex_credentials: str,
request_route: str,
):
_auth_header, vertex_project = await self._ensure_access_token_async(
credentials=vertex_credentials,
project_id=vertex_project,
)
auth_header, _ = self._get_token_and_url(
model="",
auth_header=_auth_header,
gemini_api_key=None,
vertex_credentials=vertex_credentials,
vertex_project=vertex_project,