adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changes:(controllers) Modify the statistical method of the knowledge base

Browse Source
This commit is contained in:
lanceyq
2026-04-02 12:26:20 +08:00
parent 6eca5f6cdf
commit bd48b4fdbe
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/app/controllers/memory_dashboard_controller.py
View File

@@ -611,14 +611,16 @@ async def dashboard_data(
except Exception as e: except Exception as e:
api_logger.warning(f"获取记忆总量失败: {str(e)}") api_logger.warning(f"获取记忆总量失败: {str(e)}")
# 2. 获取知识库数量（total_knowledge），排除用户知识库(permission_id='Memory') # 2. 获取知识库数量（total_knowledge）
# 逻辑：统计 knowledges 表中 workspace_id = 当前工作空间 且 parent_id = workspace_id 的记录数
# 即只统计顶层知识库（parent_id 指向所属工作空间）
try: try:
from sqlalchemy import func as _func from sqlalchemy import func as _func
from app.models.knowledge_model import Knowledge as _Knowledge from app.models.knowledge_model import Knowledge as _Knowledge
total_knowledge = db.query(_func.count(_Knowledge.id)).filter( total_knowledge = db.query(_func.count(_Knowledge.id)).filter(
_Knowledge.workspace_id == workspace_id, _Knowledge.workspace_id == workspace_id,
_Knowledge.status == 1, _Knowledge.status == 1,
_Knowledge.permission_id != "Memory" _Knowledge.parent_id == _Knowledge.workspace_id
).scalar() or 0 ).scalar() or 0
neo4j_data["total_knowledge"] = total_knowledge neo4j_data["total_knowledge"] = total_knowledge
api_logger.info(f"成功获取知识库数量: {neo4j_data['total_knowledge']}") api_logger.info(f"成功获取知识库数量: {neo4j_data['total_knowledge']}")