`gjdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ejeZd dZd dZd d Z dd Zy)uFull MoA turn trace persistence (opt-in via config ``moa.save_traces``). When enabled, every Mixture-of-Agents turn that actually runs the reference fan-out (a cache MISS in ``MoAChatCompletions.create``) appends one JSON line to ``/moa-traces/.jsonl``. The record is the TRUE FULL turn — the exact messages array each reference model received (system prompt + advisory view, not the truncated display preview), each reference's full output, and the exact messages array the aggregator received (including the injected reference-context guidance block) plus its output when available — so a run can be audited end-to-end offline: what every model saw, what every model said, and what it cost. This is a side-channel trace. It is NOT the conversation ``messages`` table and never enters message history or replay — MoA references are advisory side-calls with their own system prompt, not conversation turns, so persisting them as message rows would corrupt role alternation / replay. Traces live in their own files, keyed by session id, and are safe to delete. Cost model note: gated OFF by default. When off, the only overhead is the ``_traces_enabled()`` config read (cheap) — no file I/O, no serialization. ) annotationsN)Path)AnyOptional)get_hermes_homec r ddlm}|xsijdxsi}|jdsy|jd}|rPt t j jt j jt|}|Stdz }|S#t$rYywxYw)a1Return the trace directory if ``moa.save_traces`` is on, else None. Reads config lazily per call (config is cheap to load and this only runs on a cache-MISS MoA turn, i.e. once per user turn, not per tool iteration). ``moa.trace_dir`` overrides the default ``/moa-traces/``. r) load_configmoaN save_traces trace_dirz moa-traces) hermes_cli.configr get Exceptionrospath expandvars expanduserstrr)r moa_cfgoverridebases B/root/.hermes/venv/lib/python3.12/site-packages/agent/moa_trace.py_traces_enabled_and_dirr%s1=&B++E28b ;;} %{{;'HBGG&&rww'9'9#h-'HIJ K </ K s$B** B65B6cJ|sydjdt|DS)z/Make a session id safe as a filename component.zunknown-sessionc3LK|]}|js|dvr|ndyw)z-_._N)isalnum).0cs r z'_sanitize_session_id..@s#VU 1<Vs"$)joinr) session_ids r_sanitize_session_idr$<s!  77Vc*oV VVclt|dd}i}|?t|ddt|ddt|ddt|ddt|ddd }|t|d dt|d dt|d dt|d dt|dd|t|ddt|ddt|ddd S)uRender one reference's _RefAccounting into a full trace dict. Includes the FULL input messages the reference received and its FULL output — not the truncated display preview. usageN input_tokensr output_tokenscache_read_tokenscache_write_tokensreasoning_tokens)r(r)r*r+r,modelprovider temperaturemessagesoutputcost_usd cost_status cost_source) labelr-r.r/input_messagesr1r'r2r3r4)getattr)acctr5r' usage_dicts r _slot_tracer:Cs D'4 (E!#J #E>1=$UOQ?!(0CQ!G")%1Eq"I '/A1 E  w-D*d3t]D9!$ D9$$/D*d3t]D9t]D9  r%c 0t} | y | jdd| t|dz } t|} | sd} n| rd} nd} t j|||Dcgc]\}}}t ||c}}}||||||| | dd }| j d d 5}|jtj|d tdzdddycc}}}w#1swYyxYw#t$r!}tjd||Yd}~yd}~wwxYw)uAppend one full MoA turn record to the session's trace JSONL, if enabled. Best-effort: any failure is logged at debug and swallowed — tracing must never break a live turn. Called once per turn on a reference cache MISS. ``aggregator_output`` is the aggregator's synthesized text. On the non-streaming path (eval / quiet-mode / subagents) it was captured inline at call time. On the streaming path it is captured after the fact from the caller's resolved assistant text (``aggregator_output_fallback`` in ``consume_and_save_trace``) so the trace is self-contained either way; if that resolved text was unavailable, it falls back to None and the record points at the session store via ``output_location``. NT)parentsexist_okz.jsonlinlineinline_from_streamassistant_message_in_session_db)r5r-r.r/r6r1streamedoutput_location)tsr#preset references aggregatorazutf-8)encodingF) ensure_asciidefault z'MoA trace write failed (session=%s): %s)rmkdirr$booltimer:openwritejsondumpsrrloggerdebug)r# preset_namereference_outputsaggregator_labelaggregator_modelaggregator_provideraggregator_temperatureaggregator_input_messagesaggregator_outputaggregator_streamedrr _have_output_output_locationr5_textr8recordfexcs r save_moa_turnrdasC4 # $D |)Q 4$ /-j9:&AA-. "'  3 @ ))+$!+<&E5$D%( *)/5";+/$4 2YYsWY - P GGDJJvE3G$N O P P+* P P Q > CPPQsHAC+%C<$C+ /CC+C+C($C+(C++ D4DD)returnzOptional[Path])r# Optional[str]rer)r8rr5rrezdict[str, Any])r#rfrUrrVzlist[tuple[str, str, Any]]rWrrXrfrYrfrZrr[rr\rfr]rMreNone)__doc__ __future__rrQloggingrrNpathlibrtypingrrhermes_constantsr getLogger__name__rSrr$r:rdr%rrqs,#   ,   8 $.W