`gj&UdZddlmZddlZddlZddlmZmZmZddl m Z eje Z iZded<ej Zd dZdd Zdd Zdd Zdd Zy)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] _providersct|ts!tdt|j|j }t|t r|js tdt5tj|}|t|<ddd+tjd|t|jytjd|t|jy#1swYaxYw)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)providerrexistings K/root/.hermes/venv/lib/python3.12/site-packages/agent/image_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..@s 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'Cs9 dC  ,~~djjl+,,,s #AAcd} ddlm}|}t|tr|j dnd}t|trA|j d}t|t r |j r|j }t5tt}dddd d}|r+j |}||Stjd|jD cgc] } || s | } } t| d k(r| dS|j d } | || r| Sy#t$r }tjd|Yd}~d}~wwxYw#1swYxYwcc} w) 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 image_genrz1Could not read image_gen.provider from config: %sc t|jS#t$r+}tj d|j |Yd}~yd}~wwxYw)zDWrap ``is_available()`` so a buggy provider doesn't kill resolution.z.image_gen provider %s.is_available() raised %sNF)bool is_available Exceptionrrr)rexcs r_is_available_safez/get_active_provider.._is_available_safelsB () )  LLI166SV W s A!A  AzCimage_gen.provider='%s' configured but not registered; falling backfal)rrreturnr,)hermes_cli.configr)r dictrrrr.rrrrr"len) configuredr)cfgsectionrawr/snapshotr0rr availabler2s rget_active_providerr=KsK"!%J O1m*4S$*?#''+&T gt $++j)C#s#  YY[  $ #$<< +  O Q  %OO-Gq1CA1FGIG 9~| ,,u C -c2 K O H#NNO$$2Hs0BD+ E& E#4E#+ E4EEE cbt5tjdddy#1swYyxYw)z"Clear the registry. **Test-only.**N)rrclearrr_reset_for_testsrAs& s%.)rrr3None)r3zList[ImageGenProvider])rrr3Optional[ImageGenProvider])r3rC)r3rB)__doc__ __future__rlogging threadingtypingrrragent.image_gen_providerr getLoggerr rr__annotations__Lockrrr%r'r=rAr@rrrMsg(#''5   8 $+- ', _0/,@Fr