"""Verify API types."""

from enum import Enum
from dataclasses import dataclass
from typing import Any


class Verdict(str, Enum):
    """Safety verdict from policy verification."""

    SAFE = "safe"
    UNSAFE = "unsafe"
    ERROR = "error"
    PENDING = "pending"


@dataclass
class VerifyConfig:
    """Configuration for verification job."""

    scenario: str
    num_episodes: int = 5
    timeout: int = 300
    record_video: bool = True

    def to_dict(self) -> dict[str, Any]:
        return {
            "scenario": self.scenario,
            "num_episodes": self.num_episodes,
            "timeout": self.timeout,
            "record_video": self.record_video,
        }
