Merge remote-tracking branch 'wangcham/feat/rag' into feat/rag

pkg/core/app.py

@@ -27,6 +27,7 @@ from ..storage import mgr as storagemgr
 from ..utils import logcache
 from . import taskmgr
 from . import entities as core_entities
+from pkg.rag.knowledge.RAG_Manager import RAG_Manager
 
 
 class Application:
@@ -47,6 +48,7 @@ class Application:
 
     model_mgr: llm_model_mgr.ModelManager = None
 
+
     # TODO 移动到 pipeline 里
     tool_mgr: llm_tool_mgr.ToolManager = None
 
@@ -99,6 +101,7 @@ class Application:
 
     storage_mgr: storagemgr.StorageMgr = None
 
+
     # ========= HTTP Services =========
 
     user_service: user_service.UserService = None
@@ -111,6 +114,9 @@ class Application:
 
     bot_service: bot_service.BotService = None
 
+    knowledge_base_service: RAG_Manager = None
+
+
     def __init__(self):
         pass
 
@@ -145,6 +151,7 @@ class Application:
                 name='http-api-controller',
                 scopes=[core_entities.LifecycleControlScope.APPLICATION],
             )
+
             self.task_mgr.create_task(
                 never_ending(),
                 name='never-ending-task',

pkg/core/entities.py

@@ -19,7 +19,7 @@ class LifecycleControlScope(enum.Enum):
     APPLICATION = 'application'
     PLATFORM = 'platform'
     PLUGIN = 'plugin'
-    PROVIDER = 'provider'
+    PROVIDER = 'provider'  
 
 
 class LauncherTypes(enum.Enum):

pkg/core/stages/build_app.py

@@ -9,6 +9,7 @@ from ...command import cmdmgr
 from ...provider.session import sessionmgr as llm_session_mgr
 from ...provider.modelmgr import modelmgr as llm_model_mgr
 from ...provider.tools import toolmgr as llm_tool_mgr
+from ...rag.knowledge.RAG_Manager import RAG_Manager as knowledge_base_mgr
 from ...platform import botmgr as im_mgr
 from ...persistence import mgr as persistencemgr
 from ...api.http.controller import main as http_controller
@@ -101,6 +102,10 @@ class BuildAppStage(stage.BootingStage):
         embedding_models_service_inst = model_service.EmbeddingModelsService(ap)
         ap.embedding_models_service = embedding_models_service_inst
 
+        knowledge_base_service_inst = knowledge_base_mgr(ap)
+        await knowledge_base_service_inst.initialize_rag_system()
+        ap.knowledge_base_service = knowledge_base_service_inst
+
         pipeline_service_inst = pipeline_service.PipelineService(ap)
         ap.pipeline_service = pipeline_service_inst