chaaim123 commited on
Commit
cdd2297
·
verified ·
1 Parent(s): 1eed145

Create config/appConfig.py

Browse files
Files changed (1) hide show
  1. config/appConfig.py +13 -0
config/appConfig.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import logging
2
+ from retriever.llm_manager import LLMManager
3
+ from retriever.document_manager import DocumentManager
4
+ from retriever.chat_manager import ChatManager
5
+
6
+ class AppConfig:
7
+ def __init__(self):
8
+ # Initialize LLMManager with the default model
9
+ self.gen_llm = LLMManager() # This will initialize the default model ("gemma2-9b-it")
10
+ # Initialize DocumentManager (it will be a singleton instance shared across the app)
11
+ self.doc_manager = DocumentManager()
12
+ self.chat_manager = ChatManager(documentManager = self.doc_manager, llmManager = self.gen_llm)
13
+ logging.info("AppConfig initialized with LLMManager")