From a85a932e2502646cb1205d7e5190484cae8f8246 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 19 Aug 2024 21:09:03 -0700 Subject: [PATCH] fix(litellm_pre_call_utils.py): handle no query params in request --- litellm/proxy/auth/user_api_key_auth.py | 2 ++ litellm/proxy/litellm_pre_call_utils.py | 10 ++++++---- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/auth/user_api_key_auth.py b/litellm/proxy/auth/user_api_key_auth.py index 5dd58466b9f..8d2c728f592 100644 --- a/litellm/proxy/auth/user_api_key_auth.py +++ b/litellm/proxy/auth/user_api_key_auth.py @@ -1275,6 +1275,8 @@ def _get_request_ip_address( client_ip = request.headers["x-forwarded-for"] elif request.client is not None: client_ip = request.client.host + else: + client_ip = "" return client_ip diff --git a/litellm/proxy/litellm_pre_call_utils.py b/litellm/proxy/litellm_pre_call_utils.py index fddfcfc8da8..ff5b6dc86cd 100644 --- a/litellm/proxy/litellm_pre_call_utils.py +++ b/litellm/proxy/litellm_pre_call_utils.py @@ -171,10 +171,12 @@ async def add_litellm_data_to_request( } ## Dynamic api version (Azure OpenAI endpoints) ## - query_params = request.query_params - - # Convert query parameters to a dictionary (optional) - query_dict = dict(query_params) + try: + query_params = request.query_params + # Convert query parameters to a dictionary (optional) + query_dict = dict(query_params) + except KeyError: + query_dict = {} ## check for api version in query params dynamic_api_version: Optional[str] = query_dict.get("api-version")