"""Module for Mongodb database"""
from .mongodb import (article_collection, entity_collection, category_collection,
    keywords_collection, summary_collection, lda_collection)
from .astra import KnowledgeBase

# Initialize KnowledgeBase instance
knowledge_base = KnowledgeBase()

__all__ = [
    "article_collection",
    "entity_collection",
    "category_collection",
    "keywords_collection",
    "summary_collection",
    "lda_collection",
    "knowledge_base"
]