NQjk ddlmZddlZddlZddlmZddlmZmZddlm Z m Z m Z ddl Z ddl mZmZddlmZmZmZddlmZe rdd lmZejeZd>d Zed GddZed GddZd?dZd@dZdAdZdddBd#Z dCd&Z!dDd)Z"dEd*Z#dd+d,dFd.Z$dEd/Z%ed Gd0d1Z&dd+d,dGd2Z'dHd4Z(dId7Z) dJdKd8Z*dEd9Z+dKd:Z,ddd;dLd=Z-dS)M) annotationsN) dataclass)datetimetimezone) TYPE_CHECKINGAnyOptional)_is_oauth_tokenresolve_anthropic_token) AuthError_read_codex_tokens!resolve_codex_runtime_credentials)resolve_runtime_provider) TypeGuardreturnrc>tjtjSN)rnowrutc7/home/alina/.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)r3r/r1r2)selfs r availablezAccountUsageSnapshot.available,s$DL0DL11Q$:Q6QQr)rr3) r$r%r&r'r-r.r/r1r2propertyr6rrrr)r)!sMMMKKK!E!!!!D.0G0000!G!!!!(,,,,, RRRXRRRrr)valuer"ct|pd}|sdS|ddddS)N_ -)rstripreplacer-)r8cleaneds r_title_case_slugrA1s\%+2$$&&G t ??3 $ $ , ,S# 6 6 < < > >>rrr c|dvrdSt|ttfr-tjt|t jSt|tr|}|sdS| dr |dddz} tj |}|j r|n| t jS#t$rYdSwxYwdS)N>Nr:)tzZz+00:00)tzinfo) isinstanceintfloatr fromtimestamprrrr>endswith fromisoformatrFr? ValueError)r8textdts r _parse_dtrP8s t%#u&&E%eEllx|DDDD% {{}} 4 ==   (9x'D '--BG22 (, (G(G G   44  4s >X&&':;; > > >>rFmarkdownsnapshotOptional[AccountUsageSnapshot]rfr3 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**r:z Provider: rXrSz : unavailablerdz: z % remaining (z% used)u • resets  • z Unavailable: )r-r.appendr*r/rrmaxroundrIr!rdr#r1r2) rgrfheaderlineswindowbase remainingusedr#s rrender_account_usage_linesrv_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$@)rftimeoutrc 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_stateNreget_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_statervhermes_cli.authrgetrGrr>concurrent.futuresrrfuturesThreadPoolExecutorsubmitresultrloggerdebug) rfrrfixturergrtok 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_usdry 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/r1)rrGrHrIr~rrmrrnrr)rrrr) stater/r1ufcap_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. r3rrr0 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)rrrecg|]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)rrrrrGrr>rrrrrrrrrrrrrrvrmjoinr)rfr not_logged_inrrrrrrrrgrrenderedrrwhors 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)Nr:/z%https://chatgpt.com/backend-api/codexz/codexz /backend-apiz /wham/usagez/api/codex/usage)r>rstriprKlen)r normalizeds r_resolve_codex_usage_urlrs.b''))0055J =< 8$$2 03x==. 01 ##M)) * **rapi_keytuple[str, str, Optional[str]]c~t|pd}|r&|t|pddfS td}d} t}|dpi}t|ddpdpd}n,#t $rt ddYnwxYw|d t|d dpd|fS#t $rt d dYnwxYwd d lm }|d}| } | td| j t| j p|pddfS)aLResolve Codex quota credentials from the native runtime path. Prefer explicit live-agent credentials, then the legacy singleton OAuth state, then the credential pool. Hermes's native OAuth setup now stores device-code logins in the pool, so quota diagnostics must not depend only on the older singleton store. r:NT)refresh_if_expiringtokens account_idu5codex ▸ /usage account_id read failed (best-effort)rrru@codex ▸ /usage runtime resolver returned no creds; trying poolr) load_pool openai-codexz7No available openai-codex credential in credential pool)rr>rr rr rragent.credential_poolrselect RuntimeErrorruntime_api_keyruntime_base_url) rr explicit_keycredsr token_datarrrentrys r _resolve_codex_usage_credentialsrsw}"%%++--L?SR006688$>>" h1dKKK$(  a+--J^^H--3FVZZ b99?R@@FFHHPDJJ a a a LLP[_L ` ` ` ` ` aYUYYz2%>%>%D"!E!E!K!K!M!MzYY hhh Wbf gggggh0///// 9^ $ $D KKMME }TUUU  #e&<&N&NB"O"O"U"U"W"WY] ]]s8 D+ AB?>D+?&C(%D+'C((AD++&EEc t||\}}}d|ddd}|r||d<tjd5}|t ||}|dddn #1swxYwY|pi}|d pi} g} d D]\} } | | pi} | d }|4| t| t|t| d  g}|dpi}|dr|d}t|ttfr'|dt|dn*|dr|dtddtt|dt!| t!|S)NBearer application/jsonz codex-cli) AuthorizationAccept User-AgentzChatGPT-Account-Id.@rheaders rate_limit))primary_windowSession)secondary_windowWeeklyrr!rrr!credits has_creditsbalanceCredits balance: $.2f unlimitedzCredits balance: unlimitedr usage_api plan_type)r*r+r,r.r/r1)rhttpxClientrrraise_for_statusjsonrmrrIrPrGrHr)rrAr)rrtokenresolved_base_urlrr clientresponsepayloadrr/keyrrrrur1rrs r_fetch_codex_account_usager&s,LHV]+^+^(E j*5**$!G 3(2$% d # # #$v::67HIISZ:[[!!###$$$$$$$$$$$$$$$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    s9A??BBc 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|ttfrr r)rrrrrrrIrmrrPrGrHr)r r r"r#r$r/mappingr%rrrutilrur1extrar2r3r4s r_fetch_anthropic_account_usager:!s $ & & ," 3 3 5 5E t 5 ! ! # $zzn     +5**$*,) G d # # #$v::ISZ:[[!!###$$$$$$$$$$$$$$$mmoo#G(*GG    US!!'Rzz-(( < $)$KK1$4$4uT{{S  %++ !"6::k#:#:;;       G KK & & ,"E yyyy00  /22 99Z((1E lS%L 1 1 jQTV[P\6]6]  NNS SSS]SSSSS     ::gg    s5,B--B14B1c  td||}t|ddpd}|sdSt|ddpdd}|d}|d}d |d d }t jd 5}|||} | | pidpi} |||} | | pidpi} n#t$ri} YnwxYwdddn #1swxYwYt| dpd} t| dpd}dtd| |z dg}g}| d}| d}t| dpd}| d}t|ttfrt|dkrt|ttfrdt|cxkrt|krnnt|}t|}||z |z dz}d|dd|ddg}|r|d||td|d |!t|ttfrd"t|dd#g}| d$d%f| d&d'f| d(d)ffD]]\}}t|ttfrrrrrrrrIrnrGrHrmrrr)rr)rrruntimer r credits_urlkey_urlr r" credits_resprkey_respkey_datarArBr1r/rCrDrErF limit_valueremaining_valuer detail_parts usage_partsr8rs r_fetch_openrouter_account_usagerV^s&" G  Ir**0b 1 1 7 7 9 9E tW[[R006B77>>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*c$t|pd}|dvrdS |dkrt||S|dkrt S|dkrt ||Sn#t $rYdSwxYwdS)Nr:>r:autocustomrrWr(r<)rr>lowerr&r:rVr)r*rrrs rfetch_account_usager\s X^$$**,,2244J+++t  ' '-xQQQ Q  $ $133 3  % %28WEE E & tt 4sA?A?(A?? B  B )rr)r8r"rr")r8rrr )rOr rr)rgrhrfr3rri)rwrIrr)r{rrr|)rrh)rfr3rrIrri)rfr3rrIrr)rrrr)rr"rr"rr)NN)rr"rr"rrh)r*r"rr"rr"rrh). __future__rloggingr~ dataclassesrrrtypingrrr ragent.anthropic_adapterr r rr r rhermes_cli.runtime_providerrr getLoggerr$rrrr)rArPrdrvrzrrrrrrrrr&r:rVr\rrrrds"""""" !!!!!!''''''''////////// LLLLLLLL\\\\\\\\\\@@@@@@!       8 $ $&&&& $!!!!!!!! $ R R R R R R R R????&????(^c8XXXX]]]]@,14//////d7777t $         ,14CCCCCCL++++7^7^7^7^v#!-----`::::zEEEEV#! r