kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix None handling in knowledge tool filter

Browse Source
This commit is contained in:
jonasHanhan 2026-01-14 14:43:16 +08:00 committed by Wenjie Zhang
parent 34667e19b5
commit 9a54b9d13f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/agents/common/tools.py
View File

@ -145,7 +145,10 @@ def get_kb_based_tools(db_names: list[str] | None = None) -> list:
# 获取所有知识库 # 获取所有知识库
kb_tools = [] kb_tools = []
retrievers = knowledge_base.get_retrievers() retrievers = knowledge_base.get_retrievers()
db_ids = [kb_id for kb_id, kb in retrievers.items() if kb["name"] in db_names] or None if db_names is None:
db_ids = None
else:
db_ids = [kb_id for kb_id, kb in retrievers.items() if kb["name"] in db_names]
def _create_retriever_wrapper(db_id: str, retriever_info: dict[str, Any]): def _create_retriever_wrapper(db_id: str, retriever_info: dict[str, Any]):
"""创建检索器包装函数的工厂函数，避免闭包变量捕获问题""" """创建检索器包装函数的工厂函数，避免闭包变量捕获问题"""