From 74d59d74d4eeac197645849037b30869df59653a Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 27 Nov 2024 17:52:04 +0530 Subject: [PATCH] fix(user_api_key_auth.py): use `model` from query param --- litellm/proxy/auth/user_api_key_auth.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/auth/user_api_key_auth.py b/litellm/proxy/auth/user_api_key_auth.py index 99c7dc50e..5c90299f6 100644 --- a/litellm/proxy/auth/user_api_key_auth.py +++ b/litellm/proxy/auth/user_api_key_auth.py @@ -203,8 +203,12 @@ async def user_api_key_auth_websocket(websocket: WebSocket): request = Request(scope={"type": "http"}) request._url = websocket.url + query_params = websocket.query_params + + model = query_params.get("model") + async def return_body(): - return_string = '{{"model": "fake-openai-endpoint"}}' + return_string = f'{{"model": "{model}"}}' # return string as bytes return return_string.encode()