scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 133

fix(vertex_ai): preserve pre-fetched Authorization header

Browse source 
Allow users to pass pre-fetched Bearer tokens via extra_headers without
being overwritten by ADC tokens.

- Skip _ensure_access_token() if Authorization already in headers
- Only set Authorization header if not already present

Fixes #23572
This commit is contained in:
Jah-yee 2026-03-14 02:26:10 +08:00
parent e37efc4218
commit 6a26ce3a8a
1 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
litellm/llms/vertex_ai/vertex_ai_partner_models/main.py
View file

@ -135,11 +135,17 @@ class VertexAIPartnerModels(VertexBase):
try:
vertex_httpx_logic = VertexLLM()
access_token, project_id = vertex_httpx_logic._ensure_access_token(
credentials=vertex_credentials,
project_id=vertex_project,
custom_llm_provider="vertex_ai",
)
# Skip _ensure_access_token if Authorization already present in headers
# This allows users to pass pre-fetched tokens via extra_headers
if headers is None or "Authorization" not in headers:
access_token, project_id = vertex_httpx_logic._ensure_access_token(
credentials=vertex_credentials,
project_id=vertex_project,
custom_llm_provider="vertex_ai",
)
else:
# Use existing project_id if token is pre-provided
project_id = vertex_project
openai_like_chat_completions = OpenAILikeChatHandler()
codestral_fim_completions = CodestralTextCompletion()
@ -198,7 +204,9 @@ class VertexAIPartnerModels(VertexBase):
elif "claude" in model:
if headers is None:
headers = {}
headers.update({"Authorization": "Bearer {}".format(access_token)})
# Only set Authorization if not already present (allow pre-fetched tokens)
if "Authorization" not in headers:
headers["Authorization"] = "Bearer {}".format(access_token)
optional_params.update(
{