"""
MiniMind Max2 Enhanced Capabilities
Advanced features for reasoning, vision, agentic tasks, coding, and thinking.
"""

from .reasoning import ReasoningModule, ChainOfThoughtTrainer, ReasoningConfig
from .vision import VisionAdapter, VisionConfig, MiniMindVision
from .agentic import AgenticModule, FunctionCallingConfig, ToolRegistry
from .coding import FIMModule, CodeCompletionConfig, VibeCoder
from .speculative import SpeculativeDecoder, DraftModel
from .templates import (
    JinjaTemplateEngine,
    MDXRenderer,
    MDXComponentType,
    ExtendedCodeBlockParser,
    CodeBlockMeta,
    TemplateLibrary,
)
from .thinking import (
    ThinkingEngine,
    ThinkingConfig,
    ThinkingMode,
    ThinkingStep,
    InterleavedThinking,
    SequentialThinking,
    MultilingualThinking,
)

__all__ = [
    # Reasoning
    "ReasoningModule",
    "ChainOfThoughtTrainer",
    "ReasoningConfig",
    # Vision
    "VisionAdapter",
    "VisionConfig",
    "MiniMindVision",
    # Agentic
    "AgenticModule",
    "FunctionCallingConfig",
    "ToolRegistry",
    # Coding
    "FIMModule",
    "CodeCompletionConfig",
    "VibeCoder",
    # Optimization
    "SpeculativeDecoder",
    "DraftModel",
    # Templates & MDX
    "JinjaTemplateEngine",
    "MDXRenderer",
    "MDXComponentType",
    "ExtendedCodeBlockParser",
    "CodeBlockMeta",
    "TemplateLibrary",
    # Thinking
    "ThinkingEngine",
    "ThinkingConfig",
    "ThinkingMode",
    "ThinkingStep",
    "InterleavedThinking",
    "SequentialThinking",
    "MultilingualThinking",
]