sjUdZddlmZddlZddlZddlmZmZmZddl m Z ddl m Z ej eZiZded<iZd ed <ejZdd d dZdd d!dZdd d"dZdd d"dZdd d#dZd$dZd%dZdS)&a Video Generation Provider Registry ================================== Central map of registered providers. Populated by plugins at import-time via ``PluginContext.register_video_gen_provider()``; consumed by the ``video_generate`` tool to dispatch each call to the active backend. Active selection ---------------- The active provider is chosen by ``video_gen.provider`` in ``config.yaml``. If unset, :func:`get_active_provider` applies fallback logic: 1. If exactly one *available* provider is registered, use it. 2. Otherwise return ``None`` (the tool surfaces a helpful error pointing the user at ``hermes tools``). Mirrors ``agent/image_gen_registry.py`` so the two surfaces behave the same: the unconfigured fallback is filtered by ``is_available()`` so a box that has credentials for only one backend (e.g. DeepInfra, while the ``fal``/``xai`` plugins also register unconditionally) auto-selects it instead of returning ``None``. ) annotationsN)DictListOptional)VideoGenProvider)hermes_home_keyzDict[str, VideoGenProvider] _providersz&Dict[str, Dict[str, VideoGenProvider]]_scoped_providers)scopeproviderrr Optional[str]returnNonect|ts$tdt|j|j}t|t r|std|}t5|tnt |i}| |}|||<dddn #1swxYwY|0td|t|jdStd|t|jdS)uRegister a video generation provider. Re-registration (same ``name``) overwrites the previous entry and logs a debug message — this makes hot-reload scenarios (tests, dev loops) behave predictably. z=register_provider() expects a VideoGenProvider instance, got z3Video gen provider .name must be a non-empty stringNz.Video gen provider '%s' re-registered (was %r)z'Registered video gen provider '%s' (%s)) isinstancer TypeErrortype__name__namestrstrip ValueError_lockr r setdefaultgetloggerdebug)r r raw_namertargetexistings >* - -   }H h $ $PHNN,<,<PNOOO >>  D   $}2C2N2NuVX2Y2Y::d##t                 EtTRZ^^Mdeeeee >d8nnF]^^^^^s?CC#&C#List[VideoGenProvider]cFt5tt}|t|p t it|}dddn #1swxYwYt|dS)z0Return all registered providers, sorted by name.Nc|jSN)r)ps r!z list_providers..Jsqv)key) rdictr updater rrlistvaluessorted)r mergeditemss r!list_providersr2Ds &&j!! '++E,F_5F5FKKLLLV]]__%%&&&&&&&&&&&&&&& %-- . . ..sA2BB  B rrOptional[VideoGenProvider]cFt|tsdSt5|}t|p t i|pt|cdddS#1swxYwYdS)z5Return the provider registered under *name*, or None.N)rrrrr rrr )rr r*s r! get_providerr5Ms dC t eejjll $$U%?o.?.?DDHHMMdQ[Q_Q_`cQdQdeeeeeeeeeeeeeeeeeesA*BBBct5|tnt|i}||cdddS#1swxYwYdSr&)rr r rr)rr rs r!snapshot_registrationr7Vs (($}2C2G2Gr2R2Rzz$**,,''((((((((((((((((((sA A  A$'A$currentpreviousboolcx|}t5|tnt|i}|||ur ddddS|||dn|||<||st|ddddn #1swxYwYdS)zERestore a plugin registration only when *current* is still installed.NFT)rrr r rrpop)rr8r9r r*rs r!restore_registrationr=^s **,,C  / /$}2C2N2NuVX2Y2Y ::c??' ) ) / / / / / / / /   JJsD ! ! ! !"F3K  V   ! !% . . . / / / / / / / / / / / / / / / 4s=B/&=B//B36B3c d} ddlm}|}t|tr|dnd}t|trR|d}t|t r(|r|}n2#t$r%}t d|Yd}~nd}~wwxYw|r2 ddl m }| |krd}n#t$rYnwxYwt5tt}|t t#idddn #1swxYwY|r6||}||St d |dSdd fd|D} t'| dkr| dSdS)zResolve the currently-active provider. Reads ``video_gen.provider`` from config.yaml; falls back per the module docstring. Nr)load_config_readonly video_genr z1Could not read video_gen.provider from config: %s)NOUS_MANAGED_PROVIDERfalzEvideo_gen.provider='%s' configured but not registered; failing closedr'rrr:c t|S#t$r,}td|j|Yd}~dSd}~wwxYw)zDWrap ``is_available()`` so a buggy provider doesn't kill resolution.z.video_gen provider %s.is_available() raised %sNF)r: is_available Exceptionrrr)r'excs r!_is_available_safez/get_active_provider.._is_available_safese (()) )    LLI16SV W W W55555 s # A!AAc*g|]}| |SrI).0r'rGs r! z'get_active_provider..s*GGGq1C1CA1F1FGGGGr))r'rrr:)hermes_cli.configr?rr+rrrrErrtools.tool_backend_helpersrAlowerrr r,r rr.len) configuredr?cfgsectionrawrFrAsnapshotr availablerGs @r!get_active_providerrWts !%J O::::::""$$*4S$*?*?I#''+&&&T gt $ $ )++j))C#s## )  ) YY[[ OOO H#NNNNNNNNO   H H H H H H!!%:::"     D  FF ##)--o.?.?DDEEEFFFFFFFFFFFFFFF<< ++  O S    tHGGGHOO--GGGI 9~~| 4s=B#B)) C3CC C?? D  D AE11E58E5ct5ttddddS#1swxYwYdS)z"Clear the registry. **Test-only.**N)rr clearr rIr)r!_reset_for_testsrZs ""!!!""""""""""""""""""s3AA A )r rr r rr)r r rr#)rrr r rr3) rrr8rr9r3r r rr:)rr3)rr)__doc__ __future__rlogging threadingtypingrrragent.video_gen_providerrhermes_constantsr getLoggerrrr __annotations__r Lockrr"r2r5r7r=rWrZrIr)r!res0#"""""''''''''''555555,,,,,,  8 $ $+- ,,,,<>>>>> MQ______4.2//////7;eeeeee*.(((((( ,====@""""""r)