Update supernova/__init__.py

supernova/__init__.py

@@ -1,6 +1,15 @@
-__version__ = "0.1.0"
-
-from .config import ModelConfig
-from .model import SupernovaModel
-from .tools import ToolOrchestrator, MathEngine, SerperAPI
-from .reasoning_engine import EnhancedReasoningEngine, ReasoningType, ReasoningStep
+__version__ = "0.1.0"
+
+from .config import ModelConfig
+from .model import SupernovaModel
+from .tokenizer import load_gpt2_tokenizer
+from .data import TokenChunkDataset, load_sources_from_yaml, DataSource
+from .tools import ToolOrchestrator, MathEngine, SerperAPI
+from .reasoning_engine import EnhancedReasoningEngine, ReasoningType, ReasoningStep
+
+__all__ = [
+    'ModelConfig', 'SupernovaModel', 'load_gpt2_tokenizer', 
+    'TokenChunkDataset', 'load_sources_from_yaml', 'DataSource',
+    'ToolOrchestrator', 'MathEngine', 'SerperAPI',
+    'EnhancedReasoningEngine', 'ReasoningType', 'ReasoningStep'
+]