Initial vLLM agent strategy

scripts/load_seed.py

@@ -0,0 +1,63 @@
+from __future__ import annotations
+
+import argparse
+import csv
+
+from app.domain.priorities import ORIGIN_TO_PRIORITY, PRIORITY_LOW_CONFIDENCE
+from app.settings import load_settings
+from app.storage.db import connect
+from app.storage.repositories import Repository
+
+
+def flags_for_origin(origin: str) -> tuple[bool, bool, bool, bool]:
+    is_bounty = origin in {"manual_bounty", "bounty"}
+    is_promote = origin in {"manual_promote", "promote"}
+    downloaded = is_bounty or origin in {"manual_downloaded_by_others", "downloaded_by_others"}
+    self_download = origin in {"hf_seed", "self_download"}
+    return is_bounty, is_promote, downloaded, self_download
+
+
+def load_seed(path: str, origin: str, repo: Repository) -> int:
+    priority = ORIGIN_TO_PRIORITY.get(origin, PRIORITY_LOW_CONFIDENCE)
+    count = 0
+    with open(path, newline="", encoding="utf-8") as fh:
+        reader = csv.DictReader(fh)
+        for row in reader:
+            model_id = (row.get("model_id") or "").strip()
+            if not model_id:
+                continue
+            is_bounty, is_promote, downloaded, self_download = flags_for_origin(origin)
+            repo.upsert_candidate(
+                model_id=model_id,
+                source=(row.get("source") or "HUGGING_FACE").strip(),
+                origin=origin,
+                priority=priority,
+                is_bounty=is_bounty,
+                is_promote=is_promote,
+                known_downloaded_by_others=downloaded,
+                self_download_allowed=self_download,
+                max_model_len=int(row.get("max_model_len") or 1024),
+                target_gpu_aliases=(row.get("gpu_aliases") or "").strip() or None,
+                notes=row.get("notes"),
+            )
+            count += 1
+    return count
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--file", required=True)
+    parser.add_argument("--origin", required=True, choices=sorted(ORIGIN_TO_PRIORITY))
+    args = parser.parse_args()
+
+    settings = load_settings(require_secrets=False)
+    conn = connect(settings.db_path)
+    try:
+        count = load_seed(args.file, args.origin, Repository(conn))
+        print(f"imported {count} candidates from {args.file} as {args.origin}")
+    finally:
+        conn.close()
+
+
+if __name__ == "__main__":
+    main()