Merge branch 'feat/redis_scheduler' of https://github.com/MemTensor/MemOS into feat/redis_scheduler

Author: fridayL

examples/mem_scheduler/api_w_scheduler.py

@@ -13,12 +13,6 @@
 print(f"use_redis_queue: {mem_scheduler.use_redis_queue}")
 print(f"Queue type: {type(mem_scheduler.memos_message_queue).__name__}")
 print(f"Queue maxsize: {getattr(mem_scheduler.memos_message_queue, 'maxsize', 'N/A')}")
-
-# Check if Redis queue is connected
-if hasattr(mem_scheduler.memos_message_queue, "_is_connected"):
-    print(f"Redis connected: {mem_scheduler.memos_message_queue._is_connected}")
-if hasattr(mem_scheduler.memos_message_queue, "_redis_conn"):
-    print(f"Redis connection: {mem_scheduler.memos_message_queue._redis_conn}")
 print("=====================================\n")
 
 queue = mem_scheduler.memos_message_queue

src/memos/api/handlers/search_handler.py

@@ -105,7 +105,7 @@ def handle_search_memories(self, search_req: APISearchRequest) -> SearchResponse
         )
 
     def _get_search_mode(self, mode: str) -> str:
-        return os.getenv("SEARCH_MODE", SearchMode.FAST)
+        return mode
 
     def _search_text(
         self,

src/memos/api/routers/server_router.py

@@ -34,6 +34,7 @@
     SuggestionResponse,
 )
 from memos.log import get_logger
+from memos.mem_scheduler.base_scheduler import BaseScheduler
 
 
 logger = get_logger(__name__)
@@ -58,7 +59,7 @@
 
 # Extract commonly used components for function-based handlers
 # (These can be accessed from the components dict without unpacking all of them)
-mem_scheduler = components["mem_scheduler"]
+mem_scheduler: BaseScheduler = components["mem_scheduler"]
 llm = components["llm"]
 naive_mem_cube = components["naive_mem_cube"]