"""
Advanced Inference and Decoding Strategies for MK3

This module provides state-of-the-art reasoning and decoding techniques:
- Chain-of-Thought (CoT) prompting
- Self-Consistency for multi-path reasoning
- Contrastive Decoding (expert - amateur)
- Advanced sampling: min-p, top-k, top-p, temperature
- Speculative decoding for faster generation
"""

from .chain_of_thought import ChainOfThoughtGenerator, CoTPrompt, CoTConfig
from .self_consistency import SelfConsistencyGenerator, aggregate_answers
from .contrastive_decoding import ContrastiveDecoder, ContrastiveConfig
from .sampling import (
    SamplingStrategy,
    MinPSampler,
    TopKSampler,
    TopPSampler,
    TemperatureSampler,
    CombinedSampler,
    sample_from_logits
)
from .speculative import SpeculativeDecoder, SpeculativeConfig

__all__ = [
    # Chain-of-Thought
    'ChainOfThoughtGenerator',
    'CoTPrompt',
    'CoTConfig',

    # Self-Consistency
    'SelfConsistencyGenerator',
    'aggregate_answers',

    # Contrastive Decoding
    'ContrastiveDecoder',
    'ContrastiveConfig',

    # Sampling
    'SamplingStrategy',
    'MinPSampler',
    'TopKSampler',
    'TopPSampler',
    'TemperatureSampler',
    'CombinedSampler',
    'sample_from_logits',

    # Speculative Decoding
    'SpeculativeDecoder',
    'SpeculativeConfig',
]