diff --git a/litellm/integrations/langfuse/langfuse_handler.py b/litellm/integrations/langfuse/langfuse_handler.py index f8fec40ab84..c0088c8a1cf 100644 --- a/litellm/integrations/langfuse/langfuse_handler.py +++ b/litellm/integrations/langfuse/langfuse_handler.py @@ -36,7 +36,12 @@ 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, @@ -56,9 +61,11 @@ 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 @@ -79,17 +86,19 @@ 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 @@ -106,7 +115,8 @@ 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, ) @@ -133,7 +143,8 @@ 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"), )