,j`ddlmZddlZddlZddlmZddlmZmZddlm Z m Z m Z ddl Z ddl mZmZddlmZmZddlmZe rdd lmZejeZddZd?dZddd@d#ZdAd&Z dBd)Z!dCd*Z"dd+d,dDd.Z#dCd/Z$ed Gd0d1Z%dd+d,dEd2Z&dFd4Z'dCd5Z(dCd6Z)dGd8Z*ddd9dHd;Z+dS)I) annotationsN) dataclass)datetimetimezone) TYPE_CHECKINGAnyOptional)_is_oauth_tokenresolve_anthropic_token)_read_codex_tokens!resolve_codex_runtime_credentials)resolve_runtime_provider) TypeGuardreturnrc>tjtjSN)rnowrutc8/home/ubuntu/.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, ...]windowstuple[str, ...]detailsunavailable_reasonrboolcHt|jp|jo|j Sr)r2r.r0r1)selfs r availablezAccountUsageSnapshot.available,s$DL0DL11Q$:Q6QQr)rr2) r#r$r%r&r,r-r.r0r1propertyr5rrrr(r(!sMMMKKK!E!!!!D.0G0000!G!!!!(,,,,, RRRXRRRrr(valuer!ct|pd}|sdS|ddddS)N_ -)rstripreplacer,)r7cleaneds 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>Nr9)tzZz+00:00)tzinfo) isinstanceintfloatr fromtimestamprrrr=endswith fromisoformatrEr> ValueError)r7textdts r _parse_dtrO8s t%#u&&E%eEllx|DDDD% {{}} 4 ==   (9x'D '--BG22 (, (G(G G   44  4s >X&&':;; > > >>rFmarkdownsnapshotOptional[AccountUsageSnapshot]rer2 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**r9z Provider: rWrRz : unavailablerdz: z % remaining (z% used)u • resets  • z Unavailable: )r,r-appendr)r.rrmaxroundrHr rcr"r0r1) rfreheaderlineswindowbase remainingusedr"s rrender_account_usage_linesru_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 4G"7"7 "E"EGGHHH*+++-0_wsFD)))$##zz 'NN'NN     I &ttsK?IK?8BK??LL$@)retimeoutrc 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_stateNrdget_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_stateruhermes_cli.authrgetrFrr=concurrent.futuresrrfuturesThreadPoolExecutorsubmitresultrloggerdebug) rerrfixturerfrtok 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_usdrx 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.r0)rrFrGrHr}r~rlrrmrr(rrrr) stater.r0ufcap_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??GGcVeZdZUdZded<dZded<dZded <dZded <d Zded <dS) CreditsViewuLSurface-agnostic data for the ``/credits`` command. One portal fetch, one parse — consumed identically by the CLI panel, the gateway button, and any other money surface. Fail-open: when not logged in or the portal is unreachable, ``logged_in`` is False / ``topup_url`` is None and callers degrade gracefully. r2rrr/ balance_linesNr! identity_line topup_urlFdepleted) r#r$r%__doc__r&rrrrrrrrrUskOOO%'M''''#'M''''#I####Hrrc td} ddlm}|dpid}t |t r|s|Sn#t$r|cYSwxYw ddl}ddl m }m }|j d 5}||d | } dddn #1swxYwYn.#t$r!t dd |cYSwxYw| t%| dds|St'| } g} | t)| |} d| D} t%| dd} t%| dd}g}| r"|t | |r|d||rdd|znd}td t/| ||| t%| ddduS)uBuild the /credits view: balance block + identity line + top-up URL. Reuses the same account fetch + snapshot + URL builder as the /usage credits block, so the numbers always match. The balance block is the rendered snapshot MINUS its trailing top-up/command-hint lines (the /credits surface supplies its own affordance). Fail-open → ``CreditsView(logged_in=False)``. F)rrrrrN)rrrrTrru4credits ▸ /credits portal fetch failed (fail-open)rrrdcg|]R}|d)|dP|SS)zTop up:z(or run)lstrip startswith).0lines r z&build_credits_view..sd   ;;==++I66 KKMM,,Y77     remailorg_namezorg zTopping up as  / r)rrrrr)rrrrrFrr=rrrrrrrrrrrrrrurljoinr)rer not_logged_inrrrrrrrrfrrenderedrrwhors rbuild_credits_viewrfs %000M;;;;;;&&v..4"99.II3$$ ! !  ! !!!!          2 2q 2 A A Tkk">DkQQXXYG                 KVZ [[[gg{EBB*733H "M-hJJJ       GWd + +Ew D11HC  3u::& $($$%%%>%G    sHAA$$ A32A37'C",C C"CC"CC""(D  D base_urlc|pdd}|sd}|dr|dtd }d|vr|dzS|dzS)Nr9/z%https://chatgpt.com/backend-api/codexz/codexz /backend-apiz /wham/usagez/api/codex/usage)r=rstriprJlen)r normalizeds r_resolve_codex_usage_urlrs.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_idr9Bearer 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.r0)r r rrr=httpxClientrraise_for_statusjsonrlrrHrOrFrGr(rr@r)creds token_datarrrclientresponsepayloadrr.keyrrqrtr0rrs r_fetch_codex_account_usagers -$ 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)Nr9>r9autocustomr rr/)rr=lowerrr-rIr)r)rrrs rfetch_account_usagerNls X^$$**,,2244J+++t  ' '-// /  $ $133 3  % %28WEE E & tt 4sA<A<%A<< B  B )rr)r7r!rr!)r7rrr)rNrrr)rfrgrer2rrh)rvrHrr)rzrrr{)rrg)rer2rrHrrh)rer2rrHrr)rrrr)rr!rr!rrg)r)r!rr!rr!rrg), __future__rloggingr} dataclassesrrrtypingrrr ragent.anthropic_adapterr r rr r hermes_cli.runtime_providerrr getLoggerr#rrrr(r@rOrcruryrrrrrrrrr-rIrNrrrrVs^"""""" !!!!!!''''''''////////// LLLLLLLLQQQQQQQQ@@@@@@!       8 $ $&&&& $!!!!!!!! $ R R R R R R R R????&????(^c8XXXX]]]]@,14//////d7777t $         ,14CCCCCCL++++----`::::zEEEEV#! r