`gjUdZddlmZddlZddlZddlmZmZmZddl m Z eje Z iZded<ej Zd dZdd Zdd Zdd Zdd Zy)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] _providersct|ts!tdt|j|j }t|t r|js tdt5tj|}|t|<ddd+tjd|t|jytjd|t|jy#1swYaxYw)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)providerrexistings K/root/.hermes/venv/lib/python3.12/site-packages/agent/video_gen_registry.pyregister_providerr(s h 0 1>**+ -   ==D dC  NOO $>>$'# 4$ EtTRZ^MdMde >d8nF]F]^ $$s /C..C7ct5ttj}dddt dS#1swYxYw)z0Return all registered providers, sorted by name.Nc|jS)Nr)ps rz list_providers..Ds qvv)key)rlistrvaluessorted)itemss rlist_providersr%@s; *Z&&()* %- ..**s ;Act|tsyt5tj |j cdddS#1swYyxYw)z5Return the provider registered under *name*, or None.N)r rrrrrrs r get_providerr'Gs9 dC  ,~~djjl+,,,s #AAcd} ddlm}|}t|tr|j dnd}t|trA|j d}t|t r |j r|j }t5tt}ddd|r,j |}||Stjd|yd d}jD cgc] } || s | } } t| d k(r| dSy#t$r }tjd|Yd}~d}~wwxYw#1swYxYwcc} w) zResolve the currently-active provider. Reads ``video_gen.provider`` from config.yaml; falls back per the module docstring. Nr) load_config video_genrz1Could not read video_gen.provider from config: %szEvideo_gen.provider='%s' configured but not registered; failing closedc t|jS#t$r+}tj d|j |Yd}~yd}~wwxYw)zDWrap ``is_available()`` so a buggy provider doesn't kill resolution.z.video_gen provider %s.is_available() raised %sNF)bool is_available Exceptionrrr)rexcs r_is_available_safez/get_active_provider.._is_available_safeosB () )  LLI166SV W s A!A  A)rrreturnr,)hermes_cli.configr)r dictrrrr.rrrrr"len) configuredr)cfgsectionrawr/snapshotrr0r availables rget_active_providerr<Os. !%J O1m*4S$*?#''+&T gt $++j)C#s#  YY[  $ #$<< +  O S  %OO-Gq1CA1FGIG 9~| A O H#NNO$$2Hs0BD D;' E5E D8D33D8;Ecbt5tjdddy#1swYyxYw)z"Clear the registry. **Test-only.**N)rrclearrr_reset_for_testsr@s& s%.)rrr2None)r2zList[VideoGenProvider])rrr2Optional[VideoGenProvider])r2rB)r2rA)__doc__ __future__rlogging threadingtypingrrragent.video_gen_providerr getLoggerr rr__annotations__Lockrrr%r'r<r@r?rrrLsf0#''5   8 $+- ', _0/,0fr