Teknium
8fb678ce5d
- image_routing: capability probes -> _VISION_PROBES ordered table; magic-byte sniffing -> _MAGIC signature table; bool tokens -> dict; shared code-span / file-existence helpers for extract_image_refs. Fixture-corpus diff base vs new identical. - inline_tool_executors: 11 hand-written executor bodies -> _tool/_callback_tool arg-spec factories (kwargs mapping unchanged; corpus of calls identical). - file_safety: unify _is_under/_under_any, _resolve_target/_resolve_each, _mirror_info; deny-list tables compacted; drop retired get_cross_profile_warning stub (+ its tests). Every rule preserved (corpus). - image_gen_provider: drop _images_cache_dir (test-only wrapper), compact. - estop/delegation_context/kanban_stop/image_gen_registry: defensive-layer collapse and docstring compaction. Tool schema dump byte-identical to base.
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
"""Image generation provider registry.
|
|
|
|
Populated by plugins at import-time via ``PluginContext.register_image_gen_provider()``;
|
|
the ``image_generate`` tool dispatches to :func:`get_active_provider`. Selection is
|
|
``image_gen.provider`` in config.yaml; when unset: the single *available* provider,
|
|
else ``fal`` if registered and available (legacy default), else ``None`` (the tool
|
|
points the user at ``hermes tools``).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Optional
|
|
|
|
from agent.image_gen_provider import ImageGenProvider
|
|
from agent.provider_registry import ProviderRegistry, configured_provider_name, is_available_safe
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
_registry: ProviderRegistry[ImageGenProvider] = ProviderRegistry(
|
|
label="Image gen", provider_cls=ImageGenProvider, logger=logger,
|
|
)
|
|
_registry.export(globals())
|
|
|
|
|
|
def get_active_provider() -> Optional[ImageGenProvider]:
|
|
"""Resolve the currently-active provider. Availability semantics (mirrors
|
|
:mod:`agent.web_search_registry`): an explicitly configured provider is returned
|
|
even if ``is_available()`` is False, so the dispatcher surfaces a precise
|
|
"X_API_KEY is not set" error instead of silently switching backends; only the
|
|
unconfigured fallback path is filtered by availability."""
|
|
configured = configured_provider_name("image_gen", logger)
|
|
snapshot = _registry.merged()
|
|
if configured:
|
|
if snapshot.get(configured) is not None:
|
|
return snapshot[configured]
|
|
logger.debug("image_gen.provider='%s' configured but not registered; falling back", configured)
|
|
|
|
def _available(p: ImageGenProvider) -> bool:
|
|
return is_available_safe(p, logger, "image_gen provider %s.is_available() raised %s")
|
|
|
|
available = [p for p in snapshot.values() if _available(p)]
|
|
if len(available) == 1:
|
|
return available[0]
|
|
fal = snapshot.get("fal")
|
|
return fal if fal is not None and _available(fal) else None
|