From 25347b1f6f1441d513363d65b14d160660f62d6d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 10 Apr 2024 11:44:27 -0700 Subject: [PATCH] feat - set env vars on proxy --- litellm/proxy/proxy_server.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 691bb1adf36..6416044d8ca 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2493,13 +2493,12 @@ class ProxyConfig: if success_callback not in litellm.success_callback: litellm.success_callback.append(success_callback) _added_callback = True - if _added_callback: - # we need to set env variables too - environment_variables = config_data.get("environment_variables", None) - for k, v in environment_variables.items(): - decoded_b64 = base64.b64decode(v) - value = decrypt_value(value=decoded_b64, master_key=master_key) - os.environ[k] = value + # we need to set env variables too + environment_variables = config_data.get("environment_variables", None) + for k, v in environment_variables.items(): + decoded_b64 = base64.b64decode(v) + value = decrypt_value(value=decoded_b64, master_key=master_key) + os.environ[k] = value except Exception as e: verbose_proxy_logger.error( "{}\nTraceback:{}".format(str(e), traceback.format_exc())