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

fix: preserve pre-fetched Authorization header for vertex_ai partner models

Browse source 
- Skip _ensure_access_token if Authorization already in headers
- Only overwrite Authorization if not already present and access_token exists

Fixes #23572
This commit is contained in:
Jah-yee 2026-03-14 05:13:22 +08:00
parent 095f5e8399
commit 400bd9cb93
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -135,11 +135,15 @@ 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",
)
# Only fetch access token if Authorization header not already provided
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:
access_token, project_id = None, None
openai_like_chat_completions = OpenAILikeChatHandler()
codestral_fim_completions = CodestralTextCompletion()
@ -198,7 +202,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 and we have an access_token
if "Authorization" not in headers and access_token is not None:
headers.update({"Authorization": "Bearer {}".format(access_token)})
optional_params.update(
{