From 5edf3f2b8a698056ed1a17dc7ea2562c31888c9c Mon Sep 17 00:00:00 2001
From: lanceyq <1982376970@qq.com>
Date: Tue, 3 Mar 2026 16:16:16 +0800
Subject: [PATCH] [changes] Test the scheduled task

---
 api/app/celery_app.py                     |  8 +--
 api/app/controllers/emotion_controller.py | 88 +++++++++++------------
 redbear-mem-benchmark                     |  2 +-
 3 files changed, 47 insertions(+), 51 deletions(-)

diff --git a/api/app/celery_app.py b/api/app/celery_app.py
index e804d303..33fa1703 100644
--- a/api/app/celery_app.py
+++ b/api/app/celery_app.py
@@ -117,12 +117,8 @@ beat_schedule_config = {
             "config_id": None,  # 使用默认配置，可以通过环境变量配置
         },
     },
-}
-
-#如果配置了默认工作空间ID，则添加记忆总量统计任务
-if settings.DEFAULT_WORKSPACE_ID:
-    beat_schedule_config["write-total-memory"] = {
-        "task": "app.controllers.memory_storage_controller.search_all",
+    "write-all-workspaces-memory": {
+        "task": "app.tasks.write_all_workspaces_memory_task",
         "schedule": memory_increment_schedule,
         "args": (),
     },
diff --git a/api/app/controllers/emotion_controller.py b/api/app/controllers/emotion_controller.py
index 0a8b5fc8..8cfc5014 100644
--- a/api/app/controllers/emotion_controller.py
+++ b/api/app/controllers/emotion_controller.py
@@ -208,55 +208,55 @@ async def get_emotion_health(
 
 
 
-@router.post("/check-data", response_model=ApiResponse)
-async def check_emotion_data_exists(
-    request: EmotionSuggestionsRequest,
-    db: Session = Depends(get_db),
-    current_user: User = Depends(get_current_user),
-):
-    """检查用户情绪建议数据是否存在
+# @router.post("/check-data", response_model=ApiResponse)
+# async def check_emotion_data_exists(
+#     request: EmotionSuggestionsRequest,
+#     db: Session = Depends(get_db),
+#     current_user: User = Depends(get_current_user),
+# ):
+#     """检查用户情绪建议数据是否存在
 
-    Args:
-        request: 包含 end_user_id
-        db: 数据库会话
-        current_user: 当前用户
+#     Args:
+#         request: 包含 end_user_id
+#         db: 数据库会话
+#         current_user: 当前用户
 
-    Returns:
-        数据存在状态
-    """
-    try:
-        api_logger.info(
-            f"检查用户情绪建议数据是否存在: {request.end_user_id}",
-            extra={"end_user_id": request.end_user_id}
-        )
+#     Returns:
+#         数据存在状态
+#     """
+#     try:
+#         api_logger.info(
+#             f"检查用户情绪建议数据是否存在: {request.end_user_id}",
+#             extra={"end_user_id": request.end_user_id}
+#         )
 
-        # 从数据库获取建议
-        data = await emotion_service.get_cached_suggestions(
-            end_user_id=request.end_user_id,
-            db=db
-        )
+#         # 从数据库获取建议
+#         data = await emotion_service.get_cached_suggestions(
+#             end_user_id=request.end_user_id,
+#             db=db
+#         )
 
-        if data is None:
-            api_logger.info(f"用户 {request.end_user_id} 的情绪建议数据不存在")
-            return fail(
-                BizCode.NOT_FOUND,
-                "情绪建议数据不存在，请点击右上角刷新进行初始化",
-                {"exists": False}
-            )
+#         if data is None:
+#             api_logger.info(f"用户 {request.end_user_id} 的情绪建议数据不存在")
+#             return fail(
+#                 BizCode.NOT_FOUND,
+#                 "情绪建议数据不存在，请点击右上角刷新进行初始化",
+#                 {"exists": False}
+#             )
 
-        api_logger.info(f"用户 {request.end_user_id} 的情绪建议数据存在")
-        return success(data={"exists": True}, msg="情绪建议数据已存在")
+#         api_logger.info(f"用户 {request.end_user_id} 的情绪建议数据存在")
+#         return success(data={"exists": True}, msg="情绪建议数据已存在")
 
-    except Exception as e:
-        api_logger.error(
-            f"检查情绪建议数据失败: {str(e)}",
-            extra={"end_user_id": request.end_user_id},
-            exc_info=True
-        )
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"检查情绪建议数据失败: {str(e)}"
-        )
+#     except Exception as e:
+#         api_logger.error(
+#             f"检查情绪建议数据失败: {str(e)}",
+#             extra={"end_user_id": request.end_user_id},
+#             exc_info=True
+#         )
+#         raise HTTPException(
+#             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+#             detail=f"检查情绪建议数据失败: {str(e)}"
+#         )
 
 
 @router.post("/suggestions", response_model=ApiResponse)
@@ -383,4 +383,4 @@ async def generate_emotion_suggestions(
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail=f"生成个性化建议失败: {str(e)}"
-        )
+        )
\ No newline at end of file
diff --git a/redbear-mem-benchmark b/redbear-mem-benchmark
index 4b0257bb..8494e824 160000
--- a/redbear-mem-benchmark
+++ b/redbear-mem-benchmark
@@ -1 +1 @@
-Subproject commit 4b0257bb4e7dc384b2aaf849b0bd6eae4b39835d
+Subproject commit 8494e82498cb99c70ac67a64a544ff872432363a