From 54f8d060574dbefead8477451502e9599a429bee Mon Sep 17 00:00:00 2001 From: Rajan Paneru Date: Fri, 17 May 2024 09:55:13 +0930 Subject: [PATCH] handle exception and logged it --- litellm/proxy/proxy_server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 48a59cab8d..734ef5676a 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -367,7 +367,11 @@ def get_custom_headers(*, 'x-litellm-version': version, '"x-litellm-model-region': model_region } - return {key: value for key, value in headers.items() if value not in exclude_values} + try: + return {key: value for key, value in headers.items() if value not in exclude_values} + except Exception as e: + verbose_proxy_logger.error(f"Error setting custom headers: {e}") + return {} async def check_request_disconnection(request: Request, llm_api_call_task):