phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

vertex forward all headers from vertex

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-30 11:05:23 -07:00
parent 1bd2b2fc92
commit 50014aa78e
2 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/vertex_ai_endpoints/vertex_endpoints.py
View file

@ -102,10 +102,7 @@ async def vertex_proxy_route(
vertex_location = match.group(1) if match else None
base_target_url = f"https://{vertex_location}-aiplatform.googleapis.com/"
headers.pop("content-length", None)
_new_headers = {
"Authorization": headers.get("authorization"),
}
headers = _new_headers
headers.pop("host", None)
else:
vertex_project = default_vertex_config.get("vertex_project")
vertex_location = default_vertex_config.get("vertex_location")