From 6e97bc44044d739289c84b03ef0ddd17c9edb34e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 4 Oct 2024 12:10:25 +0530 Subject: [PATCH] fix handle case when key based logging vars are set as os.environ/ vars --- litellm/litellm_core_utils/litellm_logging.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py index 2e874a487..1915176e1 100644 --- a/litellm/litellm_core_utils/litellm_logging.py +++ b/litellm/litellm_core_utils/litellm_logging.py @@ -353,6 +353,8 @@ class Logging: 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() if kwargs: _supported_callback_params = ( @@ -360,7 +362,10 @@ class Logging: ) for param in _supported_callback_params: 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 def update_environment_variables(