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 black formatting with trailing commas

Browse source 
Signed-off-by: Aman Sachan <amansachan92905@gmail.com>
This commit is contained in:
Aman Sachan 2026-05-16 01:49:58 +00:00
parent 77d712d138
commit e1d234e170
No known key found for this signature in database
GPG key ID: D50A9E263569B417
1 changed files with 23 additions and 12 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,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"),
)