j# pUdZddlmZddlmZddlmZmZmZm Z dZ dZ dddd dddd dd dd dddd dd dd dddd dddd d ddd d Z d e d<edGddZd-dZd.dZd/dZd0dZddlZdd lmZddlZedGd!d"Zd1d'Z d2d3d,ZdS)4uODetect xAI models retired on May 15, 2026. Source: https://docs.x.ai/developers/migration/may-15-retirement Pure logic: walks a Hermes config dict, returns issues for any reference to a retired xAI model. No I/O, no CLI dependencies — testable in isolation and reusable from both `hermes doctor` and a future `hermes migrate xai`. ) annotations) dataclass)AnyDictListOptionalz8https://docs.x.ai/developers/migration/may-15-retirementz May 15, 2026zgrok-4.3N) replacementreasoning_effortnotenonezgrok-imagine-image-quality)z grok-4-0709zgrok-4-fast-reasoningzgrok-4-fast-non-reasoningzgrok-4-1-fast-reasoningzgrok-4-1-fast-non-reasoningzgrok-code-fast-1zgrok-3zgrok-imagine-image-proz#Dict[str, Dict[str, Optional[str]]]_RETIRED_MODELST)frozencNeZdZUdZded<ded<ded<dZded<dZded <dS) RetirementIssuez|d }t|tr|d|d|d}t|trI|d}t|tr|d|d|d} t| trs| d} t| trI| d} t| tr|d| d S)u'Walk all model slots in a Hermes config and return retirement issues. Slots scanned: - ``principal.model`` - ``auxiliary..model`` (introspective — covers future aux slots) - ``delegation.model`` - ``tts.xai.model`` - ``plugins.image_gen.xai.model`` pathrmodelrrNonec *t|sdSt|}t|}|dSt |||d|d|ddS)Nr r r )rrr r r )r*r&r getappendr)r/r0normentryissuess r_checkz%find_retired_xai_refs.._checkJsu%%  F%  ##D)) = F om,"YY'9::6""         r principalzprincipal.model auxiliaryz auxiliary.z.model delegationzdelegation.modelttsxaiz tts.xai.modelplugins image_genzplugins.image_gen.xai.model)r/rr0rrr1)r)dictr3items) r+r8r9aux slot_nameslot_cfgr;r<tts_xair>r?ig_xair7s @rfind_retired_xai_refsrG>s7%'F        fd # #  ;''I)T"": )--"8"8999 **[ ! !C#tN#&99;; N N Ix(D)) N5I555x||G7L7LMMML))J*d##<!:>>'#:#:;;; **U  C#t:''%.. gt $ $ : F?GKK$8$8 9 9 9jj##G'4  KKK ,, i & & K]]5))F&$'' K4fjj6I6IJJJ Mrissuec|jd|jd|jg}|jr|d|jd|jr|d|jdd|S)z8One-line human-readable rendering of a retirement issue.z: u → use z(set reasoning_effort: "z")z[note: ] )rrr r r4r join)rHpartss r format_issuerN{s  UU 3UU@QUU E L J0FJJJKKK z. ,uz,,,--- 88E??r)Pathc<eZdZUdZded<ded<ded<ded <d S) ApplyResultz#Outcome of an apply_migration call.rO file_pathzOptional[Path] backup_pathr-issues_resolvedr'config_changedN)rrrrrrrrrQrQsE--OOO****rrQyaml_docr dotted_path'tuple[Any, str]'c|d}t|dkrtd||}|ddD]8}t|tr||vrt d|d|||}9||dfS)zResolve a dotted slot path to (parent_mapping, leaf_key). Example: "auxiliary.vision.model" -> (yaml_doc["auxiliary"]["vision"], "model"). Raises KeyError if any intermediate node is missing or not a mapping. .z$Path must have at least one parent: Nz Path segment z missing in )splitr# ValueErrorr)r@KeyError)rVrWrMnodesegments r_walk_to_parentrbs   c " "E 5zzA~~O OOPPP D":$%% S)<)<Q7QQ+QQRR RG} r?rrrOr7backupcddlm}t|}|st ||st |dgdS|d}d|_|d d 5}||}dddn #1swxYwY|t |dgdSg}|D]\} t||j \} } n#t$rY(wxYw|j | | <|j r |j | d <||]|st |dgdSd} |retjd } ||jd| } t+j|| |dd 5}|||dddn #1swxYwYt || |dS)aXRewrite ``config_path`` in-place so each issue is resolved. For every issue, the model name is replaced by ``issue.replacement``. If the issue has ``reasoning_effort`` set (i.e. the migration is from a ``*-non-reasoning`` variant), a sibling ``reasoning_effort`` key is added or updated alongside the model. Uses ``ruamel.yaml`` round-trip mode so comments, key order, indentation, and type literals (booleans, ints) are preserved. A backup copy is written to ``.bak-pre-migrate-xai-YYYYMMDD-HHMMSS`` before rewriting, unless ``backup=False``. r)YAMLNF)rRrSrTrUrt)typTrzutf-8)encodingr z %Y%m%d-%H%M%Sz.bak-pre-migrate-xai-w) ruamel.yamlrerOexistsFileNotFoundErrorrQpreserve_quotesopenloadrbrr_r r r4_dtdatetimenowstrftime with_namenameshutilcopy2dump) rr7rcreyamlfhdocresolvedrHparentleafrStss rapply_migrationrs&!     {##K     - ,,,  !      4D>>>DD   #  0 0Biimm {!     ')H   *30ABBLFDD    H (t  ! @).)?F% &  !     #'K / \     ( ( 9 9!++ : :b : :    [+...   #  0 0B #r      s68BB!BC C'&C'0GGG)rrrr)rrrr')r+r,rr-)rHrrr)rVrrWrrrX)T)rrOr7r-rcr'rrQ)r __future__r dataclassesrtypingrrrrMIGRATION_GUIDE_URLRETIREMENT_DATEr rrr&r*rGrNrrrqpathlibrOrwrQrbrrrrrsG#"""""!!!!!!,,,,,,,,,,,,Q 5?TXcg$h$h4>TXcg$h$h4>TZdh$i$i4>TXcg$h$h4>TZdh$i$i4>TXcg$h$h4>TXcg$h$h4Pfjtx$y$y 8 8     $    4444 ::::z       $(RRRRRRRr