"֞iQ$UdZddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z eje Ziddd d d d dd d ddd ddd d dd dd dddd dddd dddd dddd dddd dddd dddd dddd dd d d d!d d d d"d#d d d$d%d d Zd&ed'<d(dd Zd)ed*<d+Zd2d,ZGd-d.Zdad/ed0<d3d1Zy)4aO core/observability/cost_tracker.py LLM cost attribution and tracking for Genesis observability layer. This module complements ``core/cost_tracker_v2.py`` (which gates execution against budget limits) by providing Langfuse-aligned session/agent/customer cost attribution and a JSONL audit trail. Pricing is per 1M tokens (industry-standard notation). Usage: from core.observability.cost_tracker import CostTracker tracker = CostTracker() cost = tracker.record( model="gemini-flash", input_tokens=1200, output_tokens=400, session_id="sess-abc", agent_id="scout-01", customer_id="cust-xyz", ) print(f"Cost: ${cost:.6f}") print(tracker.get_cost_summary()) VERIFICATION_STAMP Story: OBS-004 Verified By: parallel-builder Verified At: 2026-02-25 Tests: 21/21 Coverage: 100% ) annotationsN) defaultdict)datetimetimezone)Optionalzclaude-opus-4-6g.@gR@)inputoutputzclaude-sonnet-4-6g@zclaude-haiku-4-5g?g@zclaude-opus-4-5zclaude-sonnet-4z claude-haikuzgemini-2.5-prog?g$@zgemini-2.5-flashg333333?g333333?zgemini-2.0-flashzgemini-1.5-prog@zgemini-1.5-flashz gemini-proz gemini-flashztext-embedding-004znomic-embed-textztext-embedding-3-smallg{Gz?ztext-embedding-3-largegp= ף?zdict[str, dict[str, float]] MODEL_PRICINGg?dict[str, float]_DEFAULT_PRICINGz7/mnt/e/genesis-system/data/observability/cost_log.jsonlcH|jj}|tvr t|StjDcgc]\}}||vs||vst ||f}}}|rt |ddSt jd|ttScc}}w)z Return the pricing dict for a model name. Performs a case-insensitive substring match so that version suffixes and provider prefixes (e.g. ``"openrouter/gemini-flash"``) are handled. c |dS)Nr)xs 8/mnt/e/genesis-system/core/observability/cost_tracker.pyz"_resolve_pricing..cs QqT)keyu<CostTracker: unknown model '%s' — using default pricing %s) lowerstripr itemslenmaxloggerwarningr )modelrkv candidatess r_resolve_pricingr"Qs ++-   C mS!!!. 3 3 51cSAXQ J:>2155 NNF  s BBcveZdZdZefd dZ d d dZddZddZddZ ddZ dd Z dd Z y) CostTrackeraU Tracks LLM API costs with per-session, per-agent, and per-customer attribution. Instances are lightweight and can be created per-request or shared as a singleton via ``get_cost_tracker()``. Parameters ---------- log_path : str Path to the JSONL audit file. Parent directory is created on first write. c||_tt|_tt|_tt|_y)N)log_pathrfloat_session_costs _agent_costs_customer_costs)selfr&s r__init__zCostTracker.__init__zs0  0;E0B.9%.@1       +t 3 +    h '4 / '    - 5 -"hll3==?(*$ &    rc:|jj|dS)z/Return total cost accumulated for *session_id*.r )r(get)r+r3s rget_session_costzCostTracker.get_session_costs""&&z377rc:|jj|dS)z-Return total cost accumulated for *agent_id*.r )r)r@)r+r4s rget_agent_costzCostTracker.get_agent_costs  $$Xs33rc:|jj|dS)z0Return total cost accumulated for *customer_id*.r )r*r@)r+r5s rget_customer_costzCostTracker.get_customer_costs##'' S99rc\tt|jjdS)z Return total cost across all tracked sessions in this instance. Note: this reflects in-memory state only; it does not read from the JSONL file. For persistent totals, read ``cost_log.jsonl`` directly. r.)r6sumr(valuesr+s rget_daily_totalzCostTracker.get_daily_totals%S,,3356::rct|jdt|jt|jt|j dS)z Return a summary dict with all attribution breakdowns. Returns ------- dict Keys: ``daily_total_usd``, ``sessions``, ``agents``, ``customers``. )daily_total_usdsessionsagents customers)r6rJdictr(r)r*rIs rget_cost_summaryzCostTracker.get_cost_summarysK %T%9%9%;Q?T0014,,-d223   rc tjtjj|jdt |jdd5}|j tj|dzdddy#1swYyxYw#t$r#tjd|jYywxYw) zAAppend a JSONL record to the audit file, creating dirs as needed.T)exist_okazutf-8)encoding Nz+CostTracker: failed to write cost log at %s) osmakedirspathdirnamer&openwritejsondumpsOSErrorr exception)r+r=fhs rr:zCostTracker._append_logs [ KK 6 FdmmS7; 3rE*T12 3 3 3 [   JDMM Z [s0AB(B B BBB)CC)r&strreturnNone)NNN)rrcr0intr1rfr3 Optional[str]r4rgr5rgrdr')r3rcrdr')r4rcrdr')r5rcrdr')rdr')rdrQ)r=rQrdre) __name__ __module__ __qualname____doc__ COST_LOG_PATHr,r>rArCrErJrRr:rrrr$r$ms (5D %)"&%)<<< < " <  <#< <D84:; ([rr$zOptional[CostTracker] _cost_trackerc.t tatS)z2Return the module-level ``CostTracker`` singleton.)rmr$rrrget_cost_trackerros# r)rrcrdr )rdr$)rk __future__rr^loggingrX collectionsrrrtypingr getLoggerrhrr __annotations__r rlr"r$rmrorrrrvs D#  #'   8 $.DT:.CT:.DS9 . DT: . CT: .DS9.DT:.ET:.ET:.DS9.ET:.DS9.ET:."CC"@#.$CC"@%.&DC"@'.(DC"@). *003c%B"BJ 8J[J[b(, $+r