`gj2dZddlmZddlZddlmZmZmZmZddl m Z ddZ ddZ ddZ dd Zdd Zdd Zdd Zdd ZddZddZddZddZddZddZy)ug hermes fallback — manage the fallback provider chain. Fallback providers are tried in order when the primary model fails with rate-limit, overload, or connection errors. See: https://hermes-agent.nousresearch.com/docs/user-guide/features/fallback-providers Subcommands: hermes fallback [list] Show the current fallback chain (default when no subcommand) hermes fallback add Pick provider + model via the same picker as `hermes model`, then append the selection to the chain hermes fallback remove Pick an entry to delete from the chain hermes fallback clear Remove all fallback entries Storage: ``fallback_providers`` in ``~/.hermes/config.yaml`` (top-level, list of ``{provider, model, base_url?, api_mode?}`` dicts). The legacy single-dict ``fallback_model`` format is migrated to the new list format on first add. ) annotationsN)AnyDictListOptionalget_fallback_chainct|S)uyReturn the normalized fallback chain as a list of dicts. Accepts both the new list format (``fallback_providers``) and the legacy ``fallback_model`` format. When both are present, the effective chain is merged with ``fallback_providers`` entries kept first. The returned list is always a fresh copy — callers can mutate without touching the config dict. r)configs J/root/.hermes/venv/lib/python3.12/site-packages/hermes_cli/fallback_cmd.py _read_chainr s f %%c<||d<d|vr|jddyy)zAPersist the chain to ``fallback_providers`` and clear legacy key.fallback_providersfallback_modelN)pop)r chains r _write_chainr*s)#(F  6! #T*"rc|jdd}|jdd}|jd}|rd|dnd}|d|d |S) z6One-line human-readable rendering of a fallback entry.provider?modelbase_urlz [] (via ))get)entryrrbasesuffixs r _format_entryr"2sXyyS)H IIgs #E 99Z D"s4&]FWGH:Qvh //rct|tsy|jdxsdj}|jdxs|jdxsdj}|r|sy||d}|jdxsdj}|r||d<|jdxsdj}|r||d<|S) z^Pull the ``{provider, model, base_url?, api_mode?}`` dict from a ``config["model"]`` snapshot.Nrrdefaultr)rrrapi_mode) isinstancedictrstrip) model_cfgrrrrr%s r _extract_fallback_from_model_cfgr*;s i & j)/R668H ]]9 % Ew)? E2 L L NE 5)1EBE j)/R668H$j j)/R668H$j Lrc^ ddlm}|}|jdS#t$rYywxYw)zRReturn the current ``active_provider`` in auth.json, or a sentinel if unavailable.r)_load_auth_storeactive_providerN)hermes_cli.authr,r Exception)r,stores r _snapshot_auth_active_providerr1Ns44 "yy*++ s  ,,c ddlm}m}m}|5|}||d<||dddy#1swYyxYw#t$rYywxYw)z>Write back a previously snapshotted ``active_provider`` value.r)_auth_store_lockr,_save_auth_storer-N)r.r3r,r4r/)valuer3r,r4r0s r _restore_auth_active_providerr6XsX  XX   $$&E',E# $ U # $ $ $    s#=1=:== A A c ddlm}|}t|}t|s+tdttdtyt |}|rtd|ttdt |dt |d k(rd nd d t |d D]\}}td |dt|!ttdtdty)z!Print the current fallback chain.r) load_configz# No fallback providers configured.z$ Add one with: hermes fallback addNz Primary: z Fallback chain ( rentries): . zM Tried in order when the primary fails (rate-limit, 5xx, connection errors).zY Docs: https://hermes-agent.nousresearch.com/docs/user-guide/features/fallback-providers)hermes_cli.configr8r print_describe_primarylen enumerater")argsr8r rprimaryirs r cmd_fallback_listrGks- ]F  E G  34  45 'G gY'(  s5zl!s5zQGI+VVX YZeQ'25 QCr-./012 G YZ ef Grcx|jd}t|tri|jdxsdjxsd}|jdxs|jdxsdjxsd}|d|dSt|tr |jr|jSy)z?One-line description of the primary model for display purposes.rrrr$rrN)rr&r'r(str)r r)rrs r rArAs 7#I)T"MM*-4;;=Dy)JY]]7-CJsQQSZWZz++)S!ioo&7  rcddlm}m}ddlm}m}|d|}t j|jd}t}ttdtdt |||}|jd} t| } | s,t|t|ttd y t|} | rZ| d | d k(rO| d| dk(rDt|t|ttd t| d tdy t|t||} t!| } | D]U}|jd | d k(s|jd| dk(s3ttdt| dy | j#| t%| | || ttdt| tdt'| dt'| dk(rdnddttdy #t$rt|t|wxYw)zQLaunch the same picker as `hermes model`, then append the selection to the chain.r) _require_ttyselect_provider_and_modelr8 save_configz fallback addrzG Adding a fallback provider. The picker below is the same one used byuH `hermes model` — select the provider + model you want as a fallback.)rDz No fallback added.Nrz. Selected model matches the current primary (z).u; A provider cannot be a fallback for itself — no change. u. is already in the fallback chain — skipped.z Added fallback:  Chain is now r9r:rr; long.zL Run `hermes fallback list` to view, or `hermes fallback remove` to delete.)hermes_cli.mainrKrLr?r8rNcopydeepcopyrr1r@ SystemExit_restore_model_cfgr6r*r"r appendrrB)rDrKrLr8rN before_cfg model_beforeactive_provider_before after_cfg model_after new_entry primary_entry final_cfgrexistings r cmd_fallback_addras.G: J==!89L;= G ST TU G!t, I--(K0=I <(%&<=  $%5\BMz2i 6KKg&)G*<<<(%&<=  >}Y?W>XXZ[\ KL |$!"89 I  "E << #y'< <LL)Yw-?? G B}Y/00^_ `   LLE"  G }Y78 9: OCJRestore ``config["model"]`` to a previously-captured snapshot.rrMNr)r?r8rNrrSrT)rYr8rNcfgs r rVrVs:: -C }}\2G rcddlm}m}|}t|}|s t t dt y|Dcgc] }t |}}|j d ddlm}|d|d}||dks|t|k\rt t dy|j|} t||||t t d t | |r6t d t|d t|d k(rd nddt yt dt ycc}w#t$rtd|}YwxYw)z+Pick an entry from the chain and remove it.rrMu9 No fallback providers configured — nothing to remove.NCancel)_curses_prompt_choicezSelect a fallback to remove: Cancelled — no change.z Removed fallback: rPr9r:rr;rQz Fallback chain is now empty.)r?r8rNr r@r"rWhermes_cli.setuprfr/_numbered_pickrBrr) rDr8rNr rechoicesrfidxremoveds r cmd_fallback_removerns6: ]F  E   IJ )./A}Q/G/ NN8F:#$BGQO {cAgE !2  *+iinG G w!7 8 9:  E |1E aWY,WW]^_ G ./ G10 F;WEFsD6'D;;EEc ddlm}m}|}t|}|s t t dt yt t dt |dt |dk(rdnd d t |dD]\}}t d |d t|!t  td jj}|dvr t dyt|g||t t dt y#ttf$rt t dYywxYw)z0Remove all fallback entries (with confirmation).rrMu8 No fallback providers configured — nothing to clear.Nz Current fallback chain (r9r:rr;r<r=r>z Clear all entries? [y/N]: z Cancelled.>yyesrgz Fallback chain cleared.)r?r8rNr r@rBrCr"inputr(lowerKeyboardInterruptEOFErrorr)rDr8rNr rrFrresps r cmd_fallback_clearrws: ]F  E   HI  G &s5zl!s5zQGT]3^^` abeQ'25 QCr-./012 G34::<BBD  < *+ G %& G x (  ns+'D$D=<D=ct|t|dD]\}}td|d|t tdt|dj }|syt |dz }d|cxkrt|kr|Stdt|g#t $rtd Yttf$r tYywxYw) z9Fallback numbered-list picker when curses is unavailable.r:rOr>z Choice [1-z]: NrzPlease enter 1-zPlease enter a number) r@rCrrrBr(int ValueErrorrtru)questionrkrFcvalrls r riri:s (O'1%1 1#Rsm G  *S\N#67==?Cc(Q,CC&#g,& ' OCL>2 3  + ) *!8,  G s#(B#&#B# B##C9CCct|dd}|dvr t|y|dk(r t|y|dvr t|y|dk(r t |yt d|t dt d ) z:Top-level dispatcher for ``hermes fallback [subcommand]``.fallback_commandN>Nrlslistadd>rmremoveclearzUnknown fallback subcommand: z$Use one of: list, add, remove, clear)getattrrGrarnrwr@rU)rDsubs r cmd_fallbackrTst $*D 1C &&$   D! 4  -cU34 45mr)r Dict[str, Any]returnList[Dict[str, Any]])r rrrrNone)rrrrI)r)rrzOptional[Dict[str, Any]])rr)r5rrr)rr)r rrz Optional[str])rYrrr)r{rIrkz List[str]rz Optional[int])__doc__ __future__rrStypingrrrrhermes_cli.fallback_configr r rr"r*r1r6rGrArarVrnrwrirrr rsf$# ,,9&+0&  & 8 MZ` % P  F4r