"""
SAGE Data Collection Module

Infrastructure for logging interactions and collecting expert feedback
to enable future custom model training.
"""

from sage.data_collection.schemas import (
    ExpertFeedback,
    LoggedInteraction,
    TrainingExample,
)
from sage.data_collection.interaction_logger import InteractionLogger
from sage.data_collection.feedback_collector import FeedbackCollector
from sage.data_collection.export import TrainingDataExporter

__all__ = [
    "ExpertFeedback",
    "FeedbackCollector",
    "InteractionLogger",
    "LoggedInteraction",
    "TrainingDataExporter",
    "TrainingExample",
]
