diff --git a/verifiers/v1/harnesses/utils/compaction.py b/verifiers/v1/harnesses/utils/compaction.py index c352c6847..f09db7214 100644 --- a/verifiers/v1/harnesses/utils/compaction.py +++ b/verifiers/v1/harnesses/utils/compaction.py @@ -98,8 +98,10 @@ async def discover_threshold(client: AsyncOpenAI, model: str) -> int | None: if card.id != model: continue extra = card.model_extra or {} + # Prime Inference nests the window under `specs`. + specs = extra.get("specs") if isinstance(extra.get("specs"), dict) else {} for field in CONTEXT_WINDOW_FIELDS: - value = extra.get(field) + value = extra.get(field, specs.get(field)) if isinstance(value, int) and not isinstance(value, bool) and value > 0: return default_threshold(value) break