Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

plugins/model-providers/nebius-token-factory/__init__.py

@@ -0,0 +1,104 @@
+"""Nebius Token Factory provider profile."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from providers import register_provider
+from providers.base import ProviderProfile
+
+
+def _flat_model_name(model: str | None) -> str:
+    """Return a lowercase model id, tolerating vendor-prefixed IDs."""
+    return (model or "").strip().rsplit("/", 1)[-1].lower()
+
+
+def _model_supports_reasoning_effort(model: str | None) -> bool:
+    """Conservative allowlist for Nebius models that expose reasoning effort."""
+    model_name = _flat_model_name(model)
+    if not model_name:
+        return False
+    return any(
+        marker in model_name
+        for marker in (
+            "deepseek-r1",
+            "deepseek-v4",
+            "deepseek-reasoner",
+            "gpt-oss",
+            "glm-5",
+            "kimi-k2",
+            "minimax-m2",
+            "qwen3",
+        )
+    )
+
+
+class NebiusTokenFactoryProfile(ProviderProfile):
+    """Nebius Token Factory - top-level reasoning_effort."""
+
+    def build_api_kwargs_extras(
+        self,
+        *,
+        reasoning_config: dict | None = None,
+        model: str | None = None,
+        supports_reasoning: bool = False,
+        **context: Any,
+    ) -> tuple[dict[str, Any], dict[str, Any]]:
+        if not supports_reasoning and not _model_supports_reasoning_effort(model):
+            return {}, {}
+
+        if isinstance(reasoning_config, dict):
+            enabled = reasoning_config.get("enabled", True)
+            raw_effort = reasoning_config.get("effort", "medium")
+        else:
+            enabled = True
+            raw_effort = "medium"
+
+        effort = str(raw_effort or "medium").strip().lower()
+        if enabled is False or effort in {"none", "off", "disabled"}:
+            return {}, {}
+        # Canonical clamp (nearest weaker supported level, never escalate,
+        # monotonic) — the hand-rolled map this replaces inverted the ladder:
+        # ultra fell through to medium while xhigh mapped to high.
+        from agent.reasoning_effort import NEBIUS_EFFORTS, clamp_effort
+
+        effort = clamp_effort(effort, NEBIUS_EFFORTS) or "medium"
+
+        return {}, {"reasoning_effort": effort}
+
+
+nebius_token_factory = NebiusTokenFactoryProfile(
+    name="nebius-token-factory",
+    aliases=(
+        "nebius",
+        "nebius-tokenfactory",
+        "nebius-tf",
+        "token-factory",
+        "tokenfactory",
+    ),
+    display_name="Nebius Token Factory",
+    description="Nebius Token Factory — OpenAI-compatible inference",
+    signup_url="https://tokenfactory.nebius.com/",
+    env_vars=(
+        "NEBIUS_API_KEY",
+        "NEBIUS_TOKEN_FACTORY_API_KEY",
+        "NEBIUS_BASE_URL",
+    ),
+    base_url="https://api.tokenfactory.nebius.com/v1",
+    models_url="https://api.tokenfactory.nebius.com/v1/models?verbose=true",
+    auth_type="api_key",
+    default_aux_model="nvidia/NVIDIA-Nemotron-3-Nano-30B-A3B",
+    fallback_models=(
+        "Qwen/Qwen3.5-397B-A17B-fast",
+        "deepseek-ai/DeepSeek-V4-Pro",
+        "zai-org/GLM-5.1",
+        "moonshotai/Kimi-K2.5-fast",
+        "MiniMaxAI/MiniMax-M2.5-fast",
+        "deepseek-ai/DeepSeek-V3.2-fast",
+        "NousResearch/Hermes-4-70B",
+        "openai/gpt-oss-120b-fast",
+        "meta-llama/Llama-3.3-70B-Instruct",
+    ),
+)
+
+register_provider(nebius_token_factory)