__init__.py

from .default_agent_rollout_processor import AgentRolloutProcessor
from .default_dataset_adapter import default_dataset_adapter
from .default_mcp_gym_rollout_processor import MCPGymRolloutProcessor
from .default_no_op_rollout_processor import NoOpRolloutProcessor
from .default_single_turn_rollout_process import SingleTurnRolloutProcessor
from .evaluation_test import evaluation_test
from .exception_config import ExceptionHandlerConfig, BackoffConfig, get_default_exception_handler_config
from .rollout_processor import RolloutProcessor
from .types import RolloutProcessorConfig

# Conditional import for optional dependency
try:
    from .default_pydantic_ai_rollout_processor import PydanticAgentRolloutProcessor

    PYDANTIC_AI_AVAILABLE = True
except ImportError:
    PYDANTIC_AI_AVAILABLE = False
    PydanticAgentRolloutProcessor = None

__all__ = [
    "AgentRolloutProcessor",
    "MCPGymRolloutProcessor",
    "RolloutProcessor",
    "SingleTurnRolloutProcessor",
    "NoOpRolloutProcessor",
    "default_dataset_adapter",
    "RolloutProcessorConfig",
    "evaluation_test",
    "ExceptionHandlerConfig",
    "BackoffConfig",
    "get_default_exception_handler_config",
]

# Only add to __all__ if available
if PYDANTIC_AI_AVAILABLE:
    __all__.append("PydanticAgentRolloutProcessor")