410dd47c14
- 将 server/, src/, scripts/, test/ 等目录移动到 backend/ 目录下 - 使用 git rename 保留文件历史记录 - 更新 docker-compose.yml 和 api.Dockerfile 配置 WIP: 项目结构重构进行中
27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
import os
|
|
|
|
from ..config import config
|
|
from .factory import KnowledgeBaseFactory
|
|
from .graphs.upload_graph_service import UploadGraphService
|
|
from .implementations.dify import DifyKB
|
|
from .implementations.lightrag import LightRagKB
|
|
from .implementations.milvus import MilvusKB
|
|
from .manager import KnowledgeBaseManager
|
|
|
|
# 注册知识库类型
|
|
KnowledgeBaseFactory.register("milvus", MilvusKB, {"description": "基于 Milvus 的生产级向量知识库，适合高性能部署"})
|
|
KnowledgeBaseFactory.register("lightrag", LightRagKB, {"description": "基于图检索的知识库，支持实体关系构建和复杂查询"})
|
|
KnowledgeBaseFactory.register("dify", DifyKB, {"description": "连接 Dify Dataset 的只读检索知识库"})
|
|
|
|
# 创建知识库管理器
|
|
work_dir = os.path.join(config.save_dir, "knowledge_base_data")
|
|
knowledge_base = KnowledgeBaseManager(work_dir)
|
|
|
|
# 创建图数据库实例
|
|
graph_base = UploadGraphService()
|
|
|
|
# 向后兼容：让 GraphDatabase 指向 UploadGraphService
|
|
GraphDatabase = UploadGraphService
|
|
|
|
__all__ = ["GraphDatabase", "UploadGraphService", "knowledge_base", "graph_base"]
|