`gj1dZddlmZddlZddlZddlZddlmZmZddl m Z m Z eje ZdZddZddZdd Zed Gd d Zed GddZddZddddZddZy)uShared dollar-denominated usage model for the billing/subscription surfaces. The single source of truth behind the ``/usage`` and ``/subscription`` usage bars (TUI + CLI). User feedback (Jun 2026): the terminal surfaces show **dollars**, never "credits", and every usage bar must make the monthly subscription allowance and separately-purchased top-up dollars distinctly visible. Data source: the NAS account-info fetch (``NousPortalAccountInfo``), whose ``paid_service_access_info`` carries the three dollar magnitudes we render (despite the legacy ``*_credits`` field names, these are USD floats): - ``subscription_credits_remaining`` -> plan dollars left this month - ``purchased_credits_remaining`` -> top-up dollars left (rolls over) - ``total_usable_credits`` -> total spendable plus ``subscription.monthly_credits`` (the plan's monthly $ allowance, the denominator for the "% used" plan bar) and ``current_period_end`` (renewal). Design: two SEPARATE bars (decided with the user) rather than one crammed three-segment bar — at terminal widths three same-glyph density segments are unreadable. The plan bar is "spent vs allowance this month" (carries % used); the top-up bar is "money you bought, doesn't expire". Each gets full resolution and a single fill glyph, so the bar is never ambiguous and never relies on color. Fail-open everywhere: any missing/non-finite field degrades to fewer bars or a magnitudes-only view; a logged-out / unreachable portal yields ``available=False`` and the surface shows nothing. ) annotationsN) dataclassfield)AnyOptionalg@ct|tst|ttfsyt|}t j |r|SdS)zIReturn value as a float iff it's a real finite number (not bool/NaN/Inf).N) isinstanceboolintfloatmathisfinite)valuefs F/root/.hermes/venv/lib/python3.12/site-packages/agent/billing_usage.py_finiter0s;%je &E e A a 1*d*cd|xsddS)zH``$X.YY`` for display. ``None`` -> ``$0.00`` (callers gate on presence).$z,.2f)rs r_fmt_usdr8s  d# $$rcr|syddlm}t|j}|sy|jdr|dddzn|} |j |}|jd d |jd |jS#t $r- |j |ddd}n#t $r|cYcYSwxYwYbwxYw) zFormat an ISO date/timestamp as a human date, e.g. ``Jul 24, 2026``. Accepts ``2026-07-24``, ``2026-07-24T11:05:01.000Z``, etc. Returns the raw string unchanged if it can't be parsed (never raises), and ``None`` for empty input. Nr)datetimeZz+00:00 z%Y-%m-%dz%b z, ) rstrstripendswith fromisoformat ValueErrorstrptimestrftimedayyear)rrtextisodts r format_renewsr+=s ! u:   D "&--"4$s)h $C  # #C (kk$ "&&BGG9 55  ""49j9B K  s6B B6 B B6 B0+B6/B00B65B6T)frozencbeZdZUdZded<ded<ded<dZded<ed d Zed d Zy )UsageBaru+One full-resolution bar: ``spent`` of ``total``, plus a remaining figure. ``kind`` is ``"plan"`` (monthly allowance, shows % used) or ``"topup"`` (purchased dollars, no denominator — ``spent`` is 0 and ``total`` == ``remaining`` so it renders as a full bar of available balance). rkindr remaining_usd total_usdr spent_usdc |jdk7s|jdkrytdtdt |j |jz dzS)Nplanrd)r/r1maxminroundr2selfs rpct_usedzUsageBar.pct_usedfsG 99 $..A"51c#uT^^dnn%Ds%JKLMMrc||jdkrytdtd|j|jz S)z=Fraction of the bar that should read as 'remaining' (filled).rrg?)r1r6r7r0r9s r fill_fractionzUsageBar.fill_fractionls6 >>Q 3C!3!3dnn!DEFFrN)returnz Optional[int])r>r ) __name__ __module__ __qualname____doc____annotations__r2propertyr;r=rrrr.r.XsN IIu NN GGrr.ceZdZUdZded<dZded<dZded <dZded <dZded <dZ d ed <dZ d ed<dZ d ed<dZ ded<dZ ded<eddZy) UsageModelaSurface-agnostic dollar usage model shared by /usage and /subscription. ``status`` classifies the account for copy selection: - ``"free"`` : no paid access / no subscription (free models only) - ``"low"`` : paid, but total spendable < $5 (ALERT) - ``"healthy"`` : paid, total spendable >= $5 - ``"depleted"`` : paid access lost (balance exhausted) r availablefreerstatusN Optional[str] plan_name renews_atrenews_displayOptional[float]subscription_remaining_usdtopup_remaining_usdtotal_spendable_usdzOptional[UsageBar]plan_bar topup_barcNt|jxr|jdkDS)Nr)r rPr9s r has_topupzUsageModel.has_topups#D,,M1I1IA1MNNr)r>r )r?r@rArBrCrIrKrLrMrOrPrQrRrSrDrUrrrrFrFtsOFC#I}##I}#$(NM(266+//+//#'H '$(I!( OOrrFc  | t|dds tdSt|dd}t|dd}t|dd}|rtt|ddnd}|rtt|d dnd}|rtt|d dnd}| t|d dnd}| t|d dnd}|tt|d dnd} t|xs | duxr| dkD} ||} n#||fD cgc]} | |  } } | r t | nd} |durd}n| s |r|dkDsd}n| | t krd}nd}d}| 9| dkDr4|2t dt| |}td || t d| |z }d}||dkDrtd||d}td|||t|||| || Scc} w#t$r&tjddtdcYSwxYw)zBuild a :class:`UsageModel` from a ``NousPortalAccountInfo``. Fail-open. Returns ``UsageModel(available=False)`` when there's no usable account info (logged out, no entitlement block). Never raises. N logged_inFrGpaid_service_access_info subscriptionpaid_service_accesssubscription_credits_remainingpurchased_credits_remainingtotal_usable_creditsr4current_period_endmonthly_creditsrdepletedrHlowhealthyrr/r0r1r2topupT) rGrIrKrLrMrOrPrQrRrSu(usage ▸ model build failed (fail-open)exc_info) getattrrFrr sumLOW_BALANCE_THRESHOLD_USDr6r7r.r+ Exceptionloggerdebug) account_infoaccesssubpaid sub_remainingtopup_remaining total_usablerKrLmonthlyhas_subscriptiontotal_spendablevpartsrIrR remainingrSs rusage_model_from_accountr{sL J+  w|[%'P. .'A4HlND9|%:DA\b0PRV WXhl [a''&2OQU"VWgkQWwwv/EtLM]a 25/GC.t @CGC!5t -) I(3'4 / /  MSd + ?$ OE**+s0F)CF)1F$9F$=B&F)$F)),GGg$@timeoutc*t}||S ddlm}|dxsijd}t |t r|j s tdS ddl }ddl m }|jjd 5}|j|d j| }dddt!S#t$rtdcYSwxYw#1swY/xYw#t$r&t"j%dd tdcYSwxYw)aFetch account-info and build the shared usage model. Fail-open. Dev override: ``HERMES_DEV_CREDITS_FIXTURE`` short-circuits to a fixture so every usage state is testable without a live account (mirrors the existing ``/usage`` credits-block fixture path). Nr)get_provider_auth_statenous access_tokenFrX)get_nous_portal_account_info) max_workersT) force_freshr|u)usage ▸ portal fetch failed (fail-open)rf)_dev_fixture_usage_modelhermes_cli.authrgetr rr rFrkconcurrent.futureshermes_cli.nous_accountrfuturesThreadPoolExecutorsubmitresultr{rlrm)r}fixturertok concurrentrpoolaccounts rbuild_usage_modelrs '(G+;&v.4"99.I3$. .*5 +!H    2 2q 2 A jTkk">DkQXXahXiG j'00 +E**+ j j + @4 PE**+s<A B<&C#$C)C#<CCC C##,DDctjdxsdjj}|sy|dk(rt dddS|dvrt ddd d d d t d d d dS|dvr.t ddd d d ddt d d d dt dddd S|dk(rt ddd d ddt d dd dS|dk(rt ddd d ddt d dd d Sy)zMap ``HERMES_DEV_CREDITS_FIXTURE`` to a usage model for offline UX work. Recognized names: ``free | healthy | low | topup | depleted``. Returns ``None`` when the env var is unset (real portal path runs). HERMES_DEV_CREDITS_FIXTURENrHT)rGrIrK)rcmidrcPlusz 2026-07-01g,@r4g4@g@rd)rGrIrKrLrOrQrR)reztop-upg(@g:@rer) rGrIrKrLrOrPrQrRrSrbg333333 @g0@ra)osgetenvr lowerrFr.)names rrrs< II2 3 9r @ @ B H H JD  v~D4HH !!"'+ $6Y\]   """'+ $ $6Y\]G44[^_   u}"'* #6X\]   z"'* #6X\]   r)rrr>rN)rrNr>r)rrJr>rJ)rnrr>rF)r}r r>rF)r>zOptional[UsageModel])rB __future__rloggingr r dataclassesrrtypingrr getLoggerr?rlrjrrr+r.rFr{rrrrrrs># (   8 $  +% 66 $GGG6 $OOO4P+f+/+L;r