"""DeepInfra provider profile.

DeepInfra is an OpenAI-compatible inference gateway that hosts 100+ open
models (Step, GLM, Kimi, DeepSeek, MiniMax, Nemotron, Mistral, Qwen, …) as
well as image-gen / TTS / STT / embedding endpoints. The chat surface is
wired in through this profile; non-chat surfaces are wired in through
their respective plugin subsystems (``plugins/image_gen/deepinfra`` and
the TTS/STT dispatchers in ``tools/``).
"""

from providers import register_provider
from providers.base import ProviderProfile


class _DeepInfraProfile(ProviderProfile):
    """DeepInfra profile with live vision-default discovery.

    Owns its own vision default so shared vision resolution in
    ``agent/auxiliary_client.py`` stays provider-agnostic (a
    ``default_vision_model()`` hook call instead of an ``if provider ==
    "deepinfra"`` branch reaching into the catalog helpers).
    """

    def default_vision_model(self):  # type: ignore[override]
        """First vision-capable *chat* model from the live catalog, or None.

        Key-gated so a box without ``DEEPINFRA_API_KEY`` never pays the
        catalog round-trip. Requires the ``chat`` surface tag (not just the
        ``vision`` capability) so an image-gen/edit model that merely carries
        a ``vision`` tag can't be picked as a chat-completions vision backend.
        """
        import os

        if not (os.environ.get("DEEPINFRA_API_KEY") or "").strip():
            return None
        try:
            from hermes_cli.models import _fetch_deepinfra_models_by_tag
            items = _fetch_deepinfra_models_by_tag("chat")
        except Exception:
            return None
        for item in items or []:
            metadata = item.get("metadata") or {}
            tags = metadata.get("tags") if isinstance(metadata, dict) else None
            if isinstance(tags, list) and "vision" in tags:
                model_id = item.get("id")
                if model_id:
                    return model_id
        return None


deepinfra = _DeepInfraProfile(
    name="deepinfra",
    aliases=("deep-infra", "deepinfra-ai"),
    display_name="DeepInfra",
    description="DeepInfra — 100+ open models, pay-per-use",
    signup_url="https://deepinfra.com/dash/api_keys",
    env_vars=("DEEPINFRA_API_KEY", "DEEPINFRA_BASE_URL"),
    base_url="https://api.deepinfra.com/v1/openai",
    auth_type="api_key",
    # Default output cap when the user hasn't set ``agent.max_tokens``.
    # Without this the profile inherits ``None`` and the transport sends no
    # ``max_tokens`` (chat_completions.py: the ``elif profile_max`` branch
    # is skipped because ``None`` is falsy), so DeepInfra applies its small
    # server-side default (~8-16K). Tool-heavy runs (e.g. cron jobs doing
    # many web_search calls before composing) then exhaust that budget on
    # tool results, hit ``finish_reason='length'``, and fail after the
    # 3 continuation retries in conversation_loop.py.
    #
    # 128K gives ample room for tool-result processing + final output.
    # Safe across the whole catalog despite per-model limits ranging from
    # 4K (Gryphe/MythoMax) to 1M (DeepSeek-V4): DeepInfra silently CLAMPS
    # max_tokens to each model's own limit rather than rejecting it
    # (verified live). Users who set ``agent.max_tokens`` still win — their
    # value takes priority over this profile default in the transport.
    default_max_tokens=131072,
    # Auxiliary model — cheap/fast chat model the same provider uses for
    # side tasks (context compression, session search, web extract,
    # vision). This is the *only* hardcoded DeepInfra model in the
    # integration: aux resolution is synchronous (no time for a catalog
    # round-trip on every agent turn), so we need one explicit choice.
    # Every other surface (chat picker, image-gen, tts, stt, pricing)
    # discovers models live from
    # ``api.deepinfra.com/v1/openai/models?filter=true&sort_by=hermes``.
    default_aux_model="deepseek-ai/DeepSeek-V4-Flash",
    # ``fallback_models`` deliberately empty — the live catalog at
    # ``hermes_cli/models.py::_fetch_deepinfra_models`` is the source of
    # truth. When the live fetch fails (network/DNS), the picker shows
    # no options, which is preferable to silently routing the user to a
    # model that may have been retired upstream.
    fallback_models=(),
)

register_provider(deepinfra)