scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-22 00:31:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 51

style(langfuse): apply ruff formatting

Browse source 
Signed-off-by: Aman Sachan <amansachan92905@gmail.com>
This commit is contained in:
Aman Sachan 2026-05-16 00:16:09 +00:00
parent 2592f7b1f0
commit 6442bfdf64
No known key found for this signature in database
GPG key ID: D50A9E263569B417
1 changed files with 12 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
litellm/integrations/langfuse/langfuse_handler.py
View file

@ -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"),
)