fix case when gemini is used

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

View file

@ -152,7 +152,9 @@ async def vertex_proxy_route(
_auth_header, vertex_project = (
await vertex_fine_tuning_apis_instance._ensure_access_token_async(
credentials=vertex_credentials, project_id=vertex_project
credentials=vertex_credentials,
project_id=vertex_project,
custom_llm_provider="vertex_ai_beta",
)
)