phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat - set env vars on proxy

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-10 11:44:27 -07:00
parent 84c486cacb
commit 25347b1f6f
1 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/proxy_server.py
View file

@ -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())