Function _validate_draft

src/task/specs/task_draft.py:204–245

(template: TaskTemplateSpec, draft: TaskDraftSpec)

def _validate_draft(template: TaskTemplateSpec, draft: TaskDraftSpec) -> None:
    slot_specs = {slot.slot_name: slot for slot in template.slots}
    seen_counts: dict[str, int] = {}

    if not draft.base_variation_instruction.strip():
        raise ValueError("Base variation instruction cannot be empty.")
    if draft.primary_goal_relation not in template.allowed_relations:
        raise ValueError(f"Primary goal relation '{draft.primary_goal_relation}' is not allowed by template.")
    if draft.container_slot is not None and draft.container_slot not in slot_specs:
        raise ValueError(f"Unknown container slot '{draft.container_slot}'.")

    for relation in draft.fail_relations:
        if relation not in template.allowed_relations and relation not in ("on",):
            raise ValueError(f"Fail relation '{relation}' is not supported by template.")

    for draft_object in draft.objects:
        slot = slot_specs.get(draft_object.slot_name)
        if slot is None:
            raise ValueError(f"Draft references unknown slot '{draft_object.slot_name}'.")
        seen_counts[draft_object.slot_name] = seen_counts.get(draft_object.slot_name, 0) + 1
        if draft_object.group_hint not in _ALLOWED_GROUP_HINTS:
            raise ValueError(f"Unsupported group hint '{draft_object.group_hint}'.")
        if slot.group_hint != draft_object.group_hint:
            raise ValueError(
                f"Slot '{draft_object.slot_name}' expects group '{slot.group_hint}', got '{draft_object.group_hint}'."
            )
        _validate_closed_range("mass_range", draft_object.mass_range)
        _validate_closed_range("target_size_range", draft_object.target_size_range)
        if not draft_object.semantic_name.strip():
            raise ValueError(f"Slot '{draft_object.slot_name}' has empty semantic_name.")
        if not draft_object.retrieval_query.strip():
            raise ValueError(f"Slot '{draft_object.slot_name}' has empty retrieval_query.")

    for slot_name, slot in slot_specs.items():
        count = seen_counts.get(slot_name, 0)
        if count < slot.min_count or count > slot.max_count:
            raise ValueError(
                f"Slot '{slot_name}' expects count in [{slot.min_count}, {slot.max_count}], got {count}."
            )

    if draft.container_slot is None:
        raise ValueError("Current task compiler requires container_slot to be set.")


def _validate_closed_range(name: str, values: tuple[float, float]) -> None: