__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
from .data_loaders import (
    EvaluationDataLoader,
    InlineDataLoader,
    LangfuseAdapterLoader,
    LangfuseLoaderConfig,
)

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

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

# Conditional import for optional LangChain dependency
try:
    from .default_langchain_rollout_processor import LangGraphRolloutProcessor

    LANGCHAIN_AVAILABLE = True
except ImportError:
    LANGCHAIN_AVAILABLE = False
    LangGraphRolloutProcessor = None

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

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

if LANGCHAIN_AVAILABLE:
    __all__.append("LangGraphRolloutProcessor")