ei>d UddlmZddlmZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZmZddlmZddd Zed Zed Ze d Ze d ZedGddZedGddZedGddZedGddZdZideededededddd !d"eed#eded$ed%ddd !d&eed'ed(ed)dd*d+,d-eed.ed/ed0dd*d+,d1eed2ed3ed4dd*d+,d5eed6ed7ed8dd*d+,d9eed8ed6ed:dd*d+,d;eed(ed<ed'dd*d+,d=eed>ed?ed@dd*d+,dAeed#eded$ed%dddB!dCeedDedEedFedGdddB!dHeededededdddB!dIeedJed)edKed$dddB!dLeedMedNddOdPQdReed@edSddOdPQdTeed)ed(ddUdVQdWeed.ed/ddUdVQeed8ed6ddUdVQeededddXdYQeed#edddXdYQeed#edddXdYQeedDedEddXdYQeedDedZddXdYQeed[ed\ddXdYQeed]edMddXdYQd^Zd_ed`<ddeZddgZ dddnZ ddqZ!ddrZ"ddxZ# dddzZ$dhdhd{dd~Z%dhdhdhdddZ& dddZ'ddZ(ddZ)dhS)) annotations) dataclass)datetimetimezone)Decimal)AnyDictLiteralOptional)fetch_endpoint_model_metadatafetch_model_metadata)base_url_host_matchesg)inputoutput01000000)actual estimatedincludedunknown)provider_cost_apiprovider_generation_apiprovider_models_apiofficial_docs_snapshot user_overridecustom_contractnoneT)frozenceZdZUdZded<dZded<dZded<dZded<dZded<dZ ded <d Z d ed <e ddZ e ddZ d S)CanonicalUsagerint input_tokens output_tokenscache_read_tokenscache_write_tokensreasoning_tokens request_countNzOptional[dict[str, Any]] raw_usagereturnc0|j|jz|jzSN)r"r$r%selfs 5/home/div/.hermes/hermes-agent/agent/usage_pricing.py prompt_tokenszCanonicalUsage.prompt_tokens&s 4#99DtjtjSr,)rnowrutcr9r1r/r[Os8< --r1) anthropiczclaude-opus-4-20250514z15.00z75.00z1.50z18.75rzChttps://docs.anthropic.com/en/docs/build-with-claude/prompt-cachingz#anthropic-prompt-caching-2026-03-16)rErFrGrHrKrMrN)r\zclaude-sonnet-4-20250514z3.00z0.30z3.75)openaizgpt-4oz2.50z10.00z1.25zhttps://openai.com/api/pricing/zopenai-pricing-2026-03-16)rErFrGrKrMrN)r]z gpt-4o-miniz0.15z0.60z0.075)r]zgpt-4.1z2.00z8.00z0.50)r]z gpt-4.1-miniz0.40z1.60z0.10)r]z gpt-4.1-nanoz0.025)r]o3z40.00)r]zo3-miniz1.10z4.40z0.55)r\zclaude-3-5-sonnet-20241022zanthropic-pricing-2026-03-16)r\zclaude-3-5-haiku-20241022z0.80z4.00z0.08z1.00)r\zclaude-3-opus-20240229)r\zclaude-3-haiku-20240307z0.25z0.03)deepseekz deepseek-chatz0.14z0.28z1https://api-docs.deepseek.com/quick_start/pricingzdeepseek-pricing-2026-03-16)rErFrKrMrN)r_zdeepseek-reasonerz2.19)googlezgemini-2.5-prozhttps://ai.google.dev/pricingzgoogle-pricing-2026-03-16)r`zgemini-2.5-flashz'https://aws.amazon.com/bedrock/pricing/zbedrock-pricing-2026-04z3.20z0.06z0.24z0.035))r`zgemini-2.0-flash)bedrockzanthropic.claude-opus-4-6)razanthropic.claude-sonnet-4-6)razanthropic.claude-sonnet-4-5)razanthropic.claude-haiku-4-5)razamazon.nova-pro)razamazon.nova-lite)razamazon.nova-microz#Dict[tuple[str, str], PricingEntry]_OFFICIAL_DOCS_PRICINGvaluerr*rDcf|dS tt|S#t$rYdSwxYwr,)rr< Exceptionrcs r/ _to_decimalrgmsG }ts5zz""" tts " 00r!cH t|pdS#t$rYdSwxYw)Nr)r!rerfs r/_to_intrivs95:A qqs  !!N model_namer<r=rLr@c&|pd}|pd}|pd}|s%d|vr!|dd\}}|dvr|}|}|dkrtd||pddS|dkst |pdd rtd||pdd S|d kr.td |dd |pdd S|dkr.td|dd |pdd S|dvs|rd|vrt|pd||pddSt|pd|r|dd nd|pddS)Nr?/r'>r`r]r\z openai-codexsubscription_included)r=r>r@rA openrouterz openrouter.aiofficial_models_apir\rr]>localcustom localhostrrr)striplowersplitr;r)rjr=r@ provider_namebaser>inferred_provider bare_models r/resolve_billing_router{}s: ^**,,2244M N ! ! # # ) ) + +D  2 $ $ & &E SE\\(- C(;(;%:  A A A-ME&&^58>WYhAAA A $$(=hn"o(^(^$\UWf{|||| ##[ C8H8H8LW_WeceuMNNN N  XU[[5E5Eb5IT\Tb`brJKKK K++++9L9L]%>heV^Vdbds|}}}} !;)[`Ch5;;sCSCSTVCWCWfhs{tB@BQZ [ [ [[r1routeOptional[PricingEntry]crt|j|jfSr,)rbgetr=r>rur|s r/_lookup_official_docs_pricingrs+ ! % %u~u{7H7H7J7J&K L LLr1cJtt|jddS)Nz>https://openrouter.ai/docs/api/api-reference/models/get-modelszopenrouter-models-apirMrN)_pricing_entry_from_metadatar r>rs r/_openrouter_pricing_entryrs- ' S/    r1metadataDict[str, Dict[str, Any]]model_idrMrNc ||vrdS||dpi}t|d}t|d}t|d}t|dp)|dp|d}t|dp)|d p|d } |||dSdd} t| || || || | |d||t S)Npricingprompt completionrequest cache_read cached_promptinput_cache_read cache_writecache_creationinput_cache_writercrDr*c|dS|tzSr,) _ONE_MILLIONrfs r/_per_token_to_per_millionz?_pricing_entry_from_metadata.._per_token_to_per_millions =4|##r1r) rErFrGrHrIrKrMrNrP)rcrDr*rD)rrgrC_UTC_NOW) rrrMrNrrrrrrrs r/rrsxtx $$Y//52G X.. / /FW[[6677J'++i0011G L!! + ;; ' ' + ;;) * *J  M"" , ;;' ( ( , ;;* + +K ~*,t$$$$ 88@@ 9 9* E E$=$=j$I$I%>%>{%K%K$'::    r1api_keyct|||}|jdkr)tttttddS|jdkrt |S|jrMtt|j|pd|j |j d d d }|r|St|S) Nr=r@rmrincluded-route)rErFrGrHrKrNrnr?)rrlz/modelszopenai-compatible-models-apir) r{rArC_ZEROr=rr@rr r>rstriprrjr=r@rr|entrys r/get_pricing_entryrs "*x( S S SE 444#($)(-).,      ~%%(/// ~, )%.'-R P P P K.//44===:      L ( / //r1)r=api_moderesponse_usagerc|stS|pd}|pd}|dks|dkrztt |dd}tt |dd}tt |dd}tt |dd}n|d krtt |dd} tt |dd}t |d d } t| rt | d dnd}t| rt | d dnd}t d| |z |z }ntt |dd} tt |dd}t |dd } t| rt | d dnd}|stt |dd}t| rt | ddnd}|stt |dd}t d| |z |z }d} t |dd } | rtt | dd} t||||| S)uNormalize raw API response usage into canonical token buckets. Handles three API shapes: - Anthropic: input_tokens/output_tokens/cache_read_input_tokens/cache_creation_input_tokens - Codex Responses: input_tokens includes cache tokens; input_tokens_details.cached_tokens separates them - OpenAI Chat Completions: prompt_tokens includes cache tokens; prompt_tokens_details.cached_tokens separates them In both Codex and OpenAI modes, input_tokens is derived by subtracting cache tokens from the total — the API contract is that input/prompt totals include cached tokens and the details object breaks them out. r?anthropic_messagesr\r"rr#cache_read_input_tokenscache_creation_input_tokenscodex_responsesinput_tokens_detailsN cached_tokenscache_creation_tokensr0completion_tokensprompt_tokens_detailsr%output_tokens_detailsr&)r"r#r$r%r&)r rtrurigetattrmax)rr=rrwmoder"r#r$r% input_totaldetails prompt_totalr&output_detailss r/normalize_usagers"  ^**,,2244M N ! ! # # ) ) + +D ###} 'C'Cw~~qIIJJ  K KLL #GN>ASSTT ^-DdKKNS"7>;Mq#Q#QRR !#+-)    r1r=r@rusagec t|||}|jdkrttddddSt ||||}|stdddd Sg}t}|jr|jtdd|jd S|jr|j tdd|jd S|j r |j tdd|jd d S|j r |j tdd|jd d S|j'|t|j|jztz z }|j '|t|j|j ztz z }|j '|t|j |j ztz z }|j '|t|j |j ztz z }|j&|jr|t|j|jzz }d} d|d} |jdkr|tkrd} d} |jdkr|dt|| |j| |j|jt-|S)Nrrmrrr)rSrUrKrVrNrrzn/a)rSrUrKrV)z(cache-read pricing unavailable for route)rSrUrKrVrW)z)cache-write pricing unavailable for routerz~$.2frnzBOpenRouter cost is estimated from the models API until reconciled.)rSrUrKrVrPrNrW)r{rArRrrr"rErKr#rFr$rGr%rHrrrIr(r=appendrPrNtuple) rjrr=r@rr|rrWamountrUrVs r/estimate_usage_costr8s "*x( S S SE 444,      j8hX_ ` ` `E YT)FRWXXXXE F _e:BT)ELX]^^^^ _u<DT)ELX]^^^^   , 4 |C     - 5 |D   #/'%,--0LL|[[ $0'%-..1NNQ]]] (4'%122U5VVYeee )5'%233e6XX[ggg %%*=%'%-..1CCC$F    E |v&E// ~%% YZZZ |#-Ell   r1boolcnt|||}|jdkrdSt||||}|duS)uCheck whether we have pricing data for this model+route. Uses direct lookup instead of routing through the full estimation pipeline — avoids creating dummy usage objects just to check status. rrmTrN)r{rArrs r/has_known_pricingrsN "*x( S S SE 444t j8hX_ ` ` `E  r1secondsfloatc|dkr|ddS|dz }|dkr|ddS|dz }|dkr;t|dz}|rt|d|dnt|dS|dz }|dd S) N<.0fsmzh h.1fd)r!)rminuteshours remaining_mindayss r/format_duration_compactrs||     lG||     bLE rzzGbL)) 4AW#e**00 0000#e**GWGWGWW 2:D>>>>r1cdtt|}|dkrtt|S|dkrdnd}d}|D]`\}}||krU||z }|dkr|d}n|dkr|d }n|d }d |vr(|d d }|||cSa|d S)Nr-r?))iʚ;B)i@BM)rK rdrr.r,)absr!r<r)rc abs_valuesignunits thresholdsuffixscaledtexts r/format_token_count_compactrsCJJI53u::!))33D BE" + + 6  ! !*F{{ #  d{{{{3''..s33*D*&** * * * "<<r1)rcrr*rD)rcrr*r!)NN)rjr<r=rLr@rLr*r;)r|r;r*r}) rrrr<rMr<rNr<r*r})NNN) rjr<r=rLr@rLrrLr*r})rrr=rLrrLr*r ) rjr<rr r=rLr@rLrrLr*rR) rjr<r=rLr@rLrrLr*r)rrr*r<)rcr!r*r<)* __future__r dataclassesrrrdecimalrtypingrr r r agent.model_metadatar r utilsrDEFAULT_PRICINGrrrTrJr r;rCrRrrbr7rgrir{rrrrrrrrrr9r1r/rs """""""!!!!!!''''''''////////////TTTTTTTT''''''3// wy!! A B    $77777777$ $"""""""" $ * * * * * * * * $         . - V?|&ww// ' 0 0$+GFOO%,WW%5%5'X= V?|&wv ' 0 0$+GFOO%,WV__'X=!V?4|&wv ' 0 0$+GFOO'43 ;V?J|&wv '$+GG$4$4'43 QV?`|&wv '$+GFOO'43 gV?v|&wv '$+GFOO'43 }V?L|&wv '$+GG$4$4'43 SV?b|&ww// ' 0 0$+GFOO'43 iV?x|&wv '$+GFOO'43 V?P|&wv ' 0 0$+GFOO%,WV__'X6WV?h|&wv '$+GFOO%,WV__'X6oV?@|&ww// ' 0 0$+GFOO%,WW%5%5'X6GV?X|&wv '$+GFOO%,WV__'X6_V?r|&wv ''F5 yV?F|&wv ''F5 MV?\|&wv ' 0 0'23 cV?p|&wv ''23 wV?J|&wv ''23 |&ww// ' 0 0'<1 |&wv ' 0 0'<1 |&wv ' 0 0'<1 |&wv ''<1 |&wv ''<1 |&wv ''<1 |&ww// ''<1 _V?V?V?VVVVr#"[[[[[6MMMM))))\#"! 00000B#" FFFFFFZ#"! LLLLLLb#"! &    r1