`gjUdZddlmZddlZddlmZddlmZmZeje Z eddZ de d <ed d hZdd Zdd ZddZddd ddZy)u Ambient session-accounting context for auxiliary LLM calls. Auxiliary calls (vision, compression, title generation, web_extract, session_search, ...) funnel through ``agent.auxiliary_client`` which has no session handle — so their token usage was historically discarded, leaving dashboard analytics blind to aux model spend (issue #23270). Instead of threading ``session_db``/``session_id`` parameters through every aux call site, the agent loop publishes them here (mirroring the Nous Portal conversation context in ``agent.portal_tags``) and the auxiliary client records usage at its single response-validation chokepoint. ContextVar semantics give us the right isolation for free: * concurrent agents in one process (gateway sessions, delegate subagents) never see each other's accounting context; * worker threads spawned via ``tools.thread_context.propagate_context_to_thread`` (MoA fan-out, background review) inherit the parent turn's context; * asyncio tasks inherit the context of the code that created them. MoA reference/aggregator slots are explicitly EXCLUDED from recording: ``agent/conversation_loop.py`` already folds MoA advisor usage and cost into the main loop's ``update_token_counts`` delta, so recording them here would double-count (see ``_EXCLUDED_TASKS``). ) annotationsN) ContextVar)AnyOptionalaux_accounting_context)defaultzContextVar[Optional[tuple]] _accounting moa_referencemoa_aggregatorcb||stjdStj||fS)a(Publish the active session's accounting handles for aux usage recording. Called by the agent loop at turn entry. Returns the ContextVar token so callers can ``reset_accounting_context(token)`` on turn exit. Publishing ``None`` handles (no DB / no session id) clears the context. N)r set) session_db session_ids G/root/.hermes/venv/lib/python3.12/site-packages/agent/aux_accounting.pyset_accounting_contextr.s/t$$ ??J 3 44cx tj|y#t$rtjdYywxYw)z@Restore the previous accounting context (pair with ``set_...``).N)r reset Exceptionr )tokens rreset_accounting_contextr:s0% s 99c*tjS)zEReturn ``(session_db, session_id)`` for the active turn, or ``None``.)r getrrget_accounting_contextrBs ?? rproviderbase_urlc  |r|tvrytj}|y|\}}t|dd}|yddlm}m} | ||} | js1| js%| js| js | jsytt|ddxsdxsd} d} || | || } | jt| j} |j%||| ||| j| j| j| j| j| y#t$rt j#d d YqwxYw#t$rt j#dd YywxYw)uRecord an auxiliary response's token usage against the ambient session. Called from the auxiliary client's response-validation chokepoint. Strictly best-effort: any failure is swallowed (accounting must never break an aux call). No-ops when: * no accounting context is published (call is outside any agent turn), * the task is main-loop-accounted (MoA slots — see ``_EXCLUDED_TASKS``), * the response carries no usage object. The model is read from ``response.model`` (accurate even after the aux client's provider-fallback chains); *provider*/*base_url* reflect the originally-resolved route and are best-effort. Nusager)estimate_usage_costnormalize_usage)rmodelunknownrz Aux usage cost estimation failedT)exc_info) r#billing_providerbilling_base_url input_tokens output_tokenscache_read_tokenscache_write_tokensreasoning_tokensestimated_cost_usdz&Aux usage recording failed (non-fatal))_EXCLUDED_TASKSr rgetattragent.usage_pricingr!r"r)r*r+r,r-str amount_usdfloatrloggerdebugrecord_auxiliary_usage)responsetaskrrctxrr raw_usager!r"r r#estimated_costcosts rrecord_aux_usager>Gsw*.Nt. oo ; !$ JHgt4   L H=   %"5"5&&%*B*B%% GHgr28b9FY L&ux(D*!&t!7 ))  %%++--#55$77"33- *  L LL;dL K L N = MNsL E E E AE  E )-D%AE % EE EE E.-E.)rrr Optional[str])returnNone)r@zOptional[tuple]) r8rr9r?rr?rr?r@rA)__doc__ __future__rlogging contextvarsrtypingrr getLogger__name__r5r __annotations__ frozensetr/rrrr>rrrrKs4#"   8 $,6d, (_.>?@ 5#" CNCN CN CN  CN  CNr