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

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:
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