| 1 |
"""Curated base-model registry + `hf:org/name` escape hatch. |
| 2 |
|
| 3 |
Public surface: |
| 4 |
|
| 5 |
- `BASE_MODELS`, `known_keys()` — registry dict + stable ordering |
| 6 |
- `BaseModelSpec` — per-entry metadata (frozen, strict) |
| 7 |
- `resolve(spec)` — `key` or `hf:org/name` → `BaseModelSpec` |
| 8 |
- Probes live in `dlm.base_models.probes`; downloader in `downloader.py` |
| 9 |
- Errors: `BaseModelError`, `UnknownBaseModelError`, `ProbeFailedError`, |
| 10 |
`GatedModelError` |
| 11 |
""" |
| 12 |
|
| 13 |
from __future__ import annotations |
| 14 |
|
| 15 |
from dlm.base_models.downloader import DownloadResult, download_spec, sha256_of_directory |
| 16 |
from dlm.base_models.errors import ( |
| 17 |
BaseModelError, |
| 18 |
GatedModelError, |
| 19 |
ProbeFailedError, |
| 20 |
ProbeReport, |
| 21 |
ProbeResult, |
| 22 |
UnknownBaseModelError, |
| 23 |
) |
| 24 |
from dlm.base_models.license import ( |
| 25 |
AcceptanceVia, |
| 26 |
LicenseAcceptance, |
| 27 |
is_gated, |
| 28 |
require_acceptance, |
| 29 |
) |
| 30 |
from dlm.base_models.registry import BASE_MODELS, known_keys |
| 31 |
from dlm.base_models.resolver import resolve, resolve_hf |
| 32 |
from dlm.base_models.schema import BaseModelSpec |
| 33 |
|
| 34 |
__all__ = [ |
| 35 |
"BASE_MODELS", |
| 36 |
"AcceptanceVia", |
| 37 |
"BaseModelError", |
| 38 |
"BaseModelSpec", |
| 39 |
"DownloadResult", |
| 40 |
"GatedModelError", |
| 41 |
"LicenseAcceptance", |
| 42 |
"ProbeFailedError", |
| 43 |
"ProbeReport", |
| 44 |
"ProbeResult", |
| 45 |
"UnknownBaseModelError", |
| 46 |
"download_spec", |
| 47 |
"is_gated", |
| 48 |
"known_keys", |
| 49 |
"require_acceptance", |
| 50 |
"resolve", |
| 51 |
"resolve_hf", |
| 52 |
"sha256_of_directory", |
| 53 |
] |