"""
HallucinationGuard-Env: RL environment for training LLMs on hallucination avoidance.

Usage:
    from hallucination_guard_env import HallucinationEnv

    env = HallucinationEnv()
    obs = env.reset()
    result = env.step(answer="your answer", confidence=0.8)
"""

__version__ = "2.1.0"

from hallucination_guard_env.environment import HallucinationEnvironment
from hallucination_guard_env.models import (
    HallucinationAction,
    HallucinationObservation,
    HallucinationState,
    DifficultyLevel,
    HallucinationType,
    HallucinationSeverity,
)

# Convenience alias
HallucinationEnv = HallucinationEnvironment

__all__ = [
    "__version__",
    "HallucinationEnvironment",
    "HallucinationEnv",
    "HallucinationAction",
    "HallucinationObservation",
    "HallucinationState",
    "DifficultyLevel",
    "HallucinationType",
    "HallucinationSeverity",
]