yjUdZddlmZddlZddlZddlmZmZmZddl m Z ej e Z iZded<ejZdd ZddZddZddZddZdS)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)VideoGenProviderzDict[str, VideoGenProvider] _providersproviderrreturnNonec<t|ts$tdt|j|j}t|t r|stdt5t |}|t|<dddn #1swxYwY|0t d|t|jdSt d|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_lockrgetloggerdebug)r rexistings >/home/ragecks/.hermes/hermes-agent/agent/video_gen_registry.pyregister_providerr(sJ h 0 1 1  ->>* - -    =D dC P PNOOO $$>>$''# 4$$$$$$$$$$$$$$$ EtTRZ^^Mdeeeee >d8nnF]^^^^^s%B11B58B5List[VideoGenProvider]ct5tt}dddn #1swxYwYt |dS)z0Return all registered providers, sorted by name.Nc|jS)Nr)ps rz list_providers..Dsqv)key)rlistrvaluessorted)itemss rlist_providersr(@s **Z&&(())*************** %-- . . ..s ';??rrOptional[VideoGenProvider]ct|tsdSt5t|cdddS#1swxYwYdS)z5Return the provider registered under *name*, or None.N)r rrrrrrs r get_providerr+Gs dC t ,,~~djjll++,,,,,,,,,,,,,,,,,,s,AAAc d} ddlm}|}t|tr|dnd}t|trR|d}t|t r(|r|}n2#t$r%}t d|Yd}~nd}~wwxYwt5tt}dddn #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: %szEvideo_gen.provider='%s' configured but not registered; failing closedr rr boolc 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_safeose (()) )    LLI16SV W W W55555 s # A!AAc*g|]}| |Sr6).0r r4s r z'get_active_provider..{s*GGGq1C1CA1F1FGGGGr")r rr r/)hermes_cli.configr-r dictrrrr2rrrrr%len) configuredr-cfgsectionrawr3snapshotr availabler4s @rget_active_providerrCOs !%J O::::::""$$*4S$*?*?I#''+&&&T gt $ $ )++j))C#s## )  ) YY[[ OOO H#NNNNNNNNO $$ ##$$$$$$$$$$$$$$$<< ++  O S    tHGGGHOO--GGGI 9~~| 4s*B#B)) C3CC"DD Dcxt5tddddS#1swxYwYdS)z"Clear the registry. **Test-only.**N)rrclearr6r"r_reset_for_testsrFs~ s /33)r rr r )r r)rrr r))r r))r r )__doc__ __future__rlogging threadingtypingrrragent.video_gen_providerr getLoggerrrr__annotations__Lockrrr(r+rCrFr6r"rrPs0#"""""''''''''''555555  8 $ $+- ,,,, ____0////,,,,0000fr"