+ WjgRt^RIHt^RIt^RIt^RIt^RIt^RIt^RIt^RI H t H t H t H t Ht]P!R4tR tRRlt]!RR7t]!R R7tR t]!R4t!R R 4tR#)zToken/usage accounting mixin for SessionDB: the coalescing background token writer, per-model usage rows, and billing-route columns. Writer thread state lives on the instance.) annotationsN)AnyDictListOptionalTuple hermes_statec V^8dQhRRRR/#)deltaboolreturnstr)formats"FC:\Users\johnn\AppData\Local\hermes\hermes-agent\hermes_state_usage.py __annotate__rsTcc aaRV3RlloRV3RllpRPV3Rl\44pS'dRMRpR V,R V R V!R 4 R V!R4 R2,#)z``UPDATE sessions`` for one usage report: *delta* adds to the stored counters (CLI per-call path), otherwise sets them (gateway cumulative path). Cost/route columns COALESCE-fill either way (statement text is pinned by the SQL trace harness).c V^8dQhRRRR/#r colrr r)rs"rr'_token_update_sql..__annotate__s....rc"<S'dV R2#R#)z + ??rrr s&radd_token_update_sql..adds$#d|-#-rc V^8dQhRRRR/#rr)rs"rrrs;;#;#;rc$<S'dRV R2#R#)z COALESCE(z, 0) + ?rrrs&radd0_token_update_sql..add0s,13%x(:s:rc3B<"TFpRV RS!V4 R2xK R#5i)z z = z, Nr).0crs& r $_token_update_sql..s%X1,QCs3q6(#>sz0COALESCE(estimated_cost_usd, 0) + COALESCE(?, 0)zCOALESCE(?, 0)zUPDATE sessions SET z( estimated_cost_usd = z, actual_cost_usd = CASE WHEN ? IS NULL THEN actual_cost_usd ELSE actual_cost_usda END, cost_status = COALESCE(?, cost_status), cost_source = COALESCE(?, cost_source), pricing_version = COALESCE(?, pricing_version), billing_provider = COALESCE(billing_provider, ?), billing_base_url = COALESCE(billing_base_url, ?), billing_mode = COALESCE(billing_mode, ?), model = COALESCE(model, ?), api_call_count = api_call_countz WHERE id = ?)join_TOKEN_COUNTERS)r r counters estimatedrsf @r_token_update_sqlr.s..;;wwXXXHFKBQaI(*6ykB""345 6%&**:%;$<=  # #rF)r TaINSERT INTO session_model_usage ( session_id, model, billing_provider, billing_base_url, billing_mode, task, api_call_count, input_tokens, output_tokens, cache_read_tokens, cache_write_tokens, reasoning_tokens, estimated_cost_usd, actual_cost_usd, cost_status, cost_source, first_seen, last_seen ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ON CONFLICT(session_id, model, billing_provider, billing_base_url, billing_mode, task) DO UPDATE SET api_call_count = api_call_count + excluded.api_call_count, input_tokens = input_tokens + excluded.input_tokens, output_tokens = output_tokens + excluded.output_tokens, cache_read_tokens = cache_read_tokens + excluded.cache_read_tokens, cache_write_tokens = cache_write_tokens + excluded.cache_write_tokens, reasoning_tokens = reasoning_tokens + excluded.reasoning_tokens, estimated_cost_usd = estimated_cost_usd + excluded.estimated_cost_usd, actual_cost_usd = actual_cost_usd + excluded.actual_cost_usd, cost_status = COALESCE(excluded.cost_status, cost_status), cost_source = COALESCE(excluded.cost_source, cost_source), last_seen = excluded.last_seenc^]tRt^RtRtRR/RRlltRRltRR ltR5R R lltR R lt RRlt RRlt R6RRllt RRlt R7RRlltR8RRlltRRRRRRRRR^R ^R!^R"^R#^R$RR%RR&RR'RR(^R)R/R*R+lltRRRRRRR^R ^R!^R"^R#^R$RR(^/ R,R-lltR.R/ltR0^R1R/R2R3lltR4tR#)9SessionUsageMixinz!N[! !rc baaaaVP4VVVV3RlpVPV4R#)aUnconditionally set the billing route (``update_token_counts`` only COALESCE-fills NULLs) so the dashboard reflects the latest /model switch. Route writers never touch the stored prompt; ``_stored_prompt_matches_runtime`` decides staleness. See #48173, #48248. c4<VPRSSSS34R#)zUPDATE sessions SET billing_provider = ?, billing_base_url = ?, billing_mode = COALESCE(?, billing_mode) WHERE id = ?N)execute)connr5r1r4r3s&r_do;SessionUsageMixin.update_session_billing_route.._dobs# LL#&.xz$R  TrN)flush_token_counts_execute_write)selfr3r4r5r1r=s&fddd rupdate_session_billing_route.SessionUsageMixin.update_session_billing_routeUs- ! T T C rc V^8dQhRRRR/#)r r3rr r7r)rs"rrr8js;;S;t;rc aVP;_uu_4VPpVRJ;'dVP4pVP;'dV'*pV'gVPP W34V'g\ P!VPRRR7pW0nVP4VPf<\P!V4oRV3RllpW`n \P!V4VPP4RRR4X'dVP !V3/VBR#R# +'giL/;i)zEnqueue a token/cost delta for the background writer (same kwargs as :meth:`update_token_counts`). After close() stopped the writer, falls back to the synchronous path and may raise.Nzsession-db-token-writerT)targetnamedaemoncV^8dQhRR/#r r r7r)rs"rr:SessionUsageMixin.queue_token_counts..__annotate__~s@@@rcB<S!4pVeVP4R#R#N)_drain_token_queue_at_exit)dbself_refs r_drain_at_exit._drain_at_exit~s !)B!~ " = = ? .r)_token_queue_cond_token_writer_threadis_alive_token_writer_stop _token_queueappend threadingThread_token_writer_loopstart_token_atexit_hookweakrefrefatexitregister notify_allupdate_token_counts)rAr3kwargsthread writer_alivewriter_stoppedrQrPs&&, @rqueue_token_counts$SessionUsageMixin.queue_token_countsjs # # #..F!-CC&//2CL!44II\9IN!!!((*)=>#'--#66=V_ceF06-LLN..6#*;;t#4@@ 3A/7&&113/$0   $ $Z :6 : 1$ #s;E E "CE  E cV^8dQhRR/#rJr)rs"rrr8s44T4rc  zVPV4VP;_uu_4RVnVPP4RRR4R# +'giR#;i TP;_uu_4RTnTPP4RRR4i +'gii;i;i)zKApply a batch whose ``busy`` flag the caller already claimed, then release.FN)_apply_token_batchrS_token_writer_busyrb)rAbatchs&&r_apply_claimed_batch&SessionUsageMixin._apply_claimed_batchs~ 4  # #E *'''*/'&&113(''''''*/'&&113('''s.A'"A A$ 'B:<"B'  B:' B7 2B:c V^8dQhRRRR/#)r timeoutfloatr r r)rs"rrr8s%$rc VP'gVP'gR#RpVP;_uu_4\P!4V,pVP'gVP'dVP pVeVP 4'gJVP'g8RVn\VP4pVPP4MKV\P!4, pV^8:d RRR4R#VPPV4KRRR4V'dVPV4R# +'giL+;i)zBlock until every queued token delta has been applied. False on timeout (callers then read totals stale by the queued deltas). Never raises.TNF) rWrmrStime monotonicrTrUlistclearwaitro)rArrrndeadlinere remainings&& rr?$SessionUsageMixin.flush_token_countss    )@)@)@  # # #~~''1H###t'>'>'> 22N&//*;*;TE\E\E\.2D+ !2!23E%%++-$t~~'77 > $ # &&++I6!$"   % %e ,'$ #s*.E,E>%E$E6AEE E+ cV^8dQhRR/#rJr)rs"rrr8s--D-rc xVP;_uu_4\P!4VP,pVP'geVP 'gSV\P!4, pV^8:dRVnRRR4R#VPPV4KvVP'gRVnRRR4R#RVn\VP4pVPP4RRR4VPX4EK( +'giL$;i)TN) rSrurv_TOKEN_WRITER_IDLE_SECONDSrWrVrTryrmrwrxro)rA idle_deadliner{rns& rr[$SessionUsageMixin._token_writer_loops''' $ 043R3R R +++D4K4K4K -0@ @I A~591('**// :(((04D-('+/'T../!!'')#($  % %e ,%('s)8D)D)!+D).D)D)6D)) D9 c V^8dQhRRRR/#)r rn list[tuple[str, dict[str, Any]]]r r7r)rs"rrr8s i i(H iT irc &VPV4pVFwrEVP!V3/VBK R# \d#p\PRT4TpRp?LIRp?ii;i \d"p\PRYC4Rp?KvRp?ii;i)zBApply queued deltas in order, coalescing where safe. Never raises.z?async token accounting: coalesce failed, applying raw batch: %sNz5async token accounting: apply failed (session=%s): %s)_coalesce_token_deltas Exceptionloggerwarningrc)rArn coalescedexcr3rds&& rrl$SessionUsageMixin._apply_token_batchs 33E:I #, J i((>v>#,   NN\^a bI  iVXbhh is+4A$ A!AA!$ B/B  Bc V^8dQhRRRR/#)r rnrr r)rs"rrr8s44,L4Qq4rc a .pVEFwpo RpS PR4'gV.V 3RlVP4O5pV'dVeVR,^,V8XdVR,^,pVPF.pVPV^4S PV^4,WV&K0 VPF>pS PV4pVfKVPV4;'gRV,WV&K@ KVP WC\ S 434EK VUU U u.uF wrqV 3NK up p p#uup p pi)zMerge adjacent incremental deltas with an identical route, so ordering across sessions and /model switches is preserved exactly. absolute=True never merges.Nabsolutec3F<"TFpSPV4xK R#5irM)get)r$frds& rr&;SessionUsageMixin._coalesce_token_deltas..s$[#(-33Fjabb F333s&D:c V^8dQhRRRR/#)r join_timeoutrsr r7r)rs"rrr8s"-"-u"-"-rc  VP;_uu_4RVnVPP4VPpRRR4XekVP 4'dUVP VR7VP 4'd-\ PRV\VP44R#VP;_uu_4\P!4V,pVP'dvV\P!4, pV^8:d6\ PRV\VP44RRR4R#VPPV4K\VP4pV'd"RVn VPP4RRR4X'dVP!V4R#R# +'giEL;i +'giLA;i)z@Stop the writer thread and drain remaining deltas. Never raises.TN)rrzZasync token accounting: writer did not stop within %.0fs; %d queued delta(s) not persistedzfasync token accounting: concurrent drain did not finish within %.0fs; %d queued delta(s) not persisted)rSrVrbrTrUr*rrlenrWrurvrmryrwrxro)rArrerzr{rns&& r_stop_token_writer$SessionUsageMixin._stop_token_writersu  # # #&*D #  " " - - /..F$  &//"3"3 KK K -  78Dc$J[J[F\^ # # #~~',6H)))$t~~'77 >NNP$c$*;*;&<>$ #&&++I6**+E*.'!!'')$   % %e , ?$ # #$ #s%.G .G( A Fr"c(hV^8dQhRRRRRRRRRRR RR RR R R R RRRRRRRRRRRRRRRRRRRRRR/#)r r3r input_tokensint output_tokensmodel str | Nonecache_read_tokenscache_write_tokensreasoning_tokensestimated_cost_usdOptional[float]r( cost_statusr6 cost_sourcepricing_versionbilling_providerbilling_base_urlr1r)rr sourcetaskr r7r)rs"rrr8sooo-0oCFoQ[ouxoo58oP_o)o=Jo^ko' o@M ofs o $ o ;> o LP o  o +. o orc aaa\\44oSPR4SPR4SPR4SPST;'gRVR7SP VVV3Rl4R#)aUpdate totals and route attribution, ensuring the legacy missing-row backfill. *absolute*=False increments (per-API-call deltas, CLI path); *absolute*=True sets directly (gateway path, where the cached agent holds cumulative totals). ``source`` is the session's real surface for the row-existence guard; callers that don't know it leave the placeholder. ``task`` names a per-turn route (``voice_chat``): its calls count in the session totals and land in ``session_model_usage`` under that task, but never become the session's recorded route.rAr3runknown)rc,<SP!VS3/SB#rM)#_update_token_counts_in_transaction)r<rAr3valuess&r7SessionUsageMixin.update_token_counts..)s)Q)QRVXb)mfl)mrN)rlocalspop_insert_session_rowr@)rAr3rrrrrrrr(rrrrrr1r)rrrrsff&&&&&&&&&&&&&&&&&&@rrc%SessionUsageMixin.update_token_countssbfh 6 <  8   V-@-@y N mnrc&dV^8dQhRRRRRRRRRRR RR RR R R R RRRRRRRRRRRRRRRRRRRR/#)r r3rrrrrrrrrrrr(rr6rrrrr1r)rr rr r7r)rs"rrr8+s1K1K #1K361KIL1KWa1K{~1K1K581KP_1K)1K=J1K^k1K' 1K@M 1Kfs 1K $ 1K ;> 1K LP 1K ^a 1K  1Krc 2\4P4UUu/uFwppV\9gKVVbK pppV'd\M\p\ T;'g5T;'g+T;'g!T;'gT;'g T;'gT 4p\ T;'gT 4pY4YgTYYYV'dTMRV'dTMRV'dTMRV'dTMRVV3pV'*;'dTpVP RV34P4pVe \V4M/pV'*;'d\VPR4;'g^4^8H;'daT;'dW\ V4;'dD\ V4;'d1VPR4V8g;'gVPR4V8gpV'dVP RW^VVV34VP VV4V'dVP!W3RV/VBR#R#uuppi)NzISELECT model, billing_provider, api_call_count FROM sessions WHERE id = ?r)rrzUPDATE sessions SET model = ?, billing_provider = ?, billing_base_url = ?, billing_mode = ? WHERE id = ?r) ritems_MODEL_USAGE_FIELDS_TOKEN_UPDATE_ABSOLUTE_SQL_TOKEN_UPDATE_DELTA_SQLr r;fetchonerrr_record_model_usage)rAr<r3rrrrrrrr(rrrrrr1r)rrkvusagesql has_usagehas_accounted_usageparamsrecord_model_usagerowexistingfirst_accounted_routes&&&&&&&&&&&&&&&&&&&& rr5SessionUsageMixin._update_token_counts_in_transaction+s;#)(.."2O"2$!Qa;N6NA"2O,4(:QAAAA:KAAOaAAeuAA*AA.@B "9#?#?@ ):P` { 3  3 /LTDW5]a J ( #+l99 ll WZdYf (* !$49R H i iX\\*:;@@qAQF i iK^ i icghmcn i i%& i ig&%/gg8<<@R3SWg3g  ! LL#&+>NP\^h$i k S&!   $ $T JD JE J SPs HHrrrrrrrrrr(rrr)rc"\V^8dQhRRRRRRRRRRRR R R R R R R R R RRRRRRRRRR RRRR/#)r r3rrr6rrr1rrrrrrrrr(rrr)rr r7r)rs"rrr8^srr #r/<rUbr'r._writesu||?*OXXZb1$j)T $ 6 6 888 \ *-/  $ $T: JD JE Jr) patience_s)rrrrhermes_state_worker_lifecyclerr@_TRANSCRIPT_WRITE_PATIENCE_S)rAr3rrrrrrrrrrr)rrrrrsfff$$$$$$$$$$ @@rrecord_auxiliary_usage(SessionUsageMixin.record_auxiliary_usage|s#)(.."2O"2$!a;N6N"2O '5'=!3~CV T K K Ft/P/PQ%Ps BBc V^8dQhRRRR/#)r r3rr zdict[str, dict[str, float]]r)rs"rrr8s]]#]:U]rc  V'g/#VPV4p\\V4^, RR4F&pVPW#,4'gK"W#RpM VP RRP R\V4,4 R2V4pVUUu/uF9qUR,VP 4Uu/uFqfR8wgK WeV,bK upbK; upp#uupiuuppi)a Per-task auxiliary usage (``task != ''``: vision, compression, title_generation, ...) summed over the session's compression lineage. Aux calls bill to the id the turn STARTED with while compression mints child ids mid-turn, so a single-id read misses rows (#112848).NaySELECT task, COALESCE(SUM(api_call_count), 0) AS api_calls, COALESCE(SUM(input_tokens), 0) AS input_tokens, COALESCE(SUM(output_tokens), 0) AS output_tokens, COALESCE(SUM(cache_read_tokens), 0) AS cache_read_tokens, COALESCE(SUM(cache_write_tokens), 0) AS cache_write_tokens, COALESCE(SUM(reasoning_tokens), 0) AS reasoning_tokens, COALESCE(SUM(estimated_cost_usd), 0) AS estimated_cost_usd FROM session_model_usage WHERE session_id IN (,rz/) AND task != '' GROUP BY taskrr)_session_lineage_root_to_tipranger_is_explicit_branch_session _read_allr*keys)rAr3chainirowsrrs&& rauxiliary_usage_by_task)SessionUsageMixin.auxiliary_usage_by_tasksI11*=s5zA~r2.A//99b /~~ '(+xxc%j0@'A&BC "   X\\W[PSF KAF{iaQiKKW[\\K\sC%;C  C C% C%min_message_countinclude_archivedc$V^8dQhRRRRRR/#)r rrrr r zdict[str, float]r)rs"rrr8s' N N ND N]m Nrc RR.pV.pV'gVPR4VPRRPV4 R2V4pR\V^,;'g^4R\ V^,;'gR 4/#) zTokens and spend across the whole store (one scan), so the sidebar total does not shrink with paging. Spend prefers the billed figure over the estimate.zparent_session_id IS NULLzmessage_count >= ?zCOALESCE(archived, 0) = 0z SELECT COALESCE(SUM(COALESCE(input_tokens, 0) + COALESCE(output_tokens, 0)), 0), COALESCE(SUM(COALESCE(actual_cost_usd, estimated_cost_usd, 0)), 0) FROM sessions WHERE z AND z tokenscost_usdr)rX _read_oner*rrs)rArrwhererrs&$$ r usage_totalsSessionUsageMixin.usage_totalss-.BC./ LL4 5nn"LL'()    #c!fkk*Jc!fmm8LMMr)r]rmrVrT)g@)g$@)rNrrrNNNNNNNNrFNr")rrNrrrNNNNNNNNrFr")__name__ __module__ __qualname____firstlineno____doc__rBrhror?r[rlrrrNrcrrrrr__static_attributes__rrrr0r0RsF!^b!*;@48-, i4*"-H&o41Kfr=Arcgr(,rJNrbcrr67rQRrjkr-1 rTX r $( r EI r _` r ln rrs_#  33   > *r:/U;+$750 !GH wNwNr