c762564d40
- Fix 71 invalid-syntax files (class-body newline-broken assignments) - Add from/None chain to 307 B904 raise-without-from sites - Add B008 ignore to ruff.toml (already in pyproject.toml) - Noqa F401 on __init__.py re-exports (137 sites) - Noqa E402 on deferred imports (63 sites) - Bulk-add stdlib/FastAPI/project imports for F821 (127 sites) - Replace ×→x, –→-, …→... in docstrings (4093 chars) - Manual refactor of 5 SIM103/SIM116 patterns Tests: 791 passed (66 deselected due to pre-existing Redis issues in test_rag.py) Co-authored-by: opencode <opencode@rugmunch.io>
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
"""Model access map - who has tokens for what.
|
|
|
|
Updated whenever API keys change. Run with `python -m infra.model_map` from
|
|
the repo root, or `bash infra/model_map.sh` for a shell-only view.
|
|
|
|
ACTIVE providers (real keys):
|
|
- openrouter → DeepSeek, MiniMax, Gemini, Mistral, Llama (universal)
|
|
- deepseek → direct DeepSeek API
|
|
- groq → fast inference (llama-3.3-70b, llama-3.1-8b)
|
|
|
|
PLACEHOLDER (no real key, skipped):
|
|
- gemini, zai, minimax, mistral-direct, kimi
|
|
|
|
Per v4.0 master stack: all hosted APIs are routed through LiteLLM proxy at
|
|
http://rmi-litellm:4000 with the sovereign OpenAI-compatible interface.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class Provider:
|
|
name: str
|
|
active: bool
|
|
key_len: int
|
|
key_prefix: str
|
|
models: list[str]
|
|
|
|
|
|
def _read_env(name: str) -> str:
|
|
return os.environ.get(name, "")
|
|
|
|
|
|
def _check(name: str, env_var: str, models: list[str], min_len: int = 30) -> Provider:
|
|
key = _read_env(env_var)
|
|
active = len(key) >= min_len
|
|
return Provider(
|
|
name=name,
|
|
active=active,
|
|
key_len=len(key),
|
|
key_prefix=key[:8] if key else "",
|
|
models=models,
|
|
)
|
|
|
|
|
|
def get_model_map() -> list[Provider]:
|
|
"""Read all API keys from env. Returns provider status."""
|
|
return [
|
|
_check("openrouter", "OPENROUTER_API_KEY", [
|
|
"deepseek/deepseek-chat", "MiniMax/MiniMax-M3",
|
|
"google/gemini-2.5-flash", "mistralai/mistral-large-latest",
|
|
"meta-llama/llama-3-70b-instruct",
|
|
]),
|
|
_check("deepseek", "DEEPSEEK_API_KEY", ["deepseek-chat"], min_len=30),
|
|
_check("groq", "GROQ_API_KEY", [
|
|
"llama-3.3-70b-versatile", "llama-3.1-8b-instant",
|
|
]),
|
|
_check("gemini", "GEMINI_API_KEY", ["gemini-2.5-flash"], min_len=30),
|
|
_check("minimax", "MiniMax_API_KEY", ["MiniMax-M3"], min_len=30),
|
|
_check("mistral", "MISTRAL_API_KEY", ["mistral-large-latest"], min_len=30),
|
|
_check("kimi", "KIMI_API_KEY", ["kimi-k2-0711-preview"], min_len=30),
|
|
_check("ollama-cloud", "OLLAMA_API_KEY", ["llama3.2", "qwen2.5"], min_len=30),
|
|
]
|
|
|
|
|
|
def print_map() -> None:
|
|
"""Print human-readable model access map."""
|
|
print(f"{'PROVIDER':<14} {'STATUS':<10} {'LEN':<5} MODELS")
|
|
print("─" * 70)
|
|
for p in get_model_map():
|
|
status = "ACTIVE" if p.active else "PLACEHLD"
|
|
models = ", ".join(p.models[:3])
|
|
if len(p.models) > 3:
|
|
models += f" +{len(p.models) - 3}"
|
|
print(f"{p.name:<14} {status:<10} {p.key_len:<5} {models}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print_map()
|