From ab920ab4615b45084a9801bc05947797da322af6 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 6 Aug 2024 21:16:07 -0700 Subject: [PATCH] fix pass through endpoint tests --- .../pass_through_endpoints.py | 41 ++++++------------- 1 file changed, 12 insertions(+), 29 deletions(-) diff --git a/litellm/proxy/pass_through_endpoints/pass_through_endpoints.py b/litellm/proxy/pass_through_endpoints/pass_through_endpoints.py index 3ab0425a3..d71863497 100644 --- a/litellm/proxy/pass_through_endpoints/pass_through_endpoints.py +++ b/litellm/proxy/pass_through_endpoints/pass_through_endpoints.py @@ -417,36 +417,19 @@ def create_pass_through_route( except Exception: verbose_proxy_logger.warning("Defaulting to target being a url.") - if dependencies is None: - async def endpoint_func_no_auth( - request: Request, - fastapi_response: Response, - ): - return await pass_through_request( - request=request, - target=target, - custom_headers=custom_headers or {}, - user_api_key_dict=UserAPIKeyAuth(), - forward_headers=_forward_headers, - ) - - return endpoint_func_no_auth - - else: - - async def endpoint_func( - request: Request, - fastapi_response: Response, - user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), - ): - return await pass_through_request( - request=request, - target=target, - custom_headers=custom_headers or {}, - user_api_key_dict=user_api_key_dict, - forward_headers=_forward_headers, - ) + async def endpoint_func( + request: Request, + fastapi_response: Response, + user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), + ): + return await pass_through_request( + request=request, + target=target, + custom_headers=custom_headers or {}, + user_api_key_dict=user_api_key_dict, + forward_headers=_forward_headers, + ) return endpoint_func