perf: definitions

pkg/core/app.py

@@ -27,7 +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
+from ..rag.knowledge import mgr as rag_mgr
 
 
 class Application:
@@ -48,7 +48,6 @@ class Application:
 
     model_mgr: llm_model_mgr.ModelManager = None
 
-
     # TODO 移动到 pipeline 里
     tool_mgr: llm_tool_mgr.ToolManager = None
 
@@ -101,7 +100,6 @@ class Application:
 
     storage_mgr: storagemgr.StorageMgr = None
 
-
     # ========= HTTP Services =========
 
     user_service: user_service.UserService = None
@@ -114,8 +112,7 @@ class Application:
 
     bot_service: bot_service.BotService = None
 
-    knowledge_base_service: RAG_Manager = None
-
+    knowledge_base_service: rag_mgr.RAGManager = None
 
     def __init__(self):
         pass

pkg/core/stages/build_app.py

@@ -9,7 +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 ...rag.knowledge import mgr as rag_mgr
 from ...platform import botmgr as im_mgr
 from ...persistence import mgr as persistencemgr
 from ...api.http.controller import main as http_controller
@@ -102,7 +102,7 @@ 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)
+        knowledge_base_service_inst = rag_mgr.RAGManager(ap)
         await knowledge_base_service_inst.initialize_rag_system()
         ap.knowledge_base_service = knowledge_base_service_inst