)jASddlmZddlZddlZddlmZddlmZmZddlm Z m Z m Z ddl Z ddl mZmZddlmZmZddlmZe rdd lmZejeZd9d Zed GddZed GddZd:dZd;dZdd&Z d?d)Z!d@d*Z"dd+d,dAd.Z#d@d/Z$dBd1Z%d@d2Z&d@d3Z'dCd5Z(ddd6dDd8Z)dS)E) annotationsN) dataclass)datetimetimezone) TYPE_CHECKINGAnyOptional)_is_oauth_tokenresolve_anthropic_token)_read_codex_tokens!resolve_codex_runtime_credentials)resolve_runtime_provider) TypeGuardreturnrc>tjtjSN)rnowrutc:/home/wildlama/.hermes/hermes-agent/agent/account_usage.py_utc_nowrs < % %%rT)frozencDeZdZUded<dZded<dZded<dZded <dS) AccountUsageWindowstrlabelNzOptional[float] used_percentOptional[datetime]reset_at Optional[str]detail)__name__ __module__ __qualname____annotations__rr r"rrrrrsNJJJ$(L((((#'H'''' F      rrceZdZUded<ded<ded<dZded<dZd ed <d Zd ed <d Zded<dZd ed<e ddZ dS)AccountUsageSnapshotrprovidersourcer fetched_atzAccount limitstitleNr!planrztuple[AccountUsageWindow, ...]windowsztuple[str, ...]detailsunavailable_reasonrboolcHt|jp|jo|j Sr)r1r.r/r0)selfs r availablezAccountUsageSnapshot.available,s$DL0DL11Q$:Q6QQr)rr1) r#r$r%r&r,r-r.r/r0propertyr4rrrr(r(!sMMMKKK!E!!!!D.0G0000!G!!!!(,,,,, RRRXRRRrr(valuer!ct|pd}|sdS|ddddS)N_ -)rstripreplacer,)r6cleaneds r_title_case_slugr?1s\%+2$$&&G t ??3 $ $ , ,S# 6 6 < < > >>rrrc|dvrdSt|ttfr-tjt|t jSt|tr|}|sdS| dr |dddz} tj |}|j r|n| t jS#t$rYdSwxYwdS)N>Nr8)tzZz+00:00)tzinfo) isinstanceintfloatr fromtimestamprrrr<endswith fromisoformatrDr= ValueError)r6textdts r _parse_dtrN8s t%#u&&E%eEllx|DDDD% {{}} 4 ==   (9x'D '--BG22 (, (G(G G   44  4s >X&&':;; > > >>rFmarkdownsnapshotOptional[AccountUsageSnapshot]rdr1 list[str]c  |sgSd|rdnd|j|rdnd}|g}|jr'|d|jd|jdn|d|j|jD]}|j |jd}nqtdtd t|jz }tdtt|j}|jd |d |d }|j r|d t|j z }n|j r |d|j z }|||j D]}|||jr|d|j|S)Nu📈 z**r8z Provider: rVrQz : unavailablerdz: z % remaining (z% used)u • resets  • z Unavailable: )r,r-appendr)r.rrmaxroundrGr rbr"r/r0) rerdheaderlineswindowbase remainingusedr"s rrender_account_usage_linesrt_s  YX-TT2 Yx~ Yx?WttUW Y YF HE}7 G("3GGx}GGGHHHH 5("355666"     &l111DDAuS51D+E+E%EFFGGIq%f&9 : :;;<,>@@AAA &:DAAJ 86*66777|%:DAA 5== NNJ K K K w 4R+B+B<+P+PRRSSS-0_wsFD)))$##zz 'NN'NN     I &ttsK*IK*8A1K**K?>K?$@)rdtimeoutrc ddlm}|}n#t$rd}YnwxYw| t|}t ||S ddlm}|dpid}t|tr| sgSn#t$rgcYSwxYw ddl }ddl m }|jd 5} | |d | } dddn #1swxYwYt%| }t ||S#t$r!t&dd gcYSwxYw)aReturn rendered Nous-credits /usage lines, or [] when there's nothing to show. Account-independent of any live agent: gated on "a Nous account is logged in" (a cheap local auth-state check), then a wall-clock-bounded portal fetch. Shared by the CLI ``_show_usage`` and the TUI ``session.usage`` RPC so both surfaces show the same block regardless of session API-call count or resume state. Fail-open: any auth/portal hiccup or timeout returns [] (the caller shows nothing). Dev override: when HERMES_DEV_CREDITS_FIXTURE selects a fixture state, /usage renders from that fixture instead of the real portal (so the block + gauge are testable without a live account). Throwaway scaffolding. r)dev_fixture_credits_stateNrc)get_provider_auth_stater access_token)get_nous_portal_account_info) max_workersT) force_freshru9credits ▸ /usage portal fetch/render failed (fail-open))exc_info)agent.credits_trackerr Exception_snapshot_from_credits_staterthermes_cli.authrgetrErr<concurrent.futuresrrfuturesThreadPoolExecutorsubmitresultrloggerdebug) rdrrfixturerertok concurrentrpoolaccounts rnous_credits_linesrsCCCCCC++-- /88)(XFFFF;;;;;;&&v..4"99.II3$$  I   !!!!HHHHHH   2 2q 2 A A &Tkk,$"fWf%%  & & & & & & & & & & & & & & &/w77)(XFFFF   P[_ ``` sU ""AB B)(B)-%D5,D > D5 DD5D"D55(E E c  |dSg}g}t|dd}t|ttfrt j|rut|dd}t|dd}d}|r |r d|d|d}|tdtd td |d z| t|dd}|r|d |t|d d}|r|d|t|dd}|r|d|t|dddur|d|s|sdS|dtddtdt|t|S#ttf$rYdSwxYw)uMap a header-shaped CreditsState (e.g. a dev fixture) to the /usage snapshot. Renders the same magnitudes + monthly-grant % window the portal path produces, so HERMES_DEV_CREDITS_FIXTURE can exercise /usage without a live account. The *_usd strings are mock display values here (not server balance to compute on); the % comes from CreditsState.used_fraction (micros math). Fail-open → None. N used_fractionsubscription_limit_usdsubscription_usdrw of $rrrrrzSubscription credits: $ purchased_usdzTop-up credits: $ remaining_usdzTotal usable: $ paid_accessTFru,(dev fixture — HERMES_DEV_CREDITS_FIXTURE)rz dev-fixturer)r)r*r+r,r.r/)rrErFrGr|r}rkrrlrr(rrrr) stater.r/ufcap_usdsub_usdr"rrs rrrs@/ =4,. UOT 2 2 b3, ' ' DM",=,= e%=tDDGe%7>>GF :7 :9W997999 NN"(!$S#eR%Z*@*@!A!A!   %!3T::  @ NN>W>> ? ? ?==  @ NN>}>> ? ? ?==  > NN<]<< = = = 5- . .% 7 7 NNJ K K K w 4EFFF# zz 'NN'NN      I &ttsF?E'F?/AF??GGbase_urlc|pdd}|sd}|dr|dtd }d|vr|dzS|dzS)Nr8/z%https://chatgpt.com/backend-api/codexz/codexz /backend-apiz /wham/usagez/api/codex/usage)r<rstriprIlen)r normalizeds r_resolve_codex_usage_urlrTs.b''))0055J =< 8$$2 03x==. 01 ##M)) * **rc td}t}|dpi}t|ddpdpd}d|ddd d }|r||d <t jd 5}|t|dd|}|dddn #1swxYwY| pi}|dpi}g} dD]\} } || pi} | d} | 4| t| t| t| dg}|dpi}|dr|d}t|ttfr'| dt|dn*|dr| dt!ddt#t%|dt'| t'|S) NT)refresh_if_expiringtokens account_idr8Bearer api_keyapplication/jsonz codex-cli) AuthorizationAccept User-AgentzChatGPT-Account-Id.@rrheaders rate_limit))primary_windowSession)secondary_windowWeeklyrr rrr credits has_creditsbalanceCredits balance: $.2f unlimitedzCredits balance: unlimited openai-codex usage_api plan_type)r)r*r+r-r.r/)r r rrr<httpxClientrraise_for_statusjsonrkrrGrNrErFr(rr?r)creds token_datarrrclientresponsepayloadrr.keyrrprsr/rrs r_fetch_codex_account_usager_s -$ G G GE#%%J ^^H % % +FVZZ b117R88>>@@HDJ55#355$!G 3(2$% d # # #$v::6uyyR7P7PQQ[b:cc!!###$$$$$$$$$$$$$$$mmoo#G\**0bJ(*GU   U$$*zz.)) <  "4[["6::j#9#9::       Gkk)$$*G{{=!!9++i(( gU| , , 9 NNDgDDD E E E E [[ % % 9 NN7 8 8 8 :: gkk+66 7 7gg    sA C33C7:C7c tpd}|sdSt|stddt dSd|dddd d }t jd 5}|d |}|dddn #1swxYwY| pi}g}d}|D]\}}||pi} | d} | 4t| dkrt| dznt| } | t|| t| dg} |dpi} | dr| d}| d}| dpd}t|ttfr>sCCJ)))K!!!G*5**$G d # # # vzz+wz?? %%'''$$&&,"11&99?R zz'7z;;H  % % ' ' ' -2226::@bHH   HHH                '++o66=#>>M M229c::KOC][-H$I$IOOOPG(*G LL ! !Ell#455Ohll=117R88>>@@K LL ! !E53,'' %LL1   e 5 5  '' 7 7 7 75<< 7 7 7 7 7Ell 00$6+EL QOQQQkQQQQR  9    7+ 7 7 8 8 8 %)||L11      %#u&& 2B%,,BBBBC \\- ( (' 2 \\. ) ); 7 \\/ * *L 9  C CLE5 %#u.. C5<E>)rrr)ct|pd}|dvrdS |dkrtS|dkrt S|dkrt ||Sn#t $rYdSwxYwdS)Nr8>r8autocustomrrr)rr<lowerrrr/r)r)rrrs rfetch_account_usager4s X^$$**,,2244J+++t  ' '-// /  $ $133 3  % %28WEE E & tt 4sA<A<%A<< B  B )rr)r6r!rr!)r6rrr)rMrrr)rerfrdr1rrg)rurGrr)ryrrrz)rrf)rdr1rrGrrg)rrrr)rr!rr!rrf)r)r!rr!rr!rrf)* __future__rloggingr| dataclassesrrrtypingrrr ragent.anthropic_adapterr r rr r hermes_cli.runtime_providerrr getLoggerr#rrrr(r?rNrbrtrxr~rrrrrrr/r4rrrr<s"""""" !!!!!!''''''''////////// LLLLLLLLQQQQQQQQ@@@@@@!       8 $ $&&&& $!!!!!!!! $ R R R R R R R R????&????(^c8XXXX\\\\~,14//////d7777t++++----`::::zEEEEV#! r