File size: 673 Bytes
868e04b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | from __future__ import annotations
from harness.adapters.base import Adapter, Verdict
from harness.adapters.context_compression import CompressionAdapter
from harness.adapters.generic import GenericAdapter
from harness.adapters.sql_generation import SqlAdapter
ADAPTERS: dict[str, Adapter] = {
GenericAdapter.name: GenericAdapter(),
SqlAdapter.name: SqlAdapter(),
CompressionAdapter.name: CompressionAdapter(),
}
def load(name: str) -> Adapter:
try:
return ADAPTERS[name]
except KeyError as exc:
raise ValueError(f"unknown adapter {name!r}; one of {sorted(ADAPTERS)}") from exc
__all__ = ["ADAPTERS", "Adapter", "Verdict", "load"]
|