Merge pull request #2946 from BerriAI/litellm_fix_error_setting_env_vars

fix - /config/update setting env variables
This commit is contained in:
Ishaan Jaff 2024-04-10 18:22:06 -07:00 committed by GitHub
commit e5bd1b644d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2496,9 +2496,14 @@ class ProxyConfig:
# we need to set env variables too
environment_variables = config_data.get("environment_variables", {})
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
try:
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(
"Error setting env variable: %s - %s", k, str(e)
)
except Exception as e:
verbose_proxy_logger.error(
"{}\nTraceback:{}".format(str(e), traceback.format_exc())