Feat： update code

src/memos/api/handlers/search_handler.py

@@ -191,7 +191,7 @@ def _fast_search(
         """
         target_session_id = search_req.session_id or "default_session"
         search_filter = {"session_id": search_req.session_id} if search_req.session_id else None
-        plugin = bool(search_req.info is not None and search_req.info.get("origin_model"))
+        plugin = bool(search_req.source is not None and search_req.source == "plugin")
         search_results = self.naive_mem_cube.text_mem.search(
             query=search_req.query,
             user_name=user_context.mem_cube_id,

src/memos/api/product_models.py

@@ -185,7 +185,7 @@ class APISearchRequest(BaseRequest):
     )
     include_preference: bool = Field(True, description="Whether to handle preference memory")
     pref_top_k: int = Field(6, description="Number of preference results to return")
-    info: dict | None = Field(None, description="Info for search")
+    source: str | None = Field(None, description="Source of the search")
 
 
 class APIADDRequest(BaseRequest):

src/memos/memories/textual/tree_text_memory/retrieve/searcher.py

@@ -275,6 +275,7 @@ def _retrieve_simple(
             user_name=user_name,
             use_fast_graph=self.use_fast_graph,
         )
+        logger.info(f"[SIMPLESEARCH] Items count: {len(items)}")
         documents = [getattr(item, "memory", "") for item in items]
         documents_embeddings = self.embedder.embed(documents)
         similarity_matrix = cosine_similarity_matrix(documents_embeddings)