feat: Pass user_name to get_current_memory_size

Modify the call to  in  to
include . This ensures that memory sizes are retrieved
for the correct MemCube/tenant context, aligning with multi-tenant monitoring
requirements outlined in design documents. Previously, this call did not pass
the user context, potentially leading to incorrect memory size reporting in
multi-tenant environments.

Author: [email protected]

src/memos/api/routers/product_router.py

@@ -202,10 +202,11 @@ def create_memory(memory_req: MemoryCreateRequest):
             and hasattr(mos_product, "mem_scheduler")
             and mos_product.mem_scheduler
         ):
-            from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker
             from uuid import uuid4
 
-            item_id = str(uuid4()) # Generate a unique item_id for this submission
+            from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker
+
+            item_id = str(uuid4())  # Generate a unique item_id for this submission
 
             # Get Redis client from scheduler
             if (
@@ -215,14 +216,13 @@ def create_memory(memory_req: MemoryCreateRequest):
                 status_tracker = TaskStatusTracker(mos_product.mem_scheduler.redis_client)
                 # Submit task with "product_add" type
                 status_tracker.task_submitted(
-                    task_id=item_id, # Use generated item_id for internal tracking
+                    task_id=item_id,  # Use generated item_id for internal tracking
                     user_id=memory_req.user_id,
                     task_type="product_add",
                     mem_cube_id=memory_req.mem_cube_id or memory_req.user_id,
                     business_task_id=memory_req.task_id,  # Use memory_req.task_id as business_task_id
                 )
-                status_tracker.task_started(item_id, memory_req.user_id) # Use item_id here
-
+                status_tracker.task_started(item_id, memory_req.user_id)  # Use item_id here
 
         # Execute the add operation
         mos_product.add(

src/memos/mem_scheduler/general_modules/scheduler_logger.py

@@ -49,7 +49,7 @@ def create_autofilled_log_item(
         mem_cube: GeneralMemCube,
     ) -> ScheduleLogForWebItem:
         text_mem_base: TreeTextMemory = mem_cube.text_mem
-        current_memory_sizes = text_mem_base.get_current_memory_size()
+        current_memory_sizes = text_mem_base.get_current_memory_size(user_name=mem_cube_id)
         current_memory_sizes = {
             "long_term_memory_size": current_memory_sizes.get("LongTermMemory", 0),
             "user_memory_size": current_memory_sizes.get("UserMemory", 0),

src/memos/reranker/factory.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import json
+
 from typing import TYPE_CHECKING, Any
 
 # Import singleton decorator