diff --git a/rag/__init__.py b/rag/__init__.py index 8e7eac3..7a3f359 100644 --- a/rag/__init__.py +++ b/rag/__init__.py @@ -4,6 +4,6 @@ RAG业务模块 - 知识库、文档、引擎配置、订阅管理 作为独立模块被ragserver或其他应用引用 """ -from .init import init_rag_module +from .init import load_rag, init_rag_module -__all__ = ['init_rag_module'] +__all__ = ['load_rag', 'init_rag_module'] diff --git a/rag/init.py b/rag/init.py index bf2d873..0b78e19 100644 --- a/rag/init.py +++ b/rag/init.py @@ -1370,7 +1370,7 @@ async def engine_cfg_test_handler(request, params_kw, *args, **kwargs): return json.dumps({"error": str(e)[:200]}) -def init_rag_module(): +def load_rag(): env = ServerEnv() rf = RegisterFunction() rf.register("status", status_handler) @@ -1395,3 +1395,7 @@ def init_rag_module(): rf.register("engine_cfg_get", engine_cfg_get_handler) rf.register("engine_cfg_save", engine_cfg_save_handler) rf.register("engine_cfg_test", engine_cfg_test_handler) + + +# 兼容旧名(ragserver/pipeline-app 曾用 init_rag_module;规范名为 load_rag) +init_rag_module = load_rag