sj!UdZddlmZddlZddlZddlmZmZmZddl m Z ddl m Z ej eZiZded<iZd ed <ejZdd d dZdd d!dZdd d"dZdd d"dZdd d#dZd$dZd%dZdS)&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)ImageGenProvider)hermes_home_keyzDict[str, ImageGenProvider] _providersz&Dict[str, Dict[str, ImageGenProvider]]_scoped_providers)scopeproviderrr Optional[str]returnNonect|ts$tdt|j|j}t|t r|std|}t5|tnt |i}| |}|||<dddn #1swxYwY|0td|t|jdStd|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_lockr r setdefaultgetloggerdebug)r r raw_namertargetexistings >* - -   }H h $ $PHNN,<,<PNOOO >>  D   $}2C2N2NuVX2Y2Y::d##t                 EtTRZ^^Mdeeeee >d8nnF]^^^^^s?CC#&C#List[ImageGenProvider]cFt5tt}|t|p t it|}dddn #1swxYwYt|dS)z0Return all registered providers, sorted by name.Nc|jSN)r)ps r!z list_providers..Fsqv)key) rdictr updater rrlistvaluessorted)r mergeditemss r!list_providersr2@s &&j!! '++E,F_5F5FKKLLLV]]__%%&&&&&&&&&&&&&&& %-- . . ..sA2BB  B rrOptional[ImageGenProvider]cFt|tsdSt5|}t|p t i|pt|cdddS#1swxYwYdS)z5Return the provider registered under *name*, or None.N)rrrrr rrr )rr r*s r! get_providerr5Is dC t eejjll $$U%?o.?.?DDHHMMdQ[Q_Q_`cQdQdeeeeeeeeeeeeeeeeeesA*BBBct5|tnt|i}||cdddS#1swxYwYdSr&)rr r rr)rr rs r!snapshot_registrationr7Rs (($}2C2G2Gr2R2Rzz$**,,''((((((((((((((((((sA A  A$'A$currentpreviousboolcx|}t5|tnt|i}|||ur ddddS|||dn|||<||st|ddddn #1swxYwYdS)zERestore a plugin registration only when *current* is still installed.NFT)rrr r rrpop)rr8r9r r*rs r!restore_registrationr=Zs **,,C  / /$}2C2N2NuVX2Y2Y ::c??' ) ) / / / / / / / /   JJsD ! ! ! !"F3K  V   ! !% . . . / / / / / / / / / / / / / / / 4s=B/&=B//B36B3c. d} ddlm}|}t|tr|dnd}t|trR|d}t|t r(|r|}n2#t$r%}t d|Yd}~nd}~wwxYw|r2 ddl m }| |krd}n#t$rYnwxYwt5tt}|t t#idddn #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: %s)NOUS_MANAGED_PROVIDERfalr'rrr:c 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_safese (()) )    LLI16SV W W W55555 s # A!AAzCimage_gen.provider='%s' configured but not registered; falling backc*g|]}| |SrI).0r'rGs r! z'get_active_provider..s*GGGq1C1CA1F1FGGGGr))r'rrr:)hermes_cli.configr?rr+rrrrErrtools.tool_backend_helpersrAlowerrr r,r rr.len) configuredr?cfgsectionrawrFrAsnapshotr availablerBrGs @r!get_active_providerrWps"!%J O::::::""$$*4S$*?*?I#''+&&&T gt $ $ )++j))C#s## )  ) YY[[ OOO H#NNNNNNNNO   H H H H H H!!%:::"     D  FF ##)--o.?.?DDEEEFFFFFFFFFFFFFFF << ++  O Q    HGGGHOO--GGGI 9~~| ,,u  C --c22 4s=B#B)) C3CC C?? D  D AE11E58E5ct5ttddddS#1swxYwYdS)z"Clear the registry. **Test-only.**N)rr clearr rIr)r!_reset_for_testsrZs ""!!!""""""""""""""""""s3AA A )r rr r rr)r r rr#)rrr r rr3) rrr8rr9r3r r rr:)rr3)rr)__doc__ __future__rlogging threadingtypingrrragent.image_gen_providerrhermes_constantsr getLoggerrrr __annotations__r Lockrr"r2r5r7r=rWrZrIr)r!res(#"""""''''''''''555555,,,,,,  8 $ $+- ,,,,<>>>>> MQ______4.2//////7;eeeeee*.(((((( ,MMMM`""""""r)