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

fix(knowledge-tool): 修复使用的 ID 而不是知识库的 bug

Browse Source
This commit is contained in:
Wenjie Zhang 2026-01-22 02:16:08 +08:00
parent e585f796d2
commit f5fedb4dec
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/routers/chat_router.py
View File

@ -283,27 +283,27 @@ async def save_agent_config(
from src import knowledge_base
if "knowledges" in config and config["knowledges"]:
# 获取用户有权访问的知识库 ID
# 获取用户有权访问的知识库名称
try:
user_info = {"role": current_user.role, "department_id": current_user.department_id}
accessible_databases = await knowledge_base.get_databases_by_user(user_info)
accessible_db_ids = {
db.get("db_id") for db in accessible_databases.get("databases", []) if db.get("db_id")
accessible_kb_names = {
db.get("name") for db in accessible_databases.get("databases", []) if db.get("name")
}
except Exception as db_error:
logger.warning(f"获取知识库列表失败: {db_error}")
# 如果获取失败，superadmin 可以访问所有，非 superadmin 无法访问任何
if current_user.role != "superadmin":
raise HTTPException(status_code=500, detail="无法获取知识库列表")
# 回退：获取所有数据库 ID
# 回退：获取所有数据库名称
from src.repositories.knowledge_base_repository import KnowledgeBaseRepository
kb_repo = KnowledgeBaseRepository()
rows = await kb_repo.get_all()
accessible_db_ids = {row.db_id for row in rows}
accessible_kb_names = {row.name for row in rows if row.name}
# 检查配置中的知识库是否都可用
invalid_kbs = [kb for kb in config["knowledges"] if kb not in accessible_db_ids]
invalid_kbs = [kb for kb in config["knowledges"] if kb not in accessible_kb_names]
if invalid_kbs:
raise HTTPException(status_code=403, detail=f"无权访问以下知识库: {', '.join(invalid_kbs)}")
# === 校验结束 ===