| """Optional Decision Index integration.""" | |
| try: | |
| from decision_index.engines.base import Engine, NativeAbstention, Unsupported | |
| except ModuleNotFoundError as exc: | |
| if exc.name not in ("decision_index", "decision_index.engines", "decision_index.engines.base"): | |
| raise | |
| class Unsupported(ValueError): | |
| pass | |
| class NativeAbstention(Exception): | |
| def __init__(self, raw): | |
| self.raw = raw | |
| super().__init__("Native model selected its additional abstention outcome") | |
| class Engine: | |
| name = "engine" | |
| provenance = {} | |
| latency = "In-process request wall time including prompt construction; excludes model loading." | |
| def __init__(self, **options): | |
| self.options = options | |
| def __call__(self, state, questions): | |
| raise NotImplementedError | |
| def warmup(self): | |
| warm = {"warmup": {"type": "choice", "instructions": "Which color is named?", "criteria": {"red": "red", "blue": "blue"}}} | |
| for _ in range(2): | |
| try: | |
| self("The color is red.", warm) | |
| except NativeAbstention: | |
| pass | |
| def runtime(self): | |
| return {} | |
| def synchronize(self): | |
| pass | |
| def close(self): | |
| pass | |