(Gj UddlmZddlZddlmZddlmZmZddlmZddl m Z m Z m Z m Z ddlmZmZddlmZd d d Zed Zed Zd Ze dZe dZedGddZedGddZedGddZedGddZdZideededededd d!d"#d$eed%ed&ed'ed(d d)d"#d*eededededd d!d"#d+eededededd d!d"#d,eededededd d!d"#d-eededededd d!d"#d.eed/ed0ed1ed2d d!d"#d3eed/ed0ed1ed2d d!d"#d4eededededd d!d"#d5eed/ed0ed1ed2d d!d"#d6eed'eded7ed8d d!d"#d9eed0ed:ed;ed<d d!d"#d=eed/ed0ed1ed2d d!d"#d>eed?ed%ed8d d@dABdCeedDedEedFd d@dABdGeedHedIedd d@dABdJeedKedLed7d d@dABidMeed7edKedNd d@dABdOeed%edPed?d d@dABdQeedRedSedTd d@dABdUeed/ed0ed1ed2d d!d"#dVeedWedXedYed'd d!d"#dZeed0ed:ed;ed<d d!d"#d[eed\ed8ed]ed1d d!d"#d^eed_ed`d dadbcddeedTeded dadbcdfeedgedhedid dadjBdkeed8ed%d dldmcdneedDedEd dldmcdoeed7edKd dldmcdpeed0ed:ed;ed<d dqdr#dseed/ed0ed1ed2d dqdr#dteed/ed0ed1ed2d dqdr#dueedWedXedYed'd dqdr#eedWedvd dqdrceedwedxd dqdrceedyed_d dqdrceed1edzd d{|eed1edzd d{|d}Zd~ed<ddZ ddZ! dddZ"ddZ#ddZ$ddZ%ddZ&ddZ' dddZ(dddddZ)ddddddZ* dddZ+ddZ,ddZ-dS)) annotationsN) dataclass)datetimetimezone)Decimal)AnyDictLiteralOptional)fetch_endpoint_model_metadatafetch_model_metadata)base_url_host_matchesg)inputoutput01000000z)https://inference-api.nousresearch.com/v1)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 ddZd 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 9/home/rurouni/.hermes/hermes-agent/agent/usage_pricing.py prompt_tokenszCanonicalUsage.prompt_tokens(s 4#99DtjtjSr,)rnowrutcr?r1r/rads8< --r1) anthropiczclaude-opus-4-8z5.00z25.00z0.50z6.25rz8https://platform.claude.com/docs/en/about-claude/pricingzanthropic-pricing-2026-05)rKrLrMrNrQrSrT)rbzclaude-opus-4-8-fastz10.00z50.00z1.00z12.50z4https://openrouter.ai/anthropic/claude-opus-4.8-fast)rbzclaude-opus-4-7)rbzclaude-opus-4-7-20250507)rbzclaude-opus-4-6)rbzclaude-opus-4-6-20250414)rbzclaude-sonnet-4-6z3.00z15.00z0.30z3.75)rbzclaude-sonnet-4-6-20250414)rbzclaude-opus-4-5)rbzclaude-sonnet-4-5)rbzclaude-haiku-4-5z0.10z1.25)rbzclaude-opus-4-20250514z75.00z1.50z18.75)rbzclaude-sonnet-4-20250514)openaizgpt-4oz2.50zhttps://openai.com/api/pricing/zopenai-pricing-2026-03-16)rKrLrMrQrSrT)rcz gpt-4o-miniz0.15z0.60z0.075)rczgpt-4.1z2.00z8.00)rcz gpt-4.1-miniz0.40z1.60)rcz gpt-4.1-nanoz0.025)rco3z40.00)rczo3-miniz1.10z4.40z0.55)rbzclaude-3-5-sonnet-20241022)rbzclaude-3-5-haiku-20241022z0.80z4.00z0.08)rbzclaude-3-opus-20240229)rbzclaude-3-haiku-20240307z0.25z0.03)deepseekz deepseek-chatz0.14z0.28z1https://api-docs.deepseek.com/quick_start/pricingzdeepseek-pricing-2026-03-16)rKrLrQrSrT)rezdeepseek-reasonerz2.19)rezdeepseek-v4-proz1.74z3.48z0.0145zdeepseek-pricing-2026-05-12)googlezgemini-2.5-prozhttps://ai.google.dev/pricingzgoogle-pricing-2026-03-16)rfzgemini-2.5-flash)rfzgemini-2.0-flash)bedrockzanthropic.claude-opus-4-6z'https://aws.amazon.com/bedrock/pricing/zbedrock-pricing-2026-04)rgzanthropic.claude-sonnet-4-6)rgzanthropic.claude-sonnet-4-5)rgzanthropic.claude-haiku-4-5z3.20z0.06z0.24z0.035z1.20zminimax-pricing-2026-04)rKrLrQrT))rgzamazon.nova-pro)rgzamazon.nova-lite)rgzamazon.nova-micro)minimax minimax-m2.7) minimax-cnriz#Dict[tuple[str, str], PricingEntry]_OFFICIAL_DOCS_PRICINGvaluerr*rJcf|dS tt|S#t$rYdSwxYwr,)rrB Exceptionrls r/ _to_decimalrp7sG }ts5zz""" tts " 00r!cH t|pdS#t$rYdSwxYw)Nr)r!rnros r/_to_intrr@s95:A qqs  !! model_namerBrCrRrFcz|pd}|pd}|pd}|s%d|vr!|dd\}}|dvr|}|}|dkrtd||pddS|dkst |pdd rtd||pdd S|d kst |pdd rtd ||pt d S|d kr.td |dd|pddS|dkr.td|dd|pddS|dvr.t||dd|pddS|dkst |pddr.td|dd|pddS|dvs|rd|vrt|pd||pddSt|pd|r|ddnd|pddS)NrE/r'>rfrcrbz openai-codexsubscription_included)rCrDrFrG openrouterz openrouter.aiofficial_models_apinouszinference-api.nousresearch.comrbrrc>rhrjvertexzaiplatform.googleapis.comgemini>localcustom localhostr~r)striplowersplitrAr_NOUS_DEFAULT_BASE_URL)rsrCrF provider_namebaserDinferred_provider bare_models r/resolve_billing_routerGs6 ^**,,2244M N ! ! # # ) ) + +D  2 $ $ & &E SE\\(- C(;(;%:  A A A-ME&&^58>WYhAAA A $$(=hn"o(^(^$\UWf{||||"7BHh"i"iV58CeOeuJKKK K ##[ C8H8H8LW_WeceuMNNN N  XU[[5E5Eb5IT\Tb`brJKKK K111]%++c:J:J2:NYaYgegwOPPP P  $9(.bJe$f$f XU[[5E5Eb5IT\Tb`brJKKK K++++9L9L]%>heV^Vdbds|}}}} !;)[`Ch5;;sCSCSTVCWCWfhs{tB@BQZ [ [ [[r1rDc|}dD]0}||r|t|d}n1t jdd|}|S)uPNormalize a Bedrock model id to its bare foundation-model form. Bedrock cross-region inference profiles prefix the foundation model id with a region scope (``us.`` / ``global.`` / ``eu.`` / ``ap.`` / ``jp.``), e.g. ``us.anthropic.claude-opus-4-7``. The pricing table is keyed on the bare ``anthropic.claude-*`` id, so the prefix must be stripped before the lookup or every cross-region session prices as unknown. Mirrors the prefix list in ``bedrock_adapter.is_anthropic_bedrock_model``. Also normalizes dot-notation version numbers (``4.7`` → ``4-7``). )zus.zglobal.zeu.zap.zjp.N (\d+)\.(\d+)\1-\2rr startswithlenresub)rDnameprefixs r/_normalize_bedrock_model_namerksx ;;==   D9 ??6 " " F %D E  6/8T 2 2D Kr1c|}|dr|tdd}t jdd|}|S)uNormalize Anthropic model name variants to canonical form. Handles: - Dot notation: claude-opus-4.7 → claude-opus-4-7 - Short aliases: claude-opus-4.7 → claude-opus-4-7 - Strips anthropic/ prefix if present z anthropic/Nrrr)rDrs r/_normalize_anthropic_model_namersc ;;==   D |$$(C %%&&' 6/8T 2 2D Kr1routeOptional[PricingEntry]c|j}t|j|f}|r|S|jdkr:t |}||kr%t|j|f}|r|S|jdkr:t |}||kr%t|j|f}|r|SdS)Nrbrg)rDrrkgetrCrr)rrDentry normalizeds r/_lookup_official_docs_pricingrs K    E " & &'> ? ?E   ~$$4U;;   *.. /KLLE   ~""2599   *.. /KLLE  4r1cJtt|jddS)Nz>https://openrouter.ai/docs/api/api-reference/models/get-modelszopenrouter-models-apirSrT)_pricing_entry_from_metadatar rD)rs r/_openrouter_pricing_entryrs- ' S/    r1metadataDict[str, Dict[str, Any]]model_idrSrTc ||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_writerlrJr*c|dS|tzSr,) _ONE_MILLIONros r/_per_token_to_per_millionz?_pricing_entry_from_metadata.._per_token_to_per_millions =4|##r1r) rKrLrMrNrOrQrSrTrV)rlrJr*rJ)rrprI_UTC_NOW) rrrSrTrrrrrrrs 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) NrCrFrvrincluded-route)rKrLrMrNrQrTrwrE)rruz/modelszopenai-compatible-models-apir) rrGrI_ZEROrCrrFrr rDrstriprrsrCrFrrrs r/get_pricing_entryrs "*x( S S SE 444#($)(-).,      ~%%(/// ~, )%.'-R P P P K.//44===:      L ( / //r1)rCapi_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} | s1t |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. rEanthropic_messagesrbr"rr#cache_read_input_tokenscache_creation_input_tokenscodex_responsesinput_tokens_detailsN cached_tokenscache_creation_tokensr0completion_tokensprompt_tokens_detailsr%output_tokens_detailsr&completion_tokens_details)r"r#r$r%r&)r rrrrgetattrmax)rrCrrmoder"r#r$r% input_totaldetails prompt_totalr&output_detailscompletion_detailss r/normalize_usagersH"  ^**,,2244M N ! ! # # ) ) + +D ###} 'C'Cw~~qIIJJ  K KLL #GN>ASSTT ^-DdKKNS"7>;Mq#Q#QRR $^5PRVWW  &*,>BB    !#+-)    r1rCrFrusagec 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)Nrrvrrr)rYr[rQr\rTrrzn/a)rYr[rQr\)z(cache-read pricing unavailable for route)rYr[rQr\r])z)cache-write pricing unavailable for routerz~$.2frwzBOpenRouter cost is estimated from the models API until reconciled.)rYr[rQr\rVrTr])rrGrXrrr"rKrQr#rLr$rMr%rNrrrOr(rCappendrVrTtuple) rsrrCrFrrrr]amountr[r\s r/estimate_usage_costrQs "*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. rrvTrN)rrGrrs 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-rE))iʚ;B)i@BM)rK rdrr.r,)absr!rBr)rl abs_valuesignunits thresholdsuffixscaledtexts r/format_token_count_compactrsCJJI53u::!))33D BE" + + 6  ! !*F{{ #  d{{{{3''..s33*D*&** * * * "<<r1)rlrr*rJ)rlrr*r!)NN)rsrBrCrRrFrRr*rA)rDrBr*rB)rrAr*r) rrrrBrSrBrTrBr*r)NNN) rsrBrCrRrFrRrrRr*r)rrrCrRrrRr*r ) rsrBrr rCrRrFrRrrRr*rX) rsrBrCrRrFrRrrRr*r)rrr*rB)rlr!r*rB). __future__rr dataclassesrrrdecimalrtypingrr r r agent.model_metadatar r utilsrDEFAULT_PRICINGrrrrZrPr rArIrXrrkr=rprrrrrrrrrrrrrrr?r1r/rs""""""" !!!!!!''''''''////////////TTTTTTTT''''''3// wy!! D A B    $" " " " " " " " J $"""""""" $ * * * * * * * * $         . - K? |&wv ' 0 0$+GFOO%,WV__'M3K?"|&ww// ' 0 0$+GFOO%,WW%5%5'I3)K?B|&wv ' 0 0$+GFOO%,WV__'M3IK?Z|&wv ' 0 0$+GFOO%,WV__'M3aK?t|&wv ' 0 0$+GFOO%,WV__'M3{K?L|&wv ' 0 0$+GFOO%,WV__'M3SK?d|&wv ' 0 0$+GFOO%,WV__'M3kK?||&wv ' 0 0$+GFOO%,WV__'M3CK?V|&wv ' 0 0$+GFOO%,WV__'M3]K?n|&wv ' 0 0$+GFOO%,WV__'M3uK?F|&wv '$+GFOO%,WV__'M3MK?`|&ww// ' 0 0$+GFOO%,WW%5%5'M3gK?x|&wv ' 0 0$+GFOO%,WV__'M3K?R|&wv ' 0 0$+GFOO'43 YK?h|&wv '$+GG$4$4'43 oK?~|&wv '$+GFOO'43 EK?T|&wv '$+GFOO'43 [K?K?j|&wv '$+GG$4$4'43 qK?@|&ww// ' 0 0$+GFOO'43 GK?V|&wv '$+GFOO'43 ]K?n|&wv ' 0 0$+GFOO%,WV__'M3uK?F|&wv '$+GFOO%,WV__'M3MK?^|&ww// ' 0 0$+GFOO%,WW%5%5'M3eK?v|&wv '$+GFOO%,WV__'M3}K?P |&wv ''F5 W K?d |&wv ''F5 k K?x |&wv '$+GH$5$5'F5  K?P |&wv ' 0 0'23 W K?d |&wv ''23 k K?x |&wv ''23  K?T |&ww// ' 0 0$+GFOO%,WW%5%5'<1[ K?l |&wv ' 0 0$+GFOO%,WV__'<1s K?D |&wv ' 0 0$+GFOO%,WV__'<1K K?\ |&wv '$+GFOO%,WV__'<1c K?K?z |&wv ''<1 |&wv ''<1 |&ww// ''<1 |&wv ''1 |&wv ''1 KK?K?K?KKKK\#"![![![![![H("0))))\#"! 00000B#" SSSSSSt#"! LLLLLLb#"! &    r1