yj8UdZddlmZddlZddlZddlmZmZmZddl m Z ej e Z iZded<ejZdd ZddZddZddZddZdS)u Image Generation Provider Registry ================================== Central map of registered providers. Populated by plugins at import-time via ``PluginContext.register_image_gen_provider()``; consumed by the ``image_generate`` tool to dispatch each call to the active backend. Active selection ---------------- The active provider is chosen by ``image_gen.provider`` in ``config.yaml``. If unset, :func:`get_active_provider` applies fallback logic: 1. If exactly one provider is registered, use it. 2. Otherwise if a provider named ``fal`` is registered, use it (legacy default — matches pre-plugin behavior). 3. Otherwise return ``None`` (the tool surfaces a helpful error pointing the user at ``hermes tools``). ) annotationsN)DictListOptional)ImageGenProviderzDict[str, ImageGenProvider] _providersproviderrreturnNonec<t|ts$tdt|j|j}t|t r|stdt5t |}|t|<dddn #1swxYwY|0t d|t|jdSt d|t|jdS)uRegister an image 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 an ImageGenProvider instance, got z3Image gen provider .name must be a non-empty stringNz.Image gen provider '%s' re-registered (was %r)z'Registered image gen provider '%s' (%s)) isinstancer TypeErrortype__name__namestrstrip ValueError_lockrgetloggerdebug)r rexistings >/home/ragecks/.hermes/hermes-agent/agent/image_gen_registry.pyregister_providerr$sJ h 0 1 1  ->>* - -    =D dC P PNOOO $$>>$''# 4$$$$$$$$$$$$$$$ EtTRZ^^Mdeeeee >d8nnF]^^^^^s%B11B58B5List[ImageGenProvider]ct5tt}dddn #1swxYwYt |dS)z0Return all registered providers, sorted by name.Nc|jS)Nr)ps rz list_providers..@sqv)key)rlistrvaluessorted)itemss rlist_providersr(<s **Z&&(())*************** %-- . . ..s ';??rrOptional[ImageGenProvider]ct|tsdSt5t|cdddS#1swxYwYdS)z5Return the provider registered under *name*, or None.N)r rrrrrrs r get_providerr+Cs dC t ,,~~djjll++,,,,,,,,,,,,,,,,,,s,AAAcR d} ddlm}|}t|tr|dnd}t|trR|d}t|t r(|r|}n2#t$r%}t d|Yd}~nd}~wwxYwt5tt}dddn #1swxYwYdd  |r4||}||St d | fd | D}t|dkr|dS|d} |  | r| SdS)uResolve the currently-active provider. Reads ``image_gen.provider`` from config.yaml; falls back per the module docstring. **Availability semantics** (mirrors :mod:`agent.web_search_registry`): - When ``image_gen.provider`` is explicitly set, the configured provider is returned even if :meth:`ImageGenProvider.is_available` reports False — the dispatcher surfaces a precise "X_API_KEY is not set" error rather than silently switching backends. - When ``image_gen.provider`` is unset, the fallback path (single- provider shortcut and the FAL legacy preference) is filtered by ``is_available()`` so we don't pick a provider the user has no credentials for. Nr)load_config_readonly image_genr z1Could not read image_gen.provider from config: %sr rr boolc t|S#t$r,}td|j|Yd}~dSd}~wwxYw)zDWrap ``is_available()`` so a buggy provider doesn't kill resolution.z.image_gen provider %s.is_available() raised %sNF)r/ is_available Exceptionrrr)r excs r_is_available_safez/get_active_provider.._is_available_safelse (()) )    LLI16SV W W W55555 s # A!AAzCimage_gen.provider='%s' configured but not registered; falling backc*g|]}| |Sr6).0r r4s r z'get_active_provider..s*GGGq1C1CA1F1FGGGGr"fal)r rr r/)hermes_cli.configr-r dictrrrr2rrrrr%len) configuredr-cfgsectionrawr3snapshotr availabler:r4s @rget_active_providerrDKs!"!%J O::::::""$$*4S$*?*?I#''+&&&T gt $ $ )++j))C#s## )  ) YY[[ OOO H#NNNNNNNNO $$ ##$$$$$$$$$$$$$$$ << ++  O Q    HGGGHOO--GGGI 9~~| ,,u  C --c22 4s*B#B)) C3CC"DD Dcxt5tddddS#1swxYwYdS)z"Clear the registry. **Test-only.**N)rrclearr6r"r_reset_for_testsrGs~ s /33)r rr r )r r)rrr r))r r))r r )__doc__ __future__rlogging threadingtypingrrragent.image_gen_providerr getLoggerrrr__annotations__Lockrrr(r+rDrGr6r"rrQs(#"""""''''''''''555555  8 $ $+- ,,,, ____0////,,,,@@@@Fr"