| """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" | |
| ] | |