__init__.py

"""
COGNITIVE-CORE Framework
========================

Universal template for Ame Web Studio's cognitive AI architectures.
Provides standardized loading, checkpoint management, and utilities
for vision, language, world model, and multimodal cognitive systems.

Copyright © 2026 Mike Amega (Logo) - Ame Web Studio
License: Proprietary - All Rights Reserved
"""

from .cognitive_base import (
    CognitiveConfig,
    CognitiveModule,
    MemoryModule,
    TemporalModule,
    WorldModelModule,
    CognitivePreTrainedModel,
    register_cognitive_model,
)

from .cognitive_checkpoint import (
    remap_checkpoint_keys,
    validate_checkpoint,
    save_cognitive_checkpoint,
    load_cognitive_checkpoint,
)

from .cognitive_utils import (
    setup_environment,
    get_device,
    get_optimal_dtype,
    get_memory_info,
    clear_memory,
    estimate_model_memory,
    print_model_info,
    print_training_progress,
    get_hf_token,
)

from .cognitive_training import (
    CognitiveTrainingConfig,
    CognitiveTrainer,
    prepare_dataset,
    create_instruction_dataset,
    quick_train,
    CognitiveStateCallback,
)

__version__ = "1.0.0"
__author__ = "Mike Amega"
__license__ = "Proprietary"

__all__ = [
    # Base classes
    "CognitiveConfig",
    "CognitiveModule",
    "MemoryModule",
    "TemporalModule",
    "WorldModelModule",
    "CognitivePreTrainedModel",
    "register_cognitive_model",
    # Checkpoint
    "remap_checkpoint_keys",
    "validate_checkpoint",
    "save_cognitive_checkpoint",
    "load_cognitive_checkpoint",
    # Utils
    "setup_environment",
    "get_device",
    "get_optimal_dtype",
    "get_memory_info",
    "clear_memory",
    "estimate_model_memory",
    "print_model_info",
    "print_training_progress",
    "get_hf_token",
    # Training
    "CognitiveTrainingConfig",
    "CognitiveTrainer",
    "prepare_dataset",
    "create_instruction_dataset",
    "quick_train",
    "CognitiveStateCallback",
]