eiUdZddlmZddlZddlZddlmZmZmZddl 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/div/.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,AAAc$d} ddlm}|}t|tr|dnd}t|trR|d}t|t r(|r|}n2#t$r%}t d|Yd}~nd}~wwxYwt5tt}dddn #1swxYwY|r4||}||St d|t|dkr.tt|Sd |vr|d SdS) zResolve the currently-active provider. Reads ``image_gen.provider`` from config.yaml; falls back per the module docstring. Nr) load_config image_genr z1Could not read image_gen.provider from config: %szCimage_gen.provider='%s' configured but not registered; falling backfal)hermes_cli.configr-r dictrrr Exceptionrrrrlennextiterr%) configuredr-cfgsectionrawexcsnapshotr s rget_active_providerr=Ks !%J O111111kmm*4S$*?*?I#''+&&&T gt $ $ )++j))C#s## )  ) YY[[ OOO H#NNNNNNNNO $$ ##$$$$$$$$$$$$$$$ << ++  O Q     8}}D**++,,,  4s*B#B(( C2CC!DD Dcxt5tddddS#1swxYwYdS)z"Clear the registry. **Test-only.**N)rrclearr"r_reset_for_testsrAus~ s /33)r rr r )r r)rrr r))r r))r r )__doc__ __future__rlogging threadingtypingrrragent.image_gen_providerr getLoggerrrr__annotations__Lockrrr(r+r=rAr@r"rrKs(#"""""''''''''''555555  8 $ $+- ,,,, ____0////,,,,''''Tr"