From 97ff0caf70755d3673988b4352bd05ddd924e956 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 30 Nov 2023 10:56:55 -0800 Subject: [PATCH] (feat) proxy: config - azure allow users to pass in base_url --- litellm/router.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/router.py b/litellm/router.py index 9135406fda..00b3a8d125 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -842,6 +842,8 @@ class Router: api_key = os.getenv(api_key_env_name) api_base = litellm_params.get("api_base") + base_url = litellm_params.get("base_url") + api_base = api_base or base_url # allow users to pass in `api_base` or `base_url` for azure if api_base and api_base.startswith("os.environ/"): api_base_env_name = api_base.replace("os.environ/", "") api_base = os.getenv(api_base_env_name)