35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""测试引导：加载连字符目录为可导入包，使场景模块可复用内核组件。
|
|||
|
|
|
|||
|
|
- `templates/ti-cl4/llm-scenarios` → 包名 ``ti_scenarios``；
|
|||
|
|
- `core/llm-gateway` → 包名 ``llm_gateway``（场景依赖）；
|
|||
|
|
- `core/rag-kb` → 包名 ``rag_kb``（场景依赖）。
|
|||
|
|
|
|||
|
|
与仓库内各 core 模块的测试引导同款模式；这里用 importlib 完整加载包
|
|||
|
|
（执行 __init__.py），保持 ``from rag_kb import ...`` 顶层导出可用。
|
|||
|
|
"""
|
|||
|
|
import importlib.util
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
SCEN_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
REPO_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(
|
|||
|
|
os.path.dirname(os.path.dirname(os.path.abspath(__file__))))))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _load_package(name: str, path: str) -> None:
|
|||
|
|
"""按文件路径完整加载一个包（执行其 __init__.py）。"""
|
|||
|
|
if name in sys.modules:
|
|||
|
|
return
|
|||
|
|
init_py = os.path.join(path, "__init__.py")
|
|||
|
|
spec = importlib.util.spec_from_file_location(
|
|||
|
|
name, init_py, submodule_search_locations=[path])
|
|||
|
|
module = importlib.util.module_from_spec(spec)
|
|||
|
|
sys.modules[name] = module
|
|||
|
|
spec.loader.exec_module(module)
|
|||
|
|
|
|||
|
|
|
|||
|
|
_load_package("rag_kb", os.path.join(REPO_ROOT, "core", "rag-kb"))
|
|||
|
|
_load_package("llm_gateway", os.path.join(REPO_ROOT, "core", "llm-gateway"))
|
|||
|
|
_load_package("ti_scenarios", SCEN_DIR)
|