"""OpenSleuth training-side helpers (env client, dataset, reward fn)."""

from .client import EnvClient
from .dataset import (
    DEFAULT_N_BY_DIFFICULTY,
    FUNCTIONS_FOR_TRAINING,
    build_synthesis_dataset,
    discover_functions,
)
from .prompt import SYSTEM_PROMPT, build_prompt, extract_code

__all__ = [
    "EnvClient",
    "DEFAULT_N_BY_DIFFICULTY",
    "FUNCTIONS_FOR_TRAINING",
    "build_synthesis_dataset",
    "discover_functions",
    "SYSTEM_PROMPT",
    "build_prompt",
    "extract_code",
]