Upload models_dev_stub.py with huggingface_hub

models_dev_stub.py

@@ -0,0 +1,65 @@
+"""Stub: only curated models from config."""
+import logging, json
+from dataclasses import dataclass
+from typing import Any, Dict, List, Optional, Tuple
+
+logger = logging.getLogger(__name__)
+
+@dataclass
+class ProviderInfo:
+    name: str = ""
+    base_url: str = ""
+    api_doc: str = ""
+    env_vars: Tuple[str, ...] = ()
+    extra_env_vars: Tuple[str, ...] = ()
+    is_aggregator: bool = False
+
+CURATED_MODELS = [
+    {"id": "meta-llama/llama-3.3-70b-instruct:free", "provider_id": "openrouter", "name": "Llama 3.3 70B Instruct"},
+    {"id": "deepseek/deepseek-v4-flash:free", "provider_id": "openrouter", "name": "Deepseek V4 Flash"},
+    {"id": "z-ai/glm-4.5-air:free", "provider_id": "openrouter", "name": "Glm 4.5 Air"},
+    {"id": "minimax/minimax-m2.5:free", "provider_id": "openrouter", "name": "Minimax M2.5"},
+    {"id": "google/gemma-4-31b-it:free", "provider_id": "openrouter", "name": "Gemma 4 31B It"},
+    {"id": "google/gemma-4-26b-a4b-it:free", "provider_id": "openrouter", "name": "Gemma 4 26B A4B It"},
+    {"id": "qwen/qwen3-coder:free", "provider_id": "openrouter", "name": "Qwen3 Coder"},
+    {"id": "qwen/qwen3-next-80b-a3b-instruct:free", "provider_id": "openrouter", "name": "Qwen3 Next 80B A3B Instruct"},
+    {"id": "nousresearch/hermes-3-llama-3.1-405b:free", "provider_id": "openrouter", "name": "Hermes 3 Llama 3.1 405B"},
+    {"id": "openai/gpt-oss-120b:free", "provider_id": "openrouter", "name": "Gpt Oss 120B"},
+    {"id": "openai/gpt-oss-20b:free", "provider_id": "openrouter", "name": "Gpt Oss 20B"},
+    {"id": "baidu/cobuddy:free", "provider_id": "openrouter", "name": "Cobuddy"},
+    {"id": "arcee-ai/trinity-large-thinking:free", "provider_id": "openrouter", "name": "Trinity Large Thinking"},
+    {"id": "liquid/lfm-2.5-1.2b-thinking:free", "provider_id": "openrouter", "name": "Lfm 2.5 1.2B Thinking"},
+    {"id": "liquid/lfm-2.5-1.2b-instruct:free", "provider_id": "openrouter", "name": "Lfm 2.5 1.2B Instruct"},
+    {"id": "meta-llama/llama-3.2-3b-instruct:free", "provider_id": "openrouter", "name": "Llama 3.2 3B Instruct"},
+    {"id": "cognitivecomputations/dolphin-mistral-24b-venice-edition:free", "provider_id": "openrouter", "name": "Dolphin Mistral 24B Venice Edition"},
+    {"id": "nvidia/nemotron-3-super-120b-a12b:free", "provider_id": "openrouter", "name": "Nemotron 3 Super 120B A12B"},
+    {"id": "nvidia/nemotron-3-nano-30b-a3b:free", "provider_id": "openrouter", "name": "Nemotron 3 Nano 30B A3B"},
+    {"id": "nvidia/nemotron-3-nano-omni-30b-a3b-reasoning:free", "provider_id": "openrouter", "name": "Nemotron 3 Nano Omni 30B A3B Reasoning"},
+    {"id": "nvidia/nemotron-nano-9b-v2:free", "provider_id": "openrouter", "name": "Nemotron Nano 9B V2"},
+    {"id": "nvidia/nemotron-nano-12b-v2-vl:free", "provider_id": "openrouter", "name": "Nemotron Nano 12B V2 Vl"},
+    {"id": "nvidia/llama-nemotron-embed-vl-1b-v2:free", "provider_id": "openrouter", "name": "Llama Nemotron Embed Vl 1B V2"},
+    {"id": "poolside/laguna-m.1:free", "provider_id": "openrouter", "name": "Laguna M.1"},
+    {"id": "poolside/laguna-xs.2:free", "provider_id": "openrouter", "name": "Laguna Xs.2"},
+    {"id": "openrouter/auto:free", "provider_id": "openrouter", "name": "Auto"},
+]
+
+def get_provider_info(provider_id: str) -> Optional[ProviderInfo]:
+    if provider_id == "openrouter":
+        return ProviderInfo(name="OpenRouter", base_url="https://openrouter.ai/api/v1", is_aggregator=True)
+    return None
+
+def get_all_models() -> list:
+    return CURATED_MODELS
+
+def search_models(query: str = "", provider: str = "") -> list:
+    if provider and provider != "openrouter": return []
+    if query: return [m for m in CURATED_MODELS if query.lower() in m["id"].lower()]
+    return CURATED_MODELS
+
+def get_model(model_id: str) -> Optional[dict]:
+    for m in CURATED_MODELS:
+        if m["id"] == model_id: return m
+    return None
+
+def get_providers() -> dict:
+    return {"openrouter": {"name": "OpenRouter", "base_url": "https://openrouter.ai/api/v1", "is_aggregator": True}}