fix(litellm_pre_call_utils.py): handle no query params in request

This commit is contained in:
Krrish Dholakia 2024-08-19 21:09:03 -07:00
parent 2d75e7a8d4
commit 2b4d692353
2 changed files with 8 additions and 4 deletions

View file

@ -1275,6 +1275,8 @@ def _get_request_ip_address(
client_ip = request.headers["x-forwarded-for"] client_ip = request.headers["x-forwarded-for"]
elif request.client is not None: elif request.client is not None:
client_ip = request.client.host client_ip = request.client.host
else:
client_ip = ""
return client_ip return client_ip

View file

@ -171,10 +171,12 @@ async def add_litellm_data_to_request(
} }
## Dynamic api version (Azure OpenAI endpoints) ## ## Dynamic api version (Azure OpenAI endpoints) ##
try:
query_params = request.query_params query_params = request.query_params
# Convert query parameters to a dictionary (optional) # Convert query parameters to a dictionary (optional)
query_dict = dict(query_params) query_dict = dict(query_params)
except KeyError:
query_dict = {}
## check for api version in query params ## check for api version in query params
dynamic_api_version: Optional[str] = query_dict.get("api-version") dynamic_api_version: Optional[str] = query_dict.get("api-version")