phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix pass through endpoint tests

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-06 21:16:07 -07:00
parent d6db6f3ee6
commit ab920ab461
1 changed files with 12 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

41
litellm/proxy/pass_through_endpoints/pass_through_endpoints.py
View file

@ -417,36 +417,19 @@ def create_pass_through_route(
except Exception: except Exception:
verbose_proxy_logger.warning("Defaulting to target being a url.") verbose_proxy_logger.warning("Defaulting to target being a url.")
if dependencies is None:
async def endpoint_func_no_auth( async def endpoint_func(
request: Request, request: Request,
fastapi_response: Response, fastapi_response: Response,
): user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
return await pass_through_request( ):
request=request, return await pass_through_request(
target=target, request=request,
custom_headers=custom_headers or {}, target=target,
user_api_key_dict=UserAPIKeyAuth(), custom_headers=custom_headers or {},
forward_headers=_forward_headers, user_api_key_dict=user_api_key_dict,
) 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,
)
return endpoint_func return endpoint_func