ForcePilot/backend/package/yuxi/agents/middlewares/knowledge_base.py

"""知识库中间件 - 提供通用知识库工具"""

from langchain.agents.middleware import AgentMiddleware

from yuxi.agents.toolkits.kbs import get_common_kb_tools
from yuxi.utils.logging_config import logger


class KnowledgeBaseMiddleware(AgentMiddleware):
    """知识库中间件 - 提供通用知识库工具，其他没有任何作用

    提供通用知识库工具：
    - list_kbs: 列出用户可访问的知识库
    - get_mindmap: 获取指定知识库的思维导图
    - query_kb: 在指定知识库中检索
    - find_kb_document: 在指定文件内定位关键词或正则模式
    - open_kb_document: 按 file_id 分段打开知识库文档
    """

    def __init__(self):
        super().__init__()
        # 预加载通用知识库工具
        self.kb_tools = get_common_kb_tools()
        self.tools = self.kb_tools
        logger.debug(f"Initialized KnowledgeBaseMiddleware with {len(self.kb_tools)} tools")
-												refactor: 重构知识库工具并集成新工具

- 新增 KnowledgeBaseMiddleware，提供通用知识库工具（list_kbs、get_mindmap、query_kb）
- 新增 debug/tools.py 调试工具集
- 重构 kbs/tools.py，分离通用工具与 MySQL 工具
- 移除旧的 toolkits registry 简化架构
- API 变更：get_databases_by_user → get_databases_by_user_id
- 前端拆分 KnowledgeBaseTool.vue 为多个专用工具组件

											
										
										
											2026-03-04 04:06:48 +08:00
+								"""知识库中间件 - 提供通用知识库工具"""
-												refactor: Agent 运行时架构重构 - 移除 RuntimeConfigMiddleware，统一上下文准备与工具解析

- 删除 runtime_config_middleware.py，将功能合并到:
  - prepare_agent_runtime_context(): 统一上下文准备入口
  - resolve_configured_runtime_tools(): 运行时工具解析
- context.py: 规范化函数重命名 (_names → _keys)，重构 config 加载流程
- chatbot/deep_agent graph: 集成新上下文准备流程，移除 RuntimeConfigMiddleware 引用
- skills_middleware: 抽取 normalize_string_list 为共享工具函数
- subagent_service: get_subagents_from_names → get_subagents_from_slugs，支持 slug 查询
- 对应更新 repositories/services/routers 适配新接口

											
										
										
											2026-05-21 19:32:10 +08:00
+								from langchain.agents.middleware import AgentMiddleware
-												refactor: 重构知识库工具并集成新工具

- 新增 KnowledgeBaseMiddleware，提供通用知识库工具（list_kbs、get_mindmap、query_kb）
- 新增 debug/tools.py 调试工具集
- 重构 kbs/tools.py，分离通用工具与 MySQL 工具
- 移除旧的 toolkits registry 简化架构
- API 变更：get_databases_by_user → get_databases_by_user_id
- 前端拆分 KnowledgeBaseTool.vue 为多个专用工具组件

											
										
										
											2026-03-04 04:06:48 +08:00
-												refactor: 优化代码结构 将 yuxi.agents.common 移动到 yuxi.agents

											
										
										
											2026-03-20 12:08:00 +08:00
+								from yuxi.agents.toolkits.kbs import get_common_kb_tools
-												refactor: 将后端代码迁移至 backend 目录

- 将 server/, src/, scripts/, test/ 等目录移动到 backend/ 目录下
- 使用 git rename 保留文件历史记录
- 更新 docker-compose.yml 和 api.Dockerfile 配置

WIP: 项目结构重构进行中

											
										
										
											2026-03-17 10:16:44 +08:00
+								from yuxi.utils.logging_config import logger
-												refactor: 重构知识库工具并集成新工具

- 新增 KnowledgeBaseMiddleware，提供通用知识库工具（list_kbs、get_mindmap、query_kb）
- 新增 debug/tools.py 调试工具集
- 重构 kbs/tools.py，分离通用工具与 MySQL 工具
- 移除旧的 toolkits registry 简化架构
- API 变更：get_databases_by_user → get_databases_by_user_id
- 前端拆分 KnowledgeBaseTool.vue 为多个专用工具组件

											
										
										
											2026-03-04 04:06:48 +08:00
 								class KnowledgeBaseMiddleware(AgentMiddleware):
-												refactor: Agent 运行时架构重构 - 移除 RuntimeConfigMiddleware，统一上下文准备与工具解析

- 删除 runtime_config_middleware.py，将功能合并到:
  - prepare_agent_runtime_context(): 统一上下文准备入口
  - resolve_configured_runtime_tools(): 运行时工具解析
- context.py: 规范化函数重命名 (_names → _keys)，重构 config 加载流程
- chatbot/deep_agent graph: 集成新上下文准备流程，移除 RuntimeConfigMiddleware 引用
- skills_middleware: 抽取 normalize_string_list 为共享工具函数
- subagent_service: get_subagents_from_names → get_subagents_from_slugs，支持 slug 查询
- 对应更新 repositories/services/routers 适配新接口

											
										
										
											2026-05-21 19:32:10 +08:00
+								    """知识库中间件 - 提供通用知识库工具，其他没有任何作用
-												refactor: 重构知识库工具并集成新工具

- 新增 KnowledgeBaseMiddleware，提供通用知识库工具（list_kbs、get_mindmap、query_kb）
- 新增 debug/tools.py 调试工具集
- 重构 kbs/tools.py，分离通用工具与 MySQL 工具
- 移除旧的 toolkits registry 简化架构
- API 变更：get_databases_by_user → get_databases_by_user_id
- 前端拆分 KnowledgeBaseTool.vue 为多个专用工具组件

											
										
										
											2026-03-04 04:06:48 +08:00
-												feat(kb): 统一知识库并支持 search/find/open

											
										
										
											2026-05-18 14:09:42 +08:00
+								    提供通用知识库工具：
-												refactor: 重构知识库工具并集成新工具

- 新增 KnowledgeBaseMiddleware，提供通用知识库工具（list_kbs、get_mindmap、query_kb）
- 新增 debug/tools.py 调试工具集
- 重构 kbs/tools.py，分离通用工具与 MySQL 工具
- 移除旧的 toolkits registry 简化架构
- API 变更：get_databases_by_user → get_databases_by_user_id
- 前端拆分 KnowledgeBaseTool.vue 为多个专用工具组件

											
										
										
											2026-03-04 04:06:48 +08:00
+								    - list_kbs: 列出用户可访问的知识库
 								    - get_mindmap: 获取指定知识库的思维导图
 								    - query_kb: 在指定知识库中检索
-												feat(kb): 统一知识库并支持 search/find/open

											
										
										
											2026-05-18 14:09:42 +08:00
+								    - find_kb_document: 在指定文件内定位关键词或正则模式
 								    - open_kb_document: 按 file_id 分段打开知识库文档
-												refactor: 重构知识库工具并集成新工具

- 新增 KnowledgeBaseMiddleware，提供通用知识库工具（list_kbs、get_mindmap、query_kb）
- 新增 debug/tools.py 调试工具集
- 重构 kbs/tools.py，分离通用工具与 MySQL 工具
- 移除旧的 toolkits registry 简化架构
- API 变更：get_databases_by_user → get_databases_by_user_id
- 前端拆分 KnowledgeBaseTool.vue 为多个专用工具组件

											
										
										
											2026-03-04 04:06:48 +08:00
+								    """
 								    def __init__(self):
 								        super().__init__()
 								        # 预加载通用知识库工具
 								        self.kb_tools = get_common_kb_tools()
 								        self.tools = self.kb_tools
-												style(format): 格式化代码

											
										
										
											2026-03-04 04:13:05 +08:00
+								        logger.debug(f"Initialized KnowledgeBaseMiddleware with {len(self.kb_tools)} tools")