"""
Vertrauenswürdige Quellen + Kategorien für die automatische Modell-Entdeckung.
Portiert aus Mission Control v1 (sources.py). Rollen sind die EINE Quelle der
Wahrheit (vereinheitlicht): vision · coder · reasoning · agent · scout.
"""

# HF-Orgs, die zuverlässig aktuelle, hochwertige GGUF-Quants veröffentlichen.
TRUSTED_AUTHORS = ["unsloth", "bartowski", "ggml-org", "lmstudio-community"]

# Kategorien (Reihenfolge = Anzeige + Zuordnungs-Priorität). Ein Modell wird der
# ERSTEN Kategorie zugeordnet, deren Stichwort im Repo-Namen vorkommt; sonst „scout".
# Die `role` ist zugleich der Alias-Vorschlag und gehört zu ROLE_IDS.
CATEGORIES = [
    {"role": "vision", "title": "Bilder verstehen", "icon": "eye",
     "kw": ["-vl-", "-vl", "vision", "llava", "multimodal", "-mm-", "pixtral"]},
    {"role": "coder", "title": "Coden & Programmieren", "icon": "code",
     "kw": ["coder", "-code-", "code-", "codestral", "starcoder"]},
    {"role": "reasoning", "title": "Nachdenken & Logik", "icon": "pulse",
     "kw": ["-r1", "deepseek-r1", "reasoning", "qwq", "magistral", "-think", "thinking", "-o1"]},
    {"role": "agent", "title": "Agenten & Tool-Use", "icon": "layers",
     "kw": ["hermes", "-tool", "command-r", "watt", "-fc-", "function"]},
    {"role": "scout", "title": "Allrounder & Chat", "icon": "compass",
     "kw": []},  # Fallback: instruct/chat-Modelle
]

# Repo-Namensteile, die bei der Entdeckung übersprungen werden (Roh-/Spezialformate).
SKIP_TOKENS = ["-base", "-bnb-", "-gptq", "-awq", "-fp8", "draft", "tokenizer"]