setup models

harness/candidates.py

@@ -0,0 +1,33 @@
+"""Construção dos adaptadores a partir do manifesto de artefatos."""
+from __future__ import annotations
+
+import json
+from pathlib import Path
+
+from adapters import adapter_class
+
+ROOT = Path(__file__).resolve().parents[1]
+MANIFEST = ROOT / "env" / "artifacts-manifest.json"
+MODEL_REPO = {
+    "laya": "convaiinnovations/laya-multilingual",
+    "semif": "Qwen/Qwen3.5-4B",
+    "gliner": "fastino/GLiNER2.5-multi-Decide",
+    "rizzo_flow": "rizzoaiacademy/rizzo-flow",
+}
+
+
+def model_location(candidate: str, models_root: Path) -> tuple[Path, str]:
+    """Diretório local e revisão fixada do candidato, segundo o manifesto."""
+    manifest = json.loads(MANIFEST.read_text())
+    repo = MODEL_REPO[candidate]
+    entry = next(a for a in manifest["artifacts"]
+                 if a["candidate"] == candidate and a["kind"] == "hf_model" and a["origin"] == f"hf://{repo}")
+    rev = entry["revision"]
+    return models_root / candidate / repo.replace("/", "__") / rev, rev
+
+
+def build_adapter(candidate: str, models_root: str | Path, device: str = "auto",
+                  context_limit: int | None = None, **extra):
+    model_dir, revision = model_location(candidate, Path(models_root))
+    return adapter_class(candidate)(model_dir, revision, device=device,
+                                    context_limit=context_limit, **extra)