phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix handle case when key based logging vars are set as os.environ/ vars

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-04 12:10:25 +05:30
parent 670ecda4e2
commit 6e97bc4404
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/litellm_core_utils/litellm_logging.py
View file

@ -353,6 +353,8 @@ class Logging:
checks if langfuse_secret_key, gcs_bucket_name in kwargs and sets the corresponding attributes in StandardCallbackDynamicParams checks if langfuse_secret_key, gcs_bucket_name in kwargs and sets the corresponding attributes in StandardCallbackDynamicParams
""" """
from litellm.secret_managers.main import get_secret_str
standard_callback_dynamic_params = StandardCallbackDynamicParams() standard_callback_dynamic_params = StandardCallbackDynamicParams()
if kwargs: if kwargs:
_supported_callback_params = ( _supported_callback_params = (
@ -360,7 +362,10 @@ class Logging:
) )
for param in _supported_callback_params: for param in _supported_callback_params:
if param in kwargs: if param in kwargs:
standard_callback_dynamic_params[param] = kwargs.pop(param) # type: ignore _param_value = kwargs.pop(param)
if "os.environ/" in _param_value:
_param_value = get_secret_str(secret_name=_param_value)
standard_callback_dynamic_params[param] = _param_value # type: ignore
return standard_callback_dynamic_params return standard_callback_dynamic_params
def update_environment_variables( def update_environment_variables(