Update models.py

models.py

@@ -1,17 +1,27 @@
 # models.py
+# ------------------------------------------------------------------
+# Central registry of all AI models AnyCoder can route to.
+# ------------------------------------------------------------------
+from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import List, Optional
 
-@dataclass
+
+@dataclass(slots=True, frozen=True)
 class ModelInfo:
     """
-    Represents metadata for an inference model.
+    Metadata for a single model entry.
 
-    Attributes:
-        name: Human-readable name of the model.
-        id: Unique model identifier (HF/externally routed).
-        description: Short description of the model's capabilities.
-        default_provider: Preferred inference provider ("auto", "groq", "openai", "gemini", "fireworks").
+    Attributes
+    ----------
+    name            : Human‑readable label shown in the UI.
+    id              : Fully‑qualified model path, e.g. "openai/gpt‑4".
+    description     : Short capability blurb.
+    default_provider: Which provider to send inference requests to if the
+                      caller does **not** override it.  Supported values:
+                      "auto" | "groq" | "openai" | "gemini" | "fireworks".
+                      The special value "auto" lets HF Inference decide.
     """
     name: str
     id: str
@@ -19,86 +29,102 @@ class ModelInfo:
     default_provider: str = "auto"
 
 
-# Registry of supported models
+# ------------------------------------------------------------------
+# Editable list of models exposed to both back‑end & front‑end
+# ------------------------------------------------------------------
 AVAILABLE_MODELS: List[ModelInfo] = [
+    # High‑capacity HF models
+    ModelInfo(
+        name="Qwen/Qwen3‑32B",
+        id="Qwen/Qwen3-32B",
+        description="Qwen3‑32B model for high‑capacity code and text generation",
+    ),
     ModelInfo(
-        name="Moonshot Kimi-K2",
+        name="Qwen3‑235B‑A22B",
+        id="Qwen/Qwen3-235B-A22B",
+        description="Qwen3‑235B‑A22B model for code generation and general tasks",
+    ),
+    # Moonshot (Groq hardware by default)
+    ModelInfo(
+        name="Moonshot Kimi‑K2",
         id="moonshotai/Kimi-K2-Instruct",
-        description="Moonshot AI Kimi-K2-Instruct model for code generation and general tasks",
-        default_provider="groq"
+        description="Moonshot AI Kimi‑K2‑Instruct (code, chat)",
+        default_provider="groq",
     ),
+    # DeepSeek
     ModelInfo(
-        name="DeepSeek V3",
+        name="DeepSeek V3",
         id="deepseek-ai/DeepSeek-V3-0324",
-        description="DeepSeek V3 model for code generation",
+        description="DeepSeek V3 model for code generation",
     ),
     ModelInfo(
-        name="DeepSeek R1",
+        name="DeepSeek R1",
         id="deepseek-ai/DeepSeek-R1-0528",
-        description="DeepSeek R1 model for code generation",
+        description="DeepSeek R1 model for code generation",
     ),
+    # Multimodal Chinese / English models
     ModelInfo(
-        name="ERNIE-4.5-VL",
+        name="ERNIE‑4.5‑VL",
         id="baidu/ERNIE-4.5-VL-424B-A47B-Base-PT",
-        description="ERNIE-4.5-VL model for multimodal code generation with image support",
+        description="ERNIE‑4.5‑VL multimodal model (image + text)",
     ),
     ModelInfo(
-        name="MiniMax M1",
-        id="MiniMaxAI/MiniMax-M1-80k",
-        description="MiniMax M1 model for code generation and general tasks",
-    ),
-    ModelInfo(
-        name="Qwen3-235B-A22B",
-        id="Qwen/Qwen3-235B-A22B",
-        description="Qwen3-235B-A22B model for code generation and general tasks",
+        name="GLM‑4.1V‑9B‑Thinking",
+        id="THUDM/GLM-4.1V-9B-Thinking",
+        description="GLM‑4.1V‑9B multimodal reasoning model",
     ),
+    # Lightweight general‑purpose models
     ModelInfo(
-        name="SmolLM3-3B",
+        name="SmolLM3‑3B",
         id="HuggingFaceTB/SmolLM3-3B",
-        description="SmolLM3-3B model for code generation and general tasks",
+        description="SmolLM3‑3B fast, low‑latency model",
     ),
     ModelInfo(
-        name="GLM-4.1V-9B-Thinking",
-        id="THUDM/GLM-4.1V-9B-Thinking",
-        description="GLM-4.1V-9B-Thinking model for multimodal code generation with image support",
+        name="MiniMax M1",
+        id="MiniMaxAI/MiniMax-M1-80k",
+        description="MiniMax M1 80k‑context general model",
     ),
+    # External providers via HF Inference Providers
     ModelInfo(
-        name="OpenAI GPT-4",
+        name="OpenAI GPT‑4",
         id="openai/gpt-4",
-        description="OpenAI GPT-4 model via Inference Providers",
-        default_provider="openai"
+        description="OpenAI GPT‑4 accessed through HF Inference Providers",
+        default_provider="openai",
     ),
     ModelInfo(
-        name="Gemini Pro",
+        name="Gemini Pro",
         id="gemini/pro",
-        description="Google Gemini Pro model via Inference Providers",
-        default_provider="gemini"
+        description="Google Gemini Pro via HF Inference Providers",
+        default_provider="gemini",
     ),
     ModelInfo(
-        name="Fireworks AI",
+        name="Fireworks V1",
         id="fireworks-ai/fireworks-v1",
-        description="Fireworks AI model via Inference Providers",
-        default_provider="fireworks"
-    ),
-    ModelInfo(
-        name="Qwen3-32B",
-        id="Qwen/Qwen3-32B",
-        description="Qwen3-32B model for high-capacity code and text generation",
+        description="Fireworks AI flagship model",
+        default_provider="fireworks",
     ),
 ]
 
 
+# ------------------------------------------------------------------
+# Helper look‑ups
+# ------------------------------------------------------------------
 def find_model(identifier: str) -> Optional[ModelInfo]:
     """
-    Lookup a model by its human name or identifier.
+    Retrieve a `ModelInfo` either by `.id` or by case‑insensitive `.name`.
+
+    Parameters
+    ----------
+    identifier : str
+        - `"openai/gpt-4"` – exact id
+        - `"OpenAI GPT-4"`  – human label
 
-    Args:
-        identifier: ModelInfo.name (case-insensitive) or ModelInfo.id
-    Returns:
-        The matching ModelInfo or None if not found.
+    Returns
+    -------
+    Optional[ModelInfo]
     """
     key = identifier.lower()
     for m in AVAILABLE_MODELS:
         if m.id == identifier or m.name.lower() == key:
             return m
-    return None
+    return None