tPjQjdZddlZddlmZmZmZddlmZddlm Z m Z erddlm Z neZ Gdd Z y) z This file contains the LangFuseHandler class Used to get the LangFuseLogger for a given request Handles Key/Team Based Langfuse Logging N) TYPE_CHECKINGAnyFinal)StandardCallbackDynamicParams)LangFuseLoggerLangfuseLoggingConfig)DynamicLoggingCachec eZdZe d dedededzdefdZededzdedefdZede dedefd Z e d dededzde fd Z edede dzfd Zededefd 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)rr langfuse 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)r rrtemp_langfuse_logger _credentialscredentials_dicts s/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/langfuse/langfuse_handler.pyget_langfuse_logger_for_requestz/LangFuseHandler.get_langfuse_logger_for_requests 7K  C CDd ein n"AA%9/MB  .QQ!5-MR #'|"4 >GG(z H  '#2#[#[,/M$\$ $#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)rrrs r rz.LangFuseHandler._return_global_langfuse_loggerCsf +' '   >GG(# H  '#2#[#[,/M$\$ $#r"rc t|jd|jdxs|jd|jd|jd|jddu}|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_secretlangfuse_secret_key langfuse_hostlangfuse_environmentN)r%r&r(r)allow_env_credentialsr)rr logging_obj)rget set_cache)rrlangfuse_loggers r rz8LangFuseHandler._create_langfuse_logger_from_credentialsas"0 +0E F'OO,=>h+//RgBh%///:!,1G!H"-///"Bd"J "  '00##' 1 r"ct|jdxs|jd|jd|jdtj|S)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&r'r%r()r&r%r(r))r r,r _meaningful_dynamic_environment)r rs r rz3LangFuseHandler.get_dynamic_langfuse_logging_configzsa%<@@ARSK/334IJ @ D DEZ [:>>O!0!P!PQq!r   r"c|jd}|yt|j}|r/|tjdk(s|t j k(ry|S)aReturn the per-request environment only when it changes behavior. Empty/whitespace values and values equal to the deployment-wide LANGFUSE_TRACING_ENVIRONMENT fallback are treated as absent so an environment-only override that matches the default does not mint a duplicate SDK client (each client costs threads and counts against MAX_LANGFUSE_INITIALIZED_CLIENTS). r)NLANGFUSE_TRACING_ENVIRONMENT)r,strstriposgetenvrresolve_deployment_environment)r rawvalues r r0z/LangFuseHandler._meaningful_dynamic_environments[/223IJ ;C  "@AAEEGG r"c|jdH|jd7|jd&|jdtj|yy)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,r r0)r s r rz8LangFuseHandler._dynamic_langfuse_credentials_are_passedsc - 0 0 A M/334IJV/334EFR/334IJV>>?_`lr")N)__name__ __module__ __qualname__ staticmethodrr rr!rrrr rr3r0boolrr"r r r s(7;)$*G)$(;)$-t3)$  )$)$V$,t3$(;$ $$:(; 07; *G ,t3    (*G t.*G r"r ) __doc__r5typingrrr*litellm.litellm_core_utils.litellm_loggingrrrr r r r@r"r rDs3 ,,T;Nddr"