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 case when gemini is used

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-10 17:06:45 -07:00
parent 7891b3742c
commit c8fe600dbf
11 changed files with 47 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -184,10 +184,10 @@ class VertexFineTuningAPI(VertexLLM):
verbose_logger.debug(
"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,
custom_llm_provider="vertex_ai_beta",
)
auth_header, _ = self._get_token_and_url(
@ -257,10 +257,10 @@ 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,
custom_llm_provider="vertex_ai_beta",
)
auth_header, _ = self._get_token_and_url(
model="",