Teknium
72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
"""Nebius Token Factory provider profile."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from agent.reasoning_effort import NEBIUS_EFFORTS, clamp_effort
|
|
from providers import register_provider
|
|
from providers.base import ProviderProfile
|
|
|
|
# Conservative allowlist of model families that expose reasoning effort.
|
|
_REASONING_MARKERS = (
|
|
"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]]:
|
|
model_name = (model or "").strip().rsplit("/", 1)[-1].lower()
|
|
if not supports_reasoning and not any(marker in model_name for marker in _REASONING_MARKERS):
|
|
return {}, {}
|
|
rc = reasoning_config if isinstance(reasoning_config, dict) else {}
|
|
# Unset/blank effort defaults to medium (reasoning ON).
|
|
effort = str(rc.get("effort", "medium") or "medium").strip().lower()
|
|
if rc.get("enabled", True) is False or effort in {"none", "off", "disabled"}:
|
|
return {}, {}
|
|
# Canonical clamp: nearest weaker supported level, never escalate.
|
|
return {}, {"reasoning_effort": clamp_effort(effort, NEBIUS_EFFORTS) or "medium"}
|
|
|
|
|
|
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)
|