From 5baf33bf8f689de3b3835dc1035c880e48c9ccf2 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 25 Nov 2024 10:20:02 -0800 Subject: [PATCH] fix allow using vertex_ai as the primary way for pass through vertex endpoints --- litellm/proxy/vertex_ai_endpoints/vertex_endpoints.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/vertex_ai_endpoints/vertex_endpoints.py b/litellm/proxy/vertex_ai_endpoints/vertex_endpoints.py index fbf37ce8d..1a0d09a88 100644 --- a/litellm/proxy/vertex_ai_endpoints/vertex_endpoints.py +++ b/litellm/proxy/vertex_ai_endpoints/vertex_endpoints.py @@ -113,7 +113,12 @@ def construct_target_url( @router.api_route( - "/vertex-ai/{endpoint:path}", methods=["GET", "POST", "PUT", "DELETE"] + "/vertex-ai/{endpoint:path}", + methods=["GET", "POST", "PUT", "DELETE"], + include_in_schema=False, +) +@router.api_route( + "/vertex_ai/{endpoint:path}", methods=["GET", "POST", "PUT", "DELETE"] ) async def vertex_proxy_route( endpoint: str,