From 020fb54ea7c5a2510c67f36d624aaedea30bb4d6 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 15 Apr 2024 14:04:37 -0700 Subject: [PATCH] fix using wildcard models on proxy --- 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 06a51a384..9724a8c02 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -703,7 +703,11 @@ async def user_api_key_auth( verbose_proxy_logger.debug( f"model: {model}; allowed_models: {filtered_models}" ) - if model is not None and model not in filtered_models: + if ( + model is not None + and model not in filtered_models + and "*" not in filtered_models + ): raise ValueError( f"API Key not allowed to access model. This token can only access models={valid_token.models}. Tried to access {model}" )