Merge pull request #458 from SuanmoSuanyangTechnology/fix/RAG-memory

Fix/rag memory
2026-03-04 19:09:03 +08:00
parent 86f72100f0 647a978865
commit 23bfdcefef
3 changed files with 88 additions and 5 deletions
--- a/api/app/controllers/memory_dashboard_controller.py
+++ b/api/app/controllers/memory_dashboard_controller.py
@@ -606,8 +606,8 @@ async def dashboard_data(
            
            # 获取RAG相关数据
            try:
-                # total_memory: 使用 total_chunk（总chunk数）
-                total_chunk = memory_dashboard_service.get_rag_total_chunk(db, current_user)
+                # total_memory: 只统计用户知识库（permission_id='Memory'）的chunk数
+                total_chunk = memory_dashboard_service.get_rag_user_kb_total_chunk(db, current_user)
                rag_data["total_memory"] = total_chunk
                
                # total_app: 统计当前空间下的所有app数量