Update src/__init__.py

src/__init__.py

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