i64dZddlmZddlZddlmZmZmZmZd%d Z d&d Z d'dZ d(dZ d)dZ d*dZd+dZd,dZd+dZd-dZd+dZd+dZd.d#Zd+d$ZdS)/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)AnyDictListOptionalconfigDict[str, Any]returnList[Dict[str, Any]]c|dpg}t|trd|D}|r|S|d}t|tr:|dr%|drt|gSt|tr d|DSgS)uReturn the normalized fallback chain as a list of dicts. Accepts both the new list format (``fallback_providers``) and the legacy single-dict format (``fallback_model``). The returned list is always a fresh copy — callers can mutate without touching the config dict. fallback_providerscg|]P}t|t|d,|dAt|QSprovidermodel isinstancedictget.0es z_read_chain..&sSiiiaJq$,?,?iAEE*DUDUiZ[Z_Z_`gZhZhi$q''iiifallback_modelrrcg|]P}t|t|d,|dAt|QSrrrs rrz_read_chain..-sShhhA:a+>+>h155CTCThYZY^Y^_fYgYghQhhhr)rrlistr)rchainresultlegacys r _read_chainr"s JJ+ , , 2E%ii5iii  M ZZ( ) )F&$FJJz$:$:vzz'?R?RV ~&$ihhhhhh IrrNonecH||d<d|vr|dddSdS)zAPersist the chain to ``fallback_providers`` and clear legacy key.r rN)pop)rrs r _write_chainr&1s;#(F  6!! #T*****"!rentrystrc|dd}|dd}|d}|rd|dnd}|d|d |S) z6One-line human-readable rendering of a fallback entry.r?rbase_urlz [] (via ))r)r'rrbasesuffixs r _format_entryr29skyyS))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-defaultrrr+api_mode)rrrstrip)r3rrr'r+r7s r _extract_fallback_from_model_cfgr9Bs 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.authr;r Exception)r;stores r_snapshot_auth_active_providerr@Us\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_lockr;_save_auth_storer<N)r=rCr;rDr>)rArCr;rDr?s r_restore_auth_active_providerrE_s  XXXXXXXXXX     $ $$$&&E',E# $  U # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $       s1A > A AA AA AAc ddlm}|}t|}t|sRestore ``config["model"]`` to a previously-captured snapshot.rr\Nr)rNrGr]r%rbrc)rhrGr]cfgs rreresk:::::::: +--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.rr\u9 No fallback providers configured — nothing to remove.Nc,g|]}t|S)r2rs rrz'cmd_fallback_remove..s ///A}Q///rCancel)_curses_prompt_choicezSelect a fallback to remove: Cancelled — no change.z Removed fallback: r_rHrIr'rJr`z Fallback chain is now empty.)rNrGr]r"rOrfhermes_cli.setuprwr>_numbered_pickrQr%r&r2) rSrGr]rrchoicesrwidxremoveds 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).rr\u8 No fallback providers configured — nothing to clear.Nz Current fallback chain (rHrIr'rJrKrLrMz Clear all entries? [y/N]: z Cancelled.)yyesrxz Fallback chain cleared.)rNrGr]r"rOrQrRr2inputr8lowerKeyboardInterruptEOFErrorr&)rSrGr]rrrUr'resps rcmd_fallback_clearrs:::::::: []]F   E   HIII  GGG as5zz a as5zzQGGT] a a abbbeQ''225 0Q00-..001111 GGG344::<<BBDD x (  n < *+++K GGG %&&& GGGGGs3D.EEquestionr{ 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.rIr^rMTz Choice [1-z]: NrzPlease enter 1-zPlease enter a number) rOrRrrQr8int ValueErrorrr)rr{rUcvalr|s rrzrzAsJ (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-rlsadd)removermclearzUnknown fallback subcommand: z$Use one of: list, add, remove, clear)getattrrVrpr~rrOrd)rSsubs r cmd_fallbackr[s $*D 1 1C &&&$   D!!!!! 4      3c33444 4555mmr)rr r r )rr rr r r#)r'r r r()r3rr r4)r r)rArr r#)r r#)rr r rW)rhrr r#)rr(r{rr r)__doc__ __future__rrbtypingrrrrr"r&r2r9r@rErVrPrprer~rrzrrurrrs}$#""""" ,,,,,,,,,,,,(++++0000&     &    8    MZMZMZMZ`    % % % % P     F4r