yj' ~UdZddlZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZdd lmZmZmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddl m!Z!edZ"ej#Z#edZ$edZ%e dee$e%fdee$e%ffdZ&e de'deee$e%fgee$e%fffdZ& d7dddee$e%fdzde'dzdee$e%feee$e%fgee$e%ffzfdZ&dZ(eZ)ee'e*e+e'e'e,ze-ze'ffe.d<de!de'd e'd!ejde'f d"Z/de!ddfd#Z0d$e'd%e'd&e'e,ze-zd'e'ddf d(Z1d$e'd)e,d*e,d+e,d,e,ddf d-Z2 d8d/e'd$e'd0e*e+e'e'e,ze-ze'fdzd1e'ddf d2Z3d3e'de'fd4Z4d5e de'fd6Z5dS)9z Custom utility logging functions for Langfuse integration. This module provides specialized formatters for all @observe decorated functions and a conditional observe decorator that only applies when Langfuse is configured. N) OrderedDict)Callable) ParamSpecTypeVaroverload)Request)observe)box)ConsoleGroupRenderableType)Panel)Table)Text)Tree)settings)append_metrics_to_file)RepresentationT)markupPRfuncreturncdSN)rs =/DATA/AppData/hermes/projects/honcho/src/telemetry/logging.pyconditional_observer$s SnamecdSrrr s rrr*s 25rr"cdtttfdtttfffd }| ||S|S)a Conditionally apply the @observe decorator only when LANGFUSE_PUBLIC_KEY is present. Can be used in two ways: 1. As a decorator: @conditional_observe 2. As a decorator factory: @conditional_observe(name="...") Args: func: The function to potentially decorate (when used as @conditional_observe) name: Optional name for the observation (when used as @conditional_observe(name="...")) Returns: The decorated function if Langfuse is configured, otherwise the original function frchtjr$n|j}t||S|S)Nr")rLANGFUSE_PUBLIC_KEY__name__r )r$ observe_namer s r decoratorz&conditional_observe..decoratorEs>  ' #'#344L-7 ---a00 0Hr)rrr)rr r)s ` rrr1s](Xad^A yriaccumulated_metricsrepresentationhistorynew_turnmessage_created_atcg}|d|d|d|d|d|||rQ|d|||d|rQ|d|||dd|S) a! Format reasoning inputs as markdown for logging. Args: representation: Current/working representation history: Conversation history new_turn: New user message message_created_at: Message timestamp Returns: Formatted markdown string z## Reasoning Inputs z**Current Time**: z%Y-%m-%d %H:%M:%Sz### Current Representation z### Conversation History z ### New Turn  )appendstrftimeformat_as_markdownstripjoin)r+r,r-r.partss r#format_reasoning_inputs_as_markdownr8`s: E LL())) LLO/889LMMOO LL LL/000 LL2244555 1222 W]]__%%% R %&&& X^^%%&&& R 99U  rctd}|dt|jd}t |jdD] \}}|d|d|!|dt|jd}t |jdD] \}}|d|d|!t |t dS) zi Log representation in a tree structure. Args: representation: Representation to log u📊 REPRESENTATIONz[bold cyan]EXPLICIT[/] ()z[dim]z.[/] z[bold cyan]DEDUCTIVE[/] (N)raddlenexplicit enumerate deductiveconsoleprint)r+tree type_branchiobss rlog_representationrGs % & &D((Uc.:Q6R6RUUUVVKN3Q77//3-----....((Ws>;S7T7TWWWXXKN4a88//3-----.... MM$ MMOOOOOr task_namelabelvalueunitc6tttkrJ|tvrAtdtttkr |tvAt|g|||fdS)a Accumulate a metric value to be printed the next time log_performance_metrics is called. This function uses a bounded OrderedDict to prevent memory leaks. If the number of tracked tasks exceeds MAX_ACCUMULATED_TASKS, the oldest entries are evicted (FIFO). Args: task_name: Task identifier label: Metric label value: Metric value unit: Metric unit F)lastN)r=r*MAX_ACCUMULATED_TASKSpopitem setdefaultr2)rHrIrJrKs raccumulate_metricrQs*   $999 0 0 0 ###///   $999 0 0 0 ""9b1188%9MNNNNNr input_tokens output_tokenscache_read_input_tokenscache_creation_input_tokensct|d|dt|d|dt|d|d||z |z}t|d|dt|d|ddS)a Log cache-aware token usage metrics. Args: task_name: The task name for metric accumulation input_tokens: Total input tokens (cached + uncached) output_tokens: Output tokens generated cache_read_input_tokens: Tokens read from cache (90% cheaper) cache_creation_input_tokens: Tokens written to cache (25% more expensive) Returns: None rRtokensrTrUuncached_input_tokensrSN)rQ)rHrRrSrTrUrXs rlog_token_usage_metricsrYs(ixHHH!  %#  ..1LL*,A8i-JJJJJr⚡ PERFORMANCE task_slugmetricstitlec @|d|}t|s|sdS|g}t|g|z}trt |||g}g}|D]%}|ddkr|n||&t dddd}|d d d |d ddd|ddd |D]h\}}} | dkr|d} n| dkr|d} nt|} | | dd | | i|g} |rg|D]d\}}} | tj dd|ddfd| tt|ett| d|d |d!t jd"d#$} t$| t$dS)%aH Log performance metrics in a clean table and optionally send to global collector. Args: task_slug: Slug of the task that generated these metrics task_name: Name of the task that generated these metrics metrics: Dictionary of metric names and (value, unit) tuples title: Table title _NblobTz bold green)rr;) show_header header_styler paddingMetriccyan)stylewidthValuerightyellow)justifyrhriUnitdimmsz.0fsz.3f z r1:boldz z [bold green]z - z[/])r;r`P)r]r rdri)r*getpopCOLLECT_METRICS_LOCALrr2r add_columnstradd_rowreplacer]rassemblerr r ROUNDEDrArB)r[rHr\r] blob_metricsnon_blob_metricsmetrictablerJrKformatted_value content_items_unitpanels rlog_performance_metricsrs**y**I  " "9 - -g!%%i44w>G>y)W===>@LACSSf,,2BJJ6RRRR !     E  XV2666 WgXRHHH V5222/OOt 4<HH     c%jj!1!1 2 2 2 2  }5U55y555 K    E MM% MMOOOOOrpathcf|dkr*|dr|dS|S)N/)endswithrstrip)rs rnormalize_template_pathr7s2 s{{t}}S)){{{3 Krrequestc|jd}|r%t|ddrt|jSdS)Nrouterunknown)scoperxgetattrrr)rrs rget_route_templater=sG M  g & &E 3--3&uz222 9rr)NrZ)6__doc__datetime collectionsrcollections.abcrtypingrrrfastapirlangfuser richr rich.consoler r r rich.panelr rich.tabler rich.textr rich.treer src.configrsrc.telemetry.metrics_collectorrsrc.utils.representationrrArzrrrr|rNr*listtupleintfloat__annotations__r8rGrQrYrrrrrrrs ######$$$$$$//////////7777777777BBBBBB '    6 IcNN GCLL  1a4. ad^   5 5x1~A./555 5#'     1a4.4   * ad^hA/!Q$?@@    NKMM[d5cCi%6G1L+M&N!NO ("( ((!) (  ((((V" ,OO O 9u O  O  OOOO<)K)K)K)K! )K "% )K  )K)K)K)K^@D" IIII%S3Y.34 5 <I  I  IIIIX## Cr