diff --git a/litellm/integrations/langfuse/langfuse_handler.py b/litellm/integrations/langfuse/langfuse_handler.py index c0088c8a1cf..f8fec40ab84 100644 --- a/litellm/integrations/langfuse/langfuse_handler.py +++ b/litellm/integrations/langfuse/langfuse_handler.py @@ -36,12 +36,7 @@ class LangFuseHandler: """ temp_langfuse_logger: Optional[LangFuseLogger] = globalLangfuseLogger - if ( - LangFuseHandler._dynamic_langfuse_credentials_are_passed( - standard_callback_dynamic_params - ) - is False - ): + if LangFuseHandler._dynamic_langfuse_credentials_are_passed(standard_callback_dynamic_params) is False: return LangFuseHandler._return_global_langfuse_logger( globalLangfuseLogger=globalLangfuseLogger, in_memory_dynamic_logger_cache=in_memory_dynamic_logger_cache, @@ -61,11 +56,9 @@ class LangFuseHandler: # if not cached, create a new langfuse logger and cache it if temp_langfuse_logger is None: - temp_langfuse_logger = ( - LangFuseHandler._create_langfuse_logger_from_credentials( - credentials=credentials_dict, - in_memory_dynamic_logger_cache=in_memory_dynamic_logger_cache, - ) + temp_langfuse_logger = LangFuseHandler._create_langfuse_logger_from_credentials( + credentials=credentials_dict, + in_memory_dynamic_logger_cache=in_memory_dynamic_logger_cache, ) return temp_langfuse_logger @@ -86,19 +79,17 @@ class LangFuseHandler: if globalLangfuseLogger is not None: return globalLangfuseLogger - credentials_dict: Dict[str, Any] = ( - {} - ) # the global langfuse logger uses Environment Variables, there are no dynamic credentials + credentials_dict: Dict[ + str, Any + ] = {} # the global langfuse logger uses Environment Variables, there are no dynamic credentials globalLangfuseLogger = in_memory_dynamic_logger_cache.get_cache( credentials=credentials_dict, service_name="langfuse", ) if globalLangfuseLogger is None: - globalLangfuseLogger = ( - LangFuseHandler._create_langfuse_logger_from_credentials( - credentials=credentials_dict, - in_memory_dynamic_logger_cache=in_memory_dynamic_logger_cache, - ) + globalLangfuseLogger = LangFuseHandler._create_langfuse_logger_from_credentials( + credentials=credentials_dict, + in_memory_dynamic_logger_cache=in_memory_dynamic_logger_cache, ) return globalLangfuseLogger @@ -115,8 +106,7 @@ class LangFuseHandler: langfuse_logger = LangFuseLogger( langfuse_public_key=credentials.get("langfuse_public_key"), - langfuse_secret=credentials.get("langfuse_secret") - or credentials.get("langfuse_secret_key"), + langfuse_secret=credentials.get("langfuse_secret") or credentials.get("langfuse_secret_key"), langfuse_host=credentials.get("langfuse_host"), allow_env_credentials=credentials.get("langfuse_host") is None, ) @@ -143,8 +133,7 @@ class LangFuseHandler: # Guard against None to handle the case where no per-request params are passed params = standard_callback_dynamic_params or {} return LangfuseLoggingConfig( - langfuse_secret=params.get("langfuse_secret") - or params.get("langfuse_secret_key"), + langfuse_secret=params.get("langfuse_secret") or params.get("langfuse_secret_key"), langfuse_public_key=params.get("langfuse_public_key"), langfuse_host=params.get("langfuse_host"), )