gfdZddlmZmZmZmZddlmZddlm Z m Z erddlm Z neZ GddZ y ) z This file contains the LangFuseHandler class Used to get the LangFuseLogger for a given request Handles Key/Team Based Langfuse Logging ) TYPE_CHECKINGAnyDictOptional)StandardCallbackDynamicParams)LangFuseLoggerLangfuseLoggingConfig)DynamicLoggingCachec eZdZe d dededeedefdZedeededefdZ ede dedefd Z e d dedeede fd Z ededefd Zy) LangFuseHandlerN standard_callback_dynamic_paramsin_memory_dynamic_logger_cacheglobalLangfuseLoggerreturnc|}tj|durtj||Stj||}t |}|j |d}|tj ||}|S)a This function is used to get the LangFuseLogger for a given request 1. If dynamic credentials are passed - check if a LangFuseLogger is cached for the dynamic credentials - if cached LangFuseLogger is not found, create a new LangFuseLogger and cache it 2. If dynamic credentials are not passed return the globalLangfuseLogger F)rr)rrlangfuse credentials service_namerr)r (_dynamic_langfuse_credentials_are_passed_return_global_langfuse_logger#get_dynamic_langfuse_logging_configdict get_cache(_create_langfuse_logger_from_credentials)rrrtemp_langfuse_logger _credentialscredentials_dicts c/var/www/openai/venv/lib/python3.12/site-packages/litellm/integrations/langfuse/langfuse_handler.pyget_langfuse_logger_for_requestz/LangFuseHandler.get_langfuse_logger_for_requests :N  D D0   #AA%9/MB  'JJ!5-MK  - >GG(z H  'HH 03QI !$#cj||Si}|j|d}|tj||}|S)a Returns the Global LangfuseLogger set on litellm (this is the default langfuse logger - used when no dynamic credentials are passed) If no Global LangfuseLogger is set, it will check in_memory_dynamic_logger_cache for a cached LangFuseLogger This function is used to return the globalLangfuseLogger if it exists, otherwise it will check in_memory_dynamic_logger_cache for a cached LangFuseLogger rrr)rr r)rrr s r!rz.LangFuseHandler._return_global_langfuse_loggerJsh +' '   >GG(# H  'HH 03QI ! $#r#rct|jd|jd|jd}|j|d||S)z This function is used to 1. create a LangFuseLogger from the credentials 2. cache the LangFuseLogger to prevent re-creating it for the same credentials langfuse_public_keylangfuse_secret langfuse_host)r&r'r(r)rr logging_obj)r get set_cache)rrlangfuse_loggers r!rz8LangFuseHandler._create_langfuse_logger_from_credentialsjsZ) +0E F'OO,=>%///:  '00##' 1 r#ct|jdxs|jd|jd|jdS)aT This function is used to get the Langfuse logging config to use for a given request. It checks if the dynamic parameters are provided in the standard_callback_dynamic_params and uses them to get the Langfuse logging config. If no dynamic parameters are provided, it uses the `globalLangfuseLogger` values r'langfuse_secret_keyr&r()r'r&r()r r*)rrs r!rz3LangFuseHandler.get_dynamic_langfuse_logging_configsV%<@@ARSK/334IJ @ D D%!;>>O   r#c|jd3|jd"|jd|jdyy)z This function is used to check if the dynamic langfuse credentials are passed in standard_callback_dynamic_params Returns: bool: True if the dynamic langfuse credentials are passed, False otherwise r(r&r'r.TF)r*)rs r!rz8LangFuseHandler._dynamic_langfuse_credentials_are_passedsQ - 0 0 A M/334IJV/334EFR/334IJVr#)N)__name__ __module__ __qualname__ staticmethodrr rr r"rrrr rboolrr#r!r r s:>0$*G0$(;0$'~60$  0$0$d$&~6$(;$ $$>(; ,:> *G &~6    **G r#r N) __doc__typingrrrr*litellm.litellm_core_utils.litellm_loggingrrr r r r r5r#r!r9s065T;NTTr#