yj6dZddlmZddlZddlmZmZmZmZddl m Z d&d Z d'd Z d(dZ d)dZd*dZd+dZd,dZd-dZd,dZd.dZd,dZd,dZd/d$Zd,d%ZdS)0ug 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_chainconfigDict[str, Any]returnList[Dict[str, Any]]c t|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)r s =/home/ragecks/.hermes/hermes-agent/hermes_cli/fallback_cmd.py _read_chainrs f % %%chainNonecH||d<d|vr|dddSdS)zAPersist the chain to ``fallback_providers`` and clear legacy key.fallback_providersfallback_modelN)pop)r rs r _write_chainr*s;#(F  6!! #T*****"!rentrystrc|dd}|dd}|d}|rd|dnd}|d|d |S) z6One-line human-readable rendering of a fallback entry.provider?modelbase_urlz [] (via ))get)rrrbasesuffixs r _format_entryr'2skyyS))H IIgs # #E 99Z D" *]4]]]]F / /H / /v / //r model_cfgrOptional[Dict[str, Any]]ct|tsdS|dpd}|dp|dpd}|r|sdS||d}|dpd}|r||d<|dpd}|r||d<|S) z^Pull the ``{provider, model, base_url?, api_mode?}`` dict from a ``config["model"]`` snapshot.Nrr!defaultr)rrrapi_mode) isinstancedictr$strip)r(rrrrr,s r _extract_fallback_from_model_cfgr0;s i & &t j))/R6688H ]]9 % % Ew)?)? E2 L L N NE 5t)1EBBE j))/R6688H%$j j))/R6688H%$j Lrcp ddlm}|}|dS#t$rYdSwxYw)zRReturn the current ``active_provider`` in auth.json, or a sentinel if unavailable.r)_load_auth_storeactive_providerN)hermes_cli.authr2r$ Exception)r2stores r_snapshot_auth_active_providerr7Ns\444444  ""yy*+++ tts $' 55valuec ddlm}m}m}|5|}||d<||ddddS#1swxYwYdS#t$rYdSwxYw)z>Write back a previously snapshotted ``active_provider`` value.r)_auth_store_lockr2_save_auth_storer3N)r4r:r2r;r5)r8r:r2r;r6s r_restore_auth_active_providerr<Xs  XXXXXXXXXX     $ $$$&&E',E# $  U # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $       s1A > A AA AA AAc ddlm}|}t|}t|srprint_describe_primarylen enumerater')argsr>r rprimaryirs rcmd_fallback_listrMks\------ []]F   E GGG  3444  4555 ''G 'g''(((  Ys5zz Y Ys5zzQGGI Y Y YZZZeQ''225 0Q00-..001111 GGG YZZZ efff GGGGGr Optional[str]c|d}t|trs|dpdpd}|dp|dpdpd}|d|dSt|tr(|r|SdS)z?One-line description of the primary model for display purposes.rrrr+r"r#N)r$r-r.r/r)r r(rrs rrGrGs 7##I)T"",MM*--4;;==Dy))JY]]7-C-CJsQQSSZWZ++++++)S!!!ioo&7&7!   4rc ddlm}m}ddlm}m}|d|}t j|d}t}ttdtdt ||n-#t$r t|t|wxYw|}|d} t| } | s=t|t|ttd d Sdd lm} m} | | d | d| d } t|}|r| | |d |d|d | r]t|t|ttdt'| dtdd St|t||}t)|}|D]}| | |d |d|d | r1ttdt'| dd S|| t-||||ttdt'| tdt/|dt/|dkrdnddttdd S)zQLaunch the same picker as `hermes model`, then append the selection to the chain.r) _require_ttyselect_provider_and_modelr> 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.)rJz No fallback added.N)BackendIdentitysame_deploymentrr)rrrz. 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 r?r@rrA long.zL Run `hermes fallback list` to view, or `hermes fallback remove` to delete.)hermes_cli.mainrQrRrEr>rTcopydeepcopyr$r7rF SystemExit_restore_model_cfgr<r0agent.backend_identityrUrVbuildr'rappendrrH)rJrQrRr>rT before_cfg model_beforeactive_provider_before after_cfg model_after new_entryrUrV new_ident primary_entry final_cfgrexistings rcmd_fallback_addrls7GGGGGGGG::::::::L   J=!8!899L;== GGG STTT TUUU GGG!!t,,,,, <(((%&<===  I--((K0==I <(((%&<===  $%%%HGGGGGGG%%z**mmG$$z**&I 5\BBM "&&z22##G,,"&&z22      <(((%&<===  [}Y?W?W[[[\\\ KLLL |$$$!"8999 I  " "E   ?  ! !!j11ll7++!j11 "       GGG _}Y//___ ` ` ` FF  LLE"""K  GGG 9}Y77 9 9::: ZCJJ Z ZCJJ!OO Z Z Z[[[ GGG XYYYYYs  B##*C rccddlm}m}|}||ddnt j||d<||dS)z>Restore ``config["model"]`` to a previously-captured snapshot.rrSNr)rEr>rTrr[r\)rcr>rTcfgs rr^r^sk:::::::: +--C }\22G Krc0ddlm}m}|}t|}|s-t t dt dSd|D}|d ddlm}|d|d}n #t$rtd|}YnwxYw||dks|t|krt t d dS| |}t||||t t d t||r9t d t|d t|d krdnddnt dt dS)z+Pick an entry from the chain and remove it.rrSu9 No fallback providers configured — nothing to remove.Nc,g|]}t|S)r').0es r z'cmd_fallback_remove..s ///A}Q///rCancel)_curses_prompt_choicezSelect a fallback to remove: Cancelled — no change.z Removed fallback: rXr?r@rrArYz Fallback chain is now empty.)rEr>rTrrFrahermes_cli.setuprvr5_numbered_pickrHrrr') rJr>rTr rchoicesrvidxremoveds rcmd_fallback_remover}s:::::::: []]F   E   IJJJ /////G NN8F::::::##$BGQOO FFF;WEEF {cAggE !2!2  *+++iinnGK GGG 9w!7!7 9 9::: 0 ^E ^^E aWWY^^^____ ./// GGGGGs3BB$#B$c Hddlm}m}|}t|}|s-t t dt dSt t dt |dt |dkrdnd d t |dD]'\}}t d |d t|(t  td  }n5#ttf$r!t t dYdSwxYw|dvrt ddSt|g||t t dt dS)z0Remove all fallback entries (with confirmation).rrSu8 No fallback providers configured — nothing to clear.Nz Current fallback chain (r?r@rrArBrCrDz Clear all entries? [y/N]: z Cancelled.>yyesrwz Fallback chain cleared.)rEr>rTrrFrHrIr'inputr/lowerKeyboardInterruptEOFErrorr)rJr>rTr rrLrresps rcmd_fallback_clearr.s:::::::: []]F   E   HIII  GGG as5zz a as5zzQGGT] a a abbbeQ''225 0Q00-..001111 GGG344::<<BBDD x (  n < *+++K GGG %&&& GGGGGs3D.EEquestionrz List[str] Optional[int]c&t|t|dD]\}}td|d|t tdt|d}|sdSt |dz }d|cxkrt|krnn|Std t|n@#t $rtd Yn%ttf$rtYdSwxYw) z9Fallback numbered-list picker when curses is unavailable.r@rWrDTz Choice [1-z]: NrzPlease enter 1-zPlease enter a number) rFrIrrHr/int ValueErrorrr)rrzrLcvalr{s rryryQsJ (OOO'1%%1 m1mmmm GGG  6S\\66677==??C tc((Q,CC&&&&#g,,&&&&& 2CLL22 3 3 3 3 + + + ) * * * * *!8,    GGG44  s$ 4C0C2CD-DDc4t|dd}|dvrt|dS|dkrt|dS|dvrt|dS|dkrt |dSt d|t dt d ) z:Top-level dispatcher for ``hermes fallback [subcommand]``.fallback_commandN>Nr!lslistadd>rmremoveclearzUnknown fallback subcommand: z$Use one of: list, add, remove, clear)getattrrMrlr}rrFr])rJsubs r cmd_fallbackrks $*D 1 1C &&&$   D!!!!! 4      3c33444 4555mmr)r r r r )r r rr r r)rr r r)r(rr r))r r)r8rr r)r r)r r r rN)rcrr r)rrrzrr r)__doc__ __future__rr[typingrrrrhermes_cli.fallback_configr rrr'r0r7r<rMrGrlr^r}rryrrqrrrs$#""""" ,,,,,,,,,,,,999999&&&&++++0000&     &    8    dZdZdZdZN    % % % % P     F4r