This commit is contained in:
ed
2026-06-26 05:55:22 -04:00
parent cf7ef3fc66
commit 8f6ae6d983
2 changed files with 75 additions and 101 deletions
+4 -28
View File
File diff suppressed because one or more lines are too long
+2 -4
View File
@@ -35,10 +35,8 @@ def register(cap: VendorCapabilities) -> None:
_REGISTRY[(cap.vendor, cap.model)] = cap _REGISTRY[(cap.vendor, cap.model)] = cap
def get_capabilities(vendor: str, model: str) -> VendorCapabilities: def get_capabilities(vendor: str, model: str) -> VendorCapabilities:
if (vendor, model) in _REGISTRY: if (vendor, model) in _REGISTRY: return _REGISTRY[(vendor, model)]
return _REGISTRY[(vendor, model)] if (vendor, '*') in _REGISTRY: return _REGISTRY[(vendor, '*')]
if (vendor, '*') in _REGISTRY:
return _REGISTRY[(vendor, '*')]
raise KeyError(f'No capabilities registered for vendor={vendor!r} model={model!r}') raise KeyError(f'No capabilities registered for vendor={vendor!r} model={model!r}')
def list_models_for_vendor(vendor: str) -> list[str]: def list_models_for_vendor(vendor: str) -> list[str]: