"""Mosaic Stack Telemetry — Python client SDK.

A lightweight client for reporting AI coding task-completion telemetry
and querying crowd-sourced predictions from a Mosaic Stack Telemetry server.
"""

from mosaicstack_telemetry.client import TelemetryClient
from mosaicstack_telemetry.config import TelemetryConfig
from mosaicstack_telemetry.event_builder import EventBuilder
from mosaicstack_telemetry.prediction_cache import PredictionCache
from mosaicstack_telemetry.queue import EventQueue
from mosaicstack_telemetry.types.common import (
    BatchEventRequest,
    BatchEventResponse,
    BatchEventResult,
    TelemetryError,
)
from mosaicstack_telemetry.types.events import (
    Complexity,
    Harness,
    Outcome,
    Provider,
    QualityGate,
    RepoSizeCategory,
    TaskCompletionEvent,
    TaskType,
)
from mosaicstack_telemetry.types.predictions import (
    CorrectionFactors,
    PredictionData,
    PredictionMetadata,
    PredictionQuery,
    PredictionResponse,
    QualityPrediction,
    TokenDistribution,
)

__version__ = "0.1.0"

__all__ = [
    # Client
    "TelemetryClient",
    "TelemetryConfig",
    "EventBuilder",
    "EventQueue",
    "PredictionCache",
    # Types - Events
    "TaskCompletionEvent",
    "TaskType",
    "Complexity",
    "Harness",
    "Provider",
    "QualityGate",
    "Outcome",
    "RepoSizeCategory",
    # Types - Predictions
    "PredictionQuery",
    "PredictionResponse",
    "PredictionData",
    "PredictionMetadata",
    "TokenDistribution",
    "CorrectionFactors",
    "QualityPrediction",
    # Types - Common
    "BatchEventRequest",
    "BatchEventResponse",
    "BatchEventResult",
    "TelemetryError",
]