FjN^"UdZddlmZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl m Z mZmZedZdZdad ed <e d Gd dZe d GddZe d GddZdTdUdZdd ddVdZdWd ZdXd%ZdYd'Zd(dd)dZd-Zd[d1Zd\d2Zd]d3Zd^d5Z d_d8Z! dTd`d:Z"dad<Z#dbd>Z$dcdAZ%dddCZ&dddDdedJZ'dfdKZ(dgdLZ)dhdMZ*didNZ+djdPZ,dkdRZ-dhdSZ.dS)lz3Normalized Nous Portal account entitlement helpers.) annotationsN) dataclass)datetimetimezone)AnyLiteralOptional)jwt account_api inference_keynoneerror<z1tuple[str, float, 'NousPortalAccountInfo'] | None_account_info_cacheT)frozencdeZdZUdZded<dZded<dZded<dZded<dZded <dZ ded <dS) NousPortalSubscriptionInfoN Optional[str]plan Optional[int]tierOptional[float]monthly_chargecurrent_period_endcredits_remainingrollover_credits) __name__ __module__ __qualname__r__annotations__rrrrr6/usr/local/lib/hermes-agent/hermes_cli/nous_account.pyrrs{DD&*N****(,,,,,)-----(,,,,,,,r"rceZdZUdZded<dZded<dZded<dZded<dZded <dZ ded <dZ ded <dZ ded <dZ d ed<dZ d ed<dZd ed<dZd ed<dS)NousPaidServiceAccessInfoNOptional[bool]allowed paid_accessrreasonorganisation_idreffective_at_mshas_active_subscriptionactive_subscription_is_paidsubscription_tierrsubscription_monthly_chargesubscription_credits_remainingpurchased_credits_remainingtotal_usable_credits)rrrr'r r(r)r*r+r,r-r.r/r0r1r2r!r"r#r%r%s"G"""""&K&&&& F    %)O))))%)O)))).22222266666'+++++3777776:"::::377777,0000000r"r%creZdZUded<ded<ded<dZded<dZded <dZded <dZded <dZded <dZ ded <dZ ded<dZ ded<dZ ded<dZ ded<dZded<dZded<dZded<dZded<dZded<dZded<dZded<dZded<ed$d!Zed$d"Zed$d#ZdS)%NousPortalAccountInfobool logged_inNousAccountInfoSourcesourcefreshNruser_idorg_id client_id product_id nous_clientportal_base_urlinference_base_urlFinference_credential_presentcredential_sourcezOptional[datetime] expires_atemail privy_did$Optional[NousPortalSubscriptionInfo] subscriptionr&paid_service_access#Optional[NousPaidServiceAccessInfo]paid_service_access_infoOptional[dict[str, Any]] raw_claims raw_accountrreturnc|jduSNTrHselfs r#is_paidzNousPortalAccountInfo.is_paidF'4//r"c|jduS)NFrQrRs r# is_free_tierz"NousPortalAccountInfo.is_free_tierJs'500r"c|jduSrPrQrRs r#tool_gateway_entitledz+NousPortalAccountInfo.tool_gateway_entitledNrUr")rNr5)rrrr r:r;r<r=r>r?r@rArBrCrDrErGrHrJrLrMrpropertyrTrWrYr!r"r#r4r4.sOOO!!!!KKK!G!!!! F    #I#### $J$$$$!%K%%%%%)O))))(,,,,,). ....'+++++%)J))))E#I####9=L====*.....DHHHHH+/J////,0K0000E 000X0111X1000X000r"r4 account_infoOptional[NousPortalAccountInfo]rNstrc ddlm}n#t$rd}YnwxYwd}||j}t |t r|s|}|ddS)z>Return the billing URL for a normalized Nous account snapshot.r)DEFAULT_NOUS_PORTAL_URLhttps://portal.nousresearch.comN/z/billing)hermes_cli.authr_ Exceptionr? isinstancer]striprstrip)r[r_bases r#nous_portal_billing_urlrhSsD;;;;;;; DDD"CD D+ dC ' '&kk# ( ( ((s  z this feature) capabilityinclude_refresh_hintrirjr5rct|}| |jdurdS| d|d|dS|js|jr d|d|dSd|d |dS|j-d |d }|jr|d |jdz }|r|d z }|d|dz }|S|j}|r|jnd}|dkrd|dS|dks |jdur!t|||}|r |js|dz }|Sd|d|dS)aReturn user-facing guidance for a missing Nous paid entitlement. ``None`` means the account is known to have paid service access. The message intentionally works from normalized entitlement fields rather than subscription price alone: purchased credits without a subscription still count as paid access, while a paid subscription with exhausted usable credits does not. NTz9Hermes could not verify your Nous Portal entitlement, so zO is unavailable. Run `hermes model` to refresh your login, or check billing at .zeNous inference credentials are configured, but Hermes cannot verify your Nous Portal paid access for zc. Log in with `hermes model` to enable Portal-managed features. Billing and credits are managed at zLog in to Nous Portal to use z9: run `hermes model`. Billing and credits are managed at z9Hermes could not verify your Nous Portal paid access, so z is unavailable.z Account lookup failed: z, Run `hermes model` to refresh your session.z Check billing at account_missingzOHermes could not find a Nous Portal account or organisation for this login, so zi is unavailable. Run `hermes model` to authenticate again; if the problem persists, contact Nous support.no_usable_creditsFzF If you recently bought credits, run `hermes model` to refresh Hermes.zIYour Nous Portal account does not currently have paid service access, so 2 is unavailable. Add credits or update billing at ) rhrHr6rArrJr)_no_paid_access_messager9)r[rirj billing_urldetailaccessr)messages r#&format_nous_portal_entitlement_messagerubs*,77KL$D$L$Lt )  ) )% ) ) )  !   4 94>99+6999  AJ A A2= A A A '/         G F1CFFF FF  E D DF5{5555  2F$ .V]]$F """ R# R R R $$$ (HE(Q(Q), KPP  ` (: ` _ _G [ [ [LW [ [ [r"rqcZ|j}|r|jnd}|r|jnd}|r|jnd}|r|jnd}|r|jnd}|r|rt |||} d| d|d|dS|r |dur d|d|dS|durt |||} d| d|d |dSt |||} d | d|d |dS) Nz&Your Nous Portal credits are exhaustedz, so z, is unavailable. Top up or renew credits at rlFzGYour current Nous Portal plan does not include paid service access, so z+ is unavailable. Upgrade or add credits at zEYour Nous Portal account has no active subscription or usable creditsz- is unavailable. Subscribe or add credits at z3Your Nous Portal account has no usable paid creditsro)rJr,r-r2r0r1_credit_detail) r[rirqrsr,r- total_usablesubscription_creditspurchased_credits credit_details r#rprps  2F@FPf<DN::$ #> &|5IK\]]  I] I I I I:E I I I  #>%#G#G X X XIT X X X %''&|5IK\]]  ! ! !#- ! ! ! ! ! #<1EGXYYM Xm X X  X XIT X X Xr"rxrryrzcg}||d|d||d|d||d|d|sdSdd|dS) Nzusable $z.2fzsubscription $z purchased $z (z, ))appendjoin)rxryrzpartss r#rwrws E 2 222333' @&:@@@AAA$ :#4:::;;; r # %  # # ##r"Nonec dadS)z7Clear the short-lived account-info cache used by tests.N)rr!r"r#$reset_nous_portal_account_info_cachersr"F) force_freshmin_jwt_ttl_secondsrrintc ddlm}|dpi}n(#t$r}t|dcYd}~Sd}~wwxYw|d}t |}t |tr|s1D .J ;Y&&DN,<,>}t5.$77L lC ( ( 0B0B0D0Dt ($//   %UK66 E/ 6 6 0u0$77 0uj$//UK66IWUGY%G%GII  E ' , 3      O *<9IJJ    ,           *,    '++g&&,, ++g&&B*B,     &(   s* D D>"D93D>9D> Optional[Any]cddlm}|d}|r|sdSt|}|sdSd d}t || S) Nr) load_poolrrrrNtuple[float, float, int]ctt|ddpd}tt|ddpd}tt|ddpd}||| fS)Nagent_key_expires_atgrCpriorityr)_parse_iso_timestamprr)r agent_exp access_exprs r#_entry_sort_keyz0_select_nous_pool_entry.._entry_sort_keyso(8NPT)U)UVV]Z] )'%t*L*LMMTQT wuj!449:::y11r")key)rrrNr)agent.credential_poolrhas_credentialslistentriesmax)rpoolrrs r#rrs////// 9V  D t++--t4<<>>""G t2222 wO , , ,,r"rrclt|dd}t|tr|sdStt|ddpd}t|dd}|dpt |S)NrF auth_typer} refresh_tokenoauth)rrdr]relower startswithr5)rrrrs r#rrs5.$77L lC ( ( 0B0B0D0DuGE;339r::@@BBHHJJIE?D99M    ( ( ?D,?,??r"rc|pdd}|d}d|dd}tj||}tj|d 5}t j|}dddn #1swxYwYt|tr|niS) Nr`raz/api/oauth/accountzBearer zapplication/json) AuthorizationAccept)headers)timeout) rfurllibrequestRequesturlopenjsonloadsreaddecoderddict)rr?rgurlrreqresprs r#rrs   @@ H H M MD  % % %C1<11$G . g 6 6C   Q  / /34*TYY[[//1122333333333333333 $// 777R7s'9B,,B03B0tokenc  ddlm}n#t$rYdSwxYw||}|sdSt|d}|5|t jt dt|zkrdSt|d}t|d}t||t|d|} tdidd d d d d dt|ddt|ddt|dp|ddt|ddt|dd|dt|ddd dt|dpddtj|tjd|d| dt#|S)Nr)_decode_jwt_claimsexpr(r.r;)r'r(r*r.r6Tr8r r9Fr:subr<r=r>r?r@rArB auth_storerC)tzrHrJrLr!)rbrrc _coerce_floatrrrr _coerce_bool _coerce_intr% _coerce_strr4r fromtimestamprutcr) rrr?rrclaimsrr(r. access_infos r#rrs6666666 tt  & &F t  5)) * *C {cTY[[3q#6I2J2J+K+KKKKtvzz-8899K#FJJ/B$C$CDD+#FJJx$8$899+ K !   $ u e FJJu--...   6::h//000   fjj55O;9O9OPPP  vzz,77888   = 9 9::: ( 'uyy1E'F'FGGG &*T &eii0C&D&DEEU )#(,???? (K "-  <<s eT " "t % 6** + + 6** + +$UYY/?%@%@AA&uyy1E'F'FGG' 2E(F(FGG&uyy1C'D'DEE    r"rIcnt|tsdSt|d}t|d}t ||t |dt |dt |dt|dt|dt |dt|d t|d t|d t|d  S)Nr'r(r)r*r+r,r-r.r/r0r1r2) r'r(r)r*r+r,r-r.r/r0r1r2)rdrrrr%rrr)rr'r(s r#rrKsW eT " "t599Y//00Guyy7788K $599X..//#EII.?$@$@AA#EII.?$@$@AA ,UYY7P-Q-Q R R$0;X1Y1Y$Z$Z%eii0C&D&DEE$1%))r!s999"""""" !!!!!!'''''''')))))))))) VWIMMMMM $-------- $ 1 1 1 1 1 1 1 1 $!0!0!0!0!0!0!0!0H ) ) ) ) )$%!% EEEEEEP%%%%P$$$$"!555555p0 0 0 0 fBDDDDN----&@@@@&* 8 8 8 8 8 ....b""""J    2&*,0 "%%%%////     6666r"