TalkToYourDocument / config /appConfig.py
gourisankar85's picture
Upload 13 files
e6cc6f7 verified
raw
history blame
516 Bytes
import logging
from retriever.llm_manager import LLMManager
from retriever.document_manager import DocumentManager
class AppConfig:
def __init__(self):
# Initialize LLMManager with the default model
self.gen_llm = LLMManager() # This will initialize the default model ("gemma2-9b-it")
# Initialize DocumentManager (it will be a singleton instance shared across the app)
self.doc_manager = DocumentManager()
logging.info("AppConfig initialized with LLMManager")