+j,:UdZddlZddlZddlZddlmZmZmZeje Z dddddd Z ee e fe d <d Zdd ee d ee fdZ dde d ee d eejfdZdeejd ee effdZGddZdS)u LightClaw — per-turn token usage tracker. Hermes stores token counts in its SQLite ``state.db`` ``sessions`` table, but the columns are session-cumulative. We mirror openclaw's per-turn ``usage`` frame by computing ``current_cumulative − turn_start_baseline``: the whole-turn total (every LLM call of one question summed). Note: openclaw reports only the last LLM call's usage; we report the whole-turn sum. The wire format (openclaw ``UnifiedUsage``, camelCase) is identical, so front-end / ai-server need zero hermes-specific code. Baselines live in memory keyed by ``chat_id``. If lost mid-turn (e.g. process restart), we skip the frame rather than over-report the session total. N)DictOptionalTuple input_tokens output_tokenscache_read_tokenscache_write_tokensreasoning_tokens) inputTokens outputTokenscachedInputTokenscacheWriteTokensreasoningTokens_TOKEN_COLUMNShermes sessions_dirreturncfg}|rqtjtj|}|r3|tj|dtjdx}r3|tj|d|tjd ttj dD]}|d|dn#t$rYnwxYw|dt}|D]@}|r<||vr8| |tj|r|cSAdS) uLocate the Hermes ``state.db`` — never hardcoding a path or username. Resolution order (most authoritative first), kept consistent with the rest of the plugin (``adapter._sessions_dir`` / ``history._default_sessions_dir``) and the install scripts: 1. Sibling of *sessions_dir*: ``/sessions`` → ``/state.db``. This is the path the adapter already resolved at runtime (config ``extra.sessions_dir`` → ``LIGHTCLAW_SESSIONS_DIR`` → ``$HERMES_HOME/sessions``), so it honours custom install locations automatically. 2. ``$HERMES_HOME/state.db`` — the canonical env var used by every ``hermes_*.sh`` script and by the adapter. 3. ``~/.hermes/state.db`` for the current user. 4. Last-resort scan of real home dirs (``/home/*/.hermes`` and ``/root/.hermes``), mirroring ``hermes_install.sh``'s own probe — covers container setups where the gateway runs under a user that differs from ``$HOME``, without ever assuming ``ubuntu``. Returns the first existing file, or ``None`` (caller treats that as "no usage data available" and skips — never blocks the main path). zstate.db HERMES_HOMEz~/.hermes/state.dbz/homez/home/z/.hermes/state.dbz/root/.hermes/state.dbN)ospathdirnamenormpathappendjoinenvironget expandusersortedlistdirOSErrorsetaddisfile)r candidatesparent hermes_homeentryseenrs A/home/agentuser/.hermes/plugins/lightclawbot/src/usage_tracker.py_resolve_state_db_pathr+'s.J@!1!1,!?!?@@  @   bgll6:>> ? ? ?jnn]333{A"',,{J??@@@bg(()=>>??? BJw//00 A AE   ?u??? @ @ @ @ A      .///D  D$$ HHTNNNw~~d##  4s==D;; EEchat_idczt|}|stddSd} tj|} tj|_|}|d|dt|f| }|-|d|d| }|| S#| wxYw#tj $r&}t d|Yd}~dSd}~wwxYw)uRead the latest ``sessions`` row for *chat_id* (falling back to the absolute-latest row). Returns ``None`` on any error — usage is best-effort and must never break the outbound path.z%[lightclaw] usage: state.db not foundNz[input_tokens, output_tokens, cache_read_tokens, cache_write_tokens, reasoning_tokens, modelzSELECT zA FROM sessions WHERE user_id = ? ORDER BY started_at DESC LIMIT 1z/ FROM sessions ORDER BY started_at DESC LIMIT 1z,[lightclaw] usage: sessions query failed: %s)r+loggerinfosqlite3connectRow row_factorycursorexecutestrfetchonecloseErrorwarning)r,rdb_pathcolsconncurrowexcs r*_read_session_rowrA_s\ %\22G  ;<<<t  w'' &{D ++--C KKE$EEEW    ,,..C{ 7d777llnn JJLLLLDJJLLLL =EsKKKttttts0DBC,D,DDD:D55D:r?c$dtD}||St|}tD]?\}}||vr6||}t |t t frt |||<@|S)uBExtract cumulative token counters from a row (missing/NULL → 0).ci|]}|dS)r).0fields r* z$_row_to_counters..sEEEUqEEE)rr"keysitems isinstanceintfloat)r?countersrIrFcolumnvalues r*_row_to_countersrQsEEnEEEH { sxxzz??D'--//-- v T>>KE%#u.. -"%e** OrHceZdZdZd deeddfdZdeddfdZdedeee ee feffdZ dedee ee ffd Z deddfd Z dS) SessionUsageTrackerzATracks per-turn token deltas against session-cumulative counters.Nrrc0||_i|_i|_dSN) _sessions_dir _baselines_models)selfrs r*__init__zSessionUsageTracker.__init__s-95713 rHr,ct||j}t||j|<d}|+dt |vr|d}||j|<dS)zRecord cumulative counters as the baseline for a new turn (called at inbound). An unreadable DB yields an all-zero baseline, correct for a fresh session's first turn.Nmodel)rArVrQrWr"rIrX)rYr,r?r\s r*snapshot_baselinez%SessionUsageTracker.snapshot_baselinesg );<<#3C#8#8  ?w#chhjj//99LE % WrHc|j|}|td|dSt ||j}|td|dSt |}i}tD]<}||d||dz }|dkr|nd||<=|d}|d} |dkr| dkrdS|| || ztd } d D]}||dkr ||| |<|j |} t| tr| r| | d <| d fS) uClassify the round that just finished and (when real) return its usage. The cumulative token counters in Hermes' ``state.db`` are the only deterministic signal we have, so the verdict is driven entirely by whether — and by how much — they moved against the turn-start baseline. The return is ``(usage, reason)`` where ``reason`` is one of: * ``"usage"`` — baseline present, DB read OK, **delta > 0**. A real conversation turn ran an LLM call; ``usage`` is the openclaw ``UnifiedUsage`` object (whole-turn ``current − baseline``). * ``"no_llm"`` — baseline present, DB read OK, **delta == 0**. The counters did not move, so no LLM call ran. These are framework *command* rounds (``/new`` / ``/approve`` / ``/always``) that are never stored as transcript turns; the caller MUST NOT write a usage sidecar entry — a phantom line shifts the ordinal sidecar↔turn join and drops a real turn's usage on history reload. ``usage`` is ``None``. * ``"unknown"`` — we could **not measure** this turn: the baseline was lost (mid-turn restart) or the ``state.db`` row could not be read this instant. This is NOT proof the turn was free; it may be a real turn we simply failed to measure. The caller should write a placeholder to hold this turn's slot rather than silently dropping it. ``usage`` is ``None``. Crucially this distinguishes *measured zero* (``no_llm``) from *could-not-measure* (``unknown``): collapsing them — as a plain baseline check would — risks misreading a real turn that hit a transient DB read failure as a free command round. Nzi[lightclaw] usage unknown: no baseline for chat_id=%s (inbound snapshot missing, likely mid-turn restart))Nunknownzd[lightclaw] usage unknown: state.db row unreadable for chat_id=%s (cannot measure this turn's delta)rr r )Nno_llm)r r totalTokensprovider)r rrr\usage) rWrr.r/rArVrQr _PROVIDERrXrKr6) rYr,baseliner?currentdeltarFdiffrrrcr\s r* classify_turnz!SessionUsageTracker.classify_turnsD?&&w//   KKF    #?  );<< ; KK@    #?"3'' "# 3 3E;;ua((8<<q+A+AAD#'!8844E%LL]+ n-  1  !!3!3!>()'-7! $ $ R , ,EU|a$U|e    )) eS ! ! #e #"E'Ng~rHc8||dS)zReturn the per-turn usage object, or ``None`` if not emittable. Thin wrapper over :meth:`classify_turn` for callers that only need the usage value and not the reason it was (or wasn't) produced. r)rirYr,s r*compute_turn_usagez&SessionUsageTracker.compute_turn_usages !!'**1--rHcr|j|d|j|ddS)z/Drop any cached baseline / model for *chat_id*.N)rWpoprXrks r*clearzSessionUsageTracker.clears8 GT*** $'''''rHrU)__name__ __module__ __qualname____doc__rr6rZr]rrobjectrirlrorDrHr*rSrSsKK44Xc]4d4444 & & & & & &WW xS&[)*C/ 0WWWWr.#.(4V ;L2M....(S(T((((((rHrSrU)rsloggingrr0typingrrr getLoggerrpr.rr6__annotations__rdr+r2rArLrQrSrDrHr*rys}  ((((((((((  8 $ $ ))-., ""S#X  55#5(3-5555r15&& & ( & gk&&&&R (7;/ DcN    {({({({({({({({({({(rH