iAdZddlZddlZddlmZddlmZddlmZm Z m Z m Z eje ZddlmZee dedef Zd efd Zd ZdZ dd lmZej0j3ddj5dk(rej7ddZ d!dedzdededeeefdzded eegeff dZ d"dedzdededeeefdzde dded eegeffdZ d"dedzdededeeefdzde dded eegeffdZ d efdZ!d efdZ"d eeeffd Z#y#e$rCej0j3ddj5dk(rej7dd ZYwxYw)#a Observability module for browser-use This module provides observability decorators that optionally integrate with lmnr (Laminar) for tracing. If lmnr is not installed, it provides no-op wrappers that accept the same parameters. Features: - Optional lmnr integration - works with or without lmnr installed - Debug mode support - observe_debug only traces when in debug mode - Full parameter compatibility with lmnr observe decorator - No-op fallbacks when lmnr is unavailable N)Callable)wraps)AnyLiteralTypeVarcast) load_dotenvF.)boundreturncXtjddj}|dk(ryy)zMCheck if we're in debug mode based on environment variables or logging level.LMNR_LOGGING_LEVELdebugTF)osgetenvlower)lmnr_debug_modes c/mnt/e/genesis-system/.venvs/browser-army/lib/python3.12/site-packages/browser_use/observability.py_is_debug_moders,99126<<>w  F)observe!BROWSER_USE_VERBOSE_OBSERVABILITYfalsetruez#Lmnr is available for observabilityTz'Lmnr is not available for observabilityname ignore_input ignore_outputmetadatakwargsc 4ddldtdtffd }|S)zSCreate a no-op decorator that accepts all lmnr observe parameters but does nothing.rNfuncr cjr#tfd}tt|Stfd}tt|S)Nc0K|i|d{S7wNargsr r"s r async_wrapperzA_create_no_op_decorator..decorator..async_wrapperGst&v& && &s c|i|Sr%r&r's r sync_wrapperz@_create_no_op_decorator..decorator..sync_wrapperNs    r)iscoroutinefunctionrrr )r"r)r+asyncios` r decoratorz*_create_no_op_decorator..decoratorDs]   & $K'' q-   $K!! q, r)r-r )rrrrr r.r-s @r_create_no_op_decoratorr/:s# Q 1 r span_type)DEFAULTLLMTOOLc |||||ddgd|}tr-tr'tttgtftdi|St di|S)a Observability decorator that traces function execution when lmnr is available. This decorator will use lmnr's observe decorator if lmnr is installed, otherwise it will be a no-op that accepts the same parameters. Args: name: Name of the span/trace ignore_input: Whether to ignore function input parameters in tracing ignore_output: Whether to ignore function output in tracing metadata: Additional metadata to attach to the span **kwargs: Additional parameters passed to lmnr observe Returns: Decorated function that may be traced depending on lmnr availability Example: @observe(name="my_function", metadata={"version": "1.0"}) def my_function(param1, param2): return param1 + param2 r observe_debugrrrrr0tagsr&)_LMNR_AVAILABLE _lmnr_observerrr r/rrrrr0r s rrrWsc<   o &     hsAv 7 7 88 ! *6 **rc |||||dgd|}tr7tr1tr'ttt gt ftdi|St di|S)a Debug-only observability decorator that only traces when in debug mode. This decorator will use lmnr's observe decorator if both lmnr is installed AND we're in debug mode, otherwise it will be a no-op. Debug mode is determined by: - DEBUG environment variable set to 1/true/yes/on - BROWSER_USE_DEBUG environment variable set to 1/true/yes/on - Root logging level set to DEBUG or lower Args: name: Name of the span/trace ignore_input: Whether to ignore function input parameters in tracing ignore_output: Whether to ignore function output in tracing metadata: Additional metadata to attach to the span **kwargs: Additional parameters passed to lmnr observe Returns: Decorated function that may be traced only in debug mode Example: @observe_debug(ignore_input=True, ignore_output=True,name="debug_function", metadata={"debug": True}) def debug_function(param1, param2): return param1 + param2 r5r6r&)r8r9rrrr r/r:s rr5r5sgF         .*: hsAv 7 7 88 ! *6 **rctS)z'Check if lmnr is available for tracing.)r8r&rris_lmnr_availabler=srctS)z'Check if we're currently in debug mode.)rr&rr is_debug_moder?s rcPttttxr tdS)z1Get the current status of observability features.)lmnr_available debug_modeobserve_activeobserve_debug_active)r8rr&rrget_observability_statusrEs%$ #)>n.>  r)NFFN)NFFNr1)$__doc__loggingrcollections.abcr functoolsrtypingrrrr getLogger__name__loggerdotenvr r boolrr8r9lmnrrenvirongetrr ImportErrorstrdictr/r5r=r?rEr&rrrVsp  $..   8 $  CxS)*  *JJNN6@FFHFR,,45#'  Tz S>D    qc1f <#'09 ,+ Tz,+,+,+ S>D ,+ , - ,+  ,+qc1f,+`#'09 1+ Tz1+1+1+ S>D 1+ , - 1+  1+qc1f1+j4 t $sDy/cJJNN6@FFHFR,,89sAD88AF?F