Skip to content

Commit 5732303

Browse files
committed
fix lint error
1 parent f08fc45 commit 5732303

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

litellm/llms/vertex_ai/vertex_llm_base.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -239,6 +239,7 @@ def get_complete_vertex_url(
239239
stream=stream,
240240
auth_header=None,
241241
url=default_api_base,
242+
model=model,
242243
)
243244
return api_base
244245

@@ -387,31 +388,31 @@ def _handle_reauthentication(
387388
) -> Tuple[str, str]:
388389
"""
389390
Handle reauthentication when credentials refresh fails.
390-
391+
391392
This method clears the cached credentials and attempts to reload them once.
392393
It should only be called when "Reauthentication is needed" error occurs.
393-
394+
394395
Args:
395396
credentials: The original credentials
396397
project_id: The project ID
397398
credential_cache_key: The cache key to clear
398399
error: The original error that triggered reauthentication
399-
400+
400401
Returns:
401402
Tuple of (access_token, project_id)
402-
403+
403404
Raises:
404405
The original error if reauthentication fails
405406
"""
406407
verbose_logger.debug(
407408
f"Handling reauthentication for project_id: {project_id}. "
408409
f"Clearing cache and retrying once."
409410
)
410-
411+
411412
# Clear the cached credentials
412413
if credential_cache_key in self._credentials_project_mapping:
413414
del self._credentials_project_mapping[credential_cache_key]
414-
415+
415416
# Retry once with _retry_reauth=True to prevent infinite recursion
416417
try:
417418
return self.get_access_token(
@@ -441,12 +442,12 @@ def get_access_token(
441442
3. Check if loaded credentials have expired
442443
4. If expired, refresh credentials
443444
5. Return access token and project id
444-
445+
445446
Args:
446447
credentials: The credentials to use for authentication
447448
project_id: The Google Cloud project ID
448449
_retry_reauth: Internal flag to prevent infinite recursion during reauthentication
449-
450+
450451
Returns:
451452
Tuple of (access_token, project_id)
452453
"""

0 commit comments

Comments
 (0)