Update src/__init__.py

src/__init__.py

@@ -1,6 +1,5 @@
-# Initialize package
+from .history import save_conversation, load_conversation, ChatHistory
 from .retrieval import DocumentRetriever
 from .generation import ResponseGenerator
-from .history import save_conversation, load_conversation
 
-__all__ = ['DocumentRetriever', 'ResponseGenerator', 'save_conversation', 'load_conversation']
+__all__ = ['save_conversation', 'load_conversation', 'ChatHistory', 'DocumentRetriever', 'ResponseGenerator']